mindspore 2.7.0__cp39-cp39-macosx_11_0_arm64.whl → 2.7.1__cp39-cp39-macosx_11_0_arm64.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- mindspore/.commit_id +1 -1
- mindspore/__init__.py +4 -1
- mindspore/_c_dataengine.cpython-39-darwin.so +0 -0
- mindspore/_c_expression.cpython-39-darwin.so +0 -0
- mindspore/_c_mindrecord.cpython-39-darwin.so +0 -0
- mindspore/_extends/parse/compile_config.py +24 -1
- mindspore/_extends/parse/deprecated/deprecated_tensor_method.py +6 -2
- mindspore/_extends/parse/resources.py +1 -1
- mindspore/_extends/parse/standard_method.py +8 -1
- mindspore/_extends/parse/trope.py +2 -1
- mindspore/_extends/pijit/pijit_func_white_list.py +7 -22
- mindspore/boost/base.py +29 -2
- mindspore/common/_decorator.py +3 -2
- mindspore/common/_grad_function.py +3 -1
- mindspore/common/_tensor_cpp_method.py +1 -1
- mindspore/common/_tensor_docs.py +276 -65
- mindspore/common/_utils.py +0 -44
- mindspore/common/api.py +285 -35
- mindspore/common/dump.py +7 -108
- mindspore/common/dynamic_shape/auto_dynamic_shape.py +1 -3
- mindspore/common/hook_handle.py +60 -0
- mindspore/common/jit_config.py +5 -1
- mindspore/common/jit_trace.py +27 -12
- mindspore/common/lazy_inline.py +5 -3
- mindspore/common/parameter.py +13 -107
- mindspore/common/recompute.py +4 -11
- mindspore/common/tensor.py +16 -169
- mindspore/communication/_comm_helper.py +11 -1
- mindspore/communication/comm_func.py +138 -4
- mindspore/communication/management.py +85 -1
- mindspore/config/op_info.config +0 -15
- mindspore/context.py +5 -85
- mindspore/dataset/engine/datasets.py +8 -4
- mindspore/dataset/engine/datasets_vision.py +1 -1
- mindspore/dataset/engine/validators.py +1 -15
- mindspore/{experimental/llm_boost/ascend_native → graph}/__init__.py +7 -7
- mindspore/graph/custom_pass.py +55 -0
- mindspore/include/dataset/execute.h +2 -2
- mindspore/lib/libmindspore_backend_common.dylib +0 -0
- mindspore/lib/libmindspore_backend_manager.dylib +0 -0
- mindspore/lib/libmindspore_cluster.dylib +0 -0
- mindspore/lib/libmindspore_common.dylib +0 -0
- mindspore/lib/libmindspore_core.dylib +0 -0
- mindspore/lib/libmindspore_cpu.dylib +0 -0
- mindspore/lib/libmindspore_dump.dylib +0 -0
- mindspore/lib/libmindspore_frontend.dylib +0 -0
- mindspore/lib/libmindspore_hardware_abstract.dylib +0 -0
- mindspore/lib/libmindspore_memory_pool.dylib +0 -0
- mindspore/lib/libmindspore_ms_backend.dylib +0 -0
- mindspore/lib/libmindspore_ops.dylib +0 -0
- mindspore/lib/libmindspore_profiler.dylib +0 -0
- mindspore/lib/libmindspore_pyboost.dylib +0 -0
- mindspore/lib/libmindspore_pynative.dylib +0 -0
- mindspore/lib/libmindspore_runtime_pipeline.dylib +0 -0
- mindspore/lib/libmindspore_runtime_utils.dylib +0 -0
- mindspore/lib/libmindspore_tools.dylib +0 -0
- mindspore/lib/libopencv_core.411.dylib +0 -0
- mindspore/lib/libopencv_imgcodecs.411.dylib +0 -0
- mindspore/lib/libopencv_imgproc.411.dylib +0 -0
- mindspore/lib/plugin/{libmindspore_ops_host.dylib → libmindspore_ops_cpu.dylib} +0 -0
- mindspore/mindrecord/__init__.py +3 -3
- mindspore/mindrecord/common/exceptions.py +1 -0
- mindspore/mindrecord/config.py +1 -1
- mindspore/{parallel/mpi → mindrecord/core}/__init__.py +4 -1
- mindspore/mindrecord/{shardheader.py → core/shardheader.py} +2 -1
- mindspore/mindrecord/{shardindexgenerator.py → core/shardindexgenerator.py} +1 -1
- mindspore/mindrecord/{shardreader.py → core/shardreader.py} +2 -1
- mindspore/mindrecord/{shardsegment.py → core/shardsegment.py} +2 -2
- mindspore/mindrecord/{shardutils.py → core/shardutils.py} +1 -1
- mindspore/mindrecord/{shardwriter.py → core/shardwriter.py} +1 -1
- mindspore/mindrecord/filereader.py +4 -4
- mindspore/mindrecord/filewriter.py +5 -5
- mindspore/mindrecord/mindpage.py +2 -2
- mindspore/mindrecord/tools/cifar10.py +1 -1
- mindspore/mindrecord/tools/cifar100.py +1 -1
- mindspore/mindrecord/tools/cifar100_to_mr.py +1 -1
- mindspore/mindrecord/tools/cifar10_to_mr.py +1 -1
- mindspore/mindrecord/tools/csv_to_mr.py +1 -1
- mindspore/mindrecord/tools/imagenet_to_mr.py +1 -1
- mindspore/mindrecord/tools/mnist_to_mr.py +1 -1
- mindspore/mindrecord/tools/tfrecord_to_mr.py +1 -1
- mindspore/mint/__init__.py +15 -10
- mindspore/mint/distributed/distributed.py +182 -62
- mindspore/mint/nn/__init__.py +2 -16
- mindspore/mint/nn/functional.py +4 -110
- mindspore/mint/nn/layer/__init__.py +0 -2
- mindspore/mint/nn/layer/activation.py +0 -6
- mindspore/mint/nn/layer/basic.py +0 -47
- mindspore/mint/nn/layer/conv.py +4 -4
- mindspore/mint/nn/layer/normalization.py +8 -13
- mindspore/mint/nn/layer/pooling.py +0 -4
- mindspore/nn/__init__.py +1 -3
- mindspore/nn/cell.py +16 -66
- mindspore/nn/layer/basic.py +49 -1
- mindspore/nn/layer/container.py +16 -0
- mindspore/nn/layer/embedding.py +4 -169
- mindspore/nn/layer/normalization.py +2 -1
- mindspore/nn/layer/thor_layer.py +4 -85
- mindspore/nn/optim/ada_grad.py +0 -1
- mindspore/nn/optim/adafactor.py +0 -1
- mindspore/nn/optim/adam.py +31 -124
- mindspore/nn/optim/adamax.py +0 -1
- mindspore/nn/optim/asgd.py +0 -1
- mindspore/nn/optim/ftrl.py +8 -102
- mindspore/nn/optim/lamb.py +0 -1
- mindspore/nn/optim/lars.py +0 -3
- mindspore/nn/optim/lazyadam.py +25 -218
- mindspore/nn/optim/momentum.py +5 -43
- mindspore/nn/optim/optimizer.py +6 -55
- mindspore/nn/optim/proximal_ada_grad.py +0 -1
- mindspore/nn/optim/rmsprop.py +0 -1
- mindspore/nn/optim/rprop.py +0 -1
- mindspore/nn/optim/sgd.py +0 -1
- mindspore/nn/optim/tft_wrapper.py +0 -1
- mindspore/nn/optim/thor.py +0 -2
- mindspore/nn/probability/bijector/bijector.py +7 -8
- mindspore/nn/probability/bijector/gumbel_cdf.py +2 -2
- mindspore/nn/probability/bijector/power_transform.py +20 -21
- mindspore/nn/probability/bijector/scalar_affine.py +5 -5
- mindspore/nn/probability/bijector/softplus.py +13 -14
- mindspore/nn/wrap/grad_reducer.py +4 -74
- mindspore/numpy/array_creations.py +2 -2
- mindspore/numpy/fft.py +9 -9
- mindspore/{nn/reinforcement → onnx}/__init__.py +5 -8
- mindspore/onnx/onnx_export.py +137 -0
- mindspore/ops/__init__.py +2 -0
- mindspore/ops/_grad_experimental/grad_comm_ops.py +38 -2
- mindspore/ops/_op_impl/aicpu/__init__.py +0 -10
- mindspore/ops/_op_impl/cpu/__init__.py +0 -5
- mindspore/ops/auto_generate/cpp_create_prim_instance_helper.py +16 -22
- mindspore/ops/auto_generate/gen_extend_func.py +2 -7
- mindspore/ops/auto_generate/gen_ops_def.py +98 -141
- mindspore/ops/auto_generate/gen_ops_prim.py +1180 -1158
- mindspore/ops/communication.py +97 -0
- mindspore/ops/composite/__init__.py +5 -2
- mindspore/ops/composite/base.py +15 -1
- mindspore/ops/composite/multitype_ops/__init__.py +3 -1
- mindspore/ops/composite/multitype_ops/_compile_utils.py +150 -8
- mindspore/ops/composite/multitype_ops/add_impl.py +7 -0
- mindspore/ops/composite/multitype_ops/mod_impl.py +27 -0
- mindspore/ops/function/__init__.py +1 -0
- mindspore/ops/function/array_func.py +14 -12
- mindspore/ops/function/comm_func.py +3883 -0
- mindspore/ops/function/debug_func.py +3 -4
- mindspore/ops/function/math_func.py +45 -54
- mindspore/ops/function/nn_func.py +75 -294
- mindspore/ops/function/random_func.py +9 -18
- mindspore/ops/functional.py +2 -0
- mindspore/ops/functional_overload.py +354 -18
- mindspore/ops/operations/__init__.py +2 -5
- mindspore/ops/operations/_custom_ops_utils.py +7 -9
- mindspore/ops/operations/_inner_ops.py +1 -38
- mindspore/ops/operations/_rl_inner_ops.py +0 -933
- mindspore/ops/operations/array_ops.py +1 -0
- mindspore/ops/operations/comm_ops.py +94 -2
- mindspore/ops/operations/custom_ops.py +228 -19
- mindspore/ops/operations/debug_ops.py +27 -29
- mindspore/ops/operations/manually_defined/ops_def.py +27 -306
- mindspore/ops/operations/nn_ops.py +2 -2
- mindspore/ops/operations/sparse_ops.py +0 -83
- mindspore/ops/primitive.py +1 -17
- mindspore/ops/tensor_method.py +72 -3
- mindspore/ops_generate/aclnn/aclnn_kernel_register_auto_cc_generator.py +5 -5
- mindspore/ops_generate/aclnn/gen_aclnn_implement.py +8 -8
- mindspore/ops_generate/api/functions_cc_generator.py +53 -4
- mindspore/ops_generate/api/tensor_func_reg_cpp_generator.py +25 -11
- mindspore/ops_generate/common/gen_constants.py +11 -10
- mindspore/ops_generate/common/op_proto.py +18 -1
- mindspore/ops_generate/common/template.py +102 -245
- mindspore/ops_generate/common/template_utils.py +212 -0
- mindspore/ops_generate/gen_custom_ops.py +69 -0
- mindspore/ops_generate/op_def/ops_def_cc_generator.py +78 -7
- mindspore/ops_generate/op_def_py/base_op_prim_py_generator.py +360 -0
- mindspore/ops_generate/op_def_py/custom_op_prim_py_generator.py +140 -0
- mindspore/ops_generate/op_def_py/op_def_py_generator.py +54 -7
- mindspore/ops_generate/op_def_py/op_prim_py_generator.py +5 -312
- mindspore/ops_generate/pyboost/auto_grad_impl_cc_generator.py +74 -17
- mindspore/ops_generate/pyboost/auto_grad_reg_cc_generator.py +22 -5
- mindspore/ops_generate/pyboost/op_template_parser.py +3 -2
- mindspore/ops_generate/pyboost/pyboost_functions_cpp_generator.py +21 -5
- mindspore/ops_generate/pyboost/pyboost_functions_h_generator.py +2 -2
- mindspore/ops_generate/pyboost/pyboost_functions_impl_cpp_generator.py +30 -10
- mindspore/ops_generate/pyboost/pyboost_grad_function_cpp_generator.py +10 -3
- mindspore/ops_generate/pyboost/pyboost_internal_kernel_info_adapter_generator.py +1 -1
- mindspore/ops_generate/pyboost/pyboost_native_grad_functions_generator.py +19 -9
- mindspore/ops_generate/pyboost/pyboost_op_cpp_code_generator.py +71 -28
- mindspore/ops_generate/pyboost/pyboost_overload_functions_cpp_generator.py +10 -9
- mindspore/ops_generate/pyboost/pyboost_utils.py +27 -16
- mindspore/ops_generate/resources/yaml_loader.py +13 -0
- mindspore/ops_generate/tensor_py_cc_generator.py +2 -2
- mindspore/parallel/_cell_wrapper.py +1 -1
- mindspore/parallel/_parallel_serialization.py +1 -4
- mindspore/parallel/_utils.py +29 -6
- mindspore/parallel/checkpoint_transform.py +18 -2
- mindspore/parallel/cluster/process_entity/_api.py +24 -32
- mindspore/parallel/cluster/process_entity/_utils.py +9 -5
- mindspore/{experimental/llm_boost/atb → parallel/distributed}/__init__.py +21 -23
- mindspore/parallel/distributed/distributed_data_parallel.py +393 -0
- mindspore/parallel/distributed/flatten_grad_buffer.py +295 -0
- mindspore/parallel/strategy.py +336 -0
- mindspore/parallel/transform_safetensors.py +117 -16
- mindspore/profiler/analysis/viewer/ascend_kernel_details_viewer.py +3 -0
- mindspore/profiler/analysis/viewer/ms_minddata_viewer.py +1 -1
- mindspore/profiler/common/constant.py +5 -0
- mindspore/profiler/common/file_manager.py +9 -0
- mindspore/profiler/common/msprof_cmd_tool.py +38 -2
- mindspore/profiler/common/path_manager.py +56 -24
- mindspore/profiler/common/profiler_context.py +2 -12
- mindspore/profiler/common/profiler_info.py +3 -3
- mindspore/profiler/common/profiler_path_manager.py +13 -0
- mindspore/profiler/common/util.py +30 -3
- mindspore/profiler/experimental_config.py +2 -1
- mindspore/profiler/platform/npu_profiler.py +33 -6
- mindspore/run_check/_check_version.py +108 -24
- mindspore/runtime/__init__.py +3 -2
- mindspore/runtime/executor.py +11 -3
- mindspore/runtime/memory.py +112 -0
- mindspore/scipy/linalg.py +1 -1
- mindspore/{experimental/llm_boost → tools}/__init__.py +5 -5
- mindspore/tools/data_dump.py +130 -0
- mindspore/tools/sdc_detect.py +91 -0
- mindspore/tools/stress_detect.py +63 -0
- mindspore/train/__init__.py +6 -6
- mindspore/train/_utils.py +5 -18
- mindspore/train/amp.py +6 -4
- mindspore/train/callback/_checkpoint.py +0 -9
- mindspore/train/callback/_train_fault_tolerance.py +69 -18
- mindspore/train/data_sink.py +1 -5
- mindspore/train/model.py +38 -211
- mindspore/train/serialization.py +126 -387
- mindspore/utils/__init__.py +6 -3
- mindspore/utils/dlpack.py +92 -0
- mindspore/utils/dryrun.py +1 -1
- mindspore/utils/runtime_execution_order_check.py +10 -0
- mindspore/utils/sdc_detect.py +14 -12
- mindspore/utils/stress_detect.py +43 -0
- mindspore/utils/utils.py +144 -8
- mindspore/version.py +1 -1
- {mindspore-2.7.0.dist-info → mindspore-2.7.1.dist-info}/METADATA +3 -2
- {mindspore-2.7.0.dist-info → mindspore-2.7.1.dist-info}/RECORD +243 -257
- mindspore/experimental/llm_boost/ascend_native/llama_boost_ascend_native.py +0 -210
- mindspore/experimental/llm_boost/ascend_native/llm_boost.py +0 -52
- mindspore/experimental/llm_boost/atb/boost_base.py +0 -385
- mindspore/experimental/llm_boost/atb/llama_boost.py +0 -137
- mindspore/experimental/llm_boost/atb/qwen_boost.py +0 -124
- mindspore/experimental/llm_boost/register.py +0 -130
- mindspore/experimental/llm_boost/utils.py +0 -31
- mindspore/include/OWNERS +0 -7
- mindspore/lib/libmindspore_ops_kernel_common.dylib +0 -0
- mindspore/lib/libmindspore_res_manager.dylib +0 -0
- mindspore/lib/libopencv_core.4.5.dylib +0 -0
- mindspore/lib/libopencv_imgcodecs.4.5.dylib +0 -0
- mindspore/lib/libopencv_imgproc.4.5.dylib +0 -0
- mindspore/lib/libps_cache.dylib +0 -0
- mindspore/lib/plugin/cpu/libmindspore_cpu_res_manager.dylib +0 -0
- mindspore/nn/optim/_dist_optimizer_registry.py +0 -111
- mindspore/nn/reinforcement/_batch_read_write.py +0 -142
- mindspore/nn/reinforcement/_tensors_queue.py +0 -152
- mindspore/nn/reinforcement/tensor_array.py +0 -145
- mindspore/ops/_op_impl/aicpu/priority_replay_buffer.py +0 -113
- mindspore/ops/_op_impl/aicpu/reservoir_replay_buffer.py +0 -96
- mindspore/ops/_op_impl/aicpu/sparse_cross.py +0 -42
- mindspore/ops/_op_impl/cpu/buffer_append.py +0 -28
- mindspore/ops/_op_impl/cpu/buffer_get.py +0 -28
- mindspore/ops/_op_impl/cpu/buffer_sample.py +0 -28
- mindspore/ops/_op_impl/cpu/priority_replay_buffer.py +0 -42
- mindspore/ops/operations/_tensor_array.py +0 -359
- mindspore/ops/operations/rl_ops.py +0 -288
- mindspore/parallel/_offload_context.py +0 -275
- mindspore/parallel/_recovery_context.py +0 -115
- mindspore/parallel/_transformer/__init__.py +0 -35
- mindspore/parallel/_transformer/layers.py +0 -765
- mindspore/parallel/_transformer/loss.py +0 -251
- mindspore/parallel/_transformer/moe.py +0 -693
- mindspore/parallel/_transformer/op_parallel_config.py +0 -222
- mindspore/parallel/_transformer/transformer.py +0 -3124
- mindspore/parallel/mpi/_mpi_config.py +0 -116
- mindspore/train/memory_profiling_pb2.py +0 -298
- {mindspore-2.7.0.dist-info → mindspore-2.7.1.dist-info}/WHEEL +0 -0
- {mindspore-2.7.0.dist-info → mindspore-2.7.1.dist-info}/entry_points.txt +0 -0
- {mindspore-2.7.0.dist-info → mindspore-2.7.1.dist-info}/top_level.txt +0 -0
mindspore/.commit_id
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
__commit_id__ = ''[sha1]:
|
|
1
|
+
__commit_id__ = ''[sha1]:71340184,[branch]:(HEAD,origin/r2.7.1,r2.7.1)''
|
mindspore/__init__.py
CHANGED
|
@@ -30,7 +30,7 @@ from mindspore.device_manager import *
|
|
|
30
30
|
from mindspore.runtime import *
|
|
31
31
|
from mindspore.context import GRAPH_MODE, PYNATIVE_MODE, set_context, get_context, set_auto_parallel_context, \
|
|
32
32
|
get_auto_parallel_context, reset_auto_parallel_context, ParallelMode, set_ps_context, \
|
|
33
|
-
get_ps_context, reset_ps_context,
|
|
33
|
+
get_ps_context, reset_ps_context, STRICT, COMPATIBLE, LAX
|
|
34
34
|
from mindspore.version import __version__
|
|
35
35
|
from mindspore.profiler import Profiler, EnvProfiler
|
|
36
36
|
from mindspore.parallel import set_algo_parameters, get_algo_parameters, reset_algo_parameters, \
|
|
@@ -45,6 +45,8 @@ from mindspore._check_jit_forbidden_api import get_obj_module_and_name_info, is_
|
|
|
45
45
|
is_invalid_or_jit_forbidden_method
|
|
46
46
|
from mindspore import mint
|
|
47
47
|
from mindspore.ops._utils import arg_handler, arg_dtype_cast
|
|
48
|
+
from mindspore import onnx
|
|
49
|
+
from mindspore import graph
|
|
48
50
|
|
|
49
51
|
__all__ = ["run_check"]
|
|
50
52
|
__all__.extend(__version__)
|
|
@@ -57,4 +59,5 @@ __all__.extend(rewrite.__all__)
|
|
|
57
59
|
__all__.extend(safeguard.__all__)
|
|
58
60
|
__all__.extend(device_manager.__all__)
|
|
59
61
|
__all__.extend(runtime.__all__)
|
|
62
|
+
__all__.extend(graph.__all__)
|
|
60
63
|
__all__.append("Profiler")
|
|
Binary file
|
|
Binary file
|
|
Binary file
|
|
@@ -362,6 +362,27 @@ Value Range:
|
|
|
362
362
|
"""
|
|
363
363
|
JIT_ENABLE_AUGASSIGN_INPLACE = '0'
|
|
364
364
|
|
|
365
|
+
"""
|
|
366
|
+
Name: ENABLE_VIEW_INPLACE_GRAD_SCHEME_CHOOSE
|
|
367
|
+
Function: Whether enable new method for view inplace grad
|
|
368
|
+
Value Range:
|
|
369
|
+
0: Choose old or new scheme based on specific control flow scenarios.
|
|
370
|
+
1: Choose old scheme.
|
|
371
|
+
2: Choose new scheme.
|
|
372
|
+
Default: 2
|
|
373
|
+
"""
|
|
374
|
+
ENABLE_VIEW_INPLACE_GRAD_SCHEME_CHOOSE = '2'
|
|
375
|
+
|
|
376
|
+
"""
|
|
377
|
+
Name: GRAD_JIT_FILTER
|
|
378
|
+
Function: Whether to filter grad jit graph.
|
|
379
|
+
Value Range:
|
|
380
|
+
1: Enable filter grad jit output
|
|
381
|
+
2: Enable filter grad jit input and output. May cause error when input gradient information changed.
|
|
382
|
+
Default: Disable.
|
|
383
|
+
"""
|
|
384
|
+
GRAD_JIT_FILTER = '1'
|
|
385
|
+
|
|
365
386
|
__all__ = [
|
|
366
387
|
"COMPILE_PROFILE",
|
|
367
388
|
"COMPILE_PROFILE_FINISH_ACTION",
|
|
@@ -400,5 +421,7 @@ __all__ = [
|
|
|
400
421
|
"PUT_ALL_CNODE_INTO_ORDER_LIST",
|
|
401
422
|
"CHECK_PASS_NODE_SCOPE",
|
|
402
423
|
"CHECK_INVALID_VIEW_INPLACE_DOUT_LEVEL",
|
|
403
|
-
"JIT_ENABLE_AUGASSIGN_INPLACE"
|
|
424
|
+
"JIT_ENABLE_AUGASSIGN_INPLACE",
|
|
425
|
+
"ENABLE_VIEW_INPLACE_GRAD_SCHEME_CHOOSE",
|
|
426
|
+
"GRAD_JIT_FILTER"
|
|
404
427
|
]
|
|
@@ -179,7 +179,7 @@ deprecated_tensor_method_map = {
|
|
|
179
179
|
# 80 outer
|
|
180
180
|
"outer": "deprecated_tensor_outer",
|
|
181
181
|
# 81 permute
|
|
182
|
-
|
|
182
|
+
"permute": "deprecated_tensor_permute",
|
|
183
183
|
# 82 pow
|
|
184
184
|
"pow": "deprecated_tensor_pow",
|
|
185
185
|
# 83 prod
|
|
@@ -221,7 +221,7 @@ deprecated_tensor_method_map = {
|
|
|
221
221
|
# 100 square
|
|
222
222
|
|
|
223
223
|
# 101 squeeze
|
|
224
|
-
|
|
224
|
+
"squeeze": "tensor_squeeze",
|
|
225
225
|
# 102 std
|
|
226
226
|
"std": "deprecated_tensor_std",
|
|
227
227
|
# 103 sub
|
|
@@ -390,4 +390,8 @@ deprecated_tensor_method_map = {
|
|
|
390
390
|
"count_nonzero": "deprecated_tensor_count_nonzero",
|
|
391
391
|
# 1028
|
|
392
392
|
"var": "deprecated_tensor_var",
|
|
393
|
+
# 1029
|
|
394
|
+
"real": "tensor_real",
|
|
395
|
+
# 1030
|
|
396
|
+
"imag": "tensor_imag",
|
|
393
397
|
}
|
|
@@ -83,8 +83,8 @@ parse_augassign_object_map = {
|
|
|
83
83
|
ast.Mult: (trope_ns, 'augassign_mul', '*='),
|
|
84
84
|
ast.Div: (trope_ns, 'augassign_div', '/='),
|
|
85
85
|
ast.FloorDiv: (trope_ns, 'augassign_floordiv', '//='),
|
|
86
|
+
ast.Mod: (trope_ns, 'augassign_mod', '%='),
|
|
86
87
|
|
|
87
|
-
ast.Mod: (trope_ns, 'mod', '%'),
|
|
88
88
|
ast.Pow: (trope_ns, 'pow', '**'),
|
|
89
89
|
ast.MatMult: (trope_ns, 'matmul', '@'),
|
|
90
90
|
ast.LShift: (trope_ns, 'lshift', '<<'),
|
|
@@ -28,7 +28,7 @@ from mindspore.ops.composite.base import _append, _insert, _pop, _list_clear, _r
|
|
|
28
28
|
_extend, _dict_setitem, _dict_clear, _haskey, _update, _fromkeys
|
|
29
29
|
from mindspore.ops.operations._sequence_ops import TensorToTuple
|
|
30
30
|
from mindspore.ops.auto_generate import trace_v2_op, inplace_addmm_op, inplace_index_put_op, inplace_normal_op, \
|
|
31
|
-
inplace_index_add_op
|
|
31
|
+
inplace_index_add_op, inplace_sign_op
|
|
32
32
|
from mindspore.ops.auto_generate import inplace_copy_op, inplace_uniform_op, inplace_erfinv_op
|
|
33
33
|
from mindspore.ops.auto_generate import inplace_scatter_add as inplace_scatter_add_
|
|
34
34
|
from mindspore.ops.auto_generate import inplace_exponential_op
|
|
@@ -4363,6 +4363,13 @@ def sign(x):
|
|
|
4363
4363
|
return F.sign(x)
|
|
4364
4364
|
|
|
4365
4365
|
|
|
4366
|
+
def sign_(input):
|
|
4367
|
+
r"""
|
|
4368
|
+
In-place version of :func:`mindspore.mint.sign`.
|
|
4369
|
+
"""
|
|
4370
|
+
return inplace_sign_op(input)
|
|
4371
|
+
|
|
4372
|
+
|
|
4366
4373
|
def signbit(x):
|
|
4367
4374
|
"""
|
|
4368
4375
|
For details, please refer to :func:`mindspore.ops.signbit`.
|
|
@@ -42,6 +42,7 @@ from mindspore.ops.composite.multitype_ops.sub_impl import augassign_sub
|
|
|
42
42
|
from mindspore.ops.composite.multitype_ops.mul_impl import augassign_mul
|
|
43
43
|
from mindspore.ops.composite.multitype_ops.div_impl import augassign_div
|
|
44
44
|
from mindspore.ops.composite.multitype_ops.floordiv_impl import augassign_floordiv
|
|
45
|
+
from mindspore.ops.composite.multitype_ops.mod_impl import augassign_mod
|
|
45
46
|
|
|
46
47
|
from ...common import mutable
|
|
47
48
|
|
|
@@ -51,7 +52,7 @@ __all__ = ['add', 'sub', 'mul', 'truediv', 'floordiv', 'mod', 'eq', 'ne', 'lt',
|
|
|
51
52
|
'bool', 'getattr', 'setattr', 'hasattr', 'len', 'iter', 'next', 'pow', 'range', 'map', 'zip',
|
|
52
53
|
'partial', 'print', 'enumerate', 'isinstance', 'filter', 'abs', 'round', 'mutable',
|
|
53
54
|
'max', 'min', 'sum', 'list', 'tuple',
|
|
54
|
-
'augassign_add', 'augassign_sub', 'augassign_mul', 'augassign_div', 'augassign_floordiv']
|
|
55
|
+
'augassign_add', 'augassign_sub', 'augassign_mul', 'augassign_div', 'augassign_floordiv', 'augassign_mod']
|
|
55
56
|
|
|
56
57
|
|
|
57
58
|
def MakeTuple(*elts): # pragma: no cover
|
|
@@ -53,7 +53,6 @@ from mindspore.train.serialization import _exec_save, load, export_split_mindir,
|
|
|
53
53
|
_generate_front_info_for_param_data_file, _get_data_file, _encrypt_data, _split_save, _save_mindir_together, \
|
|
54
54
|
_load_into_param_dict
|
|
55
55
|
from mindspore.parallel import _cost_model_context
|
|
56
|
-
from mindspore.parallel._offload_context import offload_context
|
|
57
56
|
from mindspore.parallel._utils import _is_in_data_parallel_mode
|
|
58
57
|
from mindspore.run_check._check_version import check_version_and_env_config
|
|
59
58
|
from mindspore.dataset.callback.ds_callback import DSCallback, WaitedDSCallback
|
|
@@ -136,17 +135,13 @@ from mindspore.hal.event import Event
|
|
|
136
135
|
from mindspore.hal.memory import memory_stats, memory_reserved, max_memory_allocated, reset_peak_memory_stats, \
|
|
137
136
|
memory_summary, memory_allocated, max_memory_reserved, reset_max_memory_allocated, reset_max_memory_reserved
|
|
138
137
|
from mindspore.multiprocessing import Process
|
|
139
|
-
from mindspore.mindrecord.shardsegment import ShardSegment
|
|
140
|
-
from mindspore.mindrecord.shardreader import ShardReader
|
|
141
|
-
from mindspore.mindrecord.shardindexgenerator import ShardIndexGenerator
|
|
142
|
-
from mindspore.mindrecord.shardwriter import ShardWriter
|
|
143
|
-
from mindspore.mindrecord.shardheader import ShardHeader
|
|
144
138
|
from mindspore.mindrecord.config import encrypt, decrypt
|
|
145
|
-
from mindspore.
|
|
139
|
+
from mindspore.mindrecord import FileWriter
|
|
140
|
+
from mindspore.mindrecord import FileReader
|
|
141
|
+
from mindspore.mindrecord import MindPage
|
|
146
142
|
from mindspore.parallel._ps_context import ps_context
|
|
147
143
|
from mindspore.parallel.algo_parameter_config import _AlgoParameterConfig
|
|
148
|
-
from mindspore.parallel._utils import _reset_op_id
|
|
149
|
-
from mindspore.parallel._recovery_context import recovery_context
|
|
144
|
+
from mindspore.parallel._utils import _reset_op_id
|
|
150
145
|
from mindspore.parallel._auto_parallel_context import _AutoParallelContext
|
|
151
146
|
from mindspore.common.api import ms_memory_recycle
|
|
152
147
|
from mindspore.context import _Context
|
|
@@ -461,10 +456,6 @@ _func_map = {
|
|
|
461
456
|
function_id(dict.items): FUNC_KEY_DICT_ITEMS,
|
|
462
457
|
|
|
463
458
|
# instancemethod
|
|
464
|
-
function_id(Tensor_._flatten_tensors): FUNC_KEY_BUILTIN_FUNC, # pylint: disable=protected-access
|
|
465
|
-
function_id(Tensor_._is_flattened): FUNC_KEY_BUILTIN_FUNC, # pylint: disable=protected-access
|
|
466
|
-
function_id(Tensor_._get_flattened_tensors): FUNC_KEY_BUILTIN_FUNC, # pylint: disable=protected-access
|
|
467
|
-
function_id(Tensor_._get_fusion_size): FUNC_KEY_BUILTIN_FUNC, # pylint: disable=protected-access
|
|
468
459
|
function_id(Tensor_._is_test_stub): FUNC_KEY_BUILTIN_FUNC, # pylint: disable=protected-access
|
|
469
460
|
function_id(Tensor_.__str__): FUNC_KEY_BUILTIN_FUNC, # pylint: disable=protected-access
|
|
470
461
|
function_id(Tensor_.__repr__): FUNC_KEY_BUILTIN_FUNC, # pylint: disable=protected-access
|
|
@@ -496,7 +487,6 @@ _func_map = {
|
|
|
496
487
|
function_id(_set_elegant_exit_handle): FUNC_KEY_PIJIT_CONSTEXPR,
|
|
497
488
|
function_id(_cost_model_context.get_cost_model_context): FUNC_KEY_PIJIT_CONSTEXPR,
|
|
498
489
|
function_id(Stream.__repr__): FUNC_KEY_PIJIT_CONSTEXPR,
|
|
499
|
-
function_id(offload_context): FUNC_KEY_PIJIT_CONSTEXPR,
|
|
500
490
|
function_id(_is_in_data_parallel_mode): FUNC_KEY_PIJIT_CONSTEXPR,
|
|
501
491
|
function_id(check_version_and_env_config): FUNC_KEY_PIJIT_CONSTEXPR,
|
|
502
492
|
function_id(Tensor.tolist): FUNC_KEY_PIJIT_CONSTEXPR,
|
|
@@ -572,19 +562,14 @@ _func_map = {
|
|
|
572
562
|
function_id(reset_max_memory_allocated): FUNC_KEY_PIJIT_FORBIDDEN,
|
|
573
563
|
function_id(Process.run): FUNC_KEY_PIJIT_FORBIDDEN,
|
|
574
564
|
function_id(Process.start): FUNC_KEY_PIJIT_FORBIDDEN,
|
|
575
|
-
function_id(ShardSegment.__init__): FUNC_KEY_PIJIT_FORBIDDEN,
|
|
576
|
-
function_id(ShardReader.__init__): FUNC_KEY_PIJIT_FORBIDDEN,
|
|
577
|
-
function_id(ShardIndexGenerator.__init__): FUNC_KEY_PIJIT_FORBIDDEN,
|
|
578
|
-
function_id(ShardWriter.__init__): FUNC_KEY_PIJIT_FORBIDDEN,
|
|
579
|
-
function_id(ShardHeader.__init__): FUNC_KEY_PIJIT_FORBIDDEN,
|
|
580
565
|
function_id(encrypt): FUNC_KEY_PIJIT_FORBIDDEN,
|
|
581
566
|
function_id(decrypt): FUNC_KEY_PIJIT_FORBIDDEN,
|
|
582
|
-
function_id(
|
|
567
|
+
function_id(FileWriter.__init__): FUNC_KEY_PIJIT_FORBIDDEN,
|
|
568
|
+
function_id(FileReader.__init__): FUNC_KEY_PIJIT_FORBIDDEN,
|
|
569
|
+
function_id(MindPage.__init__): FUNC_KEY_PIJIT_FORBIDDEN,
|
|
583
570
|
function_id(ps_context): FUNC_KEY_PIJIT_FORBIDDEN,
|
|
584
571
|
function_id(_AlgoParameterConfig.__init__): FUNC_KEY_PIJIT_FORBIDDEN,
|
|
585
572
|
function_id(_reset_op_id): FUNC_KEY_PIJIT_FORBIDDEN,
|
|
586
|
-
function_id(_reset_op_id_with_offset): FUNC_KEY_PIJIT_FORBIDDEN,
|
|
587
|
-
function_id(recovery_context): FUNC_KEY_PIJIT_FORBIDDEN,
|
|
588
573
|
function_id(_AutoParallelContext.__init__): FUNC_KEY_PIJIT_FORBIDDEN,
|
|
589
574
|
function_id(ms_memory_recycle): FUNC_KEY_PIJIT_FORBIDDEN,
|
|
590
575
|
function_id(_Context.__init__): FUNC_KEY_PIJIT_FORBIDDEN,
|
mindspore/boost/base.py
CHANGED
|
@@ -16,6 +16,7 @@
|
|
|
16
16
|
from __future__ import absolute_import
|
|
17
17
|
|
|
18
18
|
import os
|
|
19
|
+
import re
|
|
19
20
|
import time
|
|
20
21
|
import math
|
|
21
22
|
import copy
|
|
@@ -29,7 +30,6 @@ from mindspore.communication.management import get_rank, get_group_size
|
|
|
29
30
|
from mindspore.train.serialization import load_checkpoint
|
|
30
31
|
from mindspore.boost.less_batch_normalization import CommonHeadLastFN
|
|
31
32
|
|
|
32
|
-
|
|
33
33
|
__all__ = ["OptimizerProcess", "ParameterProcess"]
|
|
34
34
|
|
|
35
35
|
|
|
@@ -65,6 +65,7 @@ class OptimizerProcess:
|
|
|
65
65
|
>>> optimizer_process.add_grad_centralization(network)
|
|
66
66
|
>>> optimizer = optimizer_process.generate_new_optimizer()
|
|
67
67
|
"""
|
|
68
|
+
|
|
68
69
|
def __init__(self, opt):
|
|
69
70
|
if isinstance(opt, LARS):
|
|
70
71
|
self.is_lars = True
|
|
@@ -116,7 +117,7 @@ class OptimizerProcess:
|
|
|
116
117
|
if 'beta' not in param.name and 'gamma' not in param.name and 'bias' not in param.name:
|
|
117
118
|
param_cell = params_dict[id(param)]
|
|
118
119
|
if (isinstance(param_cell, nn.Conv2d) and param_cell.group > 1) or \
|
|
119
|
-
|
|
120
|
+
isinstance(param_cell, CommonHeadLastFN):
|
|
120
121
|
params_value.append(param)
|
|
121
122
|
else:
|
|
122
123
|
params_gc_value.append(param)
|
|
@@ -203,6 +204,7 @@ class ParameterProcess:
|
|
|
203
204
|
>>> new_parameter = network.trainable_params()[:1]
|
|
204
205
|
>>> group_params = ParameterProcess.generate_group_params(new_parameter, network.trainable_params())
|
|
205
206
|
"""
|
|
207
|
+
|
|
206
208
|
def __init__(self):
|
|
207
209
|
self._parameter_indices = 1
|
|
208
210
|
|
|
@@ -486,6 +488,30 @@ def _svd_flip(u, v, transpose=True):
|
|
|
486
488
|
return u, v
|
|
487
489
|
|
|
488
490
|
|
|
491
|
+
def _check_path(path):
|
|
492
|
+
"""check if the path is valid"""
|
|
493
|
+
path_white_list_regex = re.compile(r"[^_A-Za-z0-9/.-]")
|
|
494
|
+
if not isinstance(path, str):
|
|
495
|
+
raise TypeError('Path must be str, not {}.'.format(type(path).__name__))
|
|
496
|
+
if not path:
|
|
497
|
+
raise ValueError("The value of the path cannot be empty.")
|
|
498
|
+
if path_white_list_regex.search(path):
|
|
499
|
+
raise ValueError(
|
|
500
|
+
"Input path contains invalid characters.")
|
|
501
|
+
path = os.path.expanduser(path) # Consider paths starting with "~"
|
|
502
|
+
if os.path.islink(os.path.abspath(path)): # when checking link, get rid of the "/" at the path tail if any
|
|
503
|
+
raise ValueError("The value of the path cannot be soft link: {}.".format(path))
|
|
504
|
+
|
|
505
|
+
real_path = os.path.realpath(path)
|
|
506
|
+
|
|
507
|
+
if len(real_path) > 4096:
|
|
508
|
+
raise ValueError("The length of file path should be less than 4096.")
|
|
509
|
+
|
|
510
|
+
if real_path != path and path_white_list_regex.search(real_path):
|
|
511
|
+
raise ValueError(
|
|
512
|
+
"Input path contains invalid characters.")
|
|
513
|
+
|
|
514
|
+
|
|
489
515
|
def _save_local_pca_mat(pca_mat, full_pca_mat_path, n_component):
|
|
490
516
|
"""
|
|
491
517
|
save pca mat.
|
|
@@ -495,6 +521,7 @@ def _save_local_pca_mat(pca_mat, full_pca_mat_path, n_component):
|
|
|
495
521
|
full_pca_mat_path (str): the path of full pca mat.
|
|
496
522
|
n_component (int): pca component.
|
|
497
523
|
"""
|
|
524
|
+
_check_path(full_pca_mat_path)
|
|
498
525
|
rank_size = get_group_size()
|
|
499
526
|
local_dim = math.ceil(n_component // rank_size)
|
|
500
527
|
for rank_id in range(rank_size):
|
mindspore/common/_decorator.py
CHANGED
|
@@ -22,13 +22,14 @@ from mindspore import log
|
|
|
22
22
|
DEPRECATE_SET = set()
|
|
23
23
|
|
|
24
24
|
|
|
25
|
-
def deprecated(version, substitute, use_substitute_name=False):
|
|
25
|
+
def deprecated(version, substitute, use_substitute_name=False, module_prefix=""):
|
|
26
26
|
"""deprecated warning
|
|
27
27
|
|
|
28
28
|
Args:
|
|
29
29
|
version (str): version that the operator or function is deprecated.
|
|
30
30
|
substitute (str): the substitute name for deprecated operator or function.
|
|
31
31
|
use_substitute_name (bool): flag for whether to use substitute name for deprecated operator or function
|
|
32
|
+
module_prefix (str): the module prefix of the deprecated api, such as 'mindspore.'.
|
|
32
33
|
"""
|
|
33
34
|
|
|
34
35
|
def decorate(func):
|
|
@@ -37,7 +38,7 @@ def deprecated(version, substitute, use_substitute_name=False):
|
|
|
37
38
|
cls = getattr(args[0], "__class__", None) if args else None
|
|
38
39
|
name = cls.__name__ if cls else func.__name__
|
|
39
40
|
if name + version not in DEPRECATE_SET:
|
|
40
|
-
log.warning(f"'{name}' is deprecated from version {version} and "
|
|
41
|
+
log.warning(f"'{module_prefix}{name}' is deprecated from version {version} and "
|
|
41
42
|
f"will be removed in a future version, use '{substitute}' instead.")
|
|
42
43
|
DEPRECATE_SET.add(name + version)
|
|
43
44
|
if cls and use_substitute_name:
|
|
@@ -24,7 +24,9 @@ from mindspore.common.tensor import Tensor
|
|
|
24
24
|
|
|
25
25
|
class _Function(FunctionBase_):
|
|
26
26
|
"""
|
|
27
|
-
A Class provides the ability to custom autograd function.
|
|
27
|
+
A Class provides the ability to custom autograd function. The api refers
|
|
28
|
+
to the following files from pytorch:
|
|
29
|
+
https://github.com/pytorch/pytorch/blob/main/torch/autograd/function.py
|
|
28
30
|
|
|
29
31
|
Note:
|
|
30
32
|
It is only supported in pynative mode.
|
|
@@ -14,4 +14,4 @@
|
|
|
14
14
|
# ============================================================================
|
|
15
15
|
"""Add tensor cpp methods for stub tensor"""
|
|
16
16
|
|
|
17
|
-
tensor_cpp_methods = ['atan2', 'arctan2', 'greater_equal', 'ge', 'acos', 'arccos', 'argmin', 'index_select', 'erfc', 'addcdiv', 'matmul', 'eq', 'put_', 'floor', 'addmv', 'type_as', 'reshape', 'nansum', 'logical_xor', 'logsumexp', 'topk', 'view_as', 'scatter_add', 'tan', 'narrow', 'sin', 'unique', 'copy_', 'baddbmm', 'logical_or', 'scatter_', 'logical_and', 'fill_diagonal_', 'frac', 't', 'median', 'square', 'masked_fill', 'less_equal', 'le', 'true_divide', 'outer', 'sigmoid', 'sinc', 'clamp', 'clip', 'argmax', 'atanh', 'arctanh', 'std', 'div_', '__itruediv__', 'sinh', 'subtract', '
|
|
17
|
+
tensor_cpp_methods = ['atan2', 'arctan2', 'greater_equal', 'ge', 'acos', 'arccos', 'argmin', 'index_select', 'erfc', 'addcdiv', 'matmul', 'eq', 'put_', 'floor', 'addmv', 'type_as', 'reshape', 'broadcast_to', 'nansum', 'logical_xor', 'logsumexp', 'sigmoid_', 'topk', 'view_as', 'scatter_add', 'tan', 'narrow', 'sin', 'unique', 'copy_', 'baddbmm', 'logical_or', 'scatter_', 'logical_and', 'fill_diagonal_', 'frac', 't', 'median', 'square', 'masked_fill', 'less_equal', 'le', 'true_divide', 'outer', 'sigmoid', 'sinc', 'clamp', 'clip', 'argmax', 'atanh', 'arctanh', 'std', 'div_', '__itruediv__', 'sinh', 'subtract', 'sub', '__sub__', 'floor_divide', 'rsqrt', 'not_equal', 'ne', 'dot', 'max', 'clone', 'acosh', 'arccosh', 'unsqueeze', 'lerp', 'masked_fill_', 'ceil', 'masked_select', 'index_add', 'real', 'mul_', '__imul__', 'chunk', 'index_copy_', 'unbind', 'log10', 'fmod', 'scatter', 'transpose', 'mm', 'min', 'bitwise_or', '__or__', 'masked_scatter_', 'diag', 'greater', 'gt', 'cumsum', 'expm1', 'abs', 'absolute', '__abs__', 'view', 'logical_not', 'log', 'gather', 'sqrt', 'where', 'inverse', 'bitwise_and', '__and__', 'new_full', 'flatten', 'argsort', 'xlogy', 'permute', 'triu', 'tril', 'neg', 'negative', 'erf', 'isfinite', 'repeat', 'index_fill_', 'add', '__add__', 'logaddexp', 'to', 'div', 'divide', 'floor_divide_', '__ifloordiv__', 'bincount', 'sum', 'new_zeros', 'roll', 'var', 'exp_', 'nan_to_num', 'isneginf', 'addbmm', 'minimum', 'remainder_', '__imod__', 'isinf', 'exp', 'pow', '__pow__', 'tanh', 'gcd', 'add_', '__iadd__', 'log1p', 'expand_as', 'isclose', 'allclose', 'bitwise_xor', '__xor__', 'histc', 'cosh', 'asin', 'arcsin', 'new_empty', 'maximum', 'atan', 'arctan', 'any', 'imag', 'asinh', 'arcsinh', 'masked_scatter', 'trunc', 'mul', 'select', 'bitwise_not', 'fill_', 'all', 'mean', 'sort', 'hardshrink', 'kthvalue', 'addmm', 'remainder', 'reciprocal', 'round', 'log_', 'cos', 'split', 'tile', 'squeeze', 'new_ones', 'logaddexp2', 'index', '__mod__', 'log2', 'prod', 'less', 'lt', 'sub_', '__isub__', 'take', 'count_nonzero', 'repeat_interleave']
|