mindspore 2.7.0rc1__cp311-cp311-macosx_11_0_arm64.whl → 2.7.1__cp311-cp311-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.

Potentially problematic release.


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

Files changed (367) hide show
  1. mindspore/.commit_id +1 -1
  2. mindspore/__init__.py +5 -2
  3. mindspore/_c_dataengine.cpython-311-darwin.so +0 -0
  4. mindspore/_c_expression.cpython-311-darwin.so +0 -0
  5. mindspore/_c_mindrecord.cpython-311-darwin.so +0 -0
  6. mindspore/_checkparam.py +2 -2
  7. mindspore/_extends/builtin_operations.py +3 -3
  8. mindspore/_extends/parallel_compile/akg_compiler/custom.py +1109 -0
  9. mindspore/_extends/parallel_compile/akg_compiler/gen_custom_op_files.py +1 -1
  10. mindspore/_extends/parse/__init__.py +3 -3
  11. mindspore/_extends/parse/compile_config.py +24 -1
  12. mindspore/_extends/parse/deprecated/deprecated_tensor_method.py +6 -3
  13. mindspore/_extends/parse/parser.py +28 -22
  14. mindspore/_extends/parse/resources.py +1 -1
  15. mindspore/_extends/parse/standard_method.py +23 -2
  16. mindspore/_extends/parse/trope.py +2 -1
  17. mindspore/_extends/pijit/pijit_func_white_list.py +9 -27
  18. mindspore/amp.py +0 -18
  19. mindspore/boost/base.py +29 -2
  20. mindspore/common/__init__.py +18 -12
  21. mindspore/common/_decorator.py +3 -2
  22. mindspore/common/_grad_function.py +3 -1
  23. mindspore/common/_tensor_cpp_method.py +1 -1
  24. mindspore/common/_tensor_docs.py +371 -96
  25. mindspore/common/_utils.py +7 -43
  26. mindspore/common/api.py +434 -135
  27. mindspore/common/dtype.py +98 -57
  28. mindspore/common/dump.py +7 -108
  29. mindspore/common/dynamic_shape/__init__.py +0 -0
  30. mindspore/common/{auto_dynamic_shape.py → dynamic_shape/auto_dynamic_shape.py} +15 -23
  31. mindspore/common/dynamic_shape/enable_dynamic.py +197 -0
  32. mindspore/common/file_system.py +59 -9
  33. mindspore/common/hook_handle.py +82 -3
  34. mindspore/common/jit_config.py +5 -1
  35. mindspore/common/jit_trace.py +27 -12
  36. mindspore/common/lazy_inline.py +5 -3
  37. mindspore/common/np_dtype.py +3 -3
  38. mindspore/common/parameter.py +17 -127
  39. mindspore/common/recompute.py +4 -13
  40. mindspore/common/tensor.py +50 -217
  41. mindspore/communication/_comm_helper.py +11 -1
  42. mindspore/communication/comm_func.py +138 -4
  43. mindspore/communication/management.py +85 -1
  44. mindspore/config/op_info.config +0 -15
  45. mindspore/context.py +20 -106
  46. mindspore/dataset/__init__.py +1 -1
  47. mindspore/dataset/audio/transforms.py +1 -1
  48. mindspore/dataset/core/config.py +35 -1
  49. mindspore/dataset/engine/datasets.py +338 -319
  50. mindspore/dataset/engine/datasets_user_defined.py +38 -22
  51. mindspore/dataset/engine/datasets_vision.py +1 -1
  52. mindspore/dataset/engine/validators.py +1 -15
  53. mindspore/dataset/transforms/c_transforms.py +2 -2
  54. mindspore/dataset/transforms/transforms.py +3 -3
  55. mindspore/dataset/vision/__init__.py +1 -1
  56. mindspore/dataset/vision/py_transforms.py +8 -8
  57. mindspore/dataset/vision/transforms.py +17 -5
  58. mindspore/dataset/vision/utils.py +632 -21
  59. mindspore/device_context/ascend/op_tuning.py +35 -1
  60. mindspore/{profiler/common/validator → graph}/__init__.py +9 -1
  61. mindspore/graph/custom_pass.py +55 -0
  62. mindspore/include/api/cell.h +28 -4
  63. mindspore/include/api/cfg.h +24 -7
  64. mindspore/include/api/context.h +1 -0
  65. mindspore/include/api/delegate.h +0 -2
  66. mindspore/include/api/dual_abi_helper.h +100 -19
  67. mindspore/include/api/graph.h +14 -1
  68. mindspore/include/api/kernel.h +16 -3
  69. mindspore/include/api/kernel_api.h +9 -1
  70. mindspore/include/api/metrics/accuracy.h +9 -0
  71. mindspore/include/api/model.h +5 -1
  72. mindspore/include/api/model_group.h +4 -0
  73. mindspore/include/api/model_parallel_runner.h +2 -0
  74. mindspore/include/api/status.h +48 -10
  75. mindspore/include/api/types.h +6 -1
  76. mindspore/include/dataset/constants.h +9 -0
  77. mindspore/include/dataset/execute.h +2 -2
  78. mindspore/lib/libmindspore_backend_common.dylib +0 -0
  79. mindspore/lib/libmindspore_backend_manager.dylib +0 -0
  80. mindspore/lib/libmindspore_cluster.dylib +0 -0
  81. mindspore/lib/libmindspore_common.dylib +0 -0
  82. mindspore/lib/libmindspore_core.dylib +0 -0
  83. mindspore/lib/libmindspore_cpu.dylib +0 -0
  84. mindspore/lib/libmindspore_dump.dylib +0 -0
  85. mindspore/lib/libmindspore_frontend.dylib +0 -0
  86. mindspore/lib/libmindspore_glog.0.dylib +0 -0
  87. mindspore/lib/libmindspore_gpr.15.dylib +0 -0
  88. mindspore/lib/libmindspore_grpc++.1.dylib +0 -0
  89. mindspore/lib/libmindspore_grpc.15.dylib +0 -0
  90. mindspore/lib/libmindspore_hardware_abstract.dylib +0 -0
  91. mindspore/lib/libmindspore_memory_pool.dylib +0 -0
  92. mindspore/lib/libmindspore_ms_backend.dylib +0 -0
  93. mindspore/lib/libmindspore_ops.dylib +0 -0
  94. mindspore/lib/libmindspore_profiler.dylib +0 -0
  95. mindspore/lib/libmindspore_pyboost.dylib +0 -0
  96. mindspore/lib/libmindspore_pynative.dylib +0 -0
  97. mindspore/lib/libmindspore_runtime_pipeline.dylib +0 -0
  98. mindspore/lib/libmindspore_runtime_utils.dylib +0 -0
  99. mindspore/lib/libmindspore_tools.dylib +0 -0
  100. mindspore/lib/libnnacl.dylib +0 -0
  101. mindspore/lib/libopencv_core.411.dylib +0 -0
  102. mindspore/lib/libopencv_imgcodecs.411.dylib +0 -0
  103. mindspore/lib/libopencv_imgproc.411.dylib +0 -0
  104. mindspore/lib/plugin/{libmindspore_ops_host.dylib → libmindspore_ops_cpu.dylib} +0 -0
  105. mindspore/mindrecord/__init__.py +3 -3
  106. mindspore/mindrecord/common/exceptions.py +1 -0
  107. mindspore/mindrecord/config.py +1 -1
  108. mindspore/{parallel/mpi → mindrecord/core}/__init__.py +4 -1
  109. mindspore/mindrecord/{shardheader.py → core/shardheader.py} +2 -1
  110. mindspore/mindrecord/{shardindexgenerator.py → core/shardindexgenerator.py} +1 -1
  111. mindspore/mindrecord/{shardreader.py → core/shardreader.py} +2 -1
  112. mindspore/mindrecord/{shardsegment.py → core/shardsegment.py} +2 -2
  113. mindspore/mindrecord/{shardutils.py → core/shardutils.py} +1 -1
  114. mindspore/mindrecord/{shardwriter.py → core/shardwriter.py} +1 -1
  115. mindspore/mindrecord/filereader.py +4 -4
  116. mindspore/mindrecord/filewriter.py +5 -5
  117. mindspore/mindrecord/mindpage.py +2 -2
  118. mindspore/mindrecord/tools/cifar10.py +4 -3
  119. mindspore/mindrecord/tools/cifar100.py +1 -1
  120. mindspore/mindrecord/tools/cifar100_to_mr.py +1 -1
  121. mindspore/mindrecord/tools/cifar10_to_mr.py +6 -6
  122. mindspore/mindrecord/tools/csv_to_mr.py +1 -1
  123. mindspore/mindrecord/tools/imagenet_to_mr.py +1 -1
  124. mindspore/mindrecord/tools/mnist_to_mr.py +1 -1
  125. mindspore/mindrecord/tools/tfrecord_to_mr.py +1 -1
  126. mindspore/mint/__init__.py +15 -10
  127. mindspore/mint/distributed/__init__.py +4 -0
  128. mindspore/mint/distributed/distributed.py +392 -69
  129. mindspore/mint/nn/__init__.py +2 -16
  130. mindspore/mint/nn/functional.py +4 -110
  131. mindspore/mint/nn/layer/__init__.py +0 -2
  132. mindspore/mint/nn/layer/_functions.py +1 -2
  133. mindspore/mint/nn/layer/activation.py +0 -6
  134. mindspore/mint/nn/layer/basic.py +0 -47
  135. mindspore/mint/nn/layer/conv.py +10 -10
  136. mindspore/mint/nn/layer/normalization.py +11 -16
  137. mindspore/mint/nn/layer/pooling.py +0 -4
  138. mindspore/nn/__init__.py +1 -3
  139. mindspore/nn/cell.py +231 -239
  140. mindspore/nn/layer/activation.py +4 -2
  141. mindspore/nn/layer/basic.py +56 -14
  142. mindspore/nn/layer/container.py +16 -0
  143. mindspore/nn/layer/embedding.py +4 -169
  144. mindspore/nn/layer/image.py +1 -1
  145. mindspore/nn/layer/normalization.py +2 -1
  146. mindspore/nn/layer/thor_layer.py +4 -85
  147. mindspore/nn/optim/ada_grad.py +0 -1
  148. mindspore/nn/optim/adafactor.py +0 -1
  149. mindspore/nn/optim/adam.py +32 -127
  150. mindspore/nn/optim/adamax.py +0 -1
  151. mindspore/nn/optim/asgd.py +0 -1
  152. mindspore/nn/optim/ftrl.py +8 -102
  153. mindspore/nn/optim/lamb.py +1 -4
  154. mindspore/nn/optim/lars.py +0 -3
  155. mindspore/nn/optim/lazyadam.py +25 -218
  156. mindspore/nn/optim/momentum.py +5 -43
  157. mindspore/nn/optim/optimizer.py +6 -55
  158. mindspore/nn/optim/proximal_ada_grad.py +0 -1
  159. mindspore/nn/optim/rmsprop.py +0 -1
  160. mindspore/nn/optim/rprop.py +0 -1
  161. mindspore/nn/optim/sgd.py +0 -1
  162. mindspore/nn/optim/tft_wrapper.py +2 -4
  163. mindspore/nn/optim/thor.py +0 -2
  164. mindspore/nn/probability/bijector/bijector.py +7 -8
  165. mindspore/nn/probability/bijector/gumbel_cdf.py +2 -2
  166. mindspore/nn/probability/bijector/power_transform.py +20 -21
  167. mindspore/nn/probability/bijector/scalar_affine.py +5 -5
  168. mindspore/nn/probability/bijector/softplus.py +13 -14
  169. mindspore/nn/probability/distribution/_utils/utils.py +2 -2
  170. mindspore/nn/wrap/cell_wrapper.py +39 -5
  171. mindspore/nn/wrap/grad_reducer.py +4 -89
  172. mindspore/numpy/array_creations.py +4 -4
  173. mindspore/numpy/fft.py +9 -9
  174. mindspore/numpy/utils_const.py +1 -1
  175. mindspore/{nn/reinforcement → onnx}/__init__.py +5 -8
  176. mindspore/onnx/onnx_export.py +137 -0
  177. mindspore/ops/__init__.py +2 -0
  178. mindspore/ops/_grad_experimental/grad_comm_ops.py +38 -2
  179. mindspore/ops/_grad_experimental/grad_inner_ops.py +0 -9
  180. mindspore/ops/_op_impl/aicpu/__init__.py +0 -10
  181. mindspore/ops/_op_impl/cpu/__init__.py +1 -5
  182. mindspore/ops/_op_impl/cpu/{buffer_append.py → joinedstr_op.py} +8 -8
  183. mindspore/ops/auto_generate/cpp_create_prim_instance_helper.py +29 -25
  184. mindspore/ops/auto_generate/gen_extend_func.py +6 -11
  185. mindspore/ops/auto_generate/gen_ops_def.py +381 -150
  186. mindspore/ops/auto_generate/gen_ops_prim.py +1780 -1271
  187. mindspore/ops/auto_generate/pyboost_inner_prim.py +10 -10
  188. mindspore/ops/communication.py +97 -0
  189. mindspore/ops/composite/__init__.py +5 -2
  190. mindspore/ops/composite/base.py +16 -2
  191. mindspore/ops/composite/multitype_ops/__init__.py +3 -1
  192. mindspore/ops/composite/multitype_ops/_compile_utils.py +150 -8
  193. mindspore/ops/composite/multitype_ops/_constexpr_utils.py +1 -1
  194. mindspore/ops/composite/multitype_ops/add_impl.py +7 -0
  195. mindspore/ops/composite/multitype_ops/mod_impl.py +27 -0
  196. mindspore/ops/function/__init__.py +2 -0
  197. mindspore/ops/function/array_func.py +24 -18
  198. mindspore/ops/function/comm_func.py +3883 -0
  199. mindspore/ops/function/debug_func.py +7 -6
  200. mindspore/ops/function/grad/grad_func.py +4 -12
  201. mindspore/ops/function/math_func.py +89 -86
  202. mindspore/ops/function/nn_func.py +92 -313
  203. mindspore/ops/function/random_func.py +9 -18
  204. mindspore/ops/functional.py +4 -1
  205. mindspore/ops/functional_overload.py +377 -30
  206. mindspore/ops/operations/__init__.py +2 -5
  207. mindspore/ops/operations/_custom_ops_utils.py +7 -9
  208. mindspore/ops/operations/_inner_ops.py +12 -50
  209. mindspore/ops/operations/_rl_inner_ops.py +0 -933
  210. mindspore/ops/operations/array_ops.py +5 -50
  211. mindspore/ops/operations/comm_ops.py +95 -17
  212. mindspore/ops/operations/custom_ops.py +237 -22
  213. mindspore/ops/operations/debug_ops.py +33 -35
  214. mindspore/ops/operations/manually_defined/ops_def.py +39 -318
  215. mindspore/ops/operations/math_ops.py +5 -5
  216. mindspore/ops/operations/nn_ops.py +3 -3
  217. mindspore/ops/operations/sparse_ops.py +0 -83
  218. mindspore/ops/primitive.py +4 -27
  219. mindspore/ops/tensor_method.py +88 -10
  220. mindspore/ops_generate/aclnn/aclnn_kernel_register_auto_cc_generator.py +5 -5
  221. mindspore/ops_generate/aclnn/gen_aclnn_implement.py +8 -8
  222. mindspore/ops_generate/api/functions_cc_generator.py +53 -4
  223. mindspore/ops_generate/api/tensor_func_reg_cpp_generator.py +25 -11
  224. mindspore/ops_generate/common/gen_constants.py +11 -10
  225. mindspore/ops_generate/common/op_proto.py +18 -1
  226. mindspore/ops_generate/common/template.py +102 -245
  227. mindspore/ops_generate/common/template_utils.py +212 -0
  228. mindspore/ops_generate/gen_custom_ops.py +69 -0
  229. mindspore/ops_generate/op_def/ops_def_cc_generator.py +78 -7
  230. mindspore/ops_generate/op_def_py/base_op_prim_py_generator.py +360 -0
  231. mindspore/ops_generate/op_def_py/custom_op_prim_py_generator.py +140 -0
  232. mindspore/ops_generate/op_def_py/op_def_py_generator.py +54 -7
  233. mindspore/ops_generate/op_def_py/op_prim_py_generator.py +5 -312
  234. mindspore/ops_generate/pyboost/auto_grad_impl_cc_generator.py +74 -17
  235. mindspore/ops_generate/pyboost/auto_grad_reg_cc_generator.py +22 -5
  236. mindspore/ops_generate/pyboost/gen_pyboost_func.py +0 -16
  237. mindspore/ops_generate/pyboost/op_template_parser.py +3 -2
  238. mindspore/ops_generate/pyboost/pyboost_functions_cpp_generator.py +21 -5
  239. mindspore/ops_generate/pyboost/pyboost_functions_h_generator.py +2 -2
  240. mindspore/ops_generate/pyboost/pyboost_functions_impl_cpp_generator.py +30 -10
  241. mindspore/ops_generate/pyboost/pyboost_grad_function_cpp_generator.py +10 -3
  242. mindspore/ops_generate/pyboost/pyboost_internal_kernel_info_adapter_generator.py +1 -1
  243. mindspore/ops_generate/pyboost/pyboost_native_grad_functions_generator.py +19 -9
  244. mindspore/ops_generate/pyboost/pyboost_op_cpp_code_generator.py +71 -28
  245. mindspore/ops_generate/pyboost/pyboost_overload_functions_cpp_generator.py +10 -9
  246. mindspore/ops_generate/pyboost/pyboost_utils.py +27 -16
  247. mindspore/ops_generate/resources/yaml_loader.py +13 -0
  248. mindspore/ops_generate/tensor_py_cc_generator.py +2 -2
  249. mindspore/parallel/_auto_parallel_context.py +5 -15
  250. mindspore/parallel/_cell_wrapper.py +1 -1
  251. mindspore/parallel/_parallel_serialization.py +4 -6
  252. mindspore/parallel/_ps_context.py +2 -2
  253. mindspore/parallel/_utils.py +34 -17
  254. mindspore/parallel/auto_parallel.py +23 -9
  255. mindspore/parallel/checkpoint_transform.py +20 -2
  256. mindspore/parallel/cluster/process_entity/_api.py +28 -33
  257. mindspore/parallel/cluster/process_entity/_utils.py +9 -5
  258. mindspore/parallel/cluster/run.py +5 -3
  259. mindspore/{experimental/llm_boost/ascend_native → parallel/distributed}/__init__.py +21 -22
  260. mindspore/parallel/distributed/distributed_data_parallel.py +393 -0
  261. mindspore/parallel/distributed/flatten_grad_buffer.py +295 -0
  262. mindspore/parallel/function/reshard_func.py +6 -5
  263. mindspore/parallel/nn/parallel_cell_wrapper.py +40 -3
  264. mindspore/parallel/nn/parallel_grad_reducer.py +0 -8
  265. mindspore/parallel/shard.py +7 -21
  266. mindspore/parallel/strategy.py +336 -0
  267. mindspore/parallel/transform_safetensors.py +127 -20
  268. mindspore/profiler/analysis/viewer/ascend_kernel_details_viewer.py +13 -9
  269. mindspore/profiler/analysis/viewer/ascend_op_memory_viewer.py +1 -1
  270. mindspore/profiler/analysis/viewer/ms_minddata_viewer.py +1 -1
  271. mindspore/profiler/common/constant.py +5 -0
  272. mindspore/profiler/common/file_manager.py +9 -0
  273. mindspore/profiler/common/msprof_cmd_tool.py +40 -4
  274. mindspore/profiler/common/path_manager.py +65 -24
  275. mindspore/profiler/common/profiler_context.py +27 -14
  276. mindspore/profiler/common/profiler_info.py +3 -3
  277. mindspore/profiler/common/profiler_meta_data.py +1 -0
  278. mindspore/profiler/common/profiler_op_analyse.py +10 -6
  279. mindspore/profiler/common/profiler_path_manager.py +13 -0
  280. mindspore/profiler/common/util.py +30 -3
  281. mindspore/profiler/dynamic_profiler.py +91 -46
  282. mindspore/profiler/envprofiler.py +30 -5
  283. mindspore/profiler/experimental_config.py +18 -2
  284. mindspore/profiler/platform/cpu_profiler.py +10 -4
  285. mindspore/profiler/platform/npu_profiler.py +34 -7
  286. mindspore/profiler/profiler.py +193 -145
  287. mindspore/profiler/profiler_action_controller.py +1 -1
  288. mindspore/profiler/profiler_interface.py +2 -2
  289. mindspore/rewrite/symbol_tree/symbol_tree.py +1 -1
  290. mindspore/run_check/_check_version.py +108 -24
  291. mindspore/runtime/__init__.py +9 -6
  292. mindspore/runtime/executor.py +35 -0
  293. mindspore/runtime/memory.py +113 -0
  294. mindspore/runtime/thread_bind_core.py +1 -1
  295. mindspore/scipy/linalg.py +1 -1
  296. mindspore/{experimental/llm_boost → tools}/__init__.py +5 -5
  297. mindspore/tools/data_dump.py +130 -0
  298. mindspore/tools/sdc_detect.py +91 -0
  299. mindspore/tools/stress_detect.py +63 -0
  300. mindspore/train/__init__.py +6 -6
  301. mindspore/train/_utils.py +8 -21
  302. mindspore/train/amp.py +6 -7
  303. mindspore/train/callback/_callback.py +2 -1
  304. mindspore/train/callback/_checkpoint.py +1 -17
  305. mindspore/train/callback/_flops_collector.py +10 -6
  306. mindspore/train/callback/_train_fault_tolerance.py +72 -25
  307. mindspore/train/data_sink.py +5 -9
  308. mindspore/train/dataset_helper.py +5 -5
  309. mindspore/train/model.py +41 -230
  310. mindspore/train/serialization.py +160 -401
  311. mindspore/train/train_thor/model_thor.py +2 -2
  312. mindspore/utils/__init__.py +6 -3
  313. mindspore/utils/dlpack.py +92 -0
  314. mindspore/utils/dryrun.py +1 -1
  315. mindspore/utils/runtime_execution_order_check.py +10 -0
  316. mindspore/utils/sdc_detect.py +14 -12
  317. mindspore/utils/stress_detect.py +43 -0
  318. mindspore/utils/utils.py +152 -16
  319. mindspore/version.py +1 -1
  320. {mindspore-2.7.0rc1.dist-info → mindspore-2.7.1.dist-info}/METADATA +3 -2
  321. {mindspore-2.7.0rc1.dist-info → mindspore-2.7.1.dist-info}/RECORD +325 -340
  322. mindspore/_extends/remote/kernel_build_server_ascend.py +0 -75
  323. mindspore/communication/_hccl_management.py +0 -297
  324. mindspore/experimental/llm_boost/ascend_native/llama_boost_ascend_native.py +0 -207
  325. mindspore/experimental/llm_boost/ascend_native/llm_boost.py +0 -52
  326. mindspore/experimental/llm_boost/atb/__init__.py +0 -23
  327. mindspore/experimental/llm_boost/atb/boost_base.py +0 -385
  328. mindspore/experimental/llm_boost/atb/llama_boost.py +0 -137
  329. mindspore/experimental/llm_boost/atb/qwen_boost.py +0 -124
  330. mindspore/experimental/llm_boost/register.py +0 -130
  331. mindspore/experimental/llm_boost/utils.py +0 -31
  332. mindspore/include/OWNERS +0 -7
  333. mindspore/lib/libmindspore_ops_kernel_common.dylib +0 -0
  334. mindspore/lib/libmindspore_res_manager.dylib +0 -0
  335. mindspore/lib/libopencv_core.4.5.dylib +0 -0
  336. mindspore/lib/libopencv_imgcodecs.4.5.dylib +0 -0
  337. mindspore/lib/libopencv_imgproc.4.5.dylib +0 -0
  338. mindspore/lib/libps_cache.dylib +0 -0
  339. mindspore/lib/plugin/cpu/libmindspore_cpu_res_manager.dylib +0 -0
  340. mindspore/nn/optim/_dist_optimizer_registry.py +0 -111
  341. mindspore/nn/reinforcement/_batch_read_write.py +0 -142
  342. mindspore/nn/reinforcement/_tensors_queue.py +0 -152
  343. mindspore/nn/reinforcement/tensor_array.py +0 -145
  344. mindspore/ops/_op_impl/aicpu/priority_replay_buffer.py +0 -113
  345. mindspore/ops/_op_impl/aicpu/reservoir_replay_buffer.py +0 -96
  346. mindspore/ops/_op_impl/aicpu/sparse_cross.py +0 -42
  347. mindspore/ops/_op_impl/cpu/buffer_get.py +0 -28
  348. mindspore/ops/_op_impl/cpu/buffer_sample.py +0 -28
  349. mindspore/ops/_op_impl/cpu/priority_replay_buffer.py +0 -42
  350. mindspore/ops/operations/_tensor_array.py +0 -359
  351. mindspore/ops/operations/rl_ops.py +0 -288
  352. mindspore/parallel/_offload_context.py +0 -275
  353. mindspore/parallel/_recovery_context.py +0 -115
  354. mindspore/parallel/_transformer/__init__.py +0 -35
  355. mindspore/parallel/_transformer/layers.py +0 -765
  356. mindspore/parallel/_transformer/loss.py +0 -251
  357. mindspore/parallel/_transformer/moe.py +0 -693
  358. mindspore/parallel/_transformer/op_parallel_config.py +0 -222
  359. mindspore/parallel/_transformer/transformer.py +0 -3124
  360. mindspore/parallel/mpi/_mpi_config.py +0 -116
  361. mindspore/profiler/common/validator/validate_path.py +0 -84
  362. mindspore/train/memory_profiling_pb2.py +0 -298
  363. mindspore/utils/hooks.py +0 -81
  364. /mindspore/common/{_auto_dynamic.py → dynamic_shape/_auto_dynamic.py} +0 -0
  365. {mindspore-2.7.0rc1.dist-info → mindspore-2.7.1.dist-info}/WHEEL +0 -0
  366. {mindspore-2.7.0rc1.dist-info → mindspore-2.7.1.dist-info}/entry_points.txt +0 -0
  367. {mindspore-2.7.0rc1.dist-info → mindspore-2.7.1.dist-info}/top_level.txt +0 -0
mindspore/.commit_id CHANGED
@@ -1 +1 @@
1
- __commit_id__ = ''[sha1]:e69e001d,[branch]:(HEAD,origin/r2.7.rc1,r2.7.rc1)''
1
+ __commit_id__ = ''[sha1]:71340184,[branch]:(HEAD,origin/r2.7.1,r2.7.1)''
mindspore/__init__.py CHANGED
@@ -19,7 +19,7 @@ from mindspore.run_check import run_check
19
19
  from mindspore import common, dataset, mindrecord, train, log, amp, device_manager
20
20
  from mindspore import profiler, communication, numpy, parallel, hal, runtime, device_context
21
21
  from mindspore.common import *
22
- from mindspore.common import _tensor_docs
22
+ from mindspore.common import _tensor_docs, bool, int, float
23
23
  del _tensor_docs
24
24
  from mindspore.mindrecord import *
25
25
  from mindspore.ops import _op_impl, grad, value_and_grad, vjp, jvp, jacfwd, jacrev, vmap, get_grad, constexpr
@@ -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")
mindspore/_checkparam.py CHANGED
@@ -787,9 +787,9 @@ def check_astype_dtype(dtype):
787
787
  if isinstance(dtype, str):
788
788
  if dtype.lower() not in all_types:
789
789
  raise TypeError(f"For Tensor.astype, the input type must be one of {all_types}, but got '{dtype}'.")
790
- dtype = mstype.pytype_to_dtype(np.dtype(dtype.lower()))
790
+ dtype = mstype._pytype_to_dtype(np.dtype(dtype.lower())) # pylint:disable=protected-access
791
791
  elif isinstance(dtype, type):
792
- dtype = mstype.pytype_to_dtype(dtype)
792
+ dtype = mstype._pytype_to_dtype(dtype) # pylint:disable=protected-access
793
793
  elif not dtype in mstype.number_type + (mstype.bool_,):
794
794
  raise TypeError(f"For Tensor.astype, the input type must be one of {mstype.number_type + (mstype.bool_,)}," \
795
795
  f" but got '{dtype}'.")
@@ -19,7 +19,7 @@ from mindspore.ops import functional as F
19
19
  from mindspore.ops import composite as C
20
20
  from mindspore.common.tensor import Tensor
21
21
  import mindspore.common.dtype as mstype
22
- from mindspore.common.dtype import dtype_to_nptype, get_py_obj_dtype
22
+ from mindspore.common.dtype import _dtype_to_nptype, _get_py_obj_dtype
23
23
  from mindspore._c_expression import TensorPy as CTensor
24
24
 
25
25
 
@@ -164,7 +164,7 @@ def Load(value, u=None):
164
164
 
165
165
  def scalar_cast(x, t):
166
166
  """Implement scalar_cast."""
167
- np_type = dtype_to_nptype(t)
167
+ np_type = _dtype_to_nptype(t) # pylint:disable=protected-access
168
168
  value = np_type(x)
169
169
  cast_value = np.ndarray.item(value)
170
170
  return cast_value
@@ -172,7 +172,7 @@ def scalar_cast(x, t):
172
172
 
173
173
  def typeof(x):
174
174
  """Implement typeof."""
175
- return get_py_obj_dtype(x)
175
+ return _get_py_obj_dtype(x) # pylint:disable=protected-access
176
176
 
177
177
 
178
178
  def tuple_to_array(x):