github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-05-29 18:35:00 +00:00
Code Issues Packages Projects Releases Wiki Activity
06db94fbc9
ColossalAI/extensions/flash_attention
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