binmakeswell 7696cead8d Recover kernal files

2022-07-13 12:08:21 +08:00

..

include

Recover kernal files

2022-07-13 12:08:21 +08:00

cross_entropy.cu

[NFC] polish colossalai/kernel/cuda_native/csrc/kernels/cross_entropy.cu code style (#953)

2022-05-17 10:25:06 +08:00

cublas_wrappers.cu

add colossalai kernel module (#55)

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

cuda_util.cu

[NFC] polish colossalai/kernel/cuda_native/csrc/kernels/cuda_util.cu code stype (#625)

2022-04-01 17:45:53 +08:00

dropout_kernels.cu

Recover kernal files

2022-07-13 12:08:21 +08:00

general_kernels.cu

[NFC] polish colossalai/kernel/cuda_native/csrc/kernels/general_kernels.cu code style (#963)

2022-05-17 10:25:06 +08:00

normalize_kernels.cu

[NFC] polish colossalai/kernel/cuda_native/csrc/kernels/normalize_kernels.cu code style (#974)

2022-05-17 10:25:06 +08:00

softmax_kernels.cu

[NFC] polish colossalai/kernel/cuda_native/csrc/kernels/softmax_kernels.cu code style (#954)

2022-05-17 10:25:06 +08:00

transform_kernels.cu

Recover kernal files

2022-07-13 12:08:21 +08:00