You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/kernel/cuda_native/csrc/kernels/include
Hongxin Liu 079bf3cb26
[misc] update pre-commit and run all files (#4752)
1 year ago
..
block_reduce.h
context.h
cross_entropy_layer.h
cublas_wrappers.h
cuda_util.h
dropout.h [misc] update pre-commit and run all files (#4752) 1 year ago
feed_forward.h
kernels.h [misc] update pre-commit and run all files (#4752) 1 year ago
ls_cub.cuh
normalize_layer.h [misc] update pre-commit and run all files (#4752) 1 year ago
softmax.h [misc] update pre-commit and run all files (#4752) 1 year ago
strided_batch_gemm.h