ColossalAI/extensions/flash_attention at 06db94fbc98e5f21dedd1199370bcf6ec4bd00d5 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-15 14:47:16 +00:00

Files

History

digger yu 6a3086a505 fix typo under extensions/ (#5330 )

2024-01-30 09:55:16 +08:00

..

__init__.py

[feat] refactored extension module (#5298 )

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

flash_attention_dao_cuda.py

fix typo under extensions/ (#5330 )

2024-01-30 09:55:16 +08:00

flash_attention_npu.py

[feat] refactored extension module (#5298 )

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

flash_attention_xformers_cuda.py

fix typo under extensions/ (#5330 )

2024-01-30 09:55:16 +08:00