github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/kernel/cuda_native/mha/__init__.py

4 lines
60 B
Python
Raw Normal View History

[coloattention] fix import error (#4380) fixed an import error
2023-08-04 08:28:41 +00:00
from .mha import ColoAttention
[misc] update pre-commit and run all files (#4752) * [misc] update pre-commit * [misc] run pre-commit * [misc] remove useless configuration files * [misc] ignore cuda for clang-format
2023-09-19 06:20:26 +00:00
__all__ = ["ColoAttention"]