mirror of https://github.com/hpcaitech/ColossalAI
7cfed5f076
* [feat] refactored extension module * polish * polish * polish * polish * polish * polish * polish * polish * polish * polish |
||
|---|---|---|
| .. | ||
| test_flash_attention.py | ||