github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-12-23 12:36:03 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
aabc9fb6aada9e7feb2ff8cf1f34e6ac37ade2e7
ColossalAI/tests/test_utils/test_flash_attention.py
Frank Lee 7cfed5f076 [feat] refactored extension module (#5298) 
* [feat] refactored extension module

* polish

* polish

* polish

* polish

* polish

* polish

* polish

* polish

* polish

* polish
2024-01-25 17:01:48 +08:00

6.9 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink