ColossalAI/colossalai/kernel/cuda_native/csrc/kernels/include
yuxuan-lou 44b6f8947b [NFC] polish colossalai/kernel/cuda_native/csrc/kernels/include/cuda_util.h code style (#939) 2022-05-17 10:25:06 +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
cuda_util.h [NFC] polish colossalai/kernel/cuda_native/csrc/kernels/include/cuda_util.h code style (#939) 2022-05-17 10:25:06 +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
ls_cub.cuh
normalize_layer.h
softmax.h Revert "[NFC] Hotfix/format (#984)" (#986) 2022-05-17 10:23:38 +08:00
strided_batch_gemm.h