mindspore 1.10.0__cp38-cp38-macosx_11_0_arm64.whl → 2.0.0rc1__cp38-cp38-macosx_11_0_arm64.whl

mindspore/.commit_id

@@ -1 +1 @@
-__commit_id__ = ''[sha1]:23202b67,[branch]:(HEAD,origin/r1.10,r1.10)''
+__commit_id__ = ''[sha1]:ea0b5f52,[branch]:(HEAD,origin/r2.0,r2.0)''

mindspore/__init__.py

@@ -16,19 +16,23 @@
 from __future__ import absolute_import
 
 from mindspore.run_check import run_check
-from mindspore import common, dataset, mindrecord, train, log
+from mindspore import common, dataset, mindrecord, train, log, amp
 from mindspore import profiler, communication, numpy, parallel
 from mindspore.common import *
 from mindspore.mindrecord import *
-from mindspore.ops import _op_impl
+from mindspore.ops import _op_impl, grad, value_and_grad, vjp, jvp, jacfwd, jacrev, vmap, get_grad
 from mindspore.train import *
 from mindspore.log import *
 from mindspore.context import GRAPH_MODE, PYNATIVE_MODE, set_context, get_context, set_auto_parallel_context, \
     get_auto_parallel_context, reset_auto_parallel_context, ParallelMode, set_ps_context, \
     get_ps_context, reset_ps_context
 from mindspore.version import __version__
-from mindspore.profiler import Profiler
-from mindspore.parallel import set_algo_parameters, get_algo_parameters, reset_algo_parameters
+from mindspore.profiler import Profiler, EnvProfiler
+from mindspore.parallel import set_algo_parameters, get_algo_parameters, reset_algo_parameters, \
+    rank_list_for_transform, transform_checkpoint_by_rank, transform_checkpoints, merge_pipeline_strategys, shard
+from mindspore.rewrite import SymbolTree, ScopedValue, Node, NodeType, TreeNodeHelper
+from mindspore._check_jit_forbidden_api import get_obj_module_and_name_info, is_jit_forbidden_module, \
+    is_invalid_or_jit_forbidden_method
 
 
 __all__ = ["run_check"]
@@ -38,4 +42,5 @@ __all__.extend(train.__all__)
 __all__.extend(log.__all__)
 __all__.extend(context.__all__)
 __all__.extend(parallel.__all__)
+__all__.extend(rewrite.__all__)
 __all__.append("Profiler")

mindspore/_check_jit_forbidden_api.py

@@ -0,0 +1,102 @@
+# Copyright 2023 Huawei Technologies Co., Ltd
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ============================================================================
+"""Check jit forbidden api."""
+
+import types
+from functools import wraps
+from mindspore import log as logger
+
+# module: such as "mindspore.common.initializer"
+_jit_forbidden_module = set()
+_jit_forbidden_method = set()
+
+
+def jit_forbidden_register(fn):
+    """Use @jit_forbidden_register to decoratore the api which is not supported in GRAPH_MODE."""
+    setattr(fn, '__jit_forbidden__', True)
+    if isinstance(fn, types.FunctionType):
+        fn_info = fn.__module__ + '.' + fn.__qualname__
+        add_jit_forbidden_method(fn_info)
+
+        @wraps(fn)
+        def jit_forbidden(*args, **kwargs):
+            return fn(*args, **kwargs)
+    else:
+        def jit_forbidden(*args, **kwargs):
+            return fn(*args, **kwargs)
+    return jit_forbidden
+
+
+def add_jit_forbidden_module(jit_forbidden_module):
+    """Add the jit forbidden module."""
+    logger.debug(f'add jit_forbidden_module_set: {_jit_forbidden_module}')
+    return _jit_forbidden_module.add(jit_forbidden_module)
+
+
+def remove_jit_forbidden_module(jit_forbidden_module):
+    """Remove the jit forbidden module."""
+    logger.debug(f'remove jit_forbidden_module_set: {_jit_forbidden_module}')
+    return _jit_forbidden_module.remove(jit_forbidden_module)
+
+
+def get_jit_forbidden_module():
+    """Get the jit forbidden module."""
+    logger.debug(f'get jit_forbidden_module_set: {_jit_forbidden_module}')
+    return _jit_forbidden_module
+
+
+def add_jit_forbidden_method(jit_forbidden_method):
+    """Add the jit forbidden method."""
+    logger.debug(f'add jit_forbidden_method_set: {_jit_forbidden_method}')
+    return _jit_forbidden_method.add(jit_forbidden_method)
+
+
+def remove_jit_forbidden_method(jit_forbidden_method):
+    """Remove the jit forbidden method."""
+    logger.debug(f'remove jit_forbidden_method_set: {_jit_forbidden_method}')
+    return _jit_forbidden_method.remove(jit_forbidden_method)
+
+
+def get_jit_forbidden_method():
+    """Get the jit forbidden method."""
+    logger.debug(f'get jit_forbidden_module_set: {_jit_forbidden_method}')
+    return _jit_forbidden_method
+
+
+def get_obj_module_and_name_info(obj):
+    """Return the description of the object whose type is class, function or method."""
+    if not hasattr(obj, "__module__"):
+        return None
+    if isinstance(obj, (types.FunctionType, types.MethodType)):
+        return obj.__module__, obj.__qualname__, "method or function"
+    return obj.__module__, obj.__name__, "class"
+
+
+def is_jit_forbidden_module(obj_module):
+    """Return the matching result of object module in jit forbidden module set."""
+    if obj_module in _jit_forbidden_module:
+        return True
+    return False
+
+
+def is_invalid_or_jit_forbidden_method(obj, obj_type, attr):
+    """Check obj has attribute or method."""
+    if not hasattr(obj, attr):
+        raise AttributeError(f"{obj_type} object has no attribute: {attr}")
+    method = getattr(obj, attr)
+    method_info = method.__module__ + '.' + method.__qualname__
+    return method_info in _jit_forbidden_method
+
+add_jit_forbidden_module("mindspore.common.initializer")