Jun Gao dce05da535 fix thrust-transform-reduce error (#5078)		2023-11-21 15:09:35 +08:00
..
include	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
cross_entropy.cu	[NFC] polish colossalai/kernel/cuda_native/csrc/kernels/cross_entropy.cu code style (#953)	2022-05-17 10:25:06 +08:00
cublas_wrappers.cu	[doc] add deepspeed citation and copyright (#2996)	2023-03-04 20:08:11 +08:00
cuda_util.cu	fix thrust-transform-reduce error (#5078)	2023-11-21 15:09:35 +08:00
dropout_kernels.cu	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
general_kernels.cu	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
normalize_kernels.cu	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
softmax_kernels.cu	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00
transform_kernels.cu	[misc] update pre-commit and run all files (#4752)	2023-09-19 14:20:26 +08:00