shenggan 5c3843dc98 add colossalai kernel module (#55)		2021-12-21 12:19:52 +08:00
..
block_reduce.h	add colossalai kernel module (#55)	2021-12-21 12:19:52 +08:00
context.h	add colossalai kernel module (#55)	2021-12-21 12:19:52 +08:00
cross_entropy_layer.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	add colossalai kernel module (#55)	2021-12-21 12:19:52 +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	add colossalai kernel module (#55)	2021-12-21 12:19:52 +08:00
softmax.h	add colossalai kernel module (#55)	2021-12-21 12:19:52 +08:00
strided_batch_gemm.h	add colossalai kernel module (#55)	2021-12-21 12:19:52 +08:00