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