mindspore 2.7.0__cp310-cp310-win_amd64.whl → 2.7.1__cp310-cp310-win_amd64.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.

Potentially problematic release.


This version of mindspore might be problematic. Click here for more details.

Files changed (290) hide show
  1. mindspore/.commit_id +1 -1
  2. mindspore/__init__.py +4 -1
  3. mindspore/_c_dataengine.cp310-win_amd64.pyd +0 -0
  4. mindspore/_c_expression.cp310-win_amd64.pyd +0 -0
  5. mindspore/_c_mindrecord.cp310-win_amd64.pyd +0 -0
  6. mindspore/_extends/parse/compile_config.py +24 -1
  7. mindspore/_extends/parse/deprecated/deprecated_tensor_method.py +6 -2
  8. mindspore/_extends/parse/resources.py +1 -1
  9. mindspore/_extends/parse/standard_method.py +8 -1
  10. mindspore/_extends/parse/trope.py +2 -1
  11. mindspore/_extends/pijit/pijit_func_white_list.py +7 -22
  12. mindspore/avcodec-59.dll +0 -0
  13. mindspore/avdevice-59.dll +0 -0
  14. mindspore/avfilter-8.dll +0 -0
  15. mindspore/avformat-59.dll +0 -0
  16. mindspore/avutil-57.dll +0 -0
  17. mindspore/boost/base.py +29 -2
  18. mindspore/common/_decorator.py +3 -2
  19. mindspore/common/_grad_function.py +3 -1
  20. mindspore/common/_tensor_cpp_method.py +1 -1
  21. mindspore/common/_tensor_docs.py +275 -64
  22. mindspore/common/_utils.py +0 -44
  23. mindspore/common/api.py +285 -35
  24. mindspore/common/dump.py +7 -108
  25. mindspore/common/dynamic_shape/auto_dynamic_shape.py +1 -3
  26. mindspore/common/hook_handle.py +60 -0
  27. mindspore/common/jit_config.py +5 -1
  28. mindspore/common/jit_trace.py +27 -12
  29. mindspore/common/lazy_inline.py +5 -3
  30. mindspore/common/parameter.py +13 -107
  31. mindspore/common/recompute.py +4 -11
  32. mindspore/common/tensor.py +16 -169
  33. mindspore/communication/_comm_helper.py +11 -1
  34. mindspore/communication/comm_func.py +138 -4
  35. mindspore/communication/management.py +85 -1
  36. mindspore/config/op_info.config +0 -15
  37. mindspore/context.py +5 -85
  38. mindspore/dataset/engine/datasets.py +8 -4
  39. mindspore/dataset/engine/datasets_vision.py +1 -1
  40. mindspore/dataset/engine/validators.py +1 -15
  41. mindspore/dnnl.dll +0 -0
  42. mindspore/{experimental/llm_boost/ascend_native → graph}/__init__.py +7 -7
  43. mindspore/graph/custom_pass.py +55 -0
  44. mindspore/include/dataset/execute.h +2 -2
  45. mindspore/jpeg62.dll +0 -0
  46. mindspore/mindrecord/__init__.py +3 -3
  47. mindspore/mindrecord/common/exceptions.py +1 -0
  48. mindspore/mindrecord/config.py +1 -1
  49. mindspore/{parallel/mpi → mindrecord/core}/__init__.py +4 -1
  50. mindspore/mindrecord/{shardheader.py → core/shardheader.py} +2 -1
  51. mindspore/mindrecord/{shardindexgenerator.py → core/shardindexgenerator.py} +1 -1
  52. mindspore/mindrecord/{shardreader.py → core/shardreader.py} +2 -1
  53. mindspore/mindrecord/{shardsegment.py → core/shardsegment.py} +2 -2
  54. mindspore/mindrecord/{shardutils.py → core/shardutils.py} +1 -1
  55. mindspore/mindrecord/{shardwriter.py → core/shardwriter.py} +1 -1
  56. mindspore/mindrecord/filereader.py +4 -4
  57. mindspore/mindrecord/filewriter.py +5 -5
  58. mindspore/mindrecord/mindpage.py +2 -2
  59. mindspore/mindrecord/tools/cifar10.py +1 -1
  60. mindspore/mindrecord/tools/cifar100.py +1 -1
  61. mindspore/mindrecord/tools/cifar100_to_mr.py +1 -1
  62. mindspore/mindrecord/tools/cifar10_to_mr.py +1 -1
  63. mindspore/mindrecord/tools/csv_to_mr.py +1 -1
  64. mindspore/mindrecord/tools/imagenet_to_mr.py +1 -1
  65. mindspore/mindrecord/tools/mnist_to_mr.py +1 -1
  66. mindspore/mindrecord/tools/tfrecord_to_mr.py +1 -1
  67. mindspore/mindspore_backend_common.dll +0 -0
  68. mindspore/mindspore_backend_manager.dll +0 -0
  69. mindspore/mindspore_cluster.dll +0 -0
  70. mindspore/mindspore_common.dll +0 -0
  71. mindspore/mindspore_core.dll +0 -0
  72. mindspore/mindspore_cpu.dll +0 -0
  73. mindspore/mindspore_dump.dll +0 -0
  74. mindspore/mindspore_frontend.dll +0 -0
  75. mindspore/mindspore_glog.dll +0 -0
  76. mindspore/mindspore_hardware_abstract.dll +0 -0
  77. mindspore/mindspore_memory_pool.dll +0 -0
  78. mindspore/mindspore_ms_backend.dll +0 -0
  79. mindspore/mindspore_ops.dll +0 -0
  80. mindspore/{mindspore_ops_host.dll → mindspore_ops_cpu.dll} +0 -0
  81. mindspore/mindspore_profiler.dll +0 -0
  82. mindspore/mindspore_pyboost.dll +0 -0
  83. mindspore/mindspore_pynative.dll +0 -0
  84. mindspore/mindspore_runtime_pipeline.dll +0 -0
  85. mindspore/mindspore_runtime_utils.dll +0 -0
  86. mindspore/mindspore_tools.dll +0 -0
  87. mindspore/mint/__init__.py +15 -10
  88. mindspore/mint/distributed/distributed.py +182 -62
  89. mindspore/mint/nn/__init__.py +2 -16
  90. mindspore/mint/nn/functional.py +4 -110
  91. mindspore/mint/nn/layer/__init__.py +0 -2
  92. mindspore/mint/nn/layer/activation.py +0 -6
  93. mindspore/mint/nn/layer/basic.py +0 -47
  94. mindspore/mint/nn/layer/conv.py +4 -4
  95. mindspore/mint/nn/layer/normalization.py +8 -13
  96. mindspore/mint/nn/layer/pooling.py +0 -4
  97. mindspore/nn/__init__.py +1 -3
  98. mindspore/nn/cell.py +16 -66
  99. mindspore/nn/layer/basic.py +49 -1
  100. mindspore/nn/layer/container.py +16 -0
  101. mindspore/nn/layer/embedding.py +4 -169
  102. mindspore/nn/layer/normalization.py +2 -1
  103. mindspore/nn/layer/thor_layer.py +4 -85
  104. mindspore/nn/optim/ada_grad.py +0 -1
  105. mindspore/nn/optim/adafactor.py +0 -1
  106. mindspore/nn/optim/adam.py +31 -124
  107. mindspore/nn/optim/adamax.py +0 -1
  108. mindspore/nn/optim/asgd.py +0 -1
  109. mindspore/nn/optim/ftrl.py +8 -102
  110. mindspore/nn/optim/lamb.py +0 -1
  111. mindspore/nn/optim/lars.py +0 -3
  112. mindspore/nn/optim/lazyadam.py +25 -218
  113. mindspore/nn/optim/momentum.py +5 -43
  114. mindspore/nn/optim/optimizer.py +6 -55
  115. mindspore/nn/optim/proximal_ada_grad.py +0 -1
  116. mindspore/nn/optim/rmsprop.py +0 -1
  117. mindspore/nn/optim/rprop.py +0 -1
  118. mindspore/nn/optim/sgd.py +0 -1
  119. mindspore/nn/optim/tft_wrapper.py +0 -1
  120. mindspore/nn/optim/thor.py +0 -2
  121. mindspore/nn/probability/bijector/bijector.py +7 -8
  122. mindspore/nn/probability/bijector/gumbel_cdf.py +2 -2
  123. mindspore/nn/probability/bijector/power_transform.py +20 -21
  124. mindspore/nn/probability/bijector/scalar_affine.py +5 -5
  125. mindspore/nn/probability/bijector/softplus.py +13 -14
  126. mindspore/nn/wrap/grad_reducer.py +4 -74
  127. mindspore/numpy/array_creations.py +2 -2
  128. mindspore/numpy/fft.py +9 -9
  129. mindspore/{nn/reinforcement → onnx}/__init__.py +5 -8
  130. mindspore/onnx/onnx_export.py +137 -0
  131. mindspore/opencv_core4110.dll +0 -0
  132. mindspore/opencv_imgcodecs4110.dll +0 -0
  133. mindspore/{opencv_imgproc452.dll → opencv_imgproc4110.dll} +0 -0
  134. mindspore/ops/__init__.py +2 -0
  135. mindspore/ops/_grad_experimental/grad_comm_ops.py +38 -2
  136. mindspore/ops/_op_impl/aicpu/__init__.py +0 -10
  137. mindspore/ops/_op_impl/cpu/__init__.py +0 -5
  138. mindspore/ops/auto_generate/cpp_create_prim_instance_helper.py +16 -22
  139. mindspore/ops/auto_generate/gen_extend_func.py +2 -7
  140. mindspore/ops/auto_generate/gen_ops_def.py +98 -141
  141. mindspore/ops/auto_generate/gen_ops_prim.py +12708 -12686
  142. mindspore/ops/communication.py +97 -0
  143. mindspore/ops/composite/__init__.py +5 -2
  144. mindspore/ops/composite/base.py +15 -1
  145. mindspore/ops/composite/multitype_ops/__init__.py +3 -1
  146. mindspore/ops/composite/multitype_ops/_compile_utils.py +150 -8
  147. mindspore/ops/composite/multitype_ops/add_impl.py +7 -0
  148. mindspore/ops/composite/multitype_ops/mod_impl.py +27 -0
  149. mindspore/ops/function/__init__.py +1 -0
  150. mindspore/ops/function/array_func.py +14 -12
  151. mindspore/ops/function/comm_func.py +3883 -0
  152. mindspore/ops/function/debug_func.py +3 -4
  153. mindspore/ops/function/math_func.py +45 -54
  154. mindspore/ops/function/nn_func.py +75 -294
  155. mindspore/ops/function/random_func.py +9 -18
  156. mindspore/ops/functional.py +2 -0
  157. mindspore/ops/functional_overload.py +354 -18
  158. mindspore/ops/operations/__init__.py +2 -5
  159. mindspore/ops/operations/_custom_ops_utils.py +7 -9
  160. mindspore/ops/operations/_inner_ops.py +1 -38
  161. mindspore/ops/operations/_rl_inner_ops.py +0 -933
  162. mindspore/ops/operations/array_ops.py +1 -0
  163. mindspore/ops/operations/comm_ops.py +94 -2
  164. mindspore/ops/operations/custom_ops.py +228 -19
  165. mindspore/ops/operations/debug_ops.py +27 -29
  166. mindspore/ops/operations/manually_defined/ops_def.py +27 -306
  167. mindspore/ops/operations/nn_ops.py +2 -2
  168. mindspore/ops/operations/sparse_ops.py +0 -83
  169. mindspore/ops/primitive.py +1 -17
  170. mindspore/ops/tensor_method.py +72 -3
  171. mindspore/ops_generate/aclnn/aclnn_kernel_register_auto_cc_generator.py +5 -5
  172. mindspore/ops_generate/aclnn/gen_aclnn_implement.py +8 -8
  173. mindspore/ops_generate/api/functions_cc_generator.py +53 -4
  174. mindspore/ops_generate/api/tensor_func_reg_cpp_generator.py +25 -11
  175. mindspore/ops_generate/common/gen_constants.py +11 -10
  176. mindspore/ops_generate/common/op_proto.py +18 -1
  177. mindspore/ops_generate/common/template.py +102 -245
  178. mindspore/ops_generate/common/template_utils.py +212 -0
  179. mindspore/ops_generate/gen_custom_ops.py +69 -0
  180. mindspore/ops_generate/op_def/ops_def_cc_generator.py +78 -7
  181. mindspore/ops_generate/op_def_py/base_op_prim_py_generator.py +360 -0
  182. mindspore/ops_generate/op_def_py/custom_op_prim_py_generator.py +140 -0
  183. mindspore/ops_generate/op_def_py/op_def_py_generator.py +54 -7
  184. mindspore/ops_generate/op_def_py/op_prim_py_generator.py +5 -312
  185. mindspore/ops_generate/pyboost/auto_grad_impl_cc_generator.py +74 -17
  186. mindspore/ops_generate/pyboost/auto_grad_reg_cc_generator.py +22 -5
  187. mindspore/ops_generate/pyboost/op_template_parser.py +3 -2
  188. mindspore/ops_generate/pyboost/pyboost_functions_cpp_generator.py +21 -5
  189. mindspore/ops_generate/pyboost/pyboost_functions_h_generator.py +2 -2
  190. mindspore/ops_generate/pyboost/pyboost_functions_impl_cpp_generator.py +30 -10
  191. mindspore/ops_generate/pyboost/pyboost_grad_function_cpp_generator.py +10 -3
  192. mindspore/ops_generate/pyboost/pyboost_internal_kernel_info_adapter_generator.py +1 -1
  193. mindspore/ops_generate/pyboost/pyboost_native_grad_functions_generator.py +19 -9
  194. mindspore/ops_generate/pyboost/pyboost_op_cpp_code_generator.py +71 -28
  195. mindspore/ops_generate/pyboost/pyboost_overload_functions_cpp_generator.py +10 -9
  196. mindspore/ops_generate/pyboost/pyboost_utils.py +27 -16
  197. mindspore/ops_generate/resources/yaml_loader.py +13 -0
  198. mindspore/ops_generate/tensor_py_cc_generator.py +2 -2
  199. mindspore/parallel/_cell_wrapper.py +1 -1
  200. mindspore/parallel/_parallel_serialization.py +1 -4
  201. mindspore/parallel/_utils.py +29 -6
  202. mindspore/parallel/checkpoint_transform.py +18 -2
  203. mindspore/parallel/cluster/process_entity/_api.py +24 -32
  204. mindspore/parallel/cluster/process_entity/_utils.py +9 -5
  205. mindspore/{experimental/llm_boost/atb → parallel/distributed}/__init__.py +21 -23
  206. mindspore/parallel/distributed/distributed_data_parallel.py +393 -0
  207. mindspore/parallel/distributed/flatten_grad_buffer.py +295 -0
  208. mindspore/parallel/strategy.py +336 -0
  209. mindspore/parallel/transform_safetensors.py +117 -16
  210. mindspore/profiler/analysis/viewer/ascend_kernel_details_viewer.py +3 -0
  211. mindspore/profiler/analysis/viewer/ms_minddata_viewer.py +1 -1
  212. mindspore/profiler/common/constant.py +5 -0
  213. mindspore/profiler/common/file_manager.py +9 -0
  214. mindspore/profiler/common/msprof_cmd_tool.py +38 -2
  215. mindspore/profiler/common/path_manager.py +56 -24
  216. mindspore/profiler/common/profiler_context.py +2 -12
  217. mindspore/profiler/common/profiler_info.py +3 -3
  218. mindspore/profiler/common/profiler_path_manager.py +13 -0
  219. mindspore/profiler/common/util.py +30 -3
  220. mindspore/profiler/experimental_config.py +2 -1
  221. mindspore/profiler/platform/npu_profiler.py +33 -6
  222. mindspore/run_check/_check_version.py +108 -24
  223. mindspore/runtime/__init__.py +3 -2
  224. mindspore/runtime/executor.py +11 -3
  225. mindspore/runtime/memory.py +112 -0
  226. mindspore/swresample-4.dll +0 -0
  227. mindspore/swscale-6.dll +0 -0
  228. mindspore/tinyxml2.dll +0 -0
  229. mindspore/{experimental/llm_boost → tools}/__init__.py +5 -5
  230. mindspore/tools/data_dump.py +130 -0
  231. mindspore/tools/sdc_detect.py +91 -0
  232. mindspore/tools/stress_detect.py +63 -0
  233. mindspore/train/__init__.py +6 -6
  234. mindspore/train/_utils.py +5 -18
  235. mindspore/train/amp.py +6 -4
  236. mindspore/train/callback/_checkpoint.py +0 -9
  237. mindspore/train/callback/_train_fault_tolerance.py +69 -18
  238. mindspore/train/data_sink.py +1 -5
  239. mindspore/train/model.py +38 -211
  240. mindspore/train/serialization.py +126 -387
  241. mindspore/turbojpeg.dll +0 -0
  242. mindspore/utils/__init__.py +6 -3
  243. mindspore/utils/dlpack.py +92 -0
  244. mindspore/utils/dryrun.py +1 -1
  245. mindspore/utils/runtime_execution_order_check.py +10 -0
  246. mindspore/utils/sdc_detect.py +14 -12
  247. mindspore/utils/stress_detect.py +43 -0
  248. mindspore/utils/utils.py +144 -8
  249. mindspore/version.py +1 -1
  250. {mindspore-2.7.0.dist-info → mindspore-2.7.1.dist-info}/METADATA +3 -2
  251. {mindspore-2.7.0.dist-info → mindspore-2.7.1.dist-info}/RECORD +254 -267
  252. mindspore/experimental/llm_boost/ascend_native/llama_boost_ascend_native.py +0 -210
  253. mindspore/experimental/llm_boost/ascend_native/llm_boost.py +0 -52
  254. mindspore/experimental/llm_boost/atb/boost_base.py +0 -385
  255. mindspore/experimental/llm_boost/atb/llama_boost.py +0 -137
  256. mindspore/experimental/llm_boost/atb/qwen_boost.py +0 -124
  257. mindspore/experimental/llm_boost/register.py +0 -130
  258. mindspore/experimental/llm_boost/utils.py +0 -31
  259. mindspore/include/OWNERS +0 -7
  260. mindspore/mindspore_cpu_res_manager.dll +0 -0
  261. mindspore/mindspore_ops_kernel_common.dll +0 -0
  262. mindspore/mindspore_res_manager.dll +0 -0
  263. mindspore/nn/optim/_dist_optimizer_registry.py +0 -111
  264. mindspore/nn/reinforcement/_batch_read_write.py +0 -142
  265. mindspore/nn/reinforcement/_tensors_queue.py +0 -152
  266. mindspore/nn/reinforcement/tensor_array.py +0 -145
  267. mindspore/opencv_core452.dll +0 -0
  268. mindspore/opencv_imgcodecs452.dll +0 -0
  269. mindspore/ops/_op_impl/aicpu/priority_replay_buffer.py +0 -113
  270. mindspore/ops/_op_impl/aicpu/reservoir_replay_buffer.py +0 -96
  271. mindspore/ops/_op_impl/aicpu/sparse_cross.py +0 -42
  272. mindspore/ops/_op_impl/cpu/buffer_append.py +0 -28
  273. mindspore/ops/_op_impl/cpu/buffer_get.py +0 -28
  274. mindspore/ops/_op_impl/cpu/buffer_sample.py +0 -28
  275. mindspore/ops/_op_impl/cpu/priority_replay_buffer.py +0 -42
  276. mindspore/ops/operations/_tensor_array.py +0 -359
  277. mindspore/ops/operations/rl_ops.py +0 -288
  278. mindspore/parallel/_offload_context.py +0 -275
  279. mindspore/parallel/_recovery_context.py +0 -115
  280. mindspore/parallel/_transformer/__init__.py +0 -35
  281. mindspore/parallel/_transformer/layers.py +0 -765
  282. mindspore/parallel/_transformer/loss.py +0 -251
  283. mindspore/parallel/_transformer/moe.py +0 -693
  284. mindspore/parallel/_transformer/op_parallel_config.py +0 -222
  285. mindspore/parallel/_transformer/transformer.py +0 -3124
  286. mindspore/parallel/mpi/_mpi_config.py +0 -116
  287. mindspore/train/memory_profiling_pb2.py +0 -298
  288. {mindspore-2.7.0.dist-info → mindspore-2.7.1.dist-info}/WHEEL +0 -0
  289. {mindspore-2.7.0.dist-info → mindspore-2.7.1.dist-info}/entry_points.txt +0 -0
  290. {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]:7edec76e,[branch]:(HEAD->r2.7,origin/r2.7)''
