ColossalAI/colossalai/kernel/cuda_native/csrc/kernels
Cautiousss 0b86a6345e [NFC] polish colossalai/kernel/cuda_native/csrc/kernels/cross_entropy.cu code style (#953)
Co-authored-by: 何晓昕 <cautious@hexiaoxins-MacBook-Pro.local>
2022-05-17 10:25:06 +08:00
..
include [NFC] polish colossalai/kernel/cuda_native/csrc/kernels/include/block_reduce.h code style (#938) 2022-05-17 10:25:06 +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 add colossalai kernel module (#55) 2021-12-21 12:19:52 +08:00
cuda_util.cu [NFC] polish colossalai/kernel/cuda_native/csrc/kernels/cuda_util.cu code stype (#625) 2022-04-01 17:45:53 +08:00
dropout_kernels.cu Revert "[NFC] Hotfix/format (#984)" (#986) 2022-05-17 10:23:38 +08:00
general_kernels.cu Revert "[NFC] Hotfix/format (#984)" (#986) 2022-05-17 10:23:38 +08:00
normalize_kernels.cu Revert "[NFC] Hotfix/format (#984)" (#986) 2022-05-17 10:23:38 +08:00
softmax_kernels.cu Revert "[NFC] Hotfix/format (#984)" (#986) 2022-05-17 10:23:38 +08:00
transform_kernels.cu [NFC] polish colossalai/kernel/cuda_native/csrc/kernels/transform_kernels.cu code stype (#629) 2022-04-06 11:40:59 +08:00