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