ColossalAI/colossalai/kernel/cuda_native/csrc/kernels/include
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
..
block_reduce.h
context.h
cross_entropy_layer.h
cublas_wrappers.h [doc] add deepspeed citation and copyright (#2996) 2023-03-04 20:08:11 +08:00
cuda_util.h
dropout.h [NFC] polish colossalai/kernel/cuda_native/csrc/kernels/include/dropout.h code style (#970) 2022-05-17 10:25:06 +08:00
feed_forward.h [doc] add deepspeed citation and copyright (#2996) 2023-03-04 20:08:11 +08:00
kernels.h Recover kernal files 2022-07-13 12:08:21 +08:00
ls_cub.cuh
normalize_layer.h
softmax.h [NFC] polish colossalai/kernel/cuda_native/csrc/kernels/include/softmax.h code style (#964) 2022-05-17 10:25:06 +08:00
strided_batch_gemm.h [doc] add deepspeed citation and copyright (#2996) 2023-03-04 20:08:11 +08:00