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
Xuanlei Zhao dd2c28a323
[npu] use extension for op builder (#5172)
11 months ago
..
csrc
__init__.py [npu] use extension for op builder (#5172) 11 months ago
layer_norm.py
multihead_attention.py
scaled_softmax.py