未验证 提交 afd1b372 编写于 作者: L lilong12 提交者: GitHub

hidden the auto parallel apis (#35385)

* update, test=develop
上级 39540b0e
...@@ -40,8 +40,8 @@ from .collective import get_group # noqa: F401 ...@@ -40,8 +40,8 @@ from .collective import get_group # noqa: F401
from .collective import send # noqa: F401 from .collective import send # noqa: F401
from .collective import wait # noqa: F401 from .collective import wait # noqa: F401
from .auto_parallel import shard_tensor # noqa: F401
from .auto_parallel import shard_op # noqa: F401 from .auto_parallel import shard_op # noqa: F401
from .auto_parallel import shard_tensor # noqa: F401
from .auto_parallel import set_shard_mask # noqa: F401 from .auto_parallel import set_shard_mask # noqa: F401
from .auto_parallel import set_offload_device # noqa: F401 from .auto_parallel import set_offload_device # noqa: F401
from .auto_parallel import set_pipeline_stage # noqa: F401 from .auto_parallel import set_pipeline_stage # noqa: F401
...@@ -85,10 +85,4 @@ __all__ = [ # noqa ...@@ -85,10 +85,4 @@ __all__ = [ # noqa
"wait", "wait",
"get_rank", "get_rank",
"ProbabilityEntry", "ProbabilityEntry",
"shard_tensor",
"shard_op",
"set_shard_mask",
"set_offload_device",
"set_pipeline_stage",
"ProcessMesh",
] ]
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册