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
digger-yu b9a8dff7e5
[doc] Fix typo under colossalai and doc(#3618)
2 years ago
..
csrc [doc] add deepspeed citation and copyright (#2996) 2 years ago
__init__.py [kernel] fixed repeated loading of kernels (#2549) 2 years ago
flash_attention.py updated flash attention usage 2 years ago
layer_norm.py [kernel] fixed repeated loading of kernels (#2549) 2 years ago
multihead_attention.py [doc] Fix typo under colossalai and doc(#3618) 2 years ago
scaled_softmax.py [kernel] added kernel loader to softmax autograd function (#3093) 2 years ago