extensions/flash_attention/flash_attention_npu.py · 9a3321e9f43d6688577413b419a1bfffea0c1938 · OpenDAS / ColossalAI

Frank Lee authored Jan 25, 2024

* [feat] refactored extension module

* polish

* polish

* polish

* polish

* polish

* polish

* polish

* polish

* polish

* polish

7cfed5f0

flash_attention_npu.py 2.3 KB