mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-06-21 13:11:27 +00:00
44ea461890
* use Topo class to rewrite DAG * polish code * polish code * polish code * add comment * add else to unended if Co-authored-by: Ziyue Jiang <ziyue.jiang@gmail.com>
3 lines
148 B
Python
3 lines
148 B
Python
from .topo import Topo, Partition, PartitionOutputVal, PartitionInputVal
|
|
|
|
__all__ = ['Topo', 'Partition', 'PartitionOutputVal', 'PartitionInputVal'] |