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

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-29 11:25: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