mirror of https://github.com/hpcaitech/ColossalAI
dc003c304c
* update moe module * support openmoe |
||
---|---|---|
.. | ||
model | ||
multi_tensor_apply | ||
rank_recorder | ||
tensor_detector | ||
__init__.py | ||
common.py | ||
cuda.py | ||
timer.py |