mindspore 2.4.1__cp310-cp310-win_amd64.whl → 2.5.0__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.
- mindspore/.commit_id +1 -1
- mindspore/__init__.py +8 -3
- mindspore/_c_dataengine.cp310-win_amd64.pyd +0 -0
- mindspore/_c_expression.cp310-win_amd64.pyd +0 -0
- mindspore/_c_mindrecord.cp310-win_amd64.pyd +0 -0
- mindspore/_checkparam.py +0 -5
- mindspore/_extends/parallel_compile/akg_compiler/gen_custom_op_files.py +1 -1
- mindspore/_extends/parse/compile_config.py +64 -0
- mindspore/_extends/parse/deprecated/__init__.py +0 -0
- mindspore/_extends/parse/deprecated/deprecated_tensor_method.py +375 -0
- mindspore/_extends/parse/parser.py +23 -5
- mindspore/_extends/parse/standard_method.py +123 -27
- mindspore/_extends/pijit/pijit_func_white_list.py +1 -1
- mindspore/amp.py +7 -1
- mindspore/avcodec-59.dll +0 -0
- mindspore/avdevice-59.dll +0 -0
- mindspore/avfilter-8.dll +0 -0
- mindspore/avformat-59.dll +0 -0
- mindspore/avutil-57.dll +0 -0
- mindspore/boost/boost_cell_wrapper.py +136 -41
- mindspore/common/__init__.py +3 -1
- mindspore/common/_register_for_tensor.py +0 -1
- mindspore/common/_stub_tensor.py +25 -4
- mindspore/common/_tensor_cpp_method.py +17 -0
- mindspore/common/_tensor_docs.py +6132 -0
- mindspore/common/api.py +99 -25
- mindspore/common/dtype.py +34 -34
- mindspore/common/dump.py +2 -1
- mindspore/common/file_system.py +8 -1
- mindspore/common/generator.py +2 -0
- mindspore/common/hook_handle.py +3 -1
- mindspore/common/initializer.py +3 -4
- mindspore/common/lazy_inline.py +8 -2
- mindspore/common/mindir_util.py +10 -2
- mindspore/common/parameter.py +30 -27
- mindspore/common/tensor.py +713 -1337
- mindspore/communication/__init__.py +1 -1
- mindspore/communication/_comm_helper.py +10 -0
- mindspore/communication/comm_func.py +215 -173
- mindspore/communication/management.py +23 -20
- mindspore/context.py +292 -193
- mindspore/dataset/__init__.py +23 -19
- mindspore/dataset/callback/ds_callback.py +2 -1
- mindspore/dataset/core/config.py +84 -3
- mindspore/dataset/engine/cache_admin.py +3 -3
- mindspore/dataset/engine/cache_client.py +5 -4
- mindspore/dataset/engine/datasets.py +192 -149
- mindspore/dataset/engine/datasets_audio.py +14 -0
- mindspore/dataset/engine/datasets_standard_format.py +28 -11
- mindspore/dataset/engine/datasets_text.py +38 -1
- mindspore/dataset/engine/datasets_user_defined.py +125 -65
- mindspore/dataset/engine/datasets_vision.py +81 -8
- mindspore/dataset/engine/iterators.py +281 -63
- mindspore/dataset/engine/obs/util.py +8 -0
- mindspore/dataset/engine/queue.py +40 -0
- mindspore/dataset/engine/samplers.py +26 -2
- mindspore/dataset/engine/serializer_deserializer.py +1 -1
- mindspore/dataset/engine/validators.py +43 -11
- mindspore/dataset/transforms/py_transforms_util.py +17 -0
- mindspore/dataset/transforms/transforms.py +29 -12
- mindspore/dataset/vision/validators.py +1 -2
- mindspore/device_context/__init__.py +21 -0
- mindspore/device_context/ascend/__init__.py +25 -0
- mindspore/device_context/ascend/device.py +72 -0
- mindspore/device_context/ascend/op_debug.py +94 -0
- mindspore/device_context/ascend/op_precision.py +193 -0
- mindspore/device_context/ascend/op_tuning.py +127 -0
- mindspore/device_context/cpu/__init__.py +25 -0
- mindspore/device_context/cpu/device.py +62 -0
- mindspore/device_context/cpu/op_tuning.py +43 -0
- mindspore/device_context/gpu/__init__.py +21 -0
- mindspore/device_context/gpu/device.py +70 -0
- mindspore/device_context/gpu/op_precision.py +67 -0
- mindspore/device_context/gpu/op_tuning.py +175 -0
- mindspore/device_manager.py +134 -0
- mindspore/dnnl.dll +0 -0
- mindspore/experimental/llm_boost/__init__.py +3 -2
- mindspore/experimental/llm_boost/ascend_native/__init__.py +22 -0
- mindspore/experimental/llm_boost/ascend_native/llama_boost_ascend_native.py +211 -0
- mindspore/experimental/llm_boost/ascend_native/llm_boost.py +52 -0
- mindspore/experimental/llm_boost/atb/boost_base.py +239 -64
- mindspore/experimental/llm_boost/atb/llama_boost.py +52 -30
- mindspore/experimental/llm_boost/atb/qwen_boost.py +47 -24
- mindspore/experimental/llm_boost/register.py +1 -0
- mindspore/experimental/optim/adadelta.py +26 -22
- mindspore/experimental/optim/adam.py +3 -0
- mindspore/experimental/optim/lr_scheduler.py +33 -24
- mindspore/experimental/optim/radam.py +33 -30
- mindspore/hal/device.py +28 -0
- mindspore/hal/event.py +17 -0
- mindspore/hal/memory.py +94 -3
- mindspore/hal/stream.py +91 -6
- mindspore/include/api/context.h +1 -2
- mindspore/include/dataset/constants.h +2 -2
- mindspore/jpeg62.dll +0 -0
- mindspore/log.py +12 -0
- mindspore/mindrecord/__init__.py +1 -1
- mindspore/mindrecord/config.py +17 -316
- mindspore/mindrecord/filereader.py +1 -9
- mindspore/mindrecord/filewriter.py +5 -15
- mindspore/mindrecord/mindpage.py +1 -9
- mindspore/mindspore_backend.dll +0 -0
- mindspore/mindspore_common.dll +0 -0
- mindspore/mindspore_core.dll +0 -0
- mindspore/mindspore_glog.dll +0 -0
- mindspore/mindspore_ops.dll +0 -0
- mindspore/mint/__init__.py +824 -218
- mindspore/mint/distributed/__init__.py +66 -4
- mindspore/mint/distributed/distributed.py +2594 -44
- mindspore/mint/linalg/__init__.py +6 -0
- mindspore/mint/nn/__init__.py +473 -14
- mindspore/mint/nn/functional.py +486 -11
- mindspore/mint/nn/layer/__init__.py +17 -4
- mindspore/mint/nn/layer/_functions.py +330 -0
- mindspore/mint/nn/layer/activation.py +169 -1
- mindspore/mint/nn/layer/basic.py +123 -0
- mindspore/mint/nn/layer/conv.py +727 -0
- mindspore/mint/nn/layer/normalization.py +215 -19
- mindspore/mint/nn/layer/padding.py +797 -0
- mindspore/mint/nn/layer/pooling.py +170 -0
- mindspore/mint/optim/__init__.py +2 -1
- mindspore/mint/optim/adam.py +223 -0
- mindspore/mint/optim/adamw.py +26 -19
- mindspore/mint/special/__init__.py +2 -1
- mindspore/multiprocessing/__init__.py +5 -0
- mindspore/nn/__init__.py +2 -0
- mindspore/nn/cell.py +142 -21
- mindspore/nn/dynamic_lr.py +2 -1
- mindspore/nn/layer/activation.py +6 -6
- mindspore/nn/layer/basic.py +35 -25
- mindspore/nn/layer/channel_shuffle.py +3 -3
- mindspore/nn/layer/conv.py +3 -0
- mindspore/nn/layer/embedding.py +3 -3
- mindspore/nn/layer/normalization.py +8 -7
- mindspore/nn/layer/padding.py +4 -3
- mindspore/nn/layer/pooling.py +55 -23
- mindspore/nn/layer/rnn_cells.py +1 -1
- mindspore/nn/layer/rnns.py +2 -1
- mindspore/nn/layer/timedistributed.py +5 -5
- mindspore/nn/layer/transformer.py +48 -26
- mindspore/nn/learning_rate_schedule.py +5 -3
- mindspore/nn/loss/loss.py +31 -36
- mindspore/nn/optim/ada_grad.py +1 -0
- mindspore/nn/optim/adadelta.py +2 -2
- mindspore/nn/optim/adam.py +1 -1
- mindspore/nn/optim/lars.py +1 -4
- mindspore/nn/optim/optimizer.py +1 -1
- mindspore/nn/optim/rprop.py +2 -2
- mindspore/nn/optim/thor.py +2 -1
- mindspore/nn/utils/__init__.py +22 -0
- mindspore/nn/utils/init.py +73 -0
- mindspore/nn/wrap/cell_wrapper.py +4 -6
- mindspore/nn/wrap/loss_scale.py +3 -4
- mindspore/numpy/array_creations.py +60 -62
- mindspore/numpy/array_ops.py +148 -143
- mindspore/numpy/logic_ops.py +41 -42
- mindspore/numpy/math_ops.py +361 -359
- mindspore/numpy/utils.py +16 -16
- mindspore/numpy/utils_const.py +4 -4
- mindspore/opencv_core452.dll +0 -0
- mindspore/opencv_imgcodecs452.dll +0 -0
- mindspore/opencv_imgproc452.dll +0 -0
- mindspore/ops/__init__.py +2 -1
- mindspore/ops/_grad_experimental/grad_comm_ops.py +107 -8
- mindspore/ops/_grad_experimental/grad_debug_ops.py +6 -1
- mindspore/ops/_grad_experimental/grad_inner_ops.py +9 -0
- mindspore/ops/_grad_experimental/grad_math_ops.py +2 -1
- mindspore/ops/_op_impl/cpu/__init__.py +1 -0
- mindspore/ops/_op_impl/cpu/raise_op.py +28 -0
- mindspore/ops/_vmap/vmap_array_ops.py +20 -19
- mindspore/ops/_vmap/vmap_base.py +0 -2
- mindspore/ops/_vmap/vmap_grad_nn_ops.py +19 -13
- mindspore/ops/_vmap/vmap_math_ops.py +11 -9
- mindspore/ops/_vmap/vmap_nn_ops.py +20 -34
- mindspore/ops/auto_generate/cpp_create_prim_instance_helper.py +149 -12
- mindspore/ops/auto_generate/gen_arg_handler.py +0 -61
- mindspore/ops/auto_generate/gen_extend_func.py +554 -60
- mindspore/ops/auto_generate/gen_ops_def.py +1621 -115
- mindspore/ops/auto_generate/gen_ops_prim.py +8027 -3411
- mindspore/ops/auto_generate/pyboost_inner_prim.py +183 -79
- mindspore/ops/composite/base.py +1 -1
- mindspore/ops/composite/multitype_ops/_compile_utils.py +229 -30
- mindspore/ops/composite/multitype_ops/pow_impl.py +0 -29
- mindspore/ops/function/__init__.py +12 -0
- mindspore/ops/function/array_func.py +561 -159
- mindspore/ops/function/clip_func.py +64 -0
- mindspore/ops/function/debug_func.py +28 -20
- mindspore/ops/function/image_func.py +1 -1
- mindspore/ops/function/linalg_func.py +5 -4
- mindspore/ops/function/math_func.py +1664 -294
- mindspore/ops/function/nn_func.py +988 -317
- mindspore/ops/function/parameter_func.py +3 -56
- mindspore/ops/function/random_func.py +243 -33
- mindspore/ops/function/sparse_unary_func.py +1 -1
- mindspore/ops/functional.py +18 -5
- mindspore/ops/functional_overload.py +897 -0
- mindspore/ops/operations/__init__.py +3 -2
- mindspore/ops/operations/_embedding_cache_ops.py +4 -4
- mindspore/ops/operations/_grad_ops.py +2 -34
- mindspore/ops/operations/_infer_ops.py +2 -1
- mindspore/ops/operations/_inner_ops.py +38 -8
- mindspore/ops/operations/array_ops.py +45 -303
- mindspore/ops/operations/comm_ops.py +23 -17
- mindspore/ops/operations/custom_ops.py +7 -49
- mindspore/ops/operations/debug_ops.py +42 -47
- mindspore/ops/operations/inner_ops.py +6 -4
- mindspore/ops/operations/linalg_ops.py +3 -2
- mindspore/ops/operations/manually_defined/ops_def.py +185 -104
- mindspore/ops/operations/math_ops.py +11 -216
- mindspore/ops/operations/nn_ops.py +153 -310
- mindspore/ops/primitive.py +23 -21
- mindspore/ops/tensor_method.py +1669 -0
- mindspore/ops_generate/aclnn_kernel_register_auto_cc_generator.py +110 -0
- mindspore/ops_generate/add_tensor_docs_generator.py +54 -0
- mindspore/ops_generate/arg_handler.py +0 -61
- mindspore/ops_generate/auto_grad_impl_cc_generator.py +135 -0
- mindspore/ops_generate/auto_grad_reg_cc_generator.py +93 -0
- mindspore/ops_generate/base_generator.py +11 -0
- mindspore/ops_generate/cpp_create_prim_instance_helper_generator.py +108 -0
- mindspore/ops_generate/functional_map_cpp_generator.py +491 -0
- mindspore/ops_generate/functional_overload_py_generator.py +110 -0
- mindspore/ops_generate/functions_cc_generator.py +233 -0
- mindspore/ops_generate/gen_aclnn_implement.py +110 -114
- mindspore/ops_generate/gen_constants.py +157 -3
- mindspore/ops_generate/gen_ops.py +245 -990
- mindspore/ops_generate/gen_pyboost_func.py +97 -998
- mindspore/ops_generate/gen_utils.py +119 -33
- mindspore/ops_generate/lite_ops_cpp_generator.py +155 -0
- mindspore/ops_generate/op_api_proto.py +206 -0
- mindspore/ops_generate/op_def_py_generator.py +131 -0
- mindspore/ops_generate/op_prim_py_generator.py +480 -0
- mindspore/ops_generate/op_proto.py +373 -108
- mindspore/ops_generate/op_template_parser.py +436 -0
- mindspore/ops_generate/ops_def_cc_generator.py +288 -0
- mindspore/ops_generate/ops_def_h_generator.py +74 -0
- mindspore/ops_generate/ops_name_h_generator.py +68 -0
- mindspore/ops_generate/ops_primitive_h_generator.py +81 -0
- mindspore/ops_generate/pyboost_functions_cpp_generator.py +370 -0
- mindspore/ops_generate/pyboost_functions_h_generator.py +68 -0
- mindspore/ops_generate/pyboost_functions_py_generator.py +148 -0
- mindspore/ops_generate/pyboost_grad_function_cpp_generator.py +154 -0
- mindspore/ops_generate/pyboost_inner_prim_generator.py +131 -0
- mindspore/ops_generate/pyboost_native_grad_functions_generator.py +268 -0
- mindspore/ops_generate/pyboost_op_cpp_code_generator.py +851 -0
- mindspore/ops_generate/pyboost_overload_functions_cpp_generator.py +344 -0
- mindspore/ops_generate/pyboost_utils.py +92 -33
- mindspore/ops_generate/template.py +294 -44
- mindspore/ops_generate/tensor_func_reg_cpp_generator.py +422 -0
- mindspore/parallel/__init__.py +3 -3
- mindspore/parallel/_auto_parallel_context.py +44 -34
- mindspore/parallel/_cell_wrapper.py +22 -3
- mindspore/parallel/_parallel_serialization.py +13 -2
- mindspore/parallel/_utils.py +4 -2
- mindspore/parallel/algo_parameter_config.py +1 -1
- mindspore/parallel/checkpoint_transform.py +44 -0
- mindspore/parallel/cluster/process_entity/_api.py +131 -37
- mindspore/parallel/cluster/process_entity/_utils.py +41 -6
- mindspore/parallel/cluster/run.py +20 -3
- mindspore/parallel/parameter_broadcast.py +1 -1
- mindspore/parallel/shard.py +3 -0
- mindspore/parallel/transform_safetensors.py +119 -253
- mindspore/profiler/__init__.py +17 -4
- mindspore/profiler/analysis/__init__.py +0 -0
- mindspore/profiler/analysis/parser/__init__.py +0 -0
- mindspore/profiler/analysis/parser/ascend_cann_parser.py +166 -0
- mindspore/profiler/analysis/parser/base_parser.py +158 -0
- mindspore/profiler/analysis/parser/framework_cann_relation_parser.py +45 -0
- mindspore/profiler/analysis/parser/ms_framework_parser.py +142 -0
- mindspore/profiler/analysis/parser/ms_minddata_parser.py +145 -0
- mindspore/profiler/analysis/parser/timeline_assembly_factory/__init__.py +0 -0
- mindspore/profiler/analysis/parser/timeline_assembly_factory/ascend_timeline_assembler.py +261 -0
- mindspore/profiler/analysis/parser/timeline_assembly_factory/base_timeline_assembler.py +40 -0
- mindspore/profiler/analysis/parser/timeline_assembly_factory/trace_view_container.py +84 -0
- mindspore/profiler/analysis/parser/timeline_creator/__init__.py +0 -0
- mindspore/profiler/analysis/parser/timeline_creator/base_timeline_creator.py +44 -0
- mindspore/profiler/analysis/parser/timeline_creator/cpu_op_timeline_creator.py +90 -0
- mindspore/profiler/analysis/parser/timeline_creator/fwk_timeline_creator.py +76 -0
- mindspore/profiler/analysis/parser/timeline_creator/msprof_timeline_creator.py +103 -0
- mindspore/profiler/analysis/parser/timeline_creator/scope_layer_timeline_creator.py +134 -0
- mindspore/profiler/analysis/parser/timeline_event/__init__.py +0 -0
- mindspore/profiler/analysis/parser/timeline_event/base_event.py +233 -0
- mindspore/profiler/analysis/parser/timeline_event/cpu_op_event.py +47 -0
- mindspore/profiler/analysis/parser/timeline_event/flow_event.py +36 -0
- mindspore/profiler/analysis/parser/timeline_event/fwk_event.py +260 -0
- mindspore/profiler/analysis/parser/timeline_event/msprof_event.py +73 -0
- mindspore/profiler/analysis/parser/timeline_event/scope_layer_event.py +53 -0
- mindspore/profiler/analysis/parser/timeline_event/timeline_event_pool.py +146 -0
- mindspore/profiler/analysis/task_manager.py +131 -0
- mindspore/profiler/analysis/time_converter.py +84 -0
- mindspore/profiler/analysis/viewer/__init__.py +0 -0
- mindspore/profiler/analysis/viewer/ascend_communication_viewer.py +333 -0
- mindspore/profiler/analysis/viewer/ascend_integrate_viewer.py +87 -0
- mindspore/profiler/analysis/viewer/ascend_kernel_details_viewer.py +252 -0
- mindspore/profiler/analysis/viewer/ascend_memory_viewer.py +313 -0
- mindspore/profiler/analysis/viewer/ascend_op_memory_viewer.py +322 -0
- mindspore/profiler/analysis/viewer/ascend_step_trace_time_viewer.py +265 -0
- mindspore/profiler/analysis/viewer/ascend_timeline_viewer.py +58 -0
- mindspore/profiler/analysis/viewer/base_viewer.py +26 -0
- mindspore/profiler/analysis/viewer/ms_dataset_viewer.py +97 -0
- mindspore/profiler/analysis/viewer/ms_minddata_viewer.py +581 -0
- mindspore/profiler/analysis/work_flow.py +73 -0
- mindspore/profiler/common/ascend_msprof_exporter.py +138 -0
- mindspore/profiler/common/command_executor.py +90 -0
- mindspore/profiler/common/constant.py +174 -3
- mindspore/profiler/common/file_manager.py +208 -0
- mindspore/profiler/common/log.py +130 -0
- mindspore/profiler/common/msprof_cmd_tool.py +202 -0
- mindspore/profiler/common/path_manager.py +371 -0
- mindspore/profiler/common/process_bar.py +168 -0
- mindspore/profiler/common/process_pool.py +9 -3
- mindspore/profiler/common/profiler_context.py +476 -0
- mindspore/profiler/common/profiler_info.py +304 -0
- mindspore/profiler/common/profiler_output_path.py +284 -0
- mindspore/profiler/common/profiler_parameters.py +210 -0
- mindspore/profiler/common/profiler_path_manager.py +120 -0
- mindspore/profiler/common/record_function.py +76 -0
- mindspore/profiler/common/tlv_decoder.py +76 -0
- mindspore/profiler/common/util.py +75 -2
- mindspore/profiler/dynamic_profiler.py +270 -37
- mindspore/profiler/envprofiler.py +138 -0
- mindspore/profiler/mstx.py +199 -0
- mindspore/profiler/platform/__init__.py +21 -0
- mindspore/profiler/platform/base_profiler.py +40 -0
- mindspore/profiler/platform/cpu_profiler.py +124 -0
- mindspore/profiler/platform/gpu_profiler.py +74 -0
- mindspore/profiler/platform/npu_profiler.py +309 -0
- mindspore/profiler/profiler.py +580 -93
- mindspore/profiler/profiler_action_controller.py +187 -0
- mindspore/profiler/profiler_interface.py +114 -0
- mindspore/profiler/schedule.py +208 -0
- mindspore/rewrite/api/symbol_tree.py +1 -2
- mindspore/run_check/_check_version.py +18 -13
- mindspore/runtime/__init__.py +37 -0
- mindspore/runtime/device.py +27 -0
- mindspore/runtime/event.py +209 -0
- mindspore/runtime/executor.py +148 -0
- mindspore/runtime/memory.py +392 -0
- mindspore/runtime/stream.py +460 -0
- mindspore/runtime/thread_bind_core.py +401 -0
- mindspore/swresample-4.dll +0 -0
- mindspore/swscale-6.dll +0 -0
- mindspore/tinyxml2.dll +0 -0
- mindspore/train/__init__.py +2 -2
- mindspore/train/_utils.py +53 -18
- mindspore/train/amp.py +8 -4
- mindspore/train/callback/_checkpoint.py +32 -18
- mindspore/train/callback/_early_stop.py +1 -1
- mindspore/train/callback/_flops_collector.py +105 -69
- mindspore/train/callback/_history.py +1 -1
- mindspore/train/callback/_summary_collector.py +44 -6
- mindspore/train/callback/_tft_register.py +37 -15
- mindspore/train/dataset_helper.py +11 -11
- mindspore/train/metrics/precision.py +4 -5
- mindspore/train/mind_ir_pb2.py +167 -46
- mindspore/train/model.py +13 -14
- mindspore/train/serialization.py +461 -72
- mindspore/train/summary/summary_record.py +1 -2
- mindspore/train/train_thor/model_thor.py +1 -1
- mindspore/turbojpeg.dll +0 -0
- mindspore/utils/__init__.py +4 -2
- mindspore/utils/dryrun.py +138 -0
- mindspore/utils/runtime_execution_order_check.py +550 -0
- mindspore/version.py +1 -1
- {mindspore-2.4.1.dist-info → mindspore-2.5.0.dist-info}/METADATA +3 -4
- {mindspore-2.4.1.dist-info → mindspore-2.5.0.dist-info}/RECORD +368 -242
- {mindspore-2.4.1.dist-info → mindspore-2.5.0.dist-info}/entry_points.txt +1 -1
- mindspore/common/_tensor_overload.py +0 -139
- mindspore/mindspore_np_dtype.dll +0 -0
- mindspore/profiler/envprofiling.py +0 -254
- mindspore/profiler/profiling.py +0 -1926
- {mindspore-2.4.1.dist-info → mindspore-2.5.0.dist-info}/WHEEL +0 -0
- {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
|
-
|
|
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
|
-
("
|
|
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
|
-
|
|
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
|
-
|
|
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
|
|
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),
|
|
373
|
+
stream (Stream), communication stream.
|
|
304
374
|
|
|
305
|
-
TODO
|
|
306
375
|
Examples:
|
|
307
376
|
>>> import mindspore as ms
|
|
308
|
-
>>>
|
|
309
|
-
|
|
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)}.")
|
mindspore/include/api/context.h
CHANGED
|
@@ -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/
|
|
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, ///<
|
|
112
|
-
kAscend910B, ///<
|
|
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.
|
mindspore/mindrecord/__init__.py
CHANGED
|
@@ -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',
|
|
42
|
+
'set_enc_key', 'set_enc_mode', 'set_dec_mode',
|
|
43
43
|
'SUCCESS', 'FAILED']
|