mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

add parallel strategy algo interfaces to mindspore's main interface

This commit is contained in:
Xiaoda Zhang 2022-05-11 16:36:58 +08:00
parent 751b79792b
commit 13c56e83cd
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/python/mindspore/__init__.py
View File

@ -28,6 +28,7 @@ from .context import GRAPH_MODE, PYNATIVE_MODE, set_context, get_context, set_au
from .version import __version__
from .rewrite import *
from .profiler import Profiler
from .parallel import set_algo_parameters, get_algo_parameters, reset_algo_parameters
__all__ = ["run_check"]
@ -37,4 +38,5 @@ __all__.extend(train.__all__)
__all__.extend(log.__all__)
__all__.extend(context.__all__)
__all__.extend(rewrite.__all__)
__all__.extend(parallel.__all__)
__all__.append("Profiler")