tests/test_utils/test_flash_attention.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

test_flash_attention.py 6.85 KB