github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-12 02:38:52 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
ac178ca5c17ca751ff9df38be81da2b4a005fc0d
ColossalAI/tests/test_utils/test_flash_attention.py
flybird1111 7a3dfd0c64 [shardformer] update shardformer to use flash attention 2 (#4392) 
* cherry-pick flash attention 2

cherry-pick flash attention 2

* [shardformer] update shardformer to use flash attention 2

[shardformer] update shardformer to use flash attention 2, fix

[shardformer] update shardformer to use flash attention 2, fix

[shardformer] update shardformer to use flash attention 2, fix
2023-08-15 23:25:14 +08:00

7.0 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink