mindspore 2.4.1__cp39-cp39-win_amd64.whl → 2.5.0__cp39-cp39-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 (395) hide show
  1. mindspore/.commit_id +1 -1
  2. mindspore/Microsoft.VisualStudio.Telemetry.dll +0 -0
  3. mindspore/Newtonsoft.Json.dll +0 -0
  4. mindspore/__init__.py +8 -3
  5. mindspore/_c_dataengine.cp39-win_amd64.pyd +0 -0
  6. mindspore/_c_expression.cp39-win_amd64.pyd +0 -0
  7. mindspore/_c_mindrecord.cp39-win_amd64.pyd +0 -0
  8. mindspore/_checkparam.py +0 -5
  9. mindspore/_extends/parallel_compile/akg_compiler/gen_custom_op_files.py +1 -1
  10. mindspore/_extends/parse/compile_config.py +64 -0
  11. mindspore/_extends/parse/deprecated/__init__.py +0 -0
  12. mindspore/_extends/parse/deprecated/deprecated_tensor_method.py +375 -0
  13. mindspore/_extends/parse/parser.py +23 -5
  14. mindspore/_extends/parse/standard_method.py +123 -27
  15. mindspore/_extends/pijit/pijit_func_white_list.py +1 -1
  16. mindspore/amp.py +7 -1
  17. mindspore/atlprov.dll +0 -0
  18. mindspore/avcodec-59.dll +0 -0
  19. mindspore/avdevice-59.dll +0 -0
  20. mindspore/avfilter-8.dll +0 -0
  21. mindspore/avformat-59.dll +0 -0
  22. mindspore/avutil-57.dll +0 -0
  23. mindspore/boost/boost_cell_wrapper.py +136 -41
  24. mindspore/c1.dll +0 -0
  25. mindspore/c1xx.dll +0 -0
  26. mindspore/c2.dll +0 -0
  27. mindspore/common/__init__.py +3 -1
  28. mindspore/common/_register_for_tensor.py +0 -1
  29. mindspore/common/_stub_tensor.py +25 -4
  30. mindspore/common/_tensor_cpp_method.py +17 -0
  31. mindspore/common/_tensor_docs.py +6132 -0
  32. mindspore/common/api.py +99 -25
  33. mindspore/common/dtype.py +34 -34
  34. mindspore/common/dump.py +2 -1
  35. mindspore/common/file_system.py +8 -1
  36. mindspore/common/generator.py +2 -0
  37. mindspore/common/hook_handle.py +3 -1
  38. mindspore/common/initializer.py +3 -4
  39. mindspore/common/lazy_inline.py +8 -2
  40. mindspore/common/mindir_util.py +10 -2
  41. mindspore/common/parameter.py +30 -27
  42. mindspore/common/tensor.py +713 -1337
  43. mindspore/communication/__init__.py +1 -1
  44. mindspore/communication/_comm_helper.py +10 -0
  45. mindspore/communication/comm_func.py +215 -173
  46. mindspore/communication/management.py +23 -20
  47. mindspore/context.py +292 -193
  48. mindspore/dataset/__init__.py +23 -19
  49. mindspore/dataset/callback/ds_callback.py +2 -1
  50. mindspore/dataset/core/config.py +84 -3
  51. mindspore/dataset/engine/cache_admin.py +3 -3
  52. mindspore/dataset/engine/cache_client.py +5 -4
  53. mindspore/dataset/engine/datasets.py +192 -149
  54. mindspore/dataset/engine/datasets_audio.py +14 -0
  55. mindspore/dataset/engine/datasets_standard_format.py +28 -11
  56. mindspore/dataset/engine/datasets_text.py +38 -1
  57. mindspore/dataset/engine/datasets_user_defined.py +125 -65
  58. mindspore/dataset/engine/datasets_vision.py +81 -8
  59. mindspore/dataset/engine/iterators.py +281 -63
  60. mindspore/dataset/engine/obs/util.py +8 -0
  61. mindspore/dataset/engine/queue.py +40 -0
  62. mindspore/dataset/engine/samplers.py +26 -2
  63. mindspore/dataset/engine/serializer_deserializer.py +1 -1
  64. mindspore/dataset/engine/validators.py +43 -11
  65. mindspore/dataset/transforms/py_transforms_util.py +17 -0
  66. mindspore/dataset/transforms/transforms.py +29 -12
  67. mindspore/dataset/vision/validators.py +1 -2
  68. mindspore/device_context/__init__.py +21 -0
  69. mindspore/device_context/ascend/__init__.py +25 -0
  70. mindspore/device_context/ascend/device.py +72 -0
  71. mindspore/device_context/ascend/op_debug.py +94 -0
  72. mindspore/device_context/ascend/op_precision.py +193 -0
  73. mindspore/device_context/ascend/op_tuning.py +127 -0
  74. mindspore/device_context/cpu/__init__.py +25 -0
  75. mindspore/device_context/cpu/device.py +62 -0
  76. mindspore/device_context/cpu/op_tuning.py +43 -0
  77. mindspore/device_context/gpu/__init__.py +21 -0
  78. mindspore/device_context/gpu/device.py +70 -0
  79. mindspore/device_context/gpu/op_precision.py +67 -0
  80. mindspore/device_context/gpu/op_tuning.py +175 -0
  81. mindspore/device_manager.py +134 -0
  82. mindspore/dnnl.dll +0 -0
  83. mindspore/dpcmi.dll +0 -0
  84. mindspore/experimental/llm_boost/__init__.py +3 -2
  85. mindspore/experimental/llm_boost/ascend_native/__init__.py +22 -0
  86. mindspore/experimental/llm_boost/ascend_native/llama_boost_ascend_native.py +211 -0
  87. mindspore/experimental/llm_boost/ascend_native/llm_boost.py +52 -0
  88. mindspore/experimental/llm_boost/atb/boost_base.py +239 -64
  89. mindspore/experimental/llm_boost/atb/llama_boost.py +52 -30
  90. mindspore/experimental/llm_boost/atb/qwen_boost.py +47 -24
  91. mindspore/experimental/llm_boost/register.py +1 -0
  92. mindspore/experimental/optim/adadelta.py +26 -22
  93. mindspore/experimental/optim/adam.py +3 -0
  94. mindspore/experimental/optim/lr_scheduler.py +33 -24
  95. mindspore/experimental/optim/radam.py +33 -30
  96. mindspore/hal/device.py +28 -0
  97. mindspore/hal/event.py +17 -0
  98. mindspore/hal/memory.py +94 -3
  99. mindspore/hal/stream.py +91 -6
  100. mindspore/include/api/context.h +1 -2
  101. mindspore/include/dataset/constants.h +2 -2
  102. mindspore/jpeg62.dll +0 -0
  103. mindspore/log.py +12 -0
  104. mindspore/mindrecord/__init__.py +1 -1
  105. mindspore/mindrecord/config.py +17 -316
  106. mindspore/mindrecord/filereader.py +1 -9
  107. mindspore/mindrecord/filewriter.py +5 -15
  108. mindspore/mindrecord/mindpage.py +1 -9
  109. mindspore/mindspore_backend.dll +0 -0
  110. mindspore/mindspore_common.dll +0 -0
  111. mindspore/mindspore_core.dll +0 -0
  112. mindspore/mindspore_glog.dll +0 -0
  113. mindspore/mindspore_ops.dll +0 -0
  114. mindspore/mint/__init__.py +824 -218
  115. mindspore/mint/distributed/__init__.py +66 -4
  116. mindspore/mint/distributed/distributed.py +2594 -44
  117. mindspore/mint/linalg/__init__.py +6 -0
  118. mindspore/mint/nn/__init__.py +473 -14
  119. mindspore/mint/nn/functional.py +486 -11
  120. mindspore/mint/nn/layer/__init__.py +17 -4
  121. mindspore/mint/nn/layer/_functions.py +330 -0
  122. mindspore/mint/nn/layer/activation.py +169 -1
  123. mindspore/mint/nn/layer/basic.py +123 -0
  124. mindspore/mint/nn/layer/conv.py +727 -0
  125. mindspore/mint/nn/layer/normalization.py +215 -19
  126. mindspore/mint/nn/layer/padding.py +797 -0
  127. mindspore/mint/nn/layer/pooling.py +170 -0
  128. mindspore/mint/optim/__init__.py +2 -1
  129. mindspore/mint/optim/adam.py +223 -0
  130. mindspore/mint/optim/adamw.py +26 -19
  131. mindspore/mint/special/__init__.py +2 -1
  132. mindspore/msobj140.dll +0 -0
  133. mindspore/mspdb140.dll +0 -0
  134. mindspore/mspdbcore.dll +0 -0
  135. mindspore/mspdbst.dll +0 -0
  136. mindspore/mspft140.dll +0 -0
  137. mindspore/msvcdis140.dll +0 -0
  138. mindspore/msvcp140_1.dll +0 -0
  139. mindspore/msvcp140_2.dll +0 -0
  140. mindspore/msvcp140_atomic_wait.dll +0 -0
  141. mindspore/msvcp140_codecvt_ids.dll +0 -0
  142. mindspore/multiprocessing/__init__.py +5 -0
  143. mindspore/nn/__init__.py +2 -0
  144. mindspore/nn/cell.py +142 -21
  145. mindspore/nn/dynamic_lr.py +2 -1
  146. mindspore/nn/layer/activation.py +6 -6
  147. mindspore/nn/layer/basic.py +35 -25
  148. mindspore/nn/layer/channel_shuffle.py +3 -3
  149. mindspore/nn/layer/conv.py +3 -0
  150. mindspore/nn/layer/embedding.py +3 -3
  151. mindspore/nn/layer/normalization.py +8 -7
  152. mindspore/nn/layer/padding.py +4 -3
  153. mindspore/nn/layer/pooling.py +55 -23
  154. mindspore/nn/layer/rnn_cells.py +1 -1
  155. mindspore/nn/layer/rnns.py +2 -1
  156. mindspore/nn/layer/timedistributed.py +5 -5
  157. mindspore/nn/layer/transformer.py +48 -26
  158. mindspore/nn/learning_rate_schedule.py +5 -3
  159. mindspore/nn/loss/loss.py +31 -36
  160. mindspore/nn/optim/ada_grad.py +1 -0
  161. mindspore/nn/optim/adadelta.py +2 -2
  162. mindspore/nn/optim/adam.py +1 -1
  163. mindspore/nn/optim/lars.py +1 -4
  164. mindspore/nn/optim/optimizer.py +1 -1
  165. mindspore/nn/optim/rprop.py +2 -2
  166. mindspore/nn/optim/thor.py +2 -1
  167. mindspore/nn/utils/__init__.py +22 -0
  168. mindspore/nn/utils/init.py +73 -0
  169. mindspore/nn/wrap/cell_wrapper.py +4 -6
  170. mindspore/nn/wrap/loss_scale.py +3 -4
  171. mindspore/numpy/array_creations.py +60 -62
  172. mindspore/numpy/array_ops.py +148 -143
  173. mindspore/numpy/logic_ops.py +41 -42
  174. mindspore/numpy/math_ops.py +361 -359
  175. mindspore/numpy/utils.py +16 -16
  176. mindspore/numpy/utils_const.py +4 -4
  177. mindspore/opencv_core452.dll +0 -0
  178. mindspore/opencv_imgcodecs452.dll +0 -0
  179. mindspore/opencv_imgproc452.dll +0 -0
  180. mindspore/ops/__init__.py +2 -1
  181. mindspore/ops/_grad_experimental/grad_comm_ops.py +107 -8
  182. mindspore/ops/_grad_experimental/grad_debug_ops.py +6 -1
  183. mindspore/ops/_grad_experimental/grad_inner_ops.py +9 -0
  184. mindspore/ops/_grad_experimental/grad_math_ops.py +2 -1
  185. mindspore/ops/_op_impl/cpu/__init__.py +1 -0
  186. mindspore/ops/_op_impl/cpu/raise_op.py +28 -0
  187. mindspore/ops/_vmap/vmap_array_ops.py +20 -19
  188. mindspore/ops/_vmap/vmap_base.py +0 -2
  189. mindspore/ops/_vmap/vmap_grad_nn_ops.py +19 -13
  190. mindspore/ops/_vmap/vmap_math_ops.py +11 -9
  191. mindspore/ops/_vmap/vmap_nn_ops.py +20 -34
  192. mindspore/ops/auto_generate/cpp_create_prim_instance_helper.py +149 -12
  193. mindspore/ops/auto_generate/gen_arg_handler.py +0 -61
  194. mindspore/ops/auto_generate/gen_extend_func.py +554 -60
  195. mindspore/ops/auto_generate/gen_ops_def.py +1621 -115
  196. mindspore/ops/auto_generate/gen_ops_prim.py +8027 -3411
  197. mindspore/ops/auto_generate/pyboost_inner_prim.py +183 -79
  198. mindspore/ops/composite/base.py +1 -1
  199. mindspore/ops/composite/multitype_ops/_compile_utils.py +229 -30
  200. mindspore/ops/composite/multitype_ops/pow_impl.py +0 -29
  201. mindspore/ops/function/__init__.py +12 -0
  202. mindspore/ops/function/array_func.py +561 -159
  203. mindspore/ops/function/clip_func.py +64 -0
  204. mindspore/ops/function/debug_func.py +28 -20
  205. mindspore/ops/function/image_func.py +1 -1
  206. mindspore/ops/function/linalg_func.py +5 -4
  207. mindspore/ops/function/math_func.py +1664 -294
  208. mindspore/ops/function/nn_func.py +988 -317
  209. mindspore/ops/function/parameter_func.py +3 -56
  210. mindspore/ops/function/random_func.py +243 -33
  211. mindspore/ops/function/sparse_unary_func.py +1 -1
  212. mindspore/ops/functional.py +18 -5
  213. mindspore/ops/functional_overload.py +897 -0
  214. mindspore/ops/operations/__init__.py +3 -2
  215. mindspore/ops/operations/_embedding_cache_ops.py +4 -4
  216. mindspore/ops/operations/_grad_ops.py +2 -34
  217. mindspore/ops/operations/_infer_ops.py +2 -1
  218. mindspore/ops/operations/_inner_ops.py +38 -8
  219. mindspore/ops/operations/array_ops.py +45 -303
  220. mindspore/ops/operations/comm_ops.py +23 -17
  221. mindspore/ops/operations/custom_ops.py +7 -49
  222. mindspore/ops/operations/debug_ops.py +42 -47
  223. mindspore/ops/operations/inner_ops.py +6 -4
  224. mindspore/ops/operations/linalg_ops.py +3 -2
  225. mindspore/ops/operations/manually_defined/ops_def.py +185 -104
  226. mindspore/ops/operations/math_ops.py +11 -216
  227. mindspore/ops/operations/nn_ops.py +153 -310
  228. mindspore/ops/primitive.py +23 -21
  229. mindspore/ops/tensor_method.py +1669 -0
  230. mindspore/ops_generate/aclnn_kernel_register_auto_cc_generator.py +110 -0
  231. mindspore/ops_generate/add_tensor_docs_generator.py +54 -0
  232. mindspore/ops_generate/arg_handler.py +0 -61
  233. mindspore/ops_generate/auto_grad_impl_cc_generator.py +135 -0
  234. mindspore/ops_generate/auto_grad_reg_cc_generator.py +93 -0
  235. mindspore/ops_generate/base_generator.py +11 -0
  236. mindspore/ops_generate/cpp_create_prim_instance_helper_generator.py +108 -0
  237. mindspore/ops_generate/functional_map_cpp_generator.py +491 -0
  238. mindspore/ops_generate/functional_overload_py_generator.py +110 -0
  239. mindspore/ops_generate/functions_cc_generator.py +233 -0
  240. mindspore/ops_generate/gen_aclnn_implement.py +110 -114
  241. mindspore/ops_generate/gen_constants.py +157 -3
  242. mindspore/ops_generate/gen_ops.py +245 -990
  243. mindspore/ops_generate/gen_pyboost_func.py +97 -998
  244. mindspore/ops_generate/gen_utils.py +119 -33
  245. mindspore/ops_generate/lite_ops_cpp_generator.py +155 -0
  246. mindspore/ops_generate/op_api_proto.py +206 -0
  247. mindspore/ops_generate/op_def_py_generator.py +131 -0
  248. mindspore/ops_generate/op_prim_py_generator.py +480 -0
  249. mindspore/ops_generate/op_proto.py +373 -108
  250. mindspore/ops_generate/op_template_parser.py +436 -0
  251. mindspore/ops_generate/ops_def_cc_generator.py +288 -0
  252. mindspore/ops_generate/ops_def_h_generator.py +74 -0
  253. mindspore/ops_generate/ops_name_h_generator.py +68 -0
  254. mindspore/ops_generate/ops_primitive_h_generator.py +81 -0
  255. mindspore/ops_generate/pyboost_functions_cpp_generator.py +370 -0
  256. mindspore/ops_generate/pyboost_functions_h_generator.py +68 -0
  257. mindspore/ops_generate/pyboost_functions_py_generator.py +148 -0
  258. mindspore/ops_generate/pyboost_grad_function_cpp_generator.py +154 -0
  259. mindspore/ops_generate/pyboost_inner_prim_generator.py +131 -0
  260. mindspore/ops_generate/pyboost_native_grad_functions_generator.py +268 -0
  261. mindspore/ops_generate/pyboost_op_cpp_code_generator.py +851 -0
  262. mindspore/ops_generate/pyboost_overload_functions_cpp_generator.py +344 -0
  263. mindspore/ops_generate/pyboost_utils.py +92 -33
  264. mindspore/ops_generate/template.py +294 -44
  265. mindspore/ops_generate/tensor_func_reg_cpp_generator.py +422 -0
  266. mindspore/parallel/__init__.py +3 -3
  267. mindspore/parallel/_auto_parallel_context.py +44 -34
  268. mindspore/parallel/_cell_wrapper.py +22 -3
  269. mindspore/parallel/_parallel_serialization.py +13 -2
  270. mindspore/parallel/_utils.py +4 -2
  271. mindspore/parallel/algo_parameter_config.py +1 -1
  272. mindspore/parallel/checkpoint_transform.py +44 -0
  273. mindspore/parallel/cluster/process_entity/_api.py +131 -37
  274. mindspore/parallel/cluster/process_entity/_utils.py +41 -6
  275. mindspore/parallel/cluster/run.py +20 -3
  276. mindspore/parallel/parameter_broadcast.py +1 -1
  277. mindspore/parallel/shard.py +3 -0
  278. mindspore/parallel/transform_safetensors.py +119 -253
  279. mindspore/pgodb140.dll +0 -0
  280. mindspore/pgort140.dll +0 -0
  281. mindspore/profiler/__init__.py +17 -4
  282. mindspore/profiler/analysis/__init__.py +0 -0
  283. mindspore/profiler/analysis/parser/__init__.py +0 -0
  284. mindspore/profiler/analysis/parser/ascend_cann_parser.py +166 -0
  285. mindspore/profiler/analysis/parser/base_parser.py +158 -0
  286. mindspore/profiler/analysis/parser/framework_cann_relation_parser.py +45 -0
  287. mindspore/profiler/analysis/parser/ms_framework_parser.py +142 -0
  288. mindspore/profiler/analysis/parser/ms_minddata_parser.py +145 -0
  289. mindspore/profiler/analysis/parser/timeline_assembly_factory/__init__.py +0 -0
  290. mindspore/profiler/analysis/parser/timeline_assembly_factory/ascend_timeline_assembler.py +261 -0
  291. mindspore/profiler/analysis/parser/timeline_assembly_factory/base_timeline_assembler.py +40 -0
  292. mindspore/profiler/analysis/parser/timeline_assembly_factory/trace_view_container.py +84 -0
  293. mindspore/profiler/analysis/parser/timeline_creator/__init__.py +0 -0
  294. mindspore/profiler/analysis/parser/timeline_creator/base_timeline_creator.py +44 -0
  295. mindspore/profiler/analysis/parser/timeline_creator/cpu_op_timeline_creator.py +90 -0
  296. mindspore/profiler/analysis/parser/timeline_creator/fwk_timeline_creator.py +76 -0
  297. mindspore/profiler/analysis/parser/timeline_creator/msprof_timeline_creator.py +103 -0
  298. mindspore/profiler/analysis/parser/timeline_creator/scope_layer_timeline_creator.py +134 -0
  299. mindspore/profiler/analysis/parser/timeline_event/__init__.py +0 -0
  300. mindspore/profiler/analysis/parser/timeline_event/base_event.py +233 -0
  301. mindspore/profiler/analysis/parser/timeline_event/cpu_op_event.py +47 -0
  302. mindspore/profiler/analysis/parser/timeline_event/flow_event.py +36 -0
  303. mindspore/profiler/analysis/parser/timeline_event/fwk_event.py +260 -0
  304. mindspore/profiler/analysis/parser/timeline_event/msprof_event.py +73 -0
  305. mindspore/profiler/analysis/parser/timeline_event/scope_layer_event.py +53 -0
  306. mindspore/profiler/analysis/parser/timeline_event/timeline_event_pool.py +146 -0
  307. mindspore/profiler/analysis/task_manager.py +131 -0
  308. mindspore/profiler/analysis/time_converter.py +84 -0
  309. mindspore/profiler/analysis/viewer/__init__.py +0 -0
  310. mindspore/profiler/analysis/viewer/ascend_communication_viewer.py +333 -0
  311. mindspore/profiler/analysis/viewer/ascend_integrate_viewer.py +87 -0
  312. mindspore/profiler/analysis/viewer/ascend_kernel_details_viewer.py +252 -0
  313. mindspore/profiler/analysis/viewer/ascend_memory_viewer.py +313 -0
  314. mindspore/profiler/analysis/viewer/ascend_op_memory_viewer.py +322 -0
  315. mindspore/profiler/analysis/viewer/ascend_step_trace_time_viewer.py +265 -0
  316. mindspore/profiler/analysis/viewer/ascend_timeline_viewer.py +58 -0
  317. mindspore/profiler/analysis/viewer/base_viewer.py +26 -0
  318. mindspore/profiler/analysis/viewer/ms_dataset_viewer.py +97 -0
  319. mindspore/profiler/analysis/viewer/ms_minddata_viewer.py +581 -0
  320. mindspore/profiler/analysis/work_flow.py +73 -0
  321. mindspore/profiler/common/ascend_msprof_exporter.py +138 -0
  322. mindspore/profiler/common/command_executor.py +90 -0
  323. mindspore/profiler/common/constant.py +174 -3
  324. mindspore/profiler/common/file_manager.py +208 -0
  325. mindspore/profiler/common/log.py +130 -0
  326. mindspore/profiler/common/msprof_cmd_tool.py +202 -0
  327. mindspore/profiler/common/path_manager.py +371 -0
  328. mindspore/profiler/common/process_bar.py +168 -0
  329. mindspore/profiler/common/process_pool.py +9 -3
  330. mindspore/profiler/common/profiler_context.py +476 -0
  331. mindspore/profiler/common/profiler_info.py +304 -0
  332. mindspore/profiler/common/profiler_output_path.py +284 -0
  333. mindspore/profiler/common/profiler_parameters.py +210 -0
  334. mindspore/profiler/common/profiler_path_manager.py +120 -0
  335. mindspore/profiler/common/record_function.py +76 -0
  336. mindspore/profiler/common/tlv_decoder.py +76 -0
  337. mindspore/profiler/common/util.py +75 -2
  338. mindspore/profiler/dynamic_profiler.py +270 -37
  339. mindspore/profiler/envprofiler.py +138 -0
  340. mindspore/profiler/mstx.py +199 -0
  341. mindspore/profiler/platform/__init__.py +21 -0
  342. mindspore/profiler/platform/base_profiler.py +40 -0
  343. mindspore/profiler/platform/cpu_profiler.py +124 -0
  344. mindspore/profiler/platform/gpu_profiler.py +74 -0
  345. mindspore/profiler/platform/npu_profiler.py +309 -0
  346. mindspore/profiler/profiler.py +580 -93
  347. mindspore/profiler/profiler_action_controller.py +187 -0
  348. mindspore/profiler/profiler_interface.py +114 -0
  349. mindspore/profiler/schedule.py +208 -0
  350. mindspore/rewrite/api/symbol_tree.py +1 -2
  351. mindspore/run_check/_check_version.py +18 -13
  352. mindspore/runtime/__init__.py +37 -0
  353. mindspore/runtime/device.py +27 -0
  354. mindspore/runtime/event.py +209 -0
  355. mindspore/runtime/executor.py +148 -0
  356. mindspore/runtime/memory.py +392 -0
  357. mindspore/runtime/stream.py +460 -0
  358. mindspore/runtime/thread_bind_core.py +401 -0
  359. mindspore/swresample-4.dll +0 -0
  360. mindspore/swscale-6.dll +0 -0
  361. mindspore/tbbmalloc.dll +0 -0
  362. mindspore/tinyxml2.dll +0 -0
  363. mindspore/train/__init__.py +2 -2
  364. mindspore/train/_utils.py +53 -18
  365. mindspore/train/amp.py +8 -4
  366. mindspore/train/callback/_checkpoint.py +32 -18
  367. mindspore/train/callback/_early_stop.py +1 -1
  368. mindspore/train/callback/_flops_collector.py +105 -69
  369. mindspore/train/callback/_history.py +1 -1
  370. mindspore/train/callback/_summary_collector.py +44 -6
  371. mindspore/train/callback/_tft_register.py +37 -15
  372. mindspore/train/dataset_helper.py +11 -11
  373. mindspore/train/metrics/precision.py +4 -5
  374. mindspore/train/mind_ir_pb2.py +167 -46
  375. mindspore/train/model.py +13 -14
  376. mindspore/train/serialization.py +461 -72
  377. mindspore/train/summary/summary_record.py +1 -2
  378. mindspore/train/train_thor/model_thor.py +1 -1
  379. mindspore/turbojpeg.dll +0 -0
  380. mindspore/utils/__init__.py +4 -2
  381. mindspore/utils/dryrun.py +138 -0
  382. mindspore/utils/runtime_execution_order_check.py +550 -0
  383. mindspore/vcmeta.dll +0 -0
  384. mindspore/vcruntime140.dll +0 -0
  385. mindspore/vcruntime140_1.dll +0 -0
  386. mindspore/version.py +1 -1
  387. {mindspore-2.4.1.dist-info → mindspore-2.5.0.dist-info}/METADATA +3 -4
  388. {mindspore-2.4.1.dist-info → mindspore-2.5.0.dist-info}/RECORD +391 -265
  389. {mindspore-2.4.1.dist-info → mindspore-2.5.0.dist-info}/entry_points.txt +1 -1
  390. mindspore/common/_tensor_overload.py +0 -139
  391. mindspore/mindspore_np_dtype.dll +0 -0
  392. mindspore/profiler/envprofiling.py +0 -254
  393. mindspore/profiler/profiling.py +0 -1926
  394. {mindspore-2.4.1.dist-info → mindspore-2.5.0.dist-info}/WHEEL +0 -0
  395. {mindspore-2.4.1.dist-info → mindspore-2.5.0.dist-info}/top_level.txt +0 -0
mindspore/hal/memory.py CHANGED
@@ -18,12 +18,20 @@ from mindspore._c_expression import _memory_stats, _reset_max_mem_reserved, _res
18
18
  from mindspore import log as logger
19
19
  from .device import _check_inputs_validation, is_initialized
20
20
 
21
+ function_memory_status = {'memory_stats': False, 'memory_reserved': False, 'max_memory_reserved': False,
22
+ 'empty_cache': False, 'reset_peak_memory_stats': False, 'memory_summary': False,
23
+ 'memory_allocated': False, 'max_memory_allocated': False,
24
+ 'reset_max_memory_reserved': False, 'reset_max_memory_allocated': False}
25
+
21
26
 
22
27
  @_check_inputs_validation
23
28
  def memory_stats(device_target=None):
24
29
  """
25
30
  Returns status information queried from the memory pool.
26
31
 
32
+ Note:
33
+ - The api will be deprecated, please use the api :func:`mindspore.runtime.memory_stats` instead.
34
+
27
35
  Note:
28
36
  - If `device_target` is not specified, get the device capability of the current backend set by context.
29
37
  - For the `CPU` backend, a dictionary with empty data is always returned.
@@ -49,6 +57,12 @@ def memory_stats(device_target=None):
49
57
  {<capsule object NULL at 0x7f7e8c27b030>: {'block_stream_id': 0, 'block_memory_size': 1073741824}}},
50
58
  'persistent_mem_pool_stats': {'block_unit_size': 1073741824, 'block_counts': 0, 'blocks_info': {}}}
51
59
  """
