ColossalAI/colossalai/kernel/cuda_native/csrc
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
..
kernels [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
colossal_C_frontend.cpp
compat.h [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
cpu_adam.cpp
cpu_adam.h
layer_norm_cuda.cpp [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
layer_norm_cuda_kernel.cu [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
moe_cuda.cpp [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
moe_cuda_kernel.cu [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
multi_tensor_adam.cu [doc] add deepspeed citation and copyright (#2996) 2023-03-04 20:08:11 +08:00
multi_tensor_apply.cuh [doc] add deepspeed citation and copyright (#2996) 2023-03-04 20:08:11 +08:00
multi_tensor_l2norm_kernel.cu [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
multi_tensor_lamb.cu [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
multi_tensor_scale_kernel.cu [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
multi_tensor_sgd_kernel.cu [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
multihead_attention_1d.cpp
multihead_attention_1d.h
scaled_masked_softmax.cpp [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
scaled_masked_softmax.h [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
scaled_masked_softmax_cuda.cu
scaled_upper_triang_masked_softmax.cpp
scaled_upper_triang_masked_softmax.h [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
scaled_upper_triang_masked_softmax_cuda.cu
type_shim.h [bf16] add bf16 support (#3882) 2023-06-05 15:58:31 +08:00