|
block_reduce.h
|
block_reduce.h fix format (#581)
|
2022-04-06 11:40:59 +08:00 |
|
context.h
|
add colossalai kernel module (#55)
|
2021-12-21 12:19:52 +08:00 |
|
cublas_wrappers.h
|
add colossalai kernel module (#55)
|
2021-12-21 12:19:52 +08:00 |
|
cuda_util.h
|
add colossalai kernel module (#55)
|
2021-12-21 12:19:52 +08:00 |
|
dropout.h
|
add colossalai kernel module (#55)
|
2021-12-21 12:19:52 +08:00 |
|
feed_forward.h
|
fix format (#565)
|
2022-04-06 11:40:59 +08:00 |
|
kernels.h
|
add colossalai kernel module (#55)
|
2021-12-21 12:19:52 +08:00 |
|
ls_cub.cuh
|
add colossalai kernel module (#55)
|
2021-12-21 12:19:52 +08:00 |
|
normalize_layer.h
|
fix format (#574)
|
2022-04-06 11:40:59 +08:00 |
|
softmax.h
|
fix format (#567)
|
2022-04-06 11:40:59 +08:00 |
|
strided_batch_gemm.h
|
add colossalai kernel module (#55)
|
2021-12-21 12:19:52 +08:00 |
f08fc17f2b