ColossalAI/colossalai/kernel
2022-04-06 11:40:59 +08:00
..
cuda_native [format]colossalai/kernel/cuda_native/csrc/layer_norm_cuda.cpp (#566) 2022-04-06 11:40:59 +08:00
jit adapted for sequence parallel (#163) 2022-01-20 13:44:51 +08:00
__init__.py refactor kernel (#142) 2022-01-13 16:47:17 +08:00