binmakeswell 7696cead8d Recover kernal files

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

..

block_reduce.h

[NFC] polish colossalai/kernel/cuda_native/csrc/kernels/include/block_reduce.h code style (#938)

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

context.h

[NFC] polish colossalai/kernel/cuda_native/csrc/kernels/include/context.h code style (#956)

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

cross_entropy_layer.h

[NFC] polish colossalai/kernel/cuda_native/csrc/kernels/include/cross_entropy_layer.h code style (#957)

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

cublas_wrappers.h

add colossalai kernel module (#55)

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

cuda_util.h

[NFC] polish colossalai/kernel/cuda_native/csrc/kernels/include/cuda_util.h code style (#939)

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

dropout.h

[NFC] polish colossalai/kernel/cuda_native/csrc/kernels/include/dropout.h code style (#970)

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

feed_forward.h

[NFC] polish colossalai/kernel/cuda_native/csrc/kernels/include/feed_forward.h (#968)

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

kernels.h

Recover kernal files

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

ls_cub.cuh

add colossalai kernel module (#55)

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

normalize_layer.h

fix format (#574)

2022-04-06 11:40:59 +08:00

softmax.h

[NFC] polish colossalai/kernel/cuda_native/csrc/kernels/include/softmax.h code style (#964)

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

strided_batch_gemm.h

add colossalai kernel module (#55)

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