ver217 12b4887097 [hotfix] fix CPUAdam kernel nullptr (#1410)		2022-08-05 19:45:45 +08:00
..
kernels	Recover kernal files	2022-07-13 12:08:21 +08:00
colossal_C_frontend.cpp	[NFC] polish colossalai/kernel/cuda_native/csrc/colossal_C_frontend.cpp code style	2022-05-20 23:57:38 +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	…
multi_tensor_apply.cuh	Recover kernal files	2022-07-13 12:08:21 +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	…
multihead_attention_1d.h	…
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	Recover kernal files	2022-07-13 12:08:21 +08:00