ColossalAI/colossalai/kernel/cuda_native/csrc/kernels
ver217 823f3b9cf4
[doc] add deepspeed citation and copyright (#2996)
* [doc] add deepspeed citation and copyright

* [doc] add deepspeed citation and copyright

* [doc] add deepspeed citation and copyright
2023-03-04 20:08:11 +08:00
..
include [doc] add deepspeed citation and copyright (#2996) 2023-03-04 20:08:11 +08:00
cross_entropy.cu
cublas_wrappers.cu [doc] add deepspeed citation and copyright (#2996) 2023-03-04 20:08:11 +08:00
cuda_util.cu
dropout_kernels.cu Recover kernal files 2022-07-13 12:08:21 +08:00
general_kernels.cu
normalize_kernels.cu
softmax_kernels.cu
transform_kernels.cu Recover kernal files 2022-07-13 12:08:21 +08:00