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
Jun Gao dce05da535
fix thrust-transform-reduce error (#5078)
1 year ago
..
csrc fix thrust-transform-reduce error (#5078) 1 year ago
mha [npu] add npu support for gemini and zero (#5067) 1 year ago
__init__.py [misc] update pre-commit and run all files (#4752) 1 year ago
layer_norm.py [misc] update pre-commit and run all files (#4752) 1 year ago
multihead_attention.py [misc] update pre-commit and run all files (#4752) 1 year ago
scaled_softmax.py [misc] update pre-commit and run all files (#4752) 1 year ago