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 | ||