ColossalAI/colossalai/kernel/cuda_native/csrc/kernels
ExtremeViscent 22d1df224d [NFC] polish colossalai/kernel/cuda_native/csrc/kernels/include/feed_forward.h (#968)
code style
2022-05-17 10:25:06 +08:00
..
include [NFC] polish colossalai/kernel/cuda_native/csrc/kernels/include/feed_forward.h (#968) 2022-05-17 10:25:06 +08:00
cross_entropy.cu
cublas_wrappers.cu
cuda_util.cu
dropout_kernels.cu
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
softmax_kernels.cu
transform_kernels.cu