1
+ __commit_id__ = ''[sha1]:71340184,[branch]:(HEAD->r2.7.1,origin/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, set_offload_context, get_offload_context, STRICT, COMPATIBLE, LAX
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.parallel.mpi._mpi_config import _MpiConfig
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, _reset_op_id_with_offset
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(_MpiConfig.__init__): FUNC_KEY_PIJIT_FORBIDDEN,
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/avcodec-59.dll CHANGED
Binary file
mindspore/avdevice-59.dll CHANGED
Binary file
mindspore/avfilter-8.dll CHANGED
Binary file
mindspore/avformat-59.dll CHANGED
Binary file
mindspore/avutil-57.dll CHANGED
Binary file
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
- isinstance(param_cell, CommonHeadLastFN):
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):
@@ -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 = ['abs', 'absolute', '__abs__', 'acos', 'arccos', 'acosh', 'arccosh', 'add', '__add__', 'addbmm', 'addcdiv', 'addmm', 'addmv', 'add_', '__iadd__', 'all', 'allclose', 'any', 'argmax', 'argmin', 'argsort', 'asin', 'arcsin', 'asinh', 'arcsinh', 'atan', 'arctan', 'atan2', 'arctan2', 'atanh', 'arctanh', 'baddbmm', 'bincount', 'bitwise_and', '__and__', 'bitwise_not', 'bitwise_or', '__or__', 'bitwise_xor', '__xor__', 'ceil', 'chunk', 'clamp', 'clip', 'clone', 'copy_', 'cos', 'cosh', 'count_nonzero', 'cumsum', 'diag', 'div', 'divide', 'div_', '__itruediv__', 'dot', 'eq', 'erf', 'erfc', 'exp', 'expand_as', 'expm1', 'exp_', 'fill_', 'fill_diagonal_', 'flatten', 'floor', 'floor_divide', 'floor_divide_', '__ifloordiv__', 'fmod', 'frac', 'gather', 'gcd', 'greater', 'gt', 'greater_equal', 'ge', 'hardshrink', 'histc', 'index_add', 'index_select', 'inverse', 'isclose', 'isfinite', 'isinf', 'isneginf', 'kthvalue', 'lerp', 'less', 'lt', 'less_equal', 'le', 'log', 'log10', 'log1p', 'log2', 'logaddexp', 'logaddexp2', 'logical_and', 'logical_not', 'logical_or', 'logical_xor', 'logsumexp', 'log_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_select', 'matmul', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mul', 'mul_', '__imul__', 'nansum', 'nan_to_num', 'narrow', 'neg', 'negative', 'new_empty', 'new_full', 'new_ones', 'new_zeros', 'not_equal', 'ne', 'outer', 'pow', '__pow__', 'prod', 'put_', 'reciprocal', 'remainder', 'remainder_', '__imod__', 'repeat', 'repeat_interleave', 'reshape', 'roll', 'round', 'rsqrt', 'scatter', 'scatter_', 'scatter_add', 'select', 'sigmoid', 'sin', 'sinc', 'sinh', 'sort', 'split', 'sqrt', 'square', 'std', 'sub', '__sub__', 'subtract', 'sub_', '__isub__', 'sum', 't', 'take', 'tan', 'tanh', 'tile', 'topk', 'transpose', 'tril', 'triu', 'true_divide', 'trunc', 'type_as', 'unbind', 'unique', 'unsqueeze', 'var', 'view_as', 'where', 'xlogy', '_to', '__mod__']
17
+ tensor_cpp_methods = ['abs', 'absolute', '__abs__', 'acos', 'arccos', 'acosh', 'arccosh', 'add', '__add__', 'addbmm', 'addcdiv', 'addmm', 'addmv', 'add_', '__iadd__', 'all', 'allclose', 'any', 'argmax', 'argmin', 'argsort', 'asin', 'arcsin', 'asinh', 'arcsinh', 'atan', 'arctan', 'atan2', 'arctan2', 'atanh', 'arctanh', 'baddbmm', 'bincount', 'bitwise_and', '__and__', 'bitwise_not', 'bitwise_or', '__or__', 'bitwise_xor', '__xor__', 'broadcast_to', 'ceil', 'chunk', 'clamp', 'clip', 'clone', 'copy_', 'cos', 'cosh', 'count_nonzero', 'cumsum', 'diag', 'div', 'divide', 'div_', '__itruediv__', 'dot', 'eq', 'erf', 'erfc', 'exp', 'expand_as', 'expm1', 'exp_', 'fill_', 'fill_diagonal_', 'flatten', 'floor', 'floor_divide', 'floor_divide_', '__ifloordiv__', 'fmod', 'frac', 'gather', 'gcd', 'greater', 'gt', 'greater_equal', 'ge', 'hardshrink', 'histc', 'imag', 'index', 'index_add', 'index_copy_', 'index_fill_', 'index_select', 'inverse', 'isclose', 'isfinite', 'isinf', 'isneginf', 'kthvalue', 'lerp', 'less', 'lt', 'less_equal', 'le', 'log', 'log10', 'log1p', 'log2', 'logaddexp', 'logaddexp2', 'logical_and', 'logical_not', 'logical_or', 'logical_xor', 'logsumexp', 'log_', 'masked_fill', 'masked_fill_', 'masked_scatter', 'masked_scatter_', 'masked_select', 'matmul', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mul', 'mul_', '__imul__', 'nansum', 'nan_to_num', 'narrow', 'neg', 'negative', 'new_empty', 'new_full', 'new_ones', 'new_zeros', 'not_equal', 'ne', 'outer', 'permute', 'pow', '__pow__', 'prod', 'put_', 'real', 'reciprocal', 'remainder', 'remainder_', '__imod__', 'repeat', 'repeat_interleave', 'reshape', 'roll', 'round', 'rsqrt', 'scatter', 'scatter_', 'scatter_add', 'select', 'sigmoid', 'sigmoid_', 'sin', 'sinc', 'sinh', 'sort', 'split', 'sqrt', 'square', 'squeeze', 'std', 'sub', '__sub__', 'subtract', 'sub_', '__isub__', 'sum', 't', 'take', 'tan', 'tanh', 'tile', 'to', 'topk', 'transpose', 'tril', 'triu', 'true_divide', 'trunc', 'type_as', 'unbind', 'unique', 'unsqueeze', 'var', 'view', 'view_as', 'where', 'xlogy', '__mod__']