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

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-11 14:43:10 +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