60
+ if not function_memory_status['memory_stats']:
61
+ function_memory_status['memory_stats'] = True
62
+ logger.warning(
63
+ "WARN_DEPRECATED: The usage of mindspore.hal.memory_stats() is deprecated."
64
+ " Please use mindspore.runtime.memory_stats()"
65
+ )
52
66
  if not is_initialized(device_target):
53
67
  logger.warning(f"Backend {device_target} is not initialized yet. Return empty dict.")
54
68
  return {}
@@ -60,6 +74,9 @@ def memory_reserved(device_target=None):
60
74
  """
61
75
  Returns the total amount of memory currently managed by the memory pool.
62
76
 
77
+ Note:
78
+ - The api will be deprecated, please use the api :func:`mindspore.runtime.memory_reserved` instead.
79
+
63
80
  Note:
64
81
  - If `device_target` is not specified, get the device capability of the current backend set by context.
65
82
  - For the `CPU` backend, 0 is always returned.
@@ -81,6 +98,12 @@ def memory_reserved(device_target=None):
81
98
  >>> print(ms.hal.memory_reserved())
82
99
  1073741824
83
100
  """
101
+ if not function_memory_status['memory_reserved']:
102
+ function_memory_status['memory_reserved'] = True
103
+ logger.warning(
104
+ "WARN_DEPRECATED: The usage of mindspore.hal.memory_reserved() is deprecated."
105
+ " Please use mindspore.runtime.memory_reserved()"
106
+ )
84
107
  return _memory_stats(device_target).get("total_reserved_memory", 0)
