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

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-16 07:28:45 +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