mirror of https://github.com/hpcaitech/ColossalAI
3 lines
148 B
Python
3 lines
148 B
Python
|
from .topo import Topo, Partition, PartitionOutputVal, PartitionInputVal
|
||
|
|
||
|
__all__ = ['Topo', 'Partition', 'PartitionOutputVal', 'PartitionInputVal']
|