mirror of https://github.com/hpcaitech/ColossalAI
3c08f17348
* feat: modify create_ep_hierarchical_group args * test: add ep tests * fix: remove get_process_group_ranks * fix: fix src_rank |
||
---|---|---|
.. | ||
__init__.py | ||
api.py | ||
moe_info.py |