mindspore 2.4.10__cp311-none-any.whl → 2.5.0__cp311-none-any.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/Third_Party_Open_Source_Software_Notice +39 -0
- mindspore/__init__.py +8 -3
- mindspore/_akg/akg/composite/build_module.py +6 -2
- mindspore/_akg/akg/utils/kernel_exec.py +2 -2
- mindspore/_c_dataengine.cpython-311-aarch64-linux-gnu.so +0 -0
- mindspore/_c_expression.cpython-311-aarch64-linux-gnu.so +0 -0
- mindspore/_c_mindrecord.cpython-311-aarch64-linux-gnu.so +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/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 +98 -21
- mindspore/common/dtype.py +34 -34
- mindspore/common/dump.py +2 -1
- mindspore/common/file_system.py +8 -3
- 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 +31 -15
- mindspore/common/tensor.py +713 -1337
- mindspore/communication/__init__.py +1 -1
- mindspore/communication/_comm_helper.py +5 -0
- mindspore/communication/comm_func.py +215 -173
- mindspore/communication/management.py +23 -20
- mindspore/context.py +285 -191
- 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 +11 -11
- mindspore/dataset/engine/datasets_text.py +38 -1
- mindspore/dataset/engine/datasets_user_defined.py +100 -66
- 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/experimental/llm_boost/__init__.py +1 -0
- 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 +2 -3
- mindspore/experimental/llm_boost/atb/llama_boost.py +6 -1
- 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 +0 -1
- mindspore/lib/libavcodec.so.59 +0 -0
- mindspore/lib/libavdevice.so.59 +0 -0
- mindspore/lib/libavfilter.so.8 +0 -0
- mindspore/lib/libavformat.so.59 +0 -0
- mindspore/lib/libavutil.so.57 +0 -0
- mindspore/lib/libdnnl.so.2 +0 -0
- mindspore/lib/libmindspore_backend.so +0 -0
- mindspore/lib/libmindspore_common.so +0 -0
- mindspore/lib/libmindspore_core.so +0 -0
- mindspore/lib/libmindspore_glog.so.0 +0 -0
- mindspore/lib/libmindspore_gpr.so.15 +0 -0
- mindspore/lib/libmindspore_grpc++.so.1 +0 -0
- mindspore/lib/libmindspore_grpc.so.15 +0 -0
- mindspore/lib/libmindspore_ops.so +0 -0
- mindspore/lib/libmpi_adapter.so +0 -0
- mindspore/lib/libmpi_collective.so +0 -0
- mindspore/lib/libnnacl.so +0 -0
- mindspore/lib/libopencv_core.so.4.5 +0 -0
- mindspore/lib/libopencv_imgcodecs.so.4.5 +0 -0
- mindspore/lib/libopencv_imgproc.so.4.5 +0 -0
- mindspore/lib/libps_cache.so +0 -0
- mindspore/lib/libswresample.so.4 +0 -0
- mindspore/lib/libswscale.so.6 +0 -0
- mindspore/lib/plugin/ascend/custom_aicore_ops/op_impl/ai_core/tbe/config/ascend910_93/aic-ascend910_93-ops-info.json +2048 -0
- mindspore/lib/plugin/ascend/custom_aicpu_ops/op_impl/cpu/aicpu_kernel/impl/libcust_cpu_kernels.so +0 -0
- mindspore/lib/plugin/ascend/custom_aicpu_ops/op_proto/libcust_op_proto.so +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_api/lib/libcust_opapi.so +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/custom_ascendc_910_impl/dynamic/decoder_kv_cache.py +1 -1
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/custom_ascendc_910_impl/dynamic/prompt_kv_cache.py +1 -1
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/op_tiling/lib/linux/aarch64/libcust_opmaster_rt2.0.so +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/op_tiling/liboptiling.so +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_proto/lib/linux/aarch64/libcust_opsproto_rt2.0.so +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910/version.info +1 -1
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_api/lib/libcust_opapi.so +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/config/ascend910_93/aic-ascend910_93-ops-info.json +224 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/custom_ascendc_910b_impl/dynamic/all_finite.py +1 -1
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/custom_ascendc_910b_impl/dynamic/decoder_kv_cache.py +1 -1
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/custom_ascendc_910b_impl/dynamic/prompt_kv_cache.py +1 -1
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/all_finite/AllFinite_52f59e2a65d9b1bb002de35c2819754a.json +78 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/all_finite/AllFinite_52f59e2a65d9b1bb002de35c2819754a.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/all_finite/AllFinite_6b5e50e30256d85838d6ce83514df20f.json +78 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/all_finite/AllFinite_6b5e50e30256d85838d6ce83514df20f.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/all_finite/AllFinite_74e4ac02880d452e3308c94af273562e.json +78 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/all_finite/AllFinite_74e4ac02880d452e3308c94af273562e.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_0d5520cc587ad44ce634bf3fbcffc272.json +156 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_0d5520cc587ad44ce634bf3fbcffc272.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_20390d30b3c4c0d23167ccca6c030c2b.json +156 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_20390d30b3c4c0d23167ccca6c030c2b.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_2d151f0b1d2db51faa2968d5b67544e2.json +156 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_2d151f0b1d2db51faa2968d5b67544e2.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_561690ec17cc1def3d2fcf68c1b07b56.json +156 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_561690ec17cc1def3d2fcf68c1b07b56.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_570f9aaa99e5e773b3dd0a33784363f4.json +156 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_570f9aaa99e5e773b3dd0a33784363f4.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_59668a0f0764afb98fda8ab9e84126f1.json +156 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_59668a0f0764afb98fda8ab9e84126f1.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_91d9833e4792b70b670e4e2b916abd86.json +156 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_91d9833e4792b70b670e4e2b916abd86.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_c74cdc5fef094383401856f8519504af.json +156 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_c74cdc5fef094383401856f8519504af.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_0515c7b1a4cd614449e38c5e9a7e3f8d.json +165 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_0515c7b1a4cd614449e38c5e9a7e3f8d.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_09f22d898d6358c91e7c4fc48bac48e7.json +165 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_09f22d898d6358c91e7c4fc48bac48e7.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_0cb9a6f894b925250227136e5aab7061.json +165 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_0cb9a6f894b925250227136e5aab7061.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_2fa8702ffd7ca85e9e194f62644415d5.json +165 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_2fa8702ffd7ca85e9e194f62644415d5.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_570b62f187dfd439b64613d881deedb7.json +165 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_570b62f187dfd439b64613d881deedb7.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_585218c11411ff84709b9e725b66c435.json +165 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_585218c11411ff84709b9e725b66c435.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_5c9365ccde170b358c5b126d69dae13e.json +165 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_5c9365ccde170b358c5b126d69dae13e.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_6d97c45b7c43bc16fcff8baa5dacac4e.json +165 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_6d97c45b7c43bc16fcff8baa5dacac4e.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/config/ascend910_93/all_finite.json +139 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/config/ascend910_93/binary_info_config.json +361 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/config/ascend910_93/decoder_kv_cache.json +892 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/config/ascend910_93/prompt_kv_cache.json +892 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/op_tiling/lib/linux/aarch64/libcust_opmaster_rt2.0.so +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/op_tiling/liboptiling.so +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_proto/lib/linux/aarch64/libcust_opsproto_rt2.0.so +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/version.info +1 -1
- mindspore/lib/plugin/ascend/custom_compiler/setup.py +1 -1
- mindspore/lib/plugin/ascend/libascend_collective.so +0 -0
- mindspore/lib/plugin/ascend/libdvpp_utils.so +0 -0
- mindspore/lib/plugin/ascend/liblowlatency_collective.so +0 -0
- mindspore/lib/plugin/ascend/libmindspore_cpu_kernels.so +0 -0
- mindspore/lib/plugin/ascend/libmindspore_internal_kernels.so +0 -0
- mindspore/lib/plugin/ascend/libms_ascend_native_boost.so +0 -0
- mindspore/lib/plugin/ascend/libms_atb_boost.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/device/ascend910b/bin/ascend910b.bin +957 -955
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/lib/libasdops_static.a +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/lib/liblcal_static.a +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/{acme/include/base_type.h → base_type.h} +25 -20
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/{cast/cast_tiling.h → internal.h} +6 -4
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/internal_op.h +114 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/llm/boost_kernel.h +70 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/llm/llama_impl.h +85 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/llm/model_interface.h +52 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/llm/tensor.h +81 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/op_creator.h +123 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/op_param.h +155 -110
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/{acme/include/tiling_info.h → tiling_info.h} +12 -9
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/tiling_utils.h +178 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libadd_layer_norm_op.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libadd_rms_norm_op.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libadd_rms_norm_quant_op.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libapply_rotary_pos_emb_310p_op.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libapply_rotary_pos_emb_op.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libcast_op.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libcompare_op.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libgelu_op.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libllama_op.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libmatmul_op.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libms_kernels_internal.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libms_optiling.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libmulti_weight_matmul_kernel_op.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libreshape_and_cache_nz_op.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libreshape_and_cache_op.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/librms_norm_op.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/object_kernels/internal_pp_matmul_f16_nz/internal_pp_matmul_f16_nz.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/object_kernels/internal_pp_matmul_f16_nz/internal_pp_matmul_f16_nz_0.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/object_kernels/internal_pp_matmul_i8_nz_compress/internal_pp_matmul_i8_nz_compress.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/object_kernels/internal_pp_matmul_i8_nz_compress/internal_pp_matmul_i8_nz_compress_0.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/object_kernels/internal_pp_matmul_int8_nz/internal_pp_matmul_int8_nz.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/object_kernels/internal_pp_matmul_int8_nz/internal_pp_matmul_int8_nz_0.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/so_kernels/libadd_rms_norm_quant_ascend310p.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/{lib/libapply_rotary_pos_emb_310p_impl.so → op_kernels/ascend310p/so_kernels/libapply_rotary_pos_emb_310p_ascend310p.so} +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/so_kernels/libcast_ascend310p.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/so_kernels/libcompare_ascend310p.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/so_kernels/libgelu_ascend310p.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/so_kernels/libmatmul_ascend310p.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/so_kernels/libreshape_and_cache_nz_ascend310p.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/hphol_kernels/add_rms_norm_dynamic_quant/AddRmsNormDynamicQuant_4b60f88cdc28b25a36bad2d8b0a88092.json +163 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/hphol_kernels/add_rms_norm_dynamic_quant/AddRmsNormDynamicQuant_4b60f88cdc28b25a36bad2d8b0a88092.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/hphol_kernels/add_rms_norm_dynamic_quant/AddRmsNormDynamicQuant_cde61da2bd6fededcb1ba310a6ad16ee.json +163 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/hphol_kernels/add_rms_norm_dynamic_quant/AddRmsNormDynamicQuant_cde61da2bd6fededcb1ba310a6ad16ee.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/flash_attention_score/flash_attention_score_bf16_bnsd_full_mix.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/flash_attention_score/flash_attention_score_bf16_bnsd_tri_mix.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/flash_attention_score/flash_attention_score_bf16_bsh_full_mix.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/flash_attention_score/flash_attention_score_bf16_bsh_tri_mix.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/flash_attention_score/flash_attention_score_fp16_bnsd_full_mix.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/flash_attention_score/flash_attention_score_fp16_bnsd_tri_mix.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/flash_attention_score/flash_attention_score_fp16_bsh_full_mix.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/flash_attention_score/flash_attention_score_fp16_bsh_tri_mix.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/internal_matmul_postfusion_mix/internal_matmul_postfusion_mix.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/internal_matmul_postfusion_mix/internal_matmul_postfusion_mix_mix_aic_0.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/internal_matmul_postfusion_mix/internal_matmul_postfusion_mix_mix_aiv_0.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/internal_multi_weight_matmul_postfusion_mix/internal_multi_weight_matmul_postfusion_mix.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/internal_multi_weight_matmul_postfusion_mix/internal_multi_weight_matmul_postfusion_mix_mix_aic_0.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/internal_multi_weight_matmul_postfusion_mix/internal_multi_weight_matmul_postfusion_mix_mix_aiv_0.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/{matmul_add_rmsnorm → object_kernels/matmul_add_rmsnorm}/matmul_add_rmsnorm_bf16_bf16.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/{matmul_add_rmsnorm → object_kernels/matmul_add_rmsnorm}/matmul_add_rmsnorm_bf16_fp16.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/{matmul_add_rmsnorm → object_kernels/matmul_add_rmsnorm}/matmul_add_rmsnorm_bf16_fp32.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/{matmul_add_rmsnorm → object_kernels/matmul_add_rmsnorm}/matmul_add_rmsnorm_fp16_bf16.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/{matmul_add_rmsnorm → object_kernels/matmul_add_rmsnorm}/matmul_add_rmsnorm_fp16_fp16.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/{matmul_add_rmsnorm → object_kernels/matmul_add_rmsnorm}/matmul_add_rmsnorm_fp16_fp32.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/paged_attention_v2/paged_attention_v2.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/paged_attention_v2/paged_attention_v2_mix_aic_0.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/object_kernels/paged_attention_v2/paged_attention_v2_mix_aiv_0.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/{lib/libadd_layer_norm_impl.so → op_kernels/ascend910b/so_kernels/libadd_layer_norm_ascend910b.so} +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/{lib/libadd_rms_norm_impl.so → op_kernels/ascend910b/so_kernels/libadd_rms_norm_ascend910b.so} +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/so_kernels/libadd_rms_norm_quant_ascend910b.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/{lib/libapply_rotary_pos_emb_impl.so → op_kernels/ascend910b/so_kernels/libapply_rotary_pos_emb_ascend910b.so} +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/{lib/libcast_impl.so → op_kernels/ascend910b/so_kernels/libcast_ascend910b.so} +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/{lib/libnot_equal_impl.so → op_kernels/ascend910b/so_kernels/libcompare_ascend910b.so} +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/{lib/libgelu_impl.so → op_kernels/ascend910b/so_kernels/libgelu_ascend910b.so} +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/so_kernels/libllama_ascend910b.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/{lib/libmatmul_impl.so → op_kernels/ascend910b/so_kernels/libmatmul_ascend910b.so} +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/{lib/libmulti_weight_matmul_kernel_impl.so → op_kernels/ascend910b/so_kernels/libmulti_weight_matmul_kernel_ascend910b.so} +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/{lib/libreshape_and_cache_impl.so → op_kernels/ascend910b/so_kernels/libreshape_and_cache_ascend910b.so} +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/{lib/librms_norm_impl.so → op_kernels/ascend910b/so_kernels/librms_norm_ascend910b.so} +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/lccl/lib/liblccl_wrapper.so +0 -0
- mindspore/lib/plugin/libmindspore_ascend.so.2 +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/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/cell.py +126 -19
- 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/embedding.py +3 -3
- mindspore/nn/layer/normalization.py +8 -7
- mindspore/nn/layer/padding.py +4 -3
- mindspore/nn/layer/pooling.py +47 -13
- 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 +13 -11
- 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/ops/__init__.py +2 -1
- mindspore/ops/_grad_experimental/grad_comm_ops.py +94 -13
- 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 +8024 -3409
- 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 +1659 -290
- 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 +19 -16
- mindspore/ops/operations/custom_ops.py +11 -55
- 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 +146 -308
- 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 +24 -33
- mindspore/parallel/_parallel_serialization.py +13 -2
- mindspore/parallel/_utils.py +4 -1
- 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 +2 -6
- 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/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 +31 -10
- 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 -15
- mindspore/train/serialization.py +462 -76
- mindspore/train/summary/summary_record.py +1 -2
- mindspore/train/train_thor/model_thor.py +1 -1
- mindspore/utils/__init__.py +4 -2
- mindspore/utils/bin/dataset-cache +0 -0
- mindspore/utils/bin/dataset-cache-server +0 -0
- mindspore/utils/dryrun.py +138 -0
- mindspore/utils/runtime_execution_order_check.py +550 -0
- mindspore/version.py +1 -1
- {mindspore-2.4.10.dist-info → mindspore-2.5.0.dist-info}/METADATA +2 -3
- {mindspore-2.4.10.dist-info → mindspore-2.5.0.dist-info}/RECORD +524 -458
- {mindspore-2.4.10.dist-info → mindspore-2.5.0.dist-info}/entry_points.txt +1 -1
- mindspore/_data_dump.cpython-311-aarch64-linux-gnu.so +0 -0
- mindspore/bin/cache_admin +0 -0
- mindspore/bin/cache_server +0 -0
- mindspore/common/_tensor_overload.py +0 -139
- mindspore/lib/libmindspore_np_dtype.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/include/acme.h +0 -24
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/include/acme_op.h +0 -82
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/include/op_creator.h +0 -113
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/include/op_param.h +0 -193
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/core/dtype_registry.h +0 -90
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/core/kernel_register.h +0 -46
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/core/platform/platform_configs.h +0 -89
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/core/platform/rt_funcs.h +0 -135
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/add_layer_norm_op.h +0 -60
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/add_rms_norm_op.h +0 -50
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/add_rms_norm_quant_op.h +0 -50
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/apply_rotary_pos_emb_nz_op.h +0 -42
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/apply_rotary_pos_emb_op.h +0 -55
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/asd_elewise_op.h +0 -34
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/asd_only_ops.h +0 -94
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/asd_op_base.h +0 -97
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/cast_op.h +0 -52
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/flash_attention_score_op.h +0 -97
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/gelu_op.h +0 -44
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/matmul_add_rmsnorm_op.h +0 -73
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/matmul_op.h +0 -108
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/multi_impls_op.h +0 -64
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/multi_weight_matmul_op.h +0 -91
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/paged_attention_op.h +0 -99
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/reshape_and_cache_nz_op.h +0 -44
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/reshape_and_cache_op.h +0 -44
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/ops/host_src/rms_norm_op.h +0 -64
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/utils/asd_utils.h +0 -179
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/utils/comm_utils.h +0 -69
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/acme/src/utils/profiling_util.h +0 -366
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/add/add_impl.h +0 -56
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/add/kernel/add.h +0 -21
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/add/tiling/add_tiling.h +0 -43
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/apply_rotary_pos_emb/apply_rotary_pos_emb_impl.h +0 -46
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/apply_rotary_pos_emb/kernel/apply_rotary_pos_emb.h +0 -23
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/apply_rotary_pos_emb/kernel/apply_rotary_pos_emb_base.h +0 -456
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/apply_rotary_pos_emb/kernel/apply_rotary_pos_emb_bf16.h +0 -217
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/apply_rotary_pos_emb/kernel/apply_rotary_pos_emb_fp.h +0 -391
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/apply_rotary_pos_emb/kernel/apply_rotary_pos_emb_fp16.h +0 -126
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/apply_rotary_pos_emb/kernel/apply_rotary_pos_emb_fp32.h +0 -230
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/apply_rotary_pos_emb/kernel/apply_rotary_pos_emb_tiling.h +0 -43
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/apply_rotary_pos_emb/kernel/apply_rotary_pos_emb_value.h +0 -27
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/apply_rotary_pos_emb_nz/apply_rotary_pos_emb_nz_impl.h +0 -34
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/apply_rotary_pos_emb_nz/kernel/apply_rotary_pos_emb_nz.h +0 -23
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/apply_rotary_pos_emb_nz/kernel/apply_rotary_pos_emb_nz_base.h +0 -460
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/apply_rotary_pos_emb_nz/kernel/apply_rotary_pos_emb_nz_fp16.h +0 -116
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/apply_rotary_pos_emb_nz/kernel/apply_rotary_pos_emb_nz_fp32.h +0 -230
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/apply_rotary_pos_emb_nz/kernel/apply_rotary_pos_emb_nz_tiling.h +0 -43
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/apply_rotary_pos_emb_nz/kernel/apply_rotary_pos_emb_nz_value.h +0 -27
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/asdop/asd_op_impl.h +0 -74
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/backend_param.h +0 -74
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/cast/cast_impl.h +0 -48
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/cast/kernel/cast_kernel.h +0 -21
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/compare/compare_impl.h +0 -55
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/compare/compare_tiling.h +0 -27
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/compare/kernel/compare_kernel.h +0 -23
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/and_impl.h +0 -29
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/div_impl.h +0 -29
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/elewise_binary_impl.h +0 -48
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/elewise_binary_tiling.h +0 -25
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/kernel/and_kernel.h +0 -46
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/kernel/div_kernel.h +0 -46
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/kernel/elewise_binary_base.h +0 -260
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/kernel/elewise_binary_kernel.h +0 -35
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/kernel/max_kernel.h +0 -66
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/kernel/min_kernel.h +0 -66
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/kernel/mul_kernel.h +0 -66
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/kernel/or_kernel.h +0 -46
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/max_impl.h +0 -29
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/min_impl.h +0 -29
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/mul_impl.h +0 -29
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_binary/or_impl.h +0 -29
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/abs_impl.h +0 -29
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/elewise_unary_impl.h +0 -47
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/elewise_unary_tiling.h +0 -24
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/exp_impl.h +0 -29
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/kernel/abs_kernel.h +0 -45
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/kernel/elewise_unary_base.h +0 -148
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/kernel/elewise_unary_kernel.h +0 -31
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/kernel/exp_kernel.h +0 -45
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/kernel/ln_kernel.h +0 -45
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/kernel/not_kernel.h +0 -45
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/kernel/reciprocal_kernel.h +0 -45
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/kernel/relu_kernel.h +0 -55
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/kernel/rsqrt_kernel.h +0 -45
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/kernel/sqrt_kernel.h +0 -45
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/ln_impl.h +0 -29
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/not_impl.h +0 -29
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/reciprocal_impl.h +0 -29
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/relu_impl.h +0 -29
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/rsqrt_impl.h +0 -29
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/elewise_unary/sqrt_impl.h +0 -29
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/flash_attention_score/flash_attention_score_impl.h +0 -68
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/internal_kernel.h +0 -99
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/internal_rtbackend.h +0 -21
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/lccl/lccl_wrapper.h +0 -58
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/ms_int_types.h +0 -91
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/ms_int_utils.h +0 -108
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/paged_attention/paged_attention_impl.h +0 -64
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/param/add_param.h +0 -68
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/param/attention_param.h +0 -40
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/param/cast_param.h +0 -30
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/param/compare_param.h +0 -31
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/param/elewise_param.h +0 -41
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/param/grouped_matmul_param.h +0 -40
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/param/matmul_ext_param.h +0 -38
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/param/matmul_qkv_param.h +0 -42
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/param/sub_param.h +0 -33
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/profiling_util.h +0 -377
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/reshape_and_cache_nz/kernel/reshape_and_cache_nz.h +0 -24
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/reshape_and_cache_nz/reshape_and_cache_nz_impl.h +0 -42
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/reshape_and_cache_nz/reshape_and_cache_nz_tiling.h +0 -27
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/rms_norm/rms_norm_impl.h +0 -46
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/sub/kernel/sub_kernel.h +0 -20
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/sub/sub_impl.h +0 -48
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/sub/sub_tiling.h +0 -25
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/tune_repo/matmul_table.h +0 -399
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/tune_repo/utils.h +0 -41
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/utils/backend.h +0 -45
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/utils/elewise_tiling.h +0 -29
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/utils/elewise_utils.h +0 -30
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/utils/log/log.h +0 -69
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/utils/log/log_core.h +0 -43
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/utils/log/log_entity.h +0 -38
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/utils/log/log_sink.h +0 -69
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/utils/log/log_stream.h +0 -41
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/utils/log/log_tiling.h +0 -71
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/utils/log/log_utils.h +0 -165
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/utils/math.h +0 -20
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/utils/register/kernel_creator.h +0 -39
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/utils/register/kernel_registry.h +0 -121
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/utils/utils.h +0 -106
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libAdd_impl.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libSub_impl.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libadd_rms_norm_quant_acme_impl.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libapply_rotary_pos_emb_310p_old_impl.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libapply_rotary_pos_emb_old_impl.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libreshape_and_cache_nz_impl.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libreshape_and_cache_nz_old_impl.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/AcmeMatMulPostFusionMixTactic/acme_matmul_postfusion_mix.json +0 -19
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/AcmeMatMulPostFusionMixTactic/acme_matmul_postfusion_mix.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/AcmeMatMulPostFusionMixTactic/acme_matmul_postfusion_mix_mix_aic_0.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/AcmeMatMulPostFusionMixTactic/acme_matmul_postfusion_mix_mix_aiv_0.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/AcmeMultiWeightMatMulPostFusionMixTactic/acme_multi_weight_matmul_postfusion_mix.json +0 -19
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/AcmeMultiWeightMatMulPostFusionMixTactic/acme_multi_weight_matmul_postfusion_mix.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/AcmeMultiWeightMatMulPostFusionMixTactic/acme_multi_weight_matmul_postfusion_mix_mix_aic_0.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/AcmeMultiWeightMatMulPostFusionMixTactic/acme_multi_weight_matmul_postfusion_mix_mix_aiv_0.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/flash_attention_score/flash_attention_score_bf16_bnsd_full_mix.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/flash_attention_score/flash_attention_score_bf16_bnsd_tri_mix.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/flash_attention_score/flash_attention_score_bf16_bsh_full_mix.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/flash_attention_score/flash_attention_score_bf16_bsh_tri_mix.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/flash_attention_score/flash_attention_score_fp16_bnsd_full_mix.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/flash_attention_score/flash_attention_score_fp16_bnsd_tri_mix.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/flash_attention_score/flash_attention_score_fp16_bsh_full_mix.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/flash_attention_score/flash_attention_score_fp16_bsh_tri_mix.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/paged_attention/paged_attention_bf16_bnsd_mix.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/paged_attention/paged_attention_bf16_bsh_mix.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/paged_attention/paged_attention_fp16_bnsd_mix.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/paged_attention/paged_attention_fp16_bsh_mix.o +0 -0
- mindspore/profiler/envprofiling.py +0 -254
- mindspore/profiler/profiling.py +0 -1926
- {mindspore-2.4.10.dist-info → mindspore-2.5.0.dist-info}/WHEEL +0 -0
- {mindspore-2.4.10.dist-info → mindspore-2.5.0.dist-info}/top_level.txt +0 -0
|
@@ -35,14 +35,15 @@ from mindspore.common import Tensor, CSRTensor, COOTensor
|
|
|
35
35
|
from mindspore._c_expression import Tensor as Tensor_
|
|
36
36
|
from mindspore._c_expression import CSRTensor as CSRTensor_
|
|
37
37
|
from mindspore._c_expression import COOTensor as COOTensor_
|
|
38
|
-
from ..auto_generate import (
|
|
39
|
-
|
|
40
|
-
|
|
41
|
-
|
|
42
|
-
|
|
43
|
-
|
|
44
|
-
|
|
45
|
-
|
|
38
|
+
from ..auto_generate import (
|
|
39
|
+
ExpandDims, Reshape, TensorShape, Transpose, Gather, OnesLike, ZerosLike,
|
|
40
|
+
Argmax, ArgMaxExt, ReverseV2, Diag, Eye, ScatterNd,
|
|
41
|
+
ResizeNearestNeighborV2, GatherNd, GatherD, Range, MaskedFill, RightShift,
|
|
42
|
+
NonZero, ResizeNearestNeighbor, Identity, Split, CumSum, CumProd,
|
|
43
|
+
MaskedSelect, Cummax, Cummin, Argmin, Concat, UnsortedSegmentSum, UniqueConsecutive,
|
|
44
|
+
ScalarToTensor, Triu, BroadcastTo, StridedSlice, Select, TopkExt,
|
|
45
|
+
SearchSorted, Meshgrid, Squeeze, Slice)
|
|
46
|
+
from .manually_defined import Rank, Shape, Tile, Cast, Ones, Zeros, TypeAs
|
|
46
47
|
from ..auto_generate import ArgMaxWithValue, ArgMinWithValue
|
|
47
48
|
from ..auto_generate import TensorScatterElements as TensorScatterElementsExt
|
|
48
49
|
|
|
@@ -182,8 +183,7 @@ def _check_infer_attr_reduce(axis, keep_dims, prim_name):
|
|
|
182
183
|
|
|
183
184
|
class Expand(Primitive):
|
|
184
185
|
"""
|
|
185
|
-
|
|
186
|
-
Please use :class:`mindspore.ops.BroadcastTo` instead.
|
|
186
|
+
This interface will be deprecated in the future, and use :class:`mindspore.ops.BroadcastTo` instead.
|
|
187
187
|
"""
|
|
188
188
|
|
|
189
189
|
@deprecated("2.1", "BroadcastTo", False)
|
|
@@ -474,52 +474,6 @@ class Unsqueeze(PrimitiveWithCheck):
|
|
|
474
474
|
self.axis = axis
|
|
475
475
|
|
|
476
476
|
|
|
477
|
-
class Squeeze(Primitive):
|
|
478
|
-
"""
|
|
479
|
-
Return the Tensor after deleting the dimension of size 1 in the specified `axis`.
|
|
480
|
-
|
|
481
|
-
Refer to :func:`mindspore.ops.squeeze` for more details.
|
|
482
|
-
|
|
483
|
-
Args:
|
|
484
|
-
axis (Union[int, tuple(int)]): Specifies the dimension indexes of shape to be removed, which will remove
|
|
485
|
-
all the dimensions of size 1 in the given axis parameter. If specified, it must be int32 or int64.
|
|
486
|
-
Default: ``()`` .
|
|
487
|
-
|
|
488
|
-
Inputs:
|
|
489
|
-
- **input_x** (Tensor) - The shape of tensor is :math:`(x_1, x_2, ..., x_R)`.
|
|
490
|
-
|
|
491
|
-
Outputs:
|
|
492
|
-
Tensor, the shape of tensor is :math:`(x_1, x_2, ..., x_S)`.
|
|
493
|
-
|
|
494
|
-
Supported Platforms:
|
|
495
|
-
``Ascend`` ``GPU`` ``CPU``
|
|
496
|
-
|
|
497
|
-
Examples:
|
|
498
|
-
>>> import mindspore
|
|
499
|
-
>>> import numpy as np
|
|
500
|
-
>>> from mindspore import Tensor, ops
|
|
501
|
-
>>> input_x = Tensor(np.ones(shape=[3, 2, 1]), mindspore.float32)
|
|
502
|
-
>>> squeeze = ops.Squeeze(2)
|
|
503
|
-
>>> output = squeeze(input_x)
|
|
504
|
-
>>> print(output)
|
|
505
|
-
[[1. 1.]
|
|
506
|
-
[1. 1.]
|
|
507
|
-
[1. 1.]]
|
|
508
|
-
"""
|
|
509
|
-
|
|
510
|
-
@prim_attr_register
|
|
511
|
-
def __init__(self, axis=()):
|
|
512
|
-
"""Initialize Squeeze"""
|
|
513
|
-
self.init_prim_io_names(inputs=['x'], outputs=['output'])
|
|
514
|
-
validator.check_value_type('axis', axis, [int, tuple], self.name)
|
|
515
|
-
if isinstance(axis, tuple):
|
|
516
|
-
for idx, item in enumerate(axis):
|
|
517
|
-
validator.check_value_type("axis[%d]" % idx, item, [int], self.name)
|
|
518
|
-
else:
|
|
519
|
-
self.axis = (axis,)
|
|
520
|
-
self.add_prim_attr("axis", (axis,))
|
|
521
|
-
|
|
522
|
-
|
|
523
477
|
class ConjugateTranspose(Primitive):
|
|
524
478
|
"""
|
|
525
479
|
Calculate the conjugate matrix of input x which has been transposed according to input perm.
|
|
@@ -556,8 +510,8 @@ class ConjugateTranspose(Primitive):
|
|
|
556
510
|
>>> conjugate_transpose = ops.ConjugateTranspose()
|
|
557
511
|
>>> output = conjugate_transpose(x, perm)
|
|
558
512
|
>>> print(output)
|
|
559
|
-
|
|
560
|
-
|
|
513
|
+
[[1.-1.j 3.-3.j]
|
|
514
|
+
[2.-2.j 4.-4.j]]
|
|
561
515
|
"""
|
|
562
516
|
|
|
563
517
|
@prim_attr_register
|
|
@@ -630,65 +584,6 @@ class Unique(Primitive):
|
|
|
630
584
|
self.init_prim_io_names(inputs=['x'], outputs=['output'])
|
|
631
585
|
|
|
632
586
|
|
|
633
|
-
class UniqueConsecutive(Primitive):
|
|
634
|
-
"""
|
|
635
|
-
Returns the elements that are unique in each consecutive group of equivalent elements in the input tensor.
|
|
636
|
-
|
|
637
|
-
.. warning::
|
|
638
|
-
This is an experimental API that is subject to change or deletion.
|
|
639
|
-
|
|
640
|
-
Refer to :func:`mindspore.ops.unique_consecutive` for more details.
|
|
641
|
-
|
|
642
|
-
Args:
|
|
643
|
-
return_idx (bool, optional): Whether to return the index of where the element in the original input
|
|
644
|
-
maps to the position in the output. Default: ``False`` .
|
|
645
|
-
return_counts (bool, optional): Whether to return the counts of each unique element. Default: ``False`` .
|
|
646
|
-
axis (int, optional): The dimension to apply unique. If ``None`` , the unique of the flattened input is
|
|
647
|
-
returned. If specified, it must be int32 or int64. Default: ``None`` .
|
|
648
|
-
|
|
649
|
-
Inputs:
|
|
650
|
-
- **x** (Tensor) - The input tensor.
|
|
651
|
-
|
|
652
|
-
Outputs:
|
|
653
|
-
A tensor or a tuple of tensors containing tensor objects (`output`, `idx`, `counts`).
|
|
654
|
-
|
|
655
|
-
- `output` has the same type as `x` and is used to represent the output list of unique scalar elements.
|
|
656
|
-
- If `return_idx` is True, there will be an additional returned tensor, `idx`,
|
|
657
|
-
which has the same shape as `x` and represents
|
|
658
|
-
the index of where the element in the original input maps to the position in the output.
|
|
659
|
-
- If `return_counts` is True, there will be an additional returned tensor, `counts`,
|
|
660
|
-
which represents the number of occurrences for each unique value or tensor.
|
|
661
|
-
|
|
662
|
-
Supported Platforms:
|
|
663
|
-
``Ascend`` ``GPU`` ``CPU``
|
|
664
|
-
|
|
665
|
-
Examples:
|
|
666
|
-
>>> import numpy as np
|
|
667
|
-
>>> from mindspore import Tensor, ops
|
|
668
|
-
>>> from mindspore import dtype as mstype
|
|
669
|
-
>>> x = Tensor(np.array([1, 1, 2, 2, 3, 1, 1, 2]), mstype.int32)
|
|
670
|
-
>>> unique_consecutive = ops.UniqueConsecutive(True, True, None)
|
|
671
|
-
>>> output, idx, counts = unique_consecutive(x)
|
|
672
|
-
>>> print(output)
|
|
673
|
-
[1 2 3 1 2]
|
|
674
|
-
>>> print(idx)
|
|
675
|
-
[0 0 1 1 2 3 3 4]
|
|
676
|
-
>>> print(counts)
|
|
677
|
-
[2 2 1 2 1]
|
|
678
|
-
"""
|
|
679
|
-
|
|
680
|
-
@prim_attr_register
|
|
681
|
-
def __init__(self, return_idx=False, return_counts=False, axis=None):
|
|
682
|
-
"""Initialize UniqueConsecutive"""
|
|
683
|
-
self.init_prim_io_names(inputs=['x'], outputs=['output'])
|
|
684
|
-
validator.check_value_type("return_idx", return_idx, [bool], self.name)
|
|
685
|
-
validator.check_value_type("return_counts", return_counts, [bool], self.name)
|
|
686
|
-
validator.check_value_type("axis", axis, [int, type(None)], self.name)
|
|
687
|
-
self.add_prim_attr("return_idx", return_idx)
|
|
688
|
-
self.add_prim_attr("return_counts", return_counts)
|
|
689
|
-
self.add_prim_attr("axis", axis)
|
|
690
|
-
|
|
691
|
-
|
|
692
587
|
class SparseGatherV2(Primitive):
|
|
693
588
|
"""
|
|
694
589
|
Returns a slice of input tensor based on the specified indices and axis.
|
|
@@ -1943,55 +1838,6 @@ class Unstack(Primitive):
|
|
|
1943
1838
|
validator.check_value_type("num", num, [int], self.name)
|
|
1944
1839
|
|
|
1945
1840
|
|
|
1946
|
-
class Slice(Primitive):
|
|
1947
|
-
"""
|
|
1948
|
-
Slices a tensor in the specified shape.
|
|
1949
|
-
|
|
1950
|
-
Refer to :func:`mindspore.ops.slice` for more details.
|
|
1951
|
-
|
|
1952
|
-
Inputs:
|
|
1953
|
-
- **input_x** (Tensor) - The target tensor.
|
|
1954
|
-
The shape is :math:`(N, *)` where :math:`*` means, any number of additional dimensions.
|
|
1955
|
-
- **begin** (Union[tuple, list]) - The beginning of the slice. Only constant value(>=0) is allowed.
|
|
1956
|
-
- **size** (Union[tuple, list]) - The size of the slice. Only constant value is allowed.
|
|
1957
|
-
|
|
1958
|
-
Outputs:
|
|
1959
|
-
Tensor, the shape is: input `size`, the data type is the same as `input_x`.
|
|
1960
|
-
|
|
1961
|
-
Supported Platforms:
|
|
1962
|
-
``Ascend`` ``GPU`` ``CPU``
|
|
1963
|
-
|
|
1964
|
-
Examples:
|
|
1965
|
-
>>> from mindspore import Tensor
|
|
1966
|
-
>>> from mindspore import ops
|
|
1967
|
-
>>> import numpy as np
|
|
1968
|
-
>>> data = Tensor(np.array([[[1, 1, 1], [2, 2, 2]],
|
|
1969
|
-
... [[3, 3, 3], [4, 4, 4]],
|
|
1970
|
-
... [[5, 5, 5], [6, 6, 6]]]).astype(np.int32))
|
|
1971
|
-
>>> slice_op = ops.Slice()
|
|
1972
|
-
>>> output = slice_op(data, (1, 0, 0), (1, 1, 3))
|
|
1973
|
-
>>> print(output)
|
|
1974
|
-
[[[3 3 3]]]
|
|
1975
|
-
>>> output = slice_op(data, (1, 0, 0), (1, 1, 2))
|
|
1976
|
-
>>> print(output)
|
|
1977
|
-
[[[3 3]]]
|
|
1978
|
-
>>> output = slice_op(data, (1, 0, 0), (1, 1, 1))
|
|
1979
|
-
>>> print(output)
|
|
1980
|
-
[[[3]]]
|
|
1981
|
-
>>> output = slice_op(data, (1, 1, 0), (1, 1, 3))
|
|
1982
|
-
>>> print(output)
|
|
1983
|
-
[[[4 4 4]]]
|
|
1984
|
-
>>> output = slice_op(data, (1, 0, 1), (1, 1, 2))
|
|
1985
|
-
>>> print(output)
|
|
1986
|
-
[[[3 3]]]
|
|
1987
|
-
"""
|
|
1988
|
-
|
|
1989
|
-
@prim_attr_register
|
|
1990
|
-
def __init__(self):
|
|
1991
|
-
"""Initialize slice"""
|
|
1992
|
-
self.init_prim_io_names(inputs=['x', 'begin', 'size'], outputs=['output'])
|
|
1993
|
-
|
|
1994
|
-
|
|
1995
1841
|
class Coalesce(Primitive):
|
|
1996
1842
|
"""
|
|
1997
1843
|
Returns the coalesced sparse tensor of the input.
|
|
@@ -2198,7 +2044,7 @@ class ScatterUpdate(Primitive):
|
|
|
2198
2044
|
use_locking (bool): Whether to protect the assignment by a lock. Default: ``True`` .
|
|
2199
2045
|
|
|
2200
2046
|
Inputs:
|
|
2201
|
-
- **input_x** (Parameter) - The target tensor, with data type of Parameter.
|
|
2047
|
+
- **input_x** (Union[Parameter, Tensor]) - The target tensor, with data type of Parameter or Tensor.
|
|
2202
2048
|
The shape is 0-D or :math:`(N, *)` where :math:`*` means any number of additional dimensions.
|
|
2203
2049
|
- **indices** (Tensor) - The index of input tensor. With int32 data type.
|
|
2204
2050
|
If there are duplicates in indices, the order for updating is undefined.
|
|
@@ -2212,8 +2058,8 @@ class ScatterUpdate(Primitive):
|
|
|
2212
2058
|
TypeError: If `use_locking` is not a bool.
|
|
2213
2059
|
TypeError: If `indices` is not an int32.
|
|
2214
2060
|
ValueError: If the shape of `updates` is not equal to `indices.shape + input_x.shape[1:]`.
|
|
2215
|
-
RuntimeError: If the data type of `input_x` and `updates` conversion
|
|
2216
|
-
is
|
|
2061
|
+
RuntimeError: If the data type of `input_x` and `updates` conversion is required when data type conversion
|
|
2062
|
+
is not supported.
|
|
2217
2063
|
|
|
2218
2064
|
Supported Platforms:
|
|
2219
2065
|
``Ascend`` ``GPU`` ``CPU``
|
|
@@ -2270,7 +2116,7 @@ class ScatterNdUpdate(Primitive):
|
|
|
2270
2116
|
use_locking (bool): Whether to protect the assignment by a lock. Default: ``True`` .
|
|
2271
2117
|
|
|
2272
2118
|
Inputs:
|
|
2273
|
-
- **input_x** (Parameter) - The target tensor, with data type of Parameter.
|
|
2119
|
+
- **input_x** (Union[Parameter, Tensor]) - The target tensor, with data type of Parameter or Tensor.
|
|
2274
2120
|
The shape is :math:`(N, *)` where :math:`*` means any number of additional dimensions.
|
|
2275
2121
|
- **indices** (Tensor) - The index of input tensor, with int32 or int64 data type.
|
|
2276
2122
|
- **updates** (Tensor) - N-D(2D or 3D) Tensor The tensor to be updated to the input tensor,
|
|
@@ -2282,8 +2128,8 @@ class ScatterNdUpdate(Primitive):
|
|
|
2282
2128
|
Raises:
|
|
2283
2129
|
TypeError: If `use_locking` is not a bool.
|
|
2284
2130
|
TypeError: If `indices` is not an int32 or an int64.
|
|
2285
|
-
RuntimeError: If the data type of `input_x` and `updates` conversion
|
|
2286
|
-
is
|
|
2131
|
+
RuntimeError: If the data type of `input_x` and `updates` conversion is required when data type conversion
|
|
2132
|
+
is not supported.
|
|
2287
2133
|
|
|
2288
2134
|
Supported Platforms:
|
|
2289
2135
|
``Ascend`` ``GPU`` ``CPU``
|
|
@@ -2340,7 +2186,7 @@ class ScatterMax(_ScatterOpDynamic):
|
|
|
2340
2186
|
use_locking (bool): Whether to protect the assignment by a lock. Default: ``False`` .
|
|
2341
2187
|
|
|
2342
2188
|
Inputs:
|
|
2343
|
-
- **input_x** (Parameter) - The target tensor, with data type of Parameter.
|
|
2189
|
+
- **input_x** (Union[Parameter, Tensor]) - The target tensor, with data type of Parameter or Tensor.
|
|
2344
2190
|
The shape is :math:`(N, *)` where :math:`*` means any number of additional dimensions.
|
|
2345
2191
|
- **indices** (Tensor) - The index to do max operation whose data type must be mindspore.int32 or
|
|
2346
2192
|
mindspore.int64.
|
|
@@ -2354,8 +2200,8 @@ class ScatterMax(_ScatterOpDynamic):
|
|
|
2354
2200
|
TypeError: If `use_locking` is not a bool.
|
|
2355
2201
|
TypeError: If `indices` is not an int32 or an int64.
|
|
2356
2202
|
ValueError: If the shape of `updates` is not equal to `indices.shape + x.shape[1:]`.
|
|
2357
|
-
RuntimeError: If the data type of `input_x` and `updates` conversion
|
|
2358
|
-
is
|
|
2203
|
+
RuntimeError: If the data type of `input_x` and `updates` conversion is required when data type conversion
|
|
2204
|
+
is not supported.
|
|
2359
2205
|
RuntimeError: On the Ascend platform, the input data dimension of `input_x` , `indices`
|
|
2360
2206
|
and `updates` is greater than 8 dimensions.
|
|
2361
2207
|
|
|
@@ -2401,7 +2247,7 @@ class ScatterMin(_ScatterOpDynamic):
|
|
|
2401
2247
|
use_locking (bool): Whether to protect the assignment by a lock. Default: ``False`` .
|
|
2402
2248
|
|
|
2403
2249
|
Inputs:
|
|
2404
|
-
- **input_x** (Parameter) - The target tensor, with data type of Parameter.
|
|
2250
|
+
- **input_x** (Union[Parameter, Tensor]) - The target tensor, with data type of Parameter or Tensor.
|
|
2405
2251
|
The shape is :math:`(N, *)` where :math:`*` means any number of additional dimensions.
|
|
2406
2252
|
- **indices** (Tensor) - The index to do min operation whose data type must be mindspore.int32 or
|
|
2407
2253
|
mindspore.int64.
|
|
@@ -2415,8 +2261,8 @@ class ScatterMin(_ScatterOpDynamic):
|
|
|
2415
2261
|
TypeError: If `use_locking` is not a bool.
|
|
2416
2262
|
TypeError: If `indices` is not an int32 or an int64.
|
|
2417
2263
|
ValueError: If the shape of `updates` is not equal to `indices.shape + input_x.shape[1:]`.
|
|
2418
|
-
RuntimeError: If the data type of `input_x` and `updates` conversion
|
|
2419
|
-
is
|
|
2264
|
+
RuntimeError: If the data type of `input_x` and `updates` conversion is required when data type conversion
|
|
2265
|
+
is not supported.
|
|
2420
2266
|
RuntimeError: On the Ascend platform, the input data dimension of `input_x` , `indices`
|
|
2421
2267
|
and `updates` is greater than 8 dimensions.
|
|
2422
2268
|
|
|
@@ -2465,7 +2311,7 @@ class ScatterAdd(Primitive):
|
|
|
2465
2311
|
Otherwise, the calculation result is undefined. Default: ``False`` .
|
|
2466
2312
|
|
|
2467
2313
|
Inputs:
|
|
2468
|
-
- **input_x** (Parameter) - The target tensor, with data type of Parameter.
|
|
2314
|
+
- **input_x** (Union[Parameter, Tensor]) - The target tensor, with data type of Parameter or Tensor.
|
|
2469
2315
|
- **indices** (Tensor) - The index to do min operation whose data type must be mindspore.int32 or
|
|
2470
2316
|
mindspore.int64.
|
|
2471
2317
|
- **updates** (Tensor) - The tensor doing the min operation with `input_x`,
|
|
@@ -2478,8 +2324,8 @@ class ScatterAdd(Primitive):
|
|
|
2478
2324
|
TypeError: If `use_locking` is not a bool.
|
|
2479
2325
|
TypeError: If `indices` is not an int32 or an int64.
|
|
2480
2326
|
ValueError: If the shape of `updates` is not equal to `indices.shape + x.shape[1:]`.
|
|
2481
|
-
RuntimeError: If the data type of `input_x` and `updates` conversion
|
|
2482
|
-
is
|
|
2327
|
+
RuntimeError: If the data type of `input_x` and `updates` conversion is required when data type conversion
|
|
2328
|
+
is not supported.
|
|
2483
2329
|
|
|
2484
2330
|
Supported Platforms:
|
|
2485
2331
|
``Ascend`` ``GPU`` ``CPU``
|
|
@@ -2583,7 +2429,7 @@ class ScatterSub(Primitive):
|
|
|
2583
2429
|
use_locking (bool): Whether to protect the assignment by a lock. Default: ``False`` .
|
|
2584
2430
|
|
|
2585
2431
|
Inputs:
|
|
2586
|
-
- **input_x** (Parameter) - The target tensor, with data type of Parameter.
|
|
2432
|
+
- **input_x** (Union[Parameter, Tensor]) - The target tensor, with data type of Parameter or Tensor.
|
|
2587
2433
|
The shape is :math:`(N, *)` where :math:`*` means any number of additional dimensions.
|
|
2588
2434
|
- **indices** (Tensor) - The index to do min operation whose data type must be mindspore.int32 or
|
|
2589
2435
|
mindspore.int64.
|
|
@@ -2597,8 +2443,8 @@ class ScatterSub(Primitive):
|
|
|
2597
2443
|
TypeError: If `use_locking` is not a bool.
|
|
2598
2444
|
TypeError: If `indices` is not an int32.
|
|
2599
2445
|
ValueError: If the shape of `updates` is not equal to `indices_shape + x_shape[1:]`.
|
|
2600
|
-
RuntimeError: If the data type of `input_x` and `updates` conversion
|
|
2601
|
-
is
|
|
2446
|
+
RuntimeError: If the data type of `input_x` and `updates` conversion is required when data type conversion
|
|
2447
|
+
is not supported.
|
|
2602
2448
|
|
|
2603
2449
|
Supported Platforms:
|
|
2604
2450
|
``Ascend`` ``GPU`` ``CPU``
|
|
@@ -2702,7 +2548,7 @@ class ScatterMul(_ScatterOpDynamic):
|
|
|
2702
2548
|
use_locking (bool): Whether to protect the assignment by a lock. Default: ``False`` .
|
|
2703
2549
|
|
|
2704
2550
|
Inputs:
|
|
2705
|
-
- **input_x** (Parameter) - The target tensor, with data type of Parameter.
|
|
2551
|
+
- **input_x** (Union[Parameter, Tensor]) - The target tensor, with data type of Parameter or Tensor.
|
|
2706
2552
|
The shape is :math:`(N, *)` where :math:`*` means any number of additional dimensions.
|
|
2707
2553
|
- **indices** (Tensor) - The index to do multiply operation whose data type must be mstype.int32 or
|
|
2708
2554
|
mstype.int64.
|
|
@@ -2716,8 +2562,8 @@ class ScatterMul(_ScatterOpDynamic):
|
|
|
2716
2562
|
TypeError: If `use_locking` is not a bool.
|
|
2717
2563
|
TypeError: If `indices` is not an int32 or an int64.
|
|
2718
2564
|
ValueError: If the shape of `updates` is not equal to `indices.shape + input_x.shape[1:]`.
|
|
2719
|
-
RuntimeError: If the data type of `input_x` and `updates` conversion
|
|
2720
|
-
is
|
|
2565
|
+
RuntimeError: If the data type of `input_x` and `updates` conversion is required when data type conversion
|
|
2566
|
+
is not supported.
|
|
2721
2567
|
|
|
2722
2568
|
Supported Platforms:
|
|
2723
2569
|
``Ascend`` ``GPU`` ``CPU``
|
|
@@ -2810,7 +2656,7 @@ class ScatterDiv(_ScatterOpDynamic):
|
|
|
2810
2656
|
use_locking (bool): Whether to protect the assignment by a lock. Default: ``False`` .
|
|
2811
2657
|
|
|
2812
2658
|
Inputs:
|
|
2813
|
-
- **input_x** (Parameter) - The target tensor, with data type of Parameter.
|
|
2659
|
+
- **input_x** (Union[Parameter, Tensor]) - The target tensor, with data type of Parameter or Tensor.
|
|
2814
2660
|
The shape is :math:`(N, *)` where :math:`*` means any number of additional dimensions.
|
|
2815
2661
|
- **indices** (Tensor) - The index to do divide operation whose data type must be mstype.int32 or
|
|
2816
2662
|
mstype.int64.
|
|
@@ -2824,8 +2670,8 @@ class ScatterDiv(_ScatterOpDynamic):
|
|
|
2824
2670
|
TypeError: If `use_locking` is not a bool.
|
|
2825
2671
|
TypeError: If `indices` is not an int32 or an int64.
|
|
2826
2672
|
ValueError: If the shape of `updates` is not equal to `indices.shape + input_x.shape[1:]`.
|
|
2827
|
-
RuntimeError: If the data type of `input_x` and `updates` conversion
|
|
2828
|
-
is
|
|
2673
|
+
RuntimeError: If the data type of `input_x` and `updates` conversion is required when data type conversion
|
|
2674
|
+
is not supported.
|
|
2829
2675
|
RuntimeError: On the Ascend platform, the input data dimension of `input_x` , `indices`
|
|
2830
2676
|
and `updates` is greater than 8 dimensions.
|
|
2831
2677
|
|
|
@@ -2914,7 +2760,7 @@ class ScatterNdAdd(Primitive):
|
|
|
2914
2760
|
use_locking (bool, optional): Whether to protect the assignment by a lock. Default: ``False`` .
|
|
2915
2761
|
|
|
2916
2762
|
Inputs:
|
|
2917
|
-
- **input_x** (Parameter) - The target tensor, with data type of Parameter.
|
|
2763
|
+
- **input_x** (Union[Parameter, Tensor]) - The target tensor, with data type of Parameter or Tensor.
|
|
2918
2764
|
The shape is :math:`(N, *)` where :math:`*` means any number of additional dimensions.
|
|
2919
2765
|
- **indices** (Tensor) - The index to do add operation whose data type must be mindspore.int32.
|
|
2920
2766
|
The rank of indices must be at least 2 and `indices.shape[-1] <= len(shape)`.
|
|
@@ -2991,7 +2837,7 @@ class ScatterNdSub(Primitive):
|
|
|
2991
2837
|
use_locking (bool, optional): Whether to protect the assignment by a lock. Default: ``False`` .
|
|
2992
2838
|
|
|
2993
2839
|
Inputs:
|
|
2994
|
-
- **input_x** (Parameter) - The target tensor, with data type of Parameter.
|
|
2840
|
+
- **input_x** (Union[Parameter, Tensor]) - The target tensor, with data type of Parameter or Tensor.
|
|
2995
2841
|
The shape is :math:`(N, *)` where :math:`*` means any number of additional dimensions.
|
|
2996
2842
|
- **indices** (Tensor) - The index to do sub operation whose data type must be mindspore.int32.
|
|
2997
2843
|
The rank of indices must be at least 2 and `indices.shape[-1] <= len(shape)`.
|
|
@@ -3072,7 +2918,7 @@ class ScatterNdMul(_ScatterNdOp):
|
|
|
3072
2918
|
use_locking (bool, optional): Whether to protect the assignment by a lock. Default: ``False`` .
|
|
3073
2919
|
|
|
3074
2920
|
Inputs:
|
|
3075
|
-
- **input_x** (Parameter) - The target tensor, with data type of Parameter.
|
|
2921
|
+
- **input_x** (Union[Parameter, Tensor]) - The target tensor, with data type of Parameter or Tensor.
|
|
3076
2922
|
- **indices** (Tensor) - The index to do mul operation whose data type must be int32 or int64.
|
|
3077
2923
|
The rank of indices must be at least 2 and `indices.shape[-1] <= len(shape)`.
|
|
3078
2924
|
- **updates** (Tensor) - The tensor to do the mul operation with `input_x`.
|
|
@@ -3137,7 +2983,7 @@ class ScatterNdDiv(_ScatterNdOp):
|
|
|
3137
2983
|
use_locking (bool, optional): Whether to protect the assignment by a lock. Default: ``False`` .
|
|
3138
2984
|
|
|
3139
2985
|
Inputs:
|
|
3140
|
-
- **input_x** (Parameter) - The target tensor, with data type of Parameter.
|
|
2986
|
+
- **input_x** (Union[Parameter, Tensor]) - The target tensor, with data type of Parameter or Tensor.
|
|
3141
2987
|
- **indices** (Tensor) - The index to do div operation whose data type must be int32 or int64.
|
|
3142
2988
|
The rank of indices must be at least 2 and `indices.shape[-1] <= len(shape)`.
|
|
3143
2989
|
- **updates** (Tensor) - The tensor to do the div operation with `input_x`.
|
|
@@ -3202,7 +3048,7 @@ class ScatterNdMax(_ScatterNdOp):
|
|
|
3202
3048
|
use_locking (bool, optional): Whether to protect the assignment by a lock. Default: ``False`` .
|
|
3203
3049
|
|
|
3204
3050
|
Inputs:
|
|
3205
|
-
- **input_x** (Parameter) -The target tensor, with data type of Parameter.
|
|
3051
|
+
- **input_x** (Union[Parameter, Tensor]) - The target tensor, with data type of Parameter or Tensor.
|
|
3206
3052
|
- **indices** (Tensor) - The index to do maximum operation whose data type must be int32 or int64.
|
|
3207
3053
|
The rank of indices must be at least 2 and `indices.shape[-1] <= len(shape)`.
|
|
3208
3054
|
- **updates** (Tensor) - The tensor to do the max operation with `input_x`.
|
|
@@ -3269,7 +3115,7 @@ class ScatterNdMin(_ScatterNdOp):
|
|
|
3269
3115
|
use_locking (bool, optional): Whether to protect the assignment by a lock. Default: ``False`` .
|
|
3270
3116
|
|
|
3271
3117
|
Inputs:
|
|
3272
|
-
- **input_x** (Parameter) -The target tensor, with data type of Parameter.
|
|
3118
|
+
- **input_x** (Union[Parameter, Tensor]) - The target tensor, with data type of Parameter or Tensor.
|
|
3273
3119
|
- **indices** (Tensor) - The index to do minimum operation whose data type must be int32 or int64.
|
|
3274
3120
|
The rank of indices must be at least 2 and `indices.shape[-1] <= len(shape)`.
|
|
3275
3121
|
- **updates** (Tensor) - The tensor to do the max operation with `input_x`.
|
|
@@ -3735,110 +3581,6 @@ class BatchToSpaceNDV2(Primitive):
|
|
|
3735
3581
|
self.add_prim_attr('origin_format', 'NHWC')
|
|
3736
3582
|
|
|
3737
3583
|
|
|
3738
|
-
class Meshgrid(PrimitiveWithInfer):
|
|
3739
|
-
"""
|
|
3740
|
-
Generates coordinate matrices from given coordinate tensors.
|
|
3741
|
-
|
|
3742
|
-
Refer to :func:`mindspore.ops.meshgrid` for more details.
|
|
3743
|
-
|
|
3744
|
-
Args:
|
|
3745
|
-
indexing (str, optional): Cartesian ``'xy'`` or
|
|
3746
|
-
matrix ``'ij'`` indexing of output. In the 2-D case with
|
|
3747
|
-
inputs of length `M` and `N`, the outputs are of shape :math:`(N, M)`
|
|
3748
|
-
for ``'xy'`` indexing and :math:`(M, N)` for ``'ij'`` indexing. In the 3-D
|
|
3749
|
-
case with inputs of length `M`, `N` and `P`, outputs are of shape
|
|
3750
|
-
:math:`(N, M, P)` for ``'xy'`` indexing and :math:`(M, N, P)` for ``'ij'`` indexing.
|
|
3751
|
-
Default: ``'xy'``.
|
|
3752
|
-
|
|
3753
|
-
Inputs:
|
|
3754
|
-
- **input** (Union[tuple]) - A Tuple of N 1-D Tensor objects.
|
|
3755
|
-
The length of input should be greater than 1. The data type is Number.
|
|
3756
|
-
|
|
3757
|
-
Outputs:
|
|
3758
|
-
Tensors, A Tuple of N N-D Tensor objects. The data type is the same with the Inputs.
|
|
3759
|
-
|
|
3760
|
-
Supported Platforms:
|
|
3761
|
-
``Ascend`` ``GPU`` ``CPU``
|
|
3762
|
-
|
|
3763
|
-
Examples:
|
|
3764
|
-
>>> import numpy as np
|
|
3765
|
-
>>> from mindspore import Tensor, ops
|
|
3766
|
-
>>> x = Tensor(np.array([1, 2, 3, 4]).astype(np.int32))
|
|
3767
|
-
>>> y = Tensor(np.array([5, 6, 7]).astype(np.int32))
|
|
3768
|
-
>>> z = Tensor(np.array([8, 9, 0, 1, 2]).astype(np.int32))
|
|
3769
|
-
>>> inputs = (x, y, z)
|
|
3770
|
-
>>> meshgrid = ops.Meshgrid(indexing='xy')
|
|
3771
|
-
>>> output = meshgrid(inputs)
|
|
3772
|
-
>>> print(output)
|
|
3773
|
-
(Tensor(shape=[3, 4, 5], dtype=Int32, value=
|
|
3774
|
-
[[[1, 1, 1, 1, 1],
|
|
3775
|
-
[2, 2, 2, 2, 2],
|
|
3776
|
-
[3, 3, 3, 3, 3],
|
|
3777
|
-
[4, 4, 4, 4, 4]],
|
|
3778
|
-
[[1, 1, 1, 1, 1],
|
|
3779
|
-
[2, 2, 2, 2, 2],
|
|
3780
|
-
[3, 3, 3, 3, 3],
|
|
3781
|
-
[4, 4, 4, 4, 4]],
|
|
3782
|
-
[[1, 1, 1, 1, 1],
|
|
3783
|
-
[2, 2, 2, 2, 2],
|
|
3784
|
-
[3, 3, 3, 3, 3],
|
|
3785
|
-
[4, 4, 4, 4, 4]]]),
|
|
3786
|
-
Tensor(shape=[3, 4, 5], dtype=Int32, value=
|
|
3787
|
-
[[[5, 5, 5, 5, 5],
|
|
3788
|
-
[5, 5, 5, 5, 5],
|
|
3789
|
-
[5, 5, 5, 5, 5],
|
|
3790
|
-
[5, 5, 5, 5, 5]],
|
|
3791
|
-
[[6, 6, 6, 6, 6],
|
|
3792
|
-
[6, 6, 6, 6, 6],
|
|
3793
|
-
[6, 6, 6, 6, 6],
|
|
3794
|
-
[6, 6, 6, 6, 6]],
|
|
3795
|
-
[[7, 7, 7, 7, 7],
|
|
3796
|
-
[7, 7, 7, 7, 7],
|
|
3797
|
-
[7, 7, 7, 7, 7],
|
|
3798
|
-
[7, 7, 7, 7, 7]]]),
|
|
3799
|
-
Tensor(shape=[3, 4, 5], dtype=Int32, value=
|
|
3800
|
-
[[[8, 9, 0, 1, 2],
|
|
3801
|
-
[8, 9, 0, 1, 2],
|
|
3802
|
-
[8, 9, 0, 1, 2],
|
|
3803
|
-
[8, 9, 0, 1, 2]],
|
|
3804
|
-
[[8, 9, 0, 1, 2],
|
|
3805
|
-
[8, 9, 0, 1, 2],
|
|
3806
|
-
[8, 9, 0, 1, 2],
|
|
3807
|
-
[8, 9, 0, 1, 2]],
|
|
3808
|
-
[[8, 9, 0, 1, 2],
|
|
3809
|
-
[8, 9, 0, 1, 2],
|
|
3810
|
-
[8, 9, 0, 1, 2],
|
|
3811
|
-
[8, 9, 0, 1, 2]]]))
|
|
3812
|
-
"""
|
|
3813
|
-
|
|
3814
|
-
@prim_attr_register
|
|
3815
|
-
def __init__(self, indexing="xy"):
|
|
3816
|
-
"""Initialize Meshgrid."""
|
|
3817
|
-
validator.check_value_type("indexing", indexing, (str), self.name)
|
|
3818
|
-
validator.check_string(indexing.lower(), ["xy", "ij"], "indexing", self.name)
|
|
3819
|
-
self.indexing = indexing
|
|
3820
|
-
|
|
3821
|
-
def infer_shape(self, x_shape):
|
|
3822
|
-
validator.check_value_type("shape", x_shape, [tuple], self.name)
|
|
3823
|
-
validator.check_int(len(x_shape), 2, validator.GE, "len of input", self.name)
|
|
3824
|
-
n = len(x_shape)
|
|
3825
|
-
shape_0 = []
|
|
3826
|
-
for s in x_shape:
|
|
3827
|
-
validator.check_int(len(s), 1, validator.EQ, 'each input rank', self.name)
|
|
3828
|
-
shape_0.append(s[0])
|
|
3829
|
-
if self.indexing == "xy":
|
|
3830
|
-
shape_0[0], shape_0[1] = shape_0[1], shape_0[0]
|
|
3831
|
-
out_shape = tuple(tuple(shape_0) for _ in range(n))
|
|
3832
|
-
return out_shape
|
|
3833
|
-
|
|
3834
|
-
def infer_dtype(self, x_type):
|
|
3835
|
-
validator.check_subclass("input[0]", x_type[0], mstype.tensor_type, self.name)
|
|
3836
|
-
n = len(x_type)
|
|
3837
|
-
for i in range(1, n):
|
|
3838
|
-
validator.check('x_type[%d]' % i, x_type[i], 'base', x_type[0], validator.EQ, self.name, TypeError)
|
|
3839
|
-
return x_type
|
|
3840
|
-
|
|
3841
|
-
|
|
3842
3584
|
class ReverseSequence(PrimitiveWithInfer):
|
|
3843
3585
|
r"""
|
|
3844
3586
|
Reverses variable length slices.
|
|
@@ -4326,7 +4068,7 @@ class TensorScatterUpdate(_TensorScatterOp):
|
|
|
4326
4068
|
r"""
|
|
4327
4069
|
Creates a new tensor by updating the positions in `input_x` indicated by
|
|
4328
4070
|
`indices`, with values from `update`. This operation is almost equivalent to using
|
|
4329
|
-
|
|
4071
|
+
:class:`mindspore.ops.ScatterNdUpdate` , except that the updates are applied on `input_x` instead of a zero tensor.
|
|
4330
4072
|
|
|
4331
4073
|
`indices` must have rank at least 2, the last axis is the depth of each index
|
|
4332
4074
|
vectors. For each index vector, there must be a corresponding value in `update`. If
|
|
@@ -5237,7 +4979,7 @@ class Tril(Primitive):
|
|
|
5237
4979
|
indicating the main diagonal.
|
|
5238
4980
|
|
|
5239
4981
|
Inputs:
|
|
5240
|
-
- **x** (Tensor) - The input tensor with shape :math:`(M, N
|
|
4982
|
+
- **x** (Tensor) - The input tensor with shape :math:`(*, M, N)`
|
|
5241
4983
|
where :math:`*` means any number of additional dimensions.
|
|
5242
4984
|
|
|
5243
4985
|
Outputs:
|
|
@@ -5387,7 +5129,7 @@ class IndexPut(Primitive):
|
|
|
5387
5129
|
>>> op = ops.IndexPut(accumulate = accumulate)
|
|
5388
5130
|
>>> output = op(x1, x2, indices)
|
|
5389
5131
|
>>> print(output)
|
|
5390
|
-
|
|
5132
|
+
[[4 5 3]
|
|
5391
5133
|
[4 5 6]]
|
|
5392
5134
|
"""
|
|
5393
5135
|
|