ColossalAI/colossalai/kernel/cuda_native/csrc/kernels/include
Hongxin Liu 079bf3cb26
[misc] update pre-commit and run all files (#4752)
* [misc] update pre-commit

* [misc] run pre-commit

* [misc] remove useless configuration files

* [misc] ignore cuda for clang-format
2023-09-19 14:20:26 +08:00
..
block_reduce.h [NFC] polish colossalai/kernel/cuda_native/csrc/kernels/include/block_reduce.h code style (#938) 2022-05-17 10:25:06 +08:00
context.h [NFC] polish colossalai/kernel/cuda_native/csrc/kernels/include/context.h code style (#956) 2022-05-17 10:25:06 +08:00
cross_entropy_layer.h [NFC] polish colossalai/kernel/cuda_native/csrc/kernels/include/cross_entropy_layer.h code style (#957) 2022-05-17 10:25:06 +08:00
cublas_wrappers.h [doc] add deepspeed citation and copyright (#2996) 2023-03-04 20:08:11 +08:00
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 [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
feed_forward.h [doc] add deepspeed citation and copyright (#2996) 2023-03-04 20:08:11 +08:00
kernels.h [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
ls_cub.cuh add colossalai kernel module (#55) 2021-12-21 12:19:52 +08:00
normalize_layer.h [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
softmax.h [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
strided_batch_gemm.h [doc] add deepspeed citation and copyright (#2996) 2023-03-04 20:08:11 +08:00