include
|
fix format (#565)
|
2022-04-06 11:40:59 +08:00 |
cross_entropy.cu
|
add colossalai kernel module (#55)
|
2021-12-21 12:19:52 +08:00 |
cublas_wrappers.cu
|
add colossalai kernel module (#55)
|
2021-12-21 12:19:52 +08:00 |
dropout_kernels.cu
|
add colossalai kernel module (#55)
|
2021-12-21 12:19:52 +08:00 |
general_kernels.cu
|
add colossalai kernel module (#55)
|
2021-12-21 12:19:52 +08:00 |
normalize_kernels.cu
|
add colossalai kernel module (#55)
|
2021-12-21 12:19:52 +08:00 |
softmax_kernels.cu
|
add colossalai kernel module (#55)
|
2021-12-21 12:19:52 +08:00 |
transform_kernels.cu
|
add colossalai kernel module (#55)
|
2021-12-21 12:19:52 +08:00 |