ColossalAI/colossalai/kernel/cuda_native/csrc/kernels/include
ver217 58580b50fe
Revert "[NFC] Hotfix/format (#984)" (#986)
This reverts commit 0772828fba.
2022-05-17 10:23:38 +08:00
..
block_reduce.h Revert "[NFC] Hotfix/format (#984)" (#986) 2022-05-17 10:23:38 +08:00
context.h Revert "[NFC] Hotfix/format (#984)" (#986) 2022-05-17 10:23:38 +08:00
cross_entropy_layer.h Revert "[NFC] Hotfix/format (#984)" (#986) 2022-05-17 10:23:38 +08:00
cublas_wrappers.h add colossalai kernel module (#55) 2021-12-21 12:19:52 +08:00
cuda_util.h Revert "[NFC] Hotfix/format (#984)" (#986) 2022-05-17 10:23:38 +08:00
dropout.h Revert "[NFC] Hotfix/format (#984)" (#986) 2022-05-17 10:23:38 +08:00
feed_forward.h Revert "[NFC] Hotfix/format (#984)" (#986) 2022-05-17 10:23:38 +08:00
kernels.h fix format (#613) 2022-04-06 11:40:59 +08:00
ls_cub.cuh add colossalai kernel module (#55) 2021-12-21 12:19:52 +08:00
normalize_layer.h fix format (#574) 2022-04-06 11:40:59 +08:00
softmax.h Revert "[NFC] Hotfix/format (#984)" (#986) 2022-05-17 10:23:38 +08:00
strided_batch_gemm.h add colossalai kernel module (#55) 2021-12-21 12:19:52 +08:00