mirror of https://github.com/hpcaitech/ColossalAI
823f3b9cf4
* [doc] add deepspeed citation and copyright * [doc] add deepspeed citation and copyright * [doc] add deepspeed citation and copyright |
||
---|---|---|
.. | ||
include | ||
cross_entropy.cu | ||
cublas_wrappers.cu | ||
cuda_util.cu | ||
dropout_kernels.cu | ||
general_kernels.cu | ||
normalize_kernels.cu | ||
softmax_kernels.cu | ||
transform_kernels.cu |