ColossalAI/colossalai/kernel/cuda_native/csrc
ver217 823f3b9cf4
[doc] add deepspeed citation and copyright (#2996)
* [doc] add deepspeed citation and copyright

* [doc] add deepspeed citation and copyright

* [doc] add deepspeed citation and copyright
2023-03-04 20:08:11 +08:00
..
kernels [doc] add deepspeed citation and copyright (#2996) 2023-03-04 20:08:11 +08:00
colossal_C_frontend.cpp [optimizer] add div_scale for optimizers (#2117) 2022-12-12 17:58:57 +08:00
compat.h
cpu_adam.cpp [hotfix] fix CPUAdam kernel nullptr (#1410) 2022-08-05 19:45:45 +08:00
cpu_adam.h [hotfix] fix CPUAdam kernel nullptr (#1410) 2022-08-05 19:45:45 +08:00
layer_norm_cuda.cpp [NFC] polish colossalai/kernel/cuda_native/csrc/layer_norm_cuda.cpp code style (#973) 2022-05-17 10:25:06 +08:00
layer_norm_cuda_kernel.cu
moe_cuda.cpp
moe_cuda_kernel.cu
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
multi_tensor_lamb.cu
multi_tensor_scale_kernel.cu [NFC] polish colossalai/kernel/cuda_native/csrc/multi_tensor_scale_kernel.cu code style (#977) 2022-05-17 10:25:06 +08:00
multi_tensor_sgd_kernel.cu [optim] refactor fused sgd (#1134) 2022-06-20 11:19:38 +08:00
multihead_attention_1d.cpp [hotfix] fix error for torch 2.0 (#2243) 2022-12-30 23:11:55 +08:00
multihead_attention_1d.h [hotfix] fix error for torch 2.0 (#2243) 2022-12-30 23:11:55 +08:00
scaled_masked_softmax.cpp Recover kernal files 2022-07-13 12:08:21 +08:00
scaled_masked_softmax.h Recover kernal files 2022-07-13 12:08:21 +08:00
scaled_masked_softmax_cuda.cu
scaled_upper_triang_masked_softmax.cpp
scaled_upper_triang_masked_softmax.h Recover kernal files 2022-07-13 12:08:21 +08:00
scaled_upper_triang_masked_softmax_cuda.cu
type_shim.h [doc] add deepspeed citation and copyright (#2996) 2023-03-04 20:08:11 +08:00