85
108
 
86
109
 
@@ -89,6 +112,9 @@ def max_memory_reserved(device_target=None):
89
112
  """
90
113
  Returns the peak value of the total memory managed by the memory pool since the process was started.
91
114
 
115
+ Note:
116
+ - The api will be deprecated, please use the api :func:`mindspore.runtime.max_memory_reserved` instead.
117
+
92
118
  Note:
93
119
  - If `device_target` is not specified, get the device capability of the current backend set by context.
94
120
  - For the `CPU` backend, 0 is always returned.
@@ -110,6 +136,12 @@ def max_memory_reserved(device_target=None):
110
136
  >>> print(ms.hal.max_memory_reserved())
111
137
  1073741824
112
138
  """
139
+ if not function_memory_status['max_memory_reserved']:
140
+ function_memory_status['max_memory_reserved'] = True
141
+ logger.warning(
142
+ "WARN_DEPRECATED: The usage of mindspore.hal.max_memory_reserved() is deprecated."
143
+ " Please use mindspore.runtime.max_memory_reserved()"
144
+ )
113
145
  return _memory_stats(device_target).get("max_reserved_memory", 0)
114
146
 
115
147
 
@@ -119,11 +151,16 @@ def empty_cache():
119
151
  Release all memory fragments in the memory pool, so that memory arrangement
