ColossalAI/colossalai/kernel/cuda_native/csrc/kernels at 0db43fa995a8070e69cfce3038dcdcb9a13ba90a - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-13 12:05:48 +00:00

Files

History

shenggan 5c3843dc98 add colossalai kernel module (#55 )

2021-12-21 12:19:52 +08:00

..

add colossalai kernel module (#55 )

2021-12-21 12:19:52 +08:00

cross_entropy.cu

add colossalai kernel module (#55 )

2021-12-21 12:19:52 +08:00

cublas_wrappers.cu

add colossalai kernel module (#55 )

2021-12-21 12:19:52 +08:00

cuda_util.cu

add colossalai kernel module (#55 )

2021-12-21 12:19:52 +08:00

dropout_kernels.cu

add colossalai kernel module (#55 )

2021-12-21 12:19:52 +08:00

general_kernels.cu

add colossalai kernel module (#55 )

2021-12-21 12:19:52 +08:00

normalize_kernels.cu

add colossalai kernel module (#55 )

2021-12-21 12:19:52 +08:00

softmax_kernels.cu

add colossalai kernel module (#55 )

2021-12-21 12:19:52 +08:00

transform_kernels.cu

add colossalai kernel module (#55 )

2021-12-21 12:19:52 +08:00