shenggan 5c3843dc98 add colossalai kernel module (#55)		2021-12-21 12:19:52 +08:00
..
include	add colossalai kernel module (#55)	2021-12-21 12:19:52 +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
cuda_util.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