github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/kernel/npu
History
Xuanlei Zhao d6df19bae7
[npu] support triangle attention for llama (#5130) 
* update fused attn

* update spda

* tri attn

* update triangle

* import

* fix

* fix
 		2023-11-30 14:21:30 +08:00
..
mha [npu] support triangle attention for llama (#5130) 2023-11-30 14:21:30 +08:00
__init__.py [npu] support triangle attention for llama (#5130) 2023-11-30 14:21:30 +08:00