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