120
152
  will be optimized.
121
153
 
154
+ Note:
155
+ - The api will be deprecated, please use the api :func:`mindspore.runtime.empty_cache` instead.
156
+
122
157
  Note:
123
158
  Currently, the MindSpore memory pool does not have the function of releasing memory fragments.
124
159
  This interface is reserved but implemented as an empty method and prompted in log mode.
125
160
  """
126
- logger.warning(f"The empty_cache operation is currently not supported.")
161
+ if not function_memory_status['empty_cache']:
162
+ function_memory_status['empty_cache'] = True
163
+ logger.warning(f"The empty_cache operation is currently not supported.")
127
164
 
128
165
 
129
166
  @_check_inputs_validation
@@ -131,6 +168,9 @@ def reset_peak_memory_stats(device_target=None):
131
168
  """
132
169
  Reset the "peak" stats tracked by memory manager.
133
170
 
171
+ Note:
172
+ - The api will be deprecated, please use the api :func:`mindspore.runtime.reset_peak_memory_stats` instead.
173
+
134
174
  Note:
135
175
  If `device_target` is not specified, get the device capability of the current backend set by context.
136
176
 
@@ -155,6 +195,12 @@ def reset_peak_memory_stats(device_target=None):
155
195
  >>> print(ms.hal.max_memory_allocated())
156
196
  0
