ОбзорДокументацияВойти
/
githubmirror
/
ColossalAI
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
.../csrc/cuda/
..
include
3 года назад
colossal_C_frontend.cpp
3 года назад
compat.h
3 года назад
cpu_adam.cpp
3 года назад
cpu_adam.h
3 года назад
layer_norm_cuda.cpp
3 года назад
layer_norm_cuda_kernel.cu
3 года назад
moe_cuda.cpp
3 года назад
moe_cuda_kernel.cu
3 года назад
multi_tensor_adam.cu
3 года назад
multi_tensor_apply.cuh
3 года назад
multi_tensor_l2norm_kernel.cu
3 года назад
multi_tensor_lamb.cu
3 года назад
multi_tensor_scale_kernel.cu
3 года назад
multi_tensor_sgd_kernel.cu
3 года назад
scaled_masked_softmax.cpp
3 года назад
scaled_masked_softmax.h
3 года назад
scaled_masked_softmax_cuda.cu
3 года назад
scaled_upper_triang_masked_softmax.cpp
3 года назад
scaled_upper_triang_masked_softmax.h
3 года назад
scaled_upper_triang_masked_softmax_cuda.cu
3 года назад
type_shim.h
3 года назад