ColossalAI/extensions/flash_attention at 35382a7fbf96c731ba1ed76cf5529ea3220a5b66 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-02-21 14:32:09 +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