157
197
  """
198
+ if not function_memory_status['reset_peak_memory_stats']:
199
+ function_memory_status['reset_peak_memory_stats'] = True
200
+ logger.warning(
201
+ "WARN_DEPRECATED: The usage of mindspore.hal.reset_peak_memory_stats() is deprecated."
202
+ " Please use mindspore.runtime.reset_peak_memory_stats()"
203
+ )
158
204
  _reset_max_mem_reserved(device_target)
159
205
  _reset_max_mem_allocated(device_target)
160
206
 
@@ -164,6 +210,9 @@ def memory_summary(device_target=None):
164
210
  """
165
211
  Returns readable memory pool status information.
166
212
 
213
+ Note:
214
+ - The api will be deprecated, please use the api :func:`mindspore.runtime.memory_summary` instead.
215
+
167
216
  Note:
168
217
  If `device_target` is not specified, get the device capability of the current backend set by context.
169
218
 
@@ -174,6 +223,12 @@ def memory_summary(device_target=None):
174
223
  Returns:
175
224
  str, readable memory pool status information in tabular form.
176
225
  """
226
+ if not function_memory_status['memory_summary']:
227
+ function_memory_status['memory_summary'] = True
228
+ logger.warning(
229
+ "WARN_DEPRECATED: The usage of mindspore.hal.memory_summary() is deprecated."
230
+ " Please use mindspore.runtime.memory_summary()"
231
+ )
177
232
  stats = _memory_stats(device_target)
178
233
 
179
234
  def _format_size(sz, pref_sz):
@@ -189,7 +244,7 @@ def memory_summary(device_target=None):
189
244
 
190
245
  metrics_to_display = [
191
246
  ("total_reserved_memory", "Reserved memory", _format_size),
192
- ("total_allocatd_memory", "Allocated memory", _format_size),
247
+ ("total_allocated_memory", "Allocated memory", _format_size),
193
248
  ("total_idle_memory", "Idle memory", _format_size),
194
249
  ("total_eager_free_memory", "Eager free memory", _format_size),
195
250
  ("max_reserved_memory", "Max reserved memory", _format_size),
@@ -217,6 +272,9 @@ def memory_allocated(device_target=None):
217
272
  """
218
273
  Returns the actual memory size currently occupied by Tensor.
219
274
 
275
+ Note:
276
+ - The api will be deprecated, please use the api :func:`mindspore.runtime.memory_allocated` instead.
277
+
220
278
  Note:
221
279
  - If `device_target` is not specified, get the device capability of the current backend set by context.
222
280
  - For the `CPU` backend, 0 is always returned.
@@ -238,7 +296,13 @@ def memory_allocated(device_target=None):
238
296
  >>> print(ms.hal.memory_allocated())
239
297
  1024
240
298
  """
241
- return _memory_stats(device_target).get("total_allocatd_memory", 0)
299
+ if not function_memory_status['memory_allocated']:
300
+ function_memory_status['memory_allocated'] = True
301
+ logger.warning(
302
+ "WARN_DEPRECATED: The usage of mindspore.hal.memory_allocated() is deprecated."
303
+ " Please use mindspore.runtime.memory_allocated()"
304
+ )
305
+ return _memory_stats(device_target).get("total_allocated_memory", 0)
242
306
 
243
307
 
244
308
  @_check_inputs_validation
@@ -246,6 +310,9 @@ def max_memory_allocated(device_target=None):
246
310
  """
247
311
  Returns the peak memory size of the memory pool actually occupied by Tensor since the process was started.
248
312
 
313
+ Note:
314
+ - The api will be deprecated, please use the api :func:`mindspore.runtime.max_memory_allocated` instead.
315
+
249
316
  Note:
250
317
  - If `device_target` is not specified, get the device capability of the current backend set by context.
251
318
  - For the `CPU` backend, 0 is always returned.
@@ -267,6 +334,12 @@ def max_memory_allocated(device_target=None):
267
334
  >>> print(ms.hal.max_memory_allocated())
268
335
  1536
269
336
  """
337
+ if not function_memory_status['max_memory_allocated']:
338
+ function_memory_status['max_memory_allocated'] = True
339
+ logger.warning(
340
+ "WARN_DEPRECATED: The usage of mindspore.hal.max_memory_allocated() is deprecated."
341
+ " Please use mindspore.runtime.max_memory_allocated()"
342
+ )
270
343
  return _memory_stats(device_target).get("max_allocated_memory", 0)
271
344
 
272
345
 
@@ -275,6 +348,9 @@ def reset_max_memory_reserved(device_target=None):
275
348
  """
276
349
  Reset the peak memory size managed by the memory pool.
277
350
 
351
+ Note:
352
+ - The api will be deprecated, please use the api :func:`mindspore.runtime.reset_max_memory_reserved` instead.
353
+
278
354
  Note:
279
355
  If `device_target` is not specified, get the device capability of the current backend set by context.
280
356
 
@@ -295,6 +371,12 @@ def reset_max_memory_reserved(device_target=None):
295
371
  >>> print(ms.hal.max_memory_reserved())
296
372
  0
297
373
  """
374
+ if not function_memory_status['reset_max_memory_reserved']:
375
+ function_memory_status['reset_max_memory_reserved'] = True
376
+ logger.warning(
377
+ "WARN_DEPRECATED: The usage of mindspore.hal.reset_max_memory_reserved() is deprecated."
378
+ " Please use mindspore.runtime.reset_max_memory_reserved()"
379
+ )
298
380
  _reset_max_mem_reserved(device_target)
299
381
 
300
382
 
@@ -303,6 +385,9 @@ def reset_max_memory_allocated(device_target=None):
303
385
  """
304
386
  Reset the peak memory size of the memory pool actually occupied by Tensor.
305
387
 
388
+ Note:
389
+ - The api will be deprecated, please use the api :func:`mindspore.runtime.reset_max_memory_allocated` instead.
390
+
306
391
  Note:
307
392
  If `device_target` is not specified, get the device capability of the current backend set by context.
308
393
 
@@ -323,4 +408,10 @@ def reset_max_memory_allocated(device_target=None):
323
408
  >>> print(ms.hal.max_memory_allocated())
324
409
  0
325
410
  """
411
+ if not function_memory_status['reset_max_memory_allocated']:
412
+ function_memory_status['reset_max_memory_allocated'] = True
413
+ logger.warning(
414
+ "WARN_DEPRECATED: The usage of mindspore.hal.reset_max_memory_allocated() is deprecated."
415
+ " Please use mindspore.runtime.reset_max_memory_allocated()"
416
+ )
326
417
  _reset_max_mem_allocated(device_target)
mindspore/hal/stream.py CHANGED
@@ -21,8 +21,13 @@ from mindspore._c_expression import default_stream as default_stream_
21
21
  from mindspore._c_expression import communication_stream as communication_stream_
22
22
 
23
23
  from mindspore import _checkparam as Validator
24
+ from mindspore import log as logger
24
25
  from .event import Event
25
26
 
27
+ function_stream_status = {'record_event': False, 'wait_event': False, 'wait_stream': False,
28
+ 'query': False, 'synchronize': False, 'set_cur_stream': False,
29
+ 'current_stream': False, 'default_stream': False,
30
+ 'communication_stream': False, 'StreamCtx': False}
26
31
 
27
32
  class Stream(Stream_):
28
33
  r"""
@@ -31,13 +36,15 @@ class Stream(Stream_):
31
36
  A device stream is a linear sequence of execution that belongs to a specific device,
32
37
  independent from other streams.
33
38
 
34
- For a quick start of using Stream, please refer to `Illustration of stream management <https://www.mindspore.cn/docs/en/master/api_python/samples/hal/stream_manager.html>`_ .
39
+ Note:
40
+ - The api will be deprecated, please use the api :class:`mindspore.runtime.Stream`.
35
41
 
36
42
  Args:
37
43
  priority (int, optional): priority of the stream, lower numbers represent higher priorities.
38
44
  By default, streams have priority ``0``.
39
45
  kwargs (dict): keyword arguments.
40
46
  """
47
+
41
48
  def __init__(self, priority=0, **kwargs):
42
49
  self.init_finished = False
43
50
  if 'stream' in kwargs and kwargs['stream'] is not None:
@@ -84,6 +91,12 @@ class Stream(Stream_):
84
91
  [5. 5. 5.]
85
92
  [5. 5. 5.]]
86
93
  """
94
+ if not function_stream_status['record_event']:
95
+ function_stream_status['record_event'] = True
96
+ logger.warning(
97
+ "WARN_DEPRECATED: The usage of mindspore.hal.Stream().record_event() is deprecated."
98
+ " Please use mindspore.runtime.Stream().record_event()"
99
+ )
87
100
  if event is None:
88
101
  event = Event()
89
102
  if not isinstance(event, Event):
@@ -121,6 +134,12 @@ class Stream(Stream_):
121
134
  [5. 5. 5.]
122
135
  [5. 5. 5.]]
123
136
  """
137
+ if not function_stream_status['wait_event']:
138
+ function_stream_status['wait_event'] = True
139
+ logger.warning(
140
+ "WARN_DEPRECATED: The usage of mindspore.hal.current_stream().wait_event(event) is deprecated."
141
+ " Please use mindspore.runtime.current_stream().wait_event(event)"
142
+ )
124
143
  if not isinstance(event, Event):
125
144
  raise TypeError(f"For 'wait_event', the argument 'event' should be Event,"
126
145
  f" but got {type(event)}.")
@@ -156,6 +175,12 @@ class Stream(Stream_):
156
175
  >>> print(d)
157
176
  [[4. 4.]]
158
177
  """
178
+ if not function_stream_status['wait_stream']:
179
+ function_stream_status['wait_stream'] = True
180
+ logger.warning(
181
+ "WARN_DEPRECATED: The usage of mindspore.hal.Stream() is deprecated."
182
+ " Please use mindspore.runtime.Stream()"
183
+ )
159
184
  if not isinstance(stream, Stream):
160
185
  raise TypeError(f"For 'wait_stream', the argument 'stream' should be Stream,"
161
186
  f" but got {type(stream)}.")
@@ -199,6 +224,12 @@ class Stream(Stream_):
199
224
  >>> s1.synchronize()
200
225
  >>> assert s1.query()
201
226
  """
227
+ if not function_stream_status['query']:
228
+ function_stream_status['query'] = True
229
+ logger.warning(
230
+ "WARN_DEPRECATED: The usage of mindspore.hal.Stream() is deprecated."
231
+ " Please use mindspore.runtime.Stream()"
232
+ )
202
233
  # pylint: disable=useless-super-delegation
203
234
  return super().query()
204
235
 
@@ -221,6 +252,9 @@ def synchronize():
221
252
  r"""
222
253
  Synchronize all streams on current device.(Each MindSpore process only occupies one device)
223
254
 
255
+ Note:
256
+ - The api will be deprecated, please use the api :func:`mindspore.runtime.synchronize` instead.
257
+
224
258
  Examples:
225
259
  >>> import mindspore as ms
226
260
  >>> import numpy as np
@@ -233,6 +267,12 @@ def synchronize():
233
267
  >>> ms.hal.synchronize()
234
268
  >>> assert s1.query()
235
269
  """
270
+ if not function_stream_status['synchronize']:
271
+ function_stream_status['synchronize'] = True
272
+ logger.warning(
273
+ "WARN_DEPRECATED: The usage of mindspore.hal.synchronize() is deprecated."
274
+ " Please use mindspore.runtime.synchronize()"
275
+ )
236
276
  synchronize_()
237
277
 
238
278
 
@@ -241,6 +281,9 @@ def set_cur_stream(stream):
241
281
  Sets the current stream.This is a wrapper API to set the stream.
242
282
  Usage of this function is discouraged in favor of the ``stream`` context manager.
243
283
 
284
+ Note:
285
+ - The api will be deprecated, please use the api :func:`mindspore.runtime.set_cur_stream` instead.
286
+
244
287
  Args:
245
288
  stream (Stream): selected stream. This function is a no-op
246
289
  if this argument is ``None``.
@@ -257,6 +300,12 @@ def set_cur_stream(stream):
257
300
  >>> assert ms.hal.current_stream() == s1
258
301
  >>> ms.hal.set_cur_stream(ms.hal.default_stream())
259
302
  """
303
+ if not function_stream_status['set_cur_stream']:
304
+ function_stream_status['set_cur_stream'] = True
305
+ logger.warning(
306
+ "WARN_DEPRECATED: The usage of mindspore.hal.Stream() is deprecated."
307
+ " Please use mindspore.runtime.Stream()"
308
+ )
260
309
  if stream is None:
261
310
  return
262
311
  if not isinstance(stream, Stream):
@@ -269,6 +318,9 @@ def current_stream():
269
318
  r"""
270
319
  Return current stream used on this device.
271
320
 
321
+ Note:
322
+ - The api will be deprecated, please use the api :func:`mindspore.runtime.current_stream` instead.
323
+
272
324
  Returns:
273
325
  stream (Stream), current stream.
274
326
 
@@ -277,6 +329,12 @@ def current_stream():
277
329
  >>> cur_stream = ms.hal.current_stream()
278
330
  >>> assert cur_stream == ms.hal.default_stream()
279
331
  """
332
+ if not function_stream_status['current_stream']:
333
+ function_stream_status['current_stream'] = True
334
+ logger.warning(
335
+ "WARN_DEPRECATED: The usage of mindspore.hal.current_stream() is deprecated."
336
+ " Please use mindspore.runtime.current_stream()"
337
+ )
280
338
  return Stream(stream=current_stream_())
281
339
 
282
340
 
@@ -284,6 +342,9 @@ def default_stream():
284
342
  r"""
285
343
  Return default stream on this device.
286
344
 
345
+ Note:
346
+ - The api will be deprecated, please use the api :func:`mindspore.runtime.default_stream` instead.
347
+
287
348
  Returns:
288
349
  stream (Stream), default stream.
289
350
 
@@ -292,22 +353,36 @@ def default_stream():
292
353
  >>> cur_stream = ms.hal.current_stream()
293
354
  >>> assert cur_stream == ms.hal.default_stream()
294
355
  """
356
+ if not function_stream_status['default_stream']:
357
+ function_stream_status['default_stream'] = True
358
+ logger.warning(
359
+ "WARN_DEPRECATED: The usage of mindspore.hal.default_stream() is deprecated."
360
+ " Please use mindspore.runtime.default_stream()"
361
+ )
295
362
  return Stream(stream=default_stream_())
296
363
 
297
364
 
298
365
  def communication_stream():
299
366
  r"""
300
- Return default stream on this device.
367
+ Return communication stream on this device.
368
+
369
+ Note:
370
+ - The api will be deprecated, please use the api :func:`mindspore.runtime.communication_stream` instead.
301
371
 
302
372
  Returns:
303
- stream (Stream), default stream.
373
+ stream (Stream), communication stream.
304
374
 
305
- TODO
306
375
  Examples:
307
376
  >>> import mindspore as ms
308
- >>> cur_stream = ms.hal.current_stream()
309
- >>> assert cur_stream == ms.hal.default_stream()
377
+ >>> ms.hal.communication_stream()
378
+ Stream(device_name=Ascend, device_id:0, stream id:1)
310
379
  """
380
+ if not function_stream_status['communication_stream']:
381
+ function_stream_status['communication_stream'] = True
382
+ logger.warning(
383
+ "WARN_DEPRECATED: The usage of mindspore.hal.communication_stream() is deprecated."
384
+ " Please use mindspore.runtime.communication_stream()"
385
+ )
311
386
  return Stream(stream=communication_stream_())
312
387
 
313
388
 
@@ -315,6 +390,9 @@ class StreamCtx():
315
390
  r"""
316
391
  Context-manager that selects a given stream.
317
392
 
393
+ Note:
394
+ - The api will be deprecated, please use the api :class:`mindspore.runtime.StreamCtx`.
395
+
318
396
  All kernels queued within its context will be enqueued on a selected
319
397
  stream.
320
398
 
@@ -336,7 +414,14 @@ class StreamCtx():
336
414
  >>> ms.hal.synchronize()
337
415
  >>> assert s1.query()
338
416
  """
417
+
339
418
  def __init__(self, ctx_stream):
419
+ if not function_stream_status['StreamCtx']:
420
+ function_stream_status['StreamCtx'] = True
421
+ logger.warning(
422
+ "WARN_DEPRECATED: The usage of mindspore.hal.StreamCtx(s1) is deprecated."
423
+ " Please use mindspore.runtime.StreamCtx(s1)"
424
+ )
340
425
  if ctx_stream is not None and not isinstance(ctx_stream, Stream):
341
426
  raise TypeError(f"For 'StreamCtx', the argument 'ctx_stream' should be Stream,"
342
427
  f" but got {type(ctx_stream)}.")
@@ -27,7 +27,6 @@ namespace mindspore {
27
27
  enum DelegateMode {
28
28
  kNoDelegate = 0,
29
29
  kCoreML = 1,
30
- kNNAPI = 2,
31
30
  };
32
31
 
33
32
  enum DeviceType {
@@ -236,7 +235,7 @@ std::string DeviceInfoContext::GetProviderDevice() const { return CharToString(G
236
235
  void DeviceInfoContext::SetProviderDevice(const std::string &device) { SetProviderDevice(StringToChar(device)); }
237
236
 
238
237
  /// \brief Derived from DeviceInfoContext, The configuration of the model running auto on the Host Devices, include
239
- /// CPU/GPU/NPU/Ascend310/Ascend910. This option is only valid for MindSpore Lite.
238
+ /// CPU/GPU/NPU/Ascend. This option is only valid for MindSpore Lite.
240
239
  class MS_API AutoDeviceInfo : public DeviceInfoContext {
241
240
  public:
242
241
  /// \brief Get the type of this DeviceInfoContext.
@@ -108,8 +108,8 @@ enum class DATASET_API ManualOffloadMode {
108
108
  enum class DATASET_API MapTargetDevice {
109
109
  kCpu = 0, ///< CPU Device.
110
110
  kGpu, ///< Gpu Device.
111
- kAscend310, ///< Ascend310 Device.
112
- kAscend910B, ///< Ascend910B Device.
111
+ kAscend310, ///<
112
+ kAscend910B, ///<
113
113
  kInvalid = 100
114
114
  };
115
115
 
mindspore/jpeg62.dll CHANGED
Binary file
mindspore/log.py CHANGED
@@ -26,6 +26,7 @@ from logging.handlers import RotatingFileHandler
26
26
  import traceback
27
27
  import threading
28
28
  import platform
29
+ from datetime import datetime
29
30
 
30
31
  if platform.system() != "Windows":
31
32
  import fcntl
@@ -225,6 +226,17 @@ def critical(msg, *args, **kwargs):
225
226
  _get_logger().critical(msg, *args, **kwargs)
226
227
 
227
228
 
229
+ def vlog_print(level, module, file, line, message):
230
+ '''Read environment variable VLOG_v and print to log'''
231
+ if os.environ.get("VLOG_v") == level:
232
+ now = datetime.now()
233
+ formatted_time = now.strftime("%Y-%m-%d-%H:%M:%S.%f")[:-3] + f".{now.microsecond // 1000}"
234
+ path = 'mindspore' + file.split("mindspore")[-1]
235
+ pid = os.getpid()
236
+ thread_id = threading.get_ident()
237
+ print(f"[V{level}] {module}({pid},{thread_id},python):{formatted_time} [{path}:{line}] {message}", flush=True)
238
+
239
+
228
240
  def get_level():
229
241
  """
230
242
  Get the logger level.
@@ -39,5 +39,5 @@ from .config import *
39
39
 
40
40
  __all__ = ['FileWriter', 'FileReader', 'MindPage',
41
41
  'Cifar10ToMR', 'Cifar100ToMR', 'CsvToMR', 'ImageNetToMR', 'MnistToMR', 'TFRecordToMR',
42
- 'set_enc_key', 'set_enc_mode', 'set_dec_mode', 'set_hash_mode',
42
+ 'set_enc_key', 'set_enc_mode', 'set_dec_mode',
43
43
  'SUCCESS', 'FAILED']