mindspore 2.6.0rc1__cp310-none-any.whl → 2.7.0rc1__cp310-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 +1290 -0
- mindspore/__init__.py +1 -1
- mindspore/_c_dataengine.cpython-310-aarch64-linux-gnu.so +0 -0
- mindspore/_c_expression.cpython-310-aarch64-linux-gnu.so +0 -0
- mindspore/_c_mindrecord.cpython-310-aarch64-linux-gnu.so +0 -0
- mindspore/_checkparam.py +40 -9
- mindspore/{_deprecated → _extends/optimize}/__init__.py +9 -3
- mindspore/_extends/optimize/cell_utils.py +96 -0
- mindspore/_extends/parse/__init__.py +2 -2
- mindspore/_extends/parse/compile_config.py +44 -22
- mindspore/_extends/parse/deprecated/deprecated_tensor_method.py +1 -1
- mindspore/_extends/parse/parser.py +37 -62
- mindspore/_extends/parse/resources.py +39 -0
- mindspore/_extends/parse/standard_method.py +43 -13
- mindspore/_extends/parse/trope.py +8 -1
- mindspore/_extends/pijit/__init__.py +1 -2
- mindspore/amp.py +4 -4
- mindspore/boost/adasum.py +1 -1
- mindspore/boost/boost_cell_wrapper.py +4 -4
- mindspore/common/__init__.py +27 -2
- mindspore/common/_grad_function.py +2 -1
- mindspore/common/_pijit_context.py +28 -7
- mindspore/common/_stub_tensor.py +1 -209
- mindspore/common/_tensor_cpp_method.py +1 -1
- mindspore/common/_tensor_docs.py +3096 -3035
- mindspore/common/api.py +238 -113
- mindspore/common/dtype.py +21 -11
- mindspore/common/dump.py +10 -15
- mindspore/common/generator.py +5 -3
- mindspore/common/hook_handle.py +11 -2
- mindspore/common/jit_config.py +1 -1
- mindspore/common/jit_trace.py +84 -105
- mindspore/common/parameter.py +26 -12
- mindspore/common/recompute.py +3 -3
- mindspore/common/sparse_tensor.py +0 -3
- mindspore/common/symbol.py +0 -1
- mindspore/common/tensor.py +81 -81
- mindspore/communication/_comm_helper.py +46 -4
- mindspore/communication/management.py +79 -7
- mindspore/context.py +58 -40
- mindspore/dataset/core/config.py +3 -3
- mindspore/dataset/engine/datasets.py +20 -7
- mindspore/dataset/engine/datasets_user_defined.py +33 -3
- mindspore/dataset/engine/iterators.py +2 -2
- mindspore/dataset/engine/obs/config_loader.py +2 -2
- mindspore/dataset/engine/obs/obs_mindrecord_dataset.py +8 -0
- mindspore/dataset/transforms/py_transforms.py +7 -3
- mindspore/dataset/transforms/transforms.py +7 -3
- mindspore/dataset/vision/validators.py +1 -0
- mindspore/device_context/ascend/device.py +1 -1
- mindspore/device_context/gpu/__init__.py +2 -2
- mindspore/device_context/gpu/device.py +1 -1
- mindspore/device_context/gpu/op_precision.py +4 -2
- mindspore/device_context/gpu/op_tuning.py +6 -3
- mindspore/device_manager.py +16 -9
- mindspore/experimental/llm_boost/ascend_native/llama_boost_ascend_native.py +3 -7
- mindspore/experimental/llm_boost/atb/boost_base.py +2 -3
- mindspore/experimental/optim/adadelta.py +13 -20
- mindspore/experimental/optim/adagrad.py +15 -22
- mindspore/experimental/optim/adam.py +17 -24
- mindspore/experimental/optim/adamax.py +14 -22
- mindspore/experimental/optim/adamw.py +28 -34
- mindspore/experimental/optim/asgd.py +15 -25
- mindspore/experimental/optim/lr_scheduler.py +27 -45
- mindspore/experimental/optim/nadam.py +14 -24
- mindspore/experimental/optim/optimizer.py +13 -23
- mindspore/experimental/optim/radam.py +18 -24
- mindspore/experimental/optim/rmsprop.py +14 -25
- mindspore/experimental/optim/rprop.py +15 -26
- mindspore/experimental/optim/sgd.py +9 -19
- mindspore/hal/__init__.py +4 -4
- mindspore/hal/contiguous_tensors_handle.py +2 -2
- mindspore/hal/memory.py +27 -7
- mindspore/include/api/cell.h +37 -1
- mindspore/include/api/delegate.h +10 -0
- mindspore/include/api/model.h +3 -0
- mindspore/include/api/types.h +2 -2
- mindspore/include/c_api/model_c.h +0 -58
- mindspore/include/c_api/tensor_c.h +0 -26
- mindspore/include/dataset/vision_ascend.h +1 -1
- mindspore/include/mindapi/base/type_id.h +3 -0
- mindspore/include/mindapi/base/types.h +7 -0
- mindspore/include/mindspore/ccsrc/availability/silent_check/silent_check.h +3 -4
- mindspore/include/mindspore/ccsrc/backend/common/graph_kernel/adapter/graph_kernel_cluster_cloud.h +1 -0
- mindspore/include/mindspore/ccsrc/backend/common/graph_kernel/adapter/graph_kernel_comm_info_manager.h +1 -1
- mindspore/include/mindspore/ccsrc/backend/common/graph_kernel/core/eliminate_redundant_output.h +1 -0
- mindspore/include/mindspore/ccsrc/backend/common/graph_kernel/core/graph_kernel_utils.h +17 -0
- mindspore/include/mindspore/ccsrc/backend/common/graph_kernel/core/update_state_formatter.h +2 -1
- mindspore/include/mindspore/ccsrc/backend/common/graph_kernel/depend_edge_elimination.h +61 -0
- mindspore/include/mindspore/ccsrc/backend/common/graph_kernel/graph_kernel_flags.h +9 -0
- mindspore/include/mindspore/ccsrc/backend/common/graph_kernel/model/lite_graph.h +1 -1
- mindspore/include/mindspore/ccsrc/backend/common/graph_kernel/model/node.h +5 -6
- mindspore/include/mindspore/ccsrc/backend/common/mem_reuse/mem_reuse.h +1 -2
- mindspore/include/mindspore/ccsrc/backend/common/mem_reuse/mem_reuse_checker.h +0 -1
- mindspore/include/mindspore/ccsrc/backend/common/mem_reuse/mem_swap_manager.h +0 -1
- mindspore/include/mindspore/ccsrc/backend/common/optimizer/cache_manager.h +1 -1
- mindspore/include/mindspore/ccsrc/backend/common/optimizer/dynamic_shape/convert_custom_op.h +2 -2
- mindspore/include/mindspore/ccsrc/backend/common/optimizer/dynamic_shape/link_custom_op.h +1 -1
- mindspore/include/mindspore/ccsrc/backend/common/pass/add_attr_to_node/add_attr_to_node_register.h +1 -3
- mindspore/include/mindspore/ccsrc/backend/common/pass/adjust_depend_for_parallel_optimizer_recompute_all_gather.h +1 -1
- mindspore/include/mindspore/ccsrc/backend/common/pass/convert_list_to_tuple.h +2 -1
- mindspore/include/mindspore/ccsrc/backend/common/pass/custom_defined_depend.h +1 -3
- mindspore/include/mindspore/ccsrc/backend/common/pass/gradients_allreduce_depend_last_send.h +1 -2
- mindspore/include/mindspore/ccsrc/backend/common/pass/graph_view_replace_pass.h +1 -1
- mindspore/include/mindspore/ccsrc/backend/common/pass/insert_tensor_move_for_communication.h +2 -1
- mindspore/include/mindspore/ccsrc/{plugin/device/ascend/optimizer → backend/common/pass}/ir_fusion/flash_attention_fusion.h +7 -8
- mindspore/include/mindspore/ccsrc/backend/common/pass/label_1f1b_overlap_node.h +1 -1
- mindspore/include/mindspore/ccsrc/backend/common/pass/mindir/add_depend_for_adamw.h +1 -3
- mindspore/include/mindspore/ccsrc/backend/common/pass/mindir/all_to_all_unify_mindir.h +1 -0
- mindspore/include/mindspore/ccsrc/backend/common/pass/optimize_gradients_allreduce_overlap.h +1 -1
- mindspore/include/mindspore/ccsrc/backend/common/pass/replace_node_by_proxy.h +1 -0
- mindspore/include/mindspore/ccsrc/backend/common/session/exec_order_builder.h +0 -2
- mindspore/include/mindspore/ccsrc/backend/common/session/session_basic.h +13 -16
- mindspore/include/mindspore/ccsrc/backend/common/somas/somas.h +1 -1
- mindspore/include/mindspore/ccsrc/backend/common/somas/somas_node.h +8 -2
- mindspore/include/mindspore/ccsrc/backend/ge_backend/executor/ge_device_res_manager.h +1 -2
- mindspore/include/mindspore/ccsrc/backend/ge_backend/executor/ge_graph_executor.h +13 -14
- mindspore/include/mindspore/ccsrc/backend/ge_backend/executor/ge_utils.h +0 -2
- mindspore/include/mindspore/ccsrc/backend/ge_backend/ge_backend.h +10 -8
- mindspore/include/mindspore/ccsrc/backend/ge_backend/graph_ir/convert.h +2 -1
- mindspore/include/mindspore/ccsrc/backend/ge_backend/graph_ir/graph_runner.h +2 -1
- mindspore/include/mindspore/ccsrc/backend/ge_backend/graph_ir/utils.h +12 -14
- mindspore/include/mindspore/ccsrc/backend/ge_backend/pass/matmul_allreduce_fusion.h +51 -0
- mindspore/include/mindspore/ccsrc/backend/ge_backend/runtime/actor/abstract_actor.h +26 -19
- mindspore/include/mindspore/ccsrc/backend/ge_backend/runtime/actor/actor_common.h +14 -5
- mindspore/include/mindspore/ccsrc/backend/ge_backend/runtime/actor/actor_dump.h +2 -1
- mindspore/include/mindspore/ccsrc/backend/ge_backend/runtime/actor/actor_set.h +1 -1
- mindspore/include/mindspore/ccsrc/backend/ge_backend/runtime/actor/control_flow/control_actor.h +38 -38
- mindspore/include/mindspore/ccsrc/backend/ge_backend/runtime/actor/control_flow/entrance_actor.h +9 -9
- mindspore/include/mindspore/ccsrc/backend/ge_backend/runtime/actor/control_flow/exit_actor.h +8 -8
- mindspore/include/mindspore/ccsrc/backend/ge_backend/runtime/actor/control_flow/gather_actor.h +6 -6
- mindspore/include/mindspore/ccsrc/backend/ge_backend/runtime/actor/control_flow/stack_actor.h +11 -11
- mindspore/include/mindspore/ccsrc/backend/ge_backend/runtime/actor/control_flow/switch_actor.h +2 -2
- mindspore/include/mindspore/ccsrc/backend/ge_backend/runtime/actor/data_prepare_actor.h +16 -16
- mindspore/include/mindspore/ccsrc/backend/ge_backend/runtime/actor/data_source_actor.h +9 -9
- mindspore/include/mindspore/ccsrc/backend/ge_backend/runtime/actor/debug_actor.h +8 -8
- mindspore/include/mindspore/ccsrc/backend/ge_backend/runtime/actor/debug_aware_actor.h +2 -2
- mindspore/include/mindspore/ccsrc/backend/ge_backend/runtime/actor/loop_count_actor.h +6 -6
- mindspore/include/mindspore/ccsrc/backend/ge_backend/runtime/actor/memory_aware_actor.h +6 -6
- mindspore/include/mindspore/ccsrc/backend/ge_backend/runtime/actor/memory_manager_actor.h +8 -11
- mindspore/include/mindspore/ccsrc/backend/ge_backend/runtime/actor/output_actor.h +8 -5
- mindspore/include/mindspore/ccsrc/backend/ge_backend/runtime/actor/profiler_actor.h +2 -2
- mindspore/include/mindspore/ccsrc/backend/ge_backend/runtime/actor/recorder_actor.h +2 -2
- mindspore/include/mindspore/ccsrc/backend/ge_backend/runtime/actor/super_kernel_actor.h +17 -17
- mindspore/include/mindspore/ccsrc/backend/ge_backend/runtime/device_tensor_store.h +24 -24
- mindspore/include/mindspore/ccsrc/backend/ge_backend/runtime/graph_compiler.h +6 -8
- mindspore/include/mindspore/ccsrc/backend/ge_backend/runtime/graph_partition.h +49 -0
- mindspore/include/mindspore/ccsrc/backend/ge_backend/runtime/scheduler_helper.h +1 -1
- mindspore/include/mindspore/ccsrc/backend/ge_backend/runtime/segment_runner.h +50 -0
- mindspore/include/mindspore/ccsrc/backend/ge_backend/utils/device_address_utils.h +4 -7
- mindspore/include/mindspore/ccsrc/backend/graph_compiler/op_backend.h +13 -11
- mindspore/include/mindspore/ccsrc/backend/graph_compiler/transform.h +2 -8
- mindspore/include/mindspore/ccsrc/backend/graph_compiler/vm.h +1 -5
- mindspore/include/mindspore/ccsrc/backend/ms_backend/ms_backend.h +0 -39
- mindspore/include/mindspore/ccsrc/backend/ms_backend/ms_backend_base.h +7 -4
- mindspore/include/mindspore/ccsrc/debug/checksum/checksum.h +35 -0
- mindspore/include/mindspore/ccsrc/debug/checksum/checksum_kernel.h +63 -0
- mindspore/include/mindspore/ccsrc/debug/checksum/checksum_mgr.h +50 -0
- mindspore/include/mindspore/ccsrc/debug/data_dump/device_statistic/check_overflow.h +1 -11
- mindspore/include/mindspore/ccsrc/debug/data_dump/device_statistic/common.h +0 -13
- mindspore/include/mindspore/ccsrc/debug/data_dump/device_statistic/kernel_launcher.h +3 -3
- mindspore/include/mindspore/ccsrc/debug/data_dump/device_statistic/mean.h +1 -1
- mindspore/include/mindspore/ccsrc/debug/data_dump/device_statistic/mem_manager.h +65 -0
- mindspore/include/mindspore/ccsrc/debug/data_dump/device_statistic/statistic_kernel.h +7 -8
- mindspore/include/mindspore/ccsrc/debug/data_dump/tensor_info_collect.h +21 -6
- mindspore/include/mindspore/ccsrc/debug/data_dump/tensor_statistic.h +2 -2
- mindspore/include/mindspore/ccsrc/debug/debug_services.h +1 -4
- mindspore/include/mindspore/ccsrc/debug/debugger/debugger_utils.h +7 -8
- mindspore/include/mindspore/ccsrc/debug/debugger/tensor_summary.h +0 -52
- mindspore/include/mindspore/ccsrc/debug/dump/tensordump_control.h +6 -2
- mindspore/include/mindspore/ccsrc/debug/{hooker/deprecated_env.h → dump/utils.h} +12 -9
- mindspore/include/mindspore/ccsrc/debug/profiler/mstx/mstx_impl.h +37 -24
- mindspore/include/mindspore/ccsrc/debug/profiler/mstx/mstx_symbol.h +63 -0
- mindspore/include/mindspore/ccsrc/debug/profiler/profiler.h +37 -15
- mindspore/include/mindspore/ccsrc/debug/profiler/profiling.h +9 -6
- mindspore/include/mindspore/ccsrc/debug/profiler/profiling_framework_data.h +2 -0
- mindspore/include/mindspore/ccsrc/debug/profiler/python_obj_pointer.h +7 -7
- mindspore/include/mindspore/ccsrc/debug/profiler/report_data.h +23 -0
- mindspore/include/mindspore/ccsrc/debug/profiler/thread.h +2 -2
- mindspore/include/mindspore/ccsrc/debug/utils.h +0 -5
- mindspore/include/mindspore/ccsrc/distributed/cluster/actor_route_table_proxy.h +1 -1
- mindspore/include/mindspore/ccsrc/distributed/cluster/actor_route_table_service.h +0 -2
- mindspore/include/mindspore/ccsrc/distributed/cluster/topology/meta_server_node.h +5 -5
- mindspore/include/mindspore/ccsrc/distributed/persistent/storage/local_file.h +0 -1
- mindspore/include/mindspore/ccsrc/distributed/rpc/tcp/connection.h +0 -1
- mindspore/include/mindspore/ccsrc/distributed/rpc/tcp/socket_operation.h +0 -1
- mindspore/include/mindspore/ccsrc/distributed/rpc/tcp/tcp_comm.h +0 -1
- mindspore/include/mindspore/ccsrc/distributed/rpc/tcp/tcp_socket_operation.h +0 -1
- mindspore/include/mindspore/ccsrc/frontend/ir/primitive_py.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/ir/py_execute_py.h +11 -28
- mindspore/include/mindspore/ccsrc/frontend/ir/storage.h +44 -0
- mindspore/include/mindspore/ccsrc/frontend/ir/storage_base.h +45 -0
- mindspore/include/mindspore/ccsrc/frontend/ir/tensor_py.h +10 -21
- mindspore/include/mindspore/ccsrc/frontend/operator/composite/composite.h +76 -5
- mindspore/include/mindspore/ccsrc/frontend/operator/composite/do_signature.h +2 -1
- mindspore/include/mindspore/ccsrc/frontend/operator/composite/functional_overload.h +46 -0
- mindspore/include/mindspore/ccsrc/frontend/operator/meta_dsl/common/meta_impl.h +152 -22
- mindspore/include/mindspore/ccsrc/frontend/operator/meta_dsl/common/utils.h +38 -61
- mindspore/include/mindspore/ccsrc/frontend/operator/meta_dsl/func_op/any.h +27 -0
- mindspore/include/mindspore/ccsrc/frontend/operator/meta_dsl/func_op/any_ext.h +27 -0
- mindspore/include/mindspore/ccsrc/frontend/operator/meta_dsl/func_op/conv3d_padding.h +41 -0
- mindspore/include/mindspore/ccsrc/frontend/operator/meta_dsl/func_op/einsum_ext.h +47 -0
- mindspore/include/mindspore/ccsrc/frontend/operator/meta_dsl/func_op/func_max_pool2d.h +28 -0
- mindspore/include/mindspore/ccsrc/frontend/operator/meta_dsl/func_op/gmm.h +28 -0
- mindspore/include/mindspore/ccsrc/frontend/operator/meta_dsl/func_op/gmm_backward.h +28 -0
- mindspore/include/mindspore/ccsrc/frontend/operator/meta_dsl/func_op/gmm_backward_fusion.h +28 -0
- mindspore/include/mindspore/ccsrc/frontend/operator/meta_dsl/func_op/gmm_common_utils.h +30 -0
- mindspore/include/mindspore/ccsrc/frontend/operator/meta_dsl/func_op/gmm_v2.h +28 -0
- mindspore/include/mindspore/ccsrc/frontend/operator/meta_dsl/func_op/gmm_v2_backward.h +28 -0
- mindspore/include/mindspore/ccsrc/frontend/operator/meta_dsl/func_op/gmm_v2_backward_fusion.h +28 -0
- mindspore/include/mindspore/ccsrc/frontend/operator/meta_dsl/func_op/inplace_exponential.h +31 -0
- mindspore/include/mindspore/ccsrc/frontend/operator/meta_dsl/func_op/moe_token_unpermute.h +28 -0
- mindspore/include/mindspore/ccsrc/frontend/optimizer/ad/adjoint.h +16 -2
- mindspore/include/mindspore/ccsrc/frontend/optimizer/ad/dfunctor.h +18 -10
- mindspore/include/mindspore/ccsrc/frontend/optimizer/ad/grad.h +4 -2
- mindspore/include/mindspore/ccsrc/frontend/optimizer/ad/pynative_jit_grad.h +6 -1
- mindspore/include/mindspore/ccsrc/frontend/optimizer/auto_monad_eliminate.h +3 -13
- mindspore/include/mindspore/ccsrc/frontend/optimizer/cse_pass.h +3 -7
- mindspore/include/mindspore/ccsrc/frontend/optimizer/graph_transform.h +2 -1
- mindspore/include/mindspore/ccsrc/frontend/optimizer/inplace_input_replace.h +30 -0
- mindspore/include/mindspore/ccsrc/frontend/optimizer/irpass/add_forward_monad_depend.h +47 -37
- mindspore/include/mindspore/ccsrc/frontend/optimizer/irpass/branch_culling.h +7 -104
- mindspore/include/mindspore/ccsrc/frontend/optimizer/irpass/call_graph_tuple_transform.h +2 -1
- mindspore/include/mindspore/ccsrc/frontend/optimizer/irpass/check_invalid_view_inplace_dout.h +46 -0
- mindspore/include/mindspore/ccsrc/frontend/optimizer/irpass/expand_dump_flag.h +1 -0
- mindspore/include/mindspore/ccsrc/frontend/optimizer/irpass/get_grad_eliminate.h +5 -1
- mindspore/include/mindspore/ccsrc/frontend/optimizer/irpass/inline.h +1 -0
- mindspore/include/mindspore/ccsrc/frontend/optimizer/irpass/item_dict_eliminate.h +3 -1
- mindspore/include/mindspore/ccsrc/frontend/optimizer/irpass/item_tuple_or_list_eliminate.h +1 -17
- mindspore/include/mindspore/ccsrc/frontend/optimizer/irpass/j_node_and_user_rematch.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/optimizer/irpass/loop_unroll.h +1 -0
- mindspore/include/mindspore/ccsrc/frontend/optimizer/irpass/parameter_eliminate.h +1 -0
- mindspore/include/mindspore/ccsrc/frontend/optimizer/irpass/partial_eliminate.h +1 -0
- mindspore/include/mindspore/ccsrc/frontend/optimizer/irpass/recompute_prepare.h +1 -0
- mindspore/include/mindspore/ccsrc/frontend/optimizer/irpass/special_op_eliminate.h +7 -1
- mindspore/include/mindspore/ccsrc/frontend/optimizer/irpass/stack_unstack_eliminate.h +4 -1
- mindspore/include/mindspore/ccsrc/frontend/optimizer/irpass/stopgrad_eliminate.h +3 -1
- mindspore/include/mindspore/ccsrc/frontend/optimizer/irpass/switch_or_switch_layer_defer_inline.h +10 -4
- mindspore/include/mindspore/ccsrc/frontend/optimizer/irpass.h +7 -7
- mindspore/include/mindspore/ccsrc/frontend/optimizer/opt.h +10 -5
- mindspore/include/mindspore/ccsrc/frontend/optimizer/optimizer.h +55 -280
- mindspore/include/mindspore/ccsrc/frontend/optimizer/pattern_matcher.h +8 -2
- mindspore/include/mindspore/ccsrc/frontend/parallel/auto_parallel/operator_costmodel.h +4 -0
- mindspore/include/mindspore/ccsrc/frontend/parallel/auto_parallel/stage_compute.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/came_parallel_handler.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/costmodel_context.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/dynamic_creator.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/dynamic_shape/dynamic_shape.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/graph_util/flops_collection.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/graph_util/fold_pipeline_split_utils.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/graph_util/grad_accumulation_utils.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/graph_util/graph_splitter.h +2 -0
- mindspore/include/mindspore/ccsrc/frontend/parallel/graph_util/graph_utils.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/graph_util/parallel_tensordump.h +49 -14
- mindspore/include/mindspore/ccsrc/frontend/parallel/interleaved_parallel/interleaved_parallel.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/activation_info.h +0 -11
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/addn_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/apply_rotary_pos_emb_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/arithmetic_info.h +44 -0
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/avgpool_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/cdist_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/conv3d_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/f_f_n_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/fft_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/fillv2_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/flash_attention_score_info.h +20 -20
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/fused_infer_attention_score_info.h +15 -15
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/gamma_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/grid_sampler2d.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/group_norm_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/incre_flash_attention_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/inplace_op_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/iou_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/kldiv_loss_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/kv_cache_mgr_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/kv_cache_scatter_update_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/lin_space_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/matmul_info.h +9 -0
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/max_avg_pool_3d_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/moe_compute_expert_tokens_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/moe_finalize_routing_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/moe_gating_top_k_softmax_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/moe_init_routing_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/operator_info.h +2 -0
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/ops_utils.h +4 -0
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/pad_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/paged_attention_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/paged_attention_mask_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/prompt_flash_attention_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/quant_batch_matmul_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/quant_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/quant_linear_sparse_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/reshape_and_cache_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/reshape_info.h +9 -4
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/scatter_math_ops_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/scatter_nd_ops_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/scatter_ops_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/self_define_shard_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/stand_alone_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/topkrouter_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/{decoder_k_v_cache_info.h → topprouter_info.h} +12 -15
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/tracev2_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/tril_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/weight_quant_batch_matmul_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/wkv_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/parallel_postprocessor.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/parallel_preprocessor.h +2 -2
- mindspore/include/mindspore/ccsrc/frontend/parallel/parallel_processor.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/parallel_processor_context.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/parallel_whole_graph_processor.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/parameter_manager.h +0 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/allreduce_slice_to_reducescatter.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/assign_add_opt.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/begin_end_overlap_inline.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/bias_add_comm_swap.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/dataset_reader_optimizer.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/fias_sp.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/flash_sp.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/float32_redistribution.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/full_micro_interleaved_order_control.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/handle_group_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/interleave_branches_utils.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/interleave_parallel_branches.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/interleave_split_concat_branches.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/label_fine_grained_interleaved_index.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/label_micro_interleaved_index.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/matmul_add_comm_reduction.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/merge_cast_opt.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/merge_comm.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/merge_recompute_call_nodes.h +28 -0
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/micro_interleaved_order_control.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/offloading_packed_expert.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/optimize_parallel_allgather_comm.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/overlap_grad_comm.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/overlap_grad_flash_sp.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/overlap_grad_ring_attention.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/overlap_gradmatmul_and_gradallreduce.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/overlap_opt_shard_in_pipeline.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/overlap_param_gather.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/overlap_recompute_allgather_and_flashattention_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/overlap_recompute_and_grad_model_parallel.h +1 -1
- mindspore/include/mindspore/ccsrc/{pynative/grad/jit/jit_dfunctor.h → frontend/parallel/pass/overlap_recompute_comm.h} +8 -8
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/pass_utils.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/remove_cast_before_assign_add.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/reorder_send_recv_between_fp_bp.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/slice_activation_in_cell_share_recompute.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/split_layernorm_comm_fp.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/split_matmul_comm_elementwise_fp.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pass/swap_dp_allreduce_reducescatter.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pipeline_transformer/fold_pipeline_transformer.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pipeline_transformer/gpipe_interleave_scheduler.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pipeline_transformer/pipeline_interleave.h +2 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pipeline_transformer/pipeline_scheduler.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/pipeline_transformer/seqpipe_scheduler.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/shard/shard.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/step_assigned_parallel.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/strategy_checkpoint/strategy_checkpoint_info.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/strategy_loader.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/tensor_layout/layout_utils.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/tensor_layout/shared_parameter.h +1 -1
- mindspore/include/mindspore/ccsrc/frontend/parallel/tensor_layout/tensor_transform.h +2 -2
- mindspore/include/mindspore/ccsrc/include/backend/anf_runtime_algorithm.h +29 -13
- mindspore/include/mindspore/ccsrc/include/backend/data_queue/data_queue_mgr.h +1 -1
- mindspore/include/mindspore/ccsrc/include/backend/debug/data_dump/dump_control.h +1 -0
- mindspore/include/mindspore/ccsrc/include/backend/debug/data_dump/dump_json_parser.h +4 -5
- mindspore/include/mindspore/ccsrc/include/backend/debug/data_dump/dump_utils.h +1 -0
- mindspore/include/mindspore/ccsrc/include/backend/debug/debugger/debugger.h +0 -2
- mindspore/include/mindspore/ccsrc/include/backend/debug/execute_order_tracker/execute_order_tracker.h +14 -4
- mindspore/include/mindspore/ccsrc/include/backend/debug/tensor_data.h +0 -19
- mindspore/include/mindspore/ccsrc/include/backend/distributed/cluster/tcp_store.h +53 -0
- mindspore/include/mindspore/ccsrc/include/backend/distributed/collective/collective_manager.h +17 -8
- mindspore/include/mindspore/ccsrc/include/backend/distributed/constants.h +1 -10
- mindspore/include/mindspore/ccsrc/include/backend/distributed/embedding_cache/embedding_hash_map.h +0 -2
- mindspore/include/mindspore/ccsrc/include/backend/distributed/ps/ps_context.h +0 -6
- mindspore/include/mindspore/ccsrc/include/backend/distributed/rpc/tcp/constants.h +2 -1
- mindspore/include/mindspore/ccsrc/include/backend/kernel_graph.h +0 -1
- mindspore/include/mindspore/ccsrc/include/backend/kernel_info.h +8 -10
- mindspore/include/mindspore/ccsrc/include/backend/mbuf_device_address.h +5 -5
- mindspore/include/mindspore/ccsrc/include/backend/mem_reuse/abstract_dynamic_mem_pool.h +10 -5
- mindspore/include/mindspore/ccsrc/include/backend/mem_reuse/address_discretizer.h +63 -0
- mindspore/include/mindspore/ccsrc/include/backend/mem_reuse/dynamic_mem_pool.h +12 -0
- mindspore/include/mindspore/ccsrc/include/backend/mem_reuse/max_segment_tree.h +181 -0
- mindspore/include/mindspore/ccsrc/include/backend/mem_reuse/mem_dynamic_allocator.h +0 -1
- mindspore/include/mindspore/ccsrc/include/backend/mem_reuse/mem_pool_util.h +98 -0
- mindspore/include/mindspore/ccsrc/include/backend/mem_reuse/mem_tracker.h +11 -107
- mindspore/include/mindspore/ccsrc/include/backend/mem_reuse/race_checker.h +64 -0
- mindspore/include/mindspore/ccsrc/include/backend/mem_reuse/tracker_graph.h +91 -0
- mindspore/include/mindspore/ccsrc/include/backend/optimizer/inplace_node_pass.h +1 -9
- mindspore/include/mindspore/ccsrc/include/backend/optimizer/optimizer.h +3 -6
- mindspore/include/mindspore/ccsrc/include/backend/optimizer/pattern_engine.h +0 -2
- mindspore/include/mindspore/ccsrc/include/backend/optimizer/visitor.h +2 -0
- mindspore/include/mindspore/ccsrc/include/backend/py_execute_utils.h +2 -1
- mindspore/include/mindspore/ccsrc/include/common/debug/draw.h +3 -1
- mindspore/include/mindspore/ccsrc/include/common/debug/dump_proto.h +2 -10
- mindspore/include/mindspore/ccsrc/include/common/expander/core/emitter.h +1 -1
- mindspore/include/mindspore/ccsrc/include/common/expander/core/node.h +1 -1
- mindspore/include/mindspore/ccsrc/include/common/np_dtype/np_dtypes.h +6 -15
- mindspore/include/mindspore/ccsrc/include/common/pybind_api/api_register.h +6 -1
- mindspore/include/mindspore/ccsrc/include/common/pynative/abstract_converter.h +0 -4
- mindspore/include/mindspore/ccsrc/include/common/pynative/adapter.h +2 -5
- mindspore/include/mindspore/ccsrc/include/common/pynative/common_utils.h +5 -1
- mindspore/include/mindspore/ccsrc/include/common/pynative/grad_state.h +12 -0
- mindspore/include/mindspore/ccsrc/include/common/pynative/variable.h +326 -0
- mindspore/include/mindspore/ccsrc/include/common/random.h +1 -2
- mindspore/include/mindspore/ccsrc/include/common/runtime_conf/runtime_conf.h +7 -8
- mindspore/include/mindspore/ccsrc/include/common/runtime_conf/thread_bind_core.h +12 -10
- mindspore/include/mindspore/ccsrc/include/common/utils/anfalgo.h +8 -2
- mindspore/include/mindspore/ccsrc/include/common/utils/comm_manager.h +0 -1
- mindspore/include/mindspore/ccsrc/include/common/utils/compile_cache_context.h +4 -2
- mindspore/include/mindspore/ccsrc/include/common/utils/convert_utils.h +18 -4
- mindspore/include/mindspore/ccsrc/include/common/utils/convert_utils_py.h +2 -10
- mindspore/include/mindspore/ccsrc/include/common/utils/cse.h +0 -1
- mindspore/include/mindspore/ccsrc/include/common/utils/json_operation_utils.h +1 -1
- mindspore/include/mindspore/ccsrc/include/common/utils/ms_device_shape_transfer.h +0 -1
- mindspore/include/mindspore/ccsrc/include/common/utils/parallel_context.h +8 -2
- mindspore/include/mindspore/ccsrc/include/common/utils/python_adapter.h +3 -3
- mindspore/include/mindspore/ccsrc/include/common/utils/stub_tensor.h +1 -3
- mindspore/include/mindspore/ccsrc/include/common/utils/summary/event_writer.h +1 -1
- mindspore/include/mindspore/ccsrc/include/common/utils/tensor_py.h +24 -117
- mindspore/include/mindspore/ccsrc/include/common/utils/tensor_py_wrapper.h +1 -7
- mindspore/include/mindspore/ccsrc/include/common/utils/tensor_utils.h +3 -3
- mindspore/include/mindspore/ccsrc/include/common/utils/utils.h +6 -8
- mindspore/include/mindspore/ccsrc/include/common/visible.h +0 -10
- mindspore/include/mindspore/ccsrc/kernel/framework_utils.h +3 -0
- mindspore/include/mindspore/ccsrc/kernel/graph_kernel_info.h +1 -1
- mindspore/include/mindspore/ccsrc/kernel/philox_random.h +47 -87
- mindspore/include/mindspore/ccsrc/minddata/dataset/api/python/python_mp.h +8 -0
- mindspore/include/mindspore/ccsrc/minddata/dataset/core/cv_tensor.h +1 -1
- mindspore/include/mindspore/ccsrc/minddata/dataset/core/tensor.h +26 -2
- mindspore/include/mindspore/ccsrc/minddata/dataset/engine/connector.h +7 -0
- mindspore/include/mindspore/ccsrc/minddata/dataset/engine/datasetops/map_op/map_op.h +3 -0
- mindspore/include/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/mindrecord_op.h +0 -1
- mindspore/include/mindspore/ccsrc/minddata/dataset/include/dataset/vision_ascend.h +1 -1
- mindspore/include/mindspore/ccsrc/minddata/dataset/kernels/data/parse_example_op.h +19 -15
- mindspore/include/mindspore/ccsrc/minddata/dataset/kernels/image/dvpp/utils/AclLiteType.h +6 -1
- mindspore/include/mindspore/ccsrc/minddata/dataset/kernels/image/dvpp/utils/AclLiteUtils.h +6 -1
- mindspore/include/mindspore/ccsrc/minddata/dataset/kernels/image/dvpp/utils/DvppCommon.h +1 -1
- mindspore/include/mindspore/ccsrc/minddata/dataset/kernels/image/dvpp/utils/ErrorCode.h +2 -2
- mindspore/include/mindspore/ccsrc/minddata/dataset/kernels/image/dvpp/utils/ThreadSafeQueue.h +3 -1
- mindspore/include/mindspore/ccsrc/minddata/dataset/kernels/image/dvpp/utils/VdecHelper.h +9 -12
- mindspore/include/mindspore/ccsrc/minddata/dataset/kernels/image/dvpp/utils/dvpp_video.h +29 -10
- mindspore/include/mindspore/ccsrc/minddata/dataset/kernels/image/lite_cv/image_process.h +1 -1
- mindspore/include/mindspore/ccsrc/minddata/dataset/util/btree.h +1 -1
- mindspore/include/mindspore/ccsrc/minddata/dataset/util/cond_var.h +0 -1
- mindspore/include/mindspore/ccsrc/minddata/dataset/util/json_helper.h +1 -1
- mindspore/include/mindspore/ccsrc/minddata/dataset/util/queue.h +1 -1
- mindspore/include/mindspore/ccsrc/minddata/dataset/util/sig_handler.h +2 -2
- mindspore/include/mindspore/ccsrc/minddata/dataset/util/system_pool.h +1 -1
- mindspore/include/mindspore/ccsrc/minddata/dataset/util/task_manager.h +0 -1
- mindspore/include/mindspore/ccsrc/minddata/mindrecord/include/shard_distributed_sample.h +0 -1
- mindspore/include/mindspore/ccsrc/minddata/mindrecord/include/shard_header.h +0 -1
- mindspore/include/mindspore/ccsrc/minddata/mindrecord/include/shard_index.h +0 -1
- mindspore/include/mindspore/ccsrc/minddata/mindrecord/include/shard_pk_sample.h +0 -1
- mindspore/include/mindspore/ccsrc/minddata/mindrecord/include/shard_reader.h +0 -1
- mindspore/include/mindspore/ccsrc/minddata/mindrecord/include/shard_schema.h +0 -1
- mindspore/include/mindspore/ccsrc/minddata/mindrecord/include/shard_statistics.h +0 -1
- mindspore/include/mindspore/ccsrc/minddata/mindrecord/include/shard_writer.h +0 -1
- mindspore/include/mindspore/ccsrc/minddata/utils.h +30 -2
- mindspore/include/mindspore/{ops/ops_utils/ms_extension.h → ccsrc/ms_extension/all.h} +12 -5
- mindspore/include/mindspore/ccsrc/ms_extension/api.h +32 -0
- mindspore/include/mindspore/ccsrc/ms_extension/ascend/atb/atb_common.h +98 -0
- mindspore/include/mindspore/ccsrc/ms_extension/ascend/atb/operation_cache.h +229 -0
- mindspore/include/mindspore/ccsrc/ms_extension/common/tensor.h +200 -0
- mindspore/include/mindspore/ccsrc/ms_extension/common/visible.h +28 -0
- mindspore/include/mindspore/ccsrc/ms_extension/pynative/pyboost_extension.h +312 -0
- mindspore/include/mindspore/ccsrc/pipeline/jit/pi/capture_context.h +1 -1
- mindspore/include/mindspore/ccsrc/pipeline/jit/pi/eval_frame_hook.h +2 -2
- mindspore/include/mindspore/ccsrc/pipeline/jit/pi/graph_build/build_graph_utils.h +5 -1
- mindspore/include/mindspore/ccsrc/pipeline/jit/pi/graph_build/func_graph_builder.h +22 -5
- mindspore/include/mindspore/ccsrc/pipeline/jit/pi/graph_capture/abstract_object.h +92 -19
- mindspore/include/mindspore/ccsrc/pipeline/jit/pi/graph_capture/cfg.h +19 -5
- mindspore/include/mindspore/ccsrc/pipeline/jit/pi/graph_capture/code_generator.h +53 -49
- mindspore/include/mindspore/ccsrc/pipeline/jit/pi/graph_capture/graph.h +40 -10
- mindspore/include/mindspore/ccsrc/pipeline/jit/pi/graph_capture/graph_analyzer.h +3 -19
- mindspore/include/mindspore/ccsrc/pipeline/jit/pi/graph_capture/graph_arguments_optimizer.h +145 -0
- mindspore/include/mindspore/ccsrc/pipeline/jit/pi/graph_capture/graph_build.h +13 -24
- mindspore/include/mindspore/ccsrc/pipeline/jit/pi/graph_capture/node.h +33 -3
- mindspore/include/mindspore/ccsrc/pipeline/jit/pi/graph_capture/side_effect.h +116 -27
- mindspore/include/mindspore/ccsrc/pipeline/jit/pi/graph_compiler/abstract_type.h +2 -2
- mindspore/include/mindspore/ccsrc/pipeline/jit/pi/graph_compiler/compiler.h +4 -0
- mindspore/include/mindspore/ccsrc/pipeline/jit/pi/graph_compiler/utils.h +2 -0
- mindspore/include/mindspore/ccsrc/pipeline/jit/pi/graph_guard/cache.h +32 -8
- mindspore/include/mindspore/ccsrc/pipeline/jit/pi/graph_guard/guard.h +11 -15
- mindspore/include/mindspore/ccsrc/pipeline/jit/pi/graph_guard/guard_utils.h +17 -8
- mindspore/include/mindspore/ccsrc/pipeline/jit/pi/graph_guard/infer.h +0 -2
- mindspore/include/mindspore/ccsrc/pipeline/jit/pi/graph_guard/shape_ctx.h +3 -3
- mindspore/include/mindspore/ccsrc/pipeline/jit/pi/graph_guard/trace.h +50 -93
- mindspore/include/mindspore/ccsrc/pipeline/jit/pi/jit_compile_results.h +1 -16
- mindspore/include/mindspore/ccsrc/pipeline/jit/pi/pi_jit_config.h +11 -22
- mindspore/include/mindspore/ccsrc/pipeline/jit/pi/python_adapter/py_code.h +56 -5
- mindspore/include/mindspore/ccsrc/pipeline/jit/pi/python_adapter/py_frame.h +6 -7
- mindspore/include/mindspore/ccsrc/pipeline/jit/pi/python_adapter/pydef.h +0 -6
- mindspore/include/mindspore/ccsrc/pipeline/jit/pi/runtime.h +1 -1
- mindspore/include/mindspore/ccsrc/pipeline/jit/pi/utils/opcode_util.h +5 -0
- mindspore/include/mindspore/ccsrc/pipeline/jit/pi/utils/stop_trace_reason.h +26 -17
- mindspore/include/mindspore/ccsrc/pipeline/jit/pi/utils/utils.h +15 -11
- mindspore/include/mindspore/ccsrc/pipeline/jit/ps/action.h +2 -4
- mindspore/include/mindspore/ccsrc/pipeline/jit/ps/executor/executor_py.h +116 -0
- mindspore/include/mindspore/ccsrc/pipeline/jit/ps/executor/graph_executor_py.h +118 -0
- mindspore/include/mindspore/ccsrc/pipeline/jit/ps/{pipeline_jit.h → executor/jit_executor_py.h} +6 -6
- mindspore/include/mindspore/ccsrc/pipeline/jit/ps/graph_circle_handler.h +35 -0
- mindspore/include/mindspore/ccsrc/pipeline/jit/ps/parse/data_converter.h +6 -0
- mindspore/include/mindspore/ccsrc/pipeline/jit/ps/parse/function_block.h +1 -0
- mindspore/include/mindspore/ccsrc/pipeline/jit/ps/parse/parse.h +10 -0
- mindspore/include/mindspore/ccsrc/pipeline/jit/ps/parse/parse_base.h +1 -6
- mindspore/include/mindspore/ccsrc/pipeline/jit/ps/parse/resolve.h +9 -0
- mindspore/include/mindspore/ccsrc/pipeline/jit/ps/pass.h +3 -1
- mindspore/include/mindspore/ccsrc/pipeline/jit/ps/pass_config.h +4 -0
- mindspore/include/mindspore/ccsrc/pipeline/jit/ps/pipeline.h +12 -170
- mindspore/include/mindspore/ccsrc/pipeline/jit/ps/resource.h +7 -16
- mindspore/include/mindspore/ccsrc/pipeline/jit/ps/static_analysis/async_eval_result.h +2 -2
- mindspore/include/mindspore/ccsrc/pipeline/jit/ps/static_analysis/auto_monad.h +1 -1
- mindspore/include/mindspore/ccsrc/pipeline/jit/ps/static_analysis/builtin_prim.h +3 -3
- mindspore/include/mindspore/ccsrc/pipeline/jit/ps/static_analysis/evaluator.h +12 -12
- mindspore/include/mindspore/ccsrc/pipeline/jit/ps/static_analysis/order_enforce.h +1 -1
- mindspore/include/mindspore/ccsrc/pipeline/jit/ps/static_analysis/prim.h +13 -76
- mindspore/include/mindspore/ccsrc/pipeline/jit/ps/static_analysis/prim_utils.h +78 -0
- mindspore/include/mindspore/ccsrc/pipeline/jit/ps/static_analysis/program_specialize.h +1 -1
- mindspore/include/mindspore/ccsrc/pipeline/jit/ps/validator.h +3 -0
- mindspore/include/mindspore/ccsrc/pipeline/jit/trace/trace_recorder.h +19 -4
- mindspore/include/mindspore/ccsrc/pipeline/llm_boost/llm_boost_binder.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/hal/device/ascend_kernel_runtime.h +0 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/hal/device/kernel_select_ascend.h +3 -4
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/hal/hardware/acl_stream_assign.h +2 -2
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/hal/hardware/ascend_deprecated_interface.h +0 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/hal/hardware/{ge_device_context.h → ascend_device_context.h} +10 -19
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/hal/hardware/ascend_device_res_manager.h +14 -10
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/hal/hardware/ge_graph_optimization.h +0 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/hal/hardware/ge_kernel_executor.h +10 -2
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/hal/hardware/stress_detect.h +40 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/hal/profiler/ascend_profiling.h +32 -7
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/hal/profiler/feature_mgr.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/hal/profiler/mstx/mstx_dispatcher.h +10 -8
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/atb/add_atb_kernel.h +2 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/atb/atb_adapter.h +12 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/atb/atb_kernel_mod.h +3 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/atb/inplace_grouped_matmul_add_atb_kernel.h +2 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/atb/inplace_matmul_add_atb_kernel.h +39 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/dvm/dvm_comm_info.h +1 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/dvm/dvm_kernel_mod.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/dvm/lazy_fusion_kernel.h +15 -16
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/dvm/lazy_fusion_op.h +118 -93
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/hccl_kernel.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/hcom_all_gather_v.h +51 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/hcom_receive.h +3 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/hcom_reduce_scatter_v.h +51 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/hcom_util.h +26 -11
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/auto_generate/dist_comm_all_gather.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/auto_generate/dist_comm_all_gather_into_tensor.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/auto_generate/dist_comm_all_reduce.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/auto_generate/dist_comm_all_to_all_v.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/auto_generate/dist_comm_all_to_all_v_single.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/auto_generate/dist_comm_barrier.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/auto_generate/dist_comm_batch_isend_irecv.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/auto_generate/dist_comm_broadcast.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/auto_generate/dist_comm_gather.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/auto_generate/dist_comm_gather_into_tensor.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/auto_generate/dist_comm_irecv.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/auto_generate/dist_comm_isend.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/auto_generate/dist_comm_reduce.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/auto_generate/dist_comm_reduce_scatter.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/auto_generate/dist_comm_reduce_scatter_tensor.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/auto_generate/dist_comm_scatter.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/auto_generate/dist_comm_scatter_tensor.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/auto_generate/inner_comm_all_gather.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/auto_generate/inner_comm_all_reduce.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/auto_generate/inner_comm_all_to_all_v.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/auto_generate/inner_comm_irecv.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/auto_generate/inner_comm_isend.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/auto_generate/inner_comm_reduce_scatter.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/comm_common.h +4 -4
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/dist_comm_all_gather.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/dist_comm_all_gather_into_tensor.h +2 -2
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/dist_comm_all_reduce.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/dist_comm_all_to_all_v.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/dist_comm_all_to_all_v_single.h +2 -2
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/dist_comm_broadcast.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/dist_comm_gather.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/dist_comm_gather_into_tensor.h +2 -2
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/dist_comm_irecv.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/dist_comm_isend.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/dist_comm_reduce.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/dist_comm_reduce_scatter.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/dist_comm_reduce_scatter_tensor.h +2 -2
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/dist_comm_scatter.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/dist_comm_scatter_tensor.h +2 -2
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/inner_comm_all_gather.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/inner_comm_all_reduce.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/inner_comm_all_to_all_v.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/inner_comm_isend.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/pyboost/inner_comm_reduce_scatter.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/internal/add_rms_norm_quant.h +15 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/{ge/ge_kernel_build.h → internal/dynamic_ntk.h} +12 -9
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/internal/fused_add_topk_div.h +31 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/internal/internal_tiling_cache.h +3 -3
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/internal/pyboost/apply_rotary_pos_emb.h +46 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/internal/pyboost/auto_gen/internal_kernel_info_adapter.h +82 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/internal/pyboost/auto_gen/kernel_info_adapter.h +69 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/internal/pyboost/flash_attention_score.h +54 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/internal/pyboost/internal_kernel_info.h +162 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/internal/pyboost/internal_pyboost_utils.h +121 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/internal/pyboost/paged_attention.h +91 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/internal/pyboost/reshape_and_cache.h +43 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/internal/swiglu_dynamic_quant.h +32 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/rts/reshape_ext.h +5 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/boost_model_atb.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb/atb_infer.h +7 -9
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb/comm.h +82 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb/common_op_params.h +77 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb/context.h +50 -10
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb/graph_op_builder.h +24 -18
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb/infer_op_params.h +2331 -671
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb/operation.h +29 -11
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb/operation_infra.h +78 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb/svector.h +19 -22
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb/train_op_params.h +215 -24
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb/types.h +39 -24
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb/utils.h +7 -9
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb_speed/base/context_factory.h +1 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb_speed/base/event_manager.h +156 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb_speed/base/external_comm_manager.h +68 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb_speed/base/hosttensor_binder.h +0 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb_speed/base/model.h +33 -29
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb_speed/log/error.h +49 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb_speed/log/file_utils.h +86 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb_speed/log/log_config.h +84 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb_speed/log/log_error.h +20 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb_speed/log/log_utils.h +86 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb_speed/log.h +128 -52
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb_speed/utils/ModelTaskExecutor.h +64 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb_speed/{log/log_sink_stdout.h → utils/TaskQueue.h} +21 -11
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb_speed/utils/check_util.h +80 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb_speed/utils/config.h +2 -12
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb_speed/utils/{filesystem.h → file_system.h} +1 -11
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb_speed/utils/hccl_runner.h +48 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb_speed/utils/model_factory.h +1 -3
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb_speed/utils/operation_factory.h +3 -3
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb_speed/utils/operation_util.h +6 -5
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb_speed/utils/share_memory.h +46 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb_speed/utils/singleton.h +7 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb_speed/utils/statistic.h +1 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb_speed/utils/str_split.h +0 -2
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/optimizer/expander_fallback.h +33 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/optimizer/ge_backend_optimization.h +1 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/matmul_assignadd_fusion.h +61 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion_infer/inference_swiglu_fusion_v2.h +52 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion_infer/inference_weight_preprocess_utils.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion_infer/moe_init_routing_dyn_quantv2_fusion.h +66 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion_infer/rms_norm_quant_fusion.h +24 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion_infer/swiglu_dynamic_quant_fusion.h +47 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion_infer/swiglu_reshape_dynamic_quant_fusion.h +47 -0
- mindspore/include/mindspore/ccsrc/plugin/device/cpu/hal/device/cpu_common.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/cpu/hal/device/cpu_kernel_runtime.h +2 -1
- mindspore/include/mindspore/ccsrc/plugin/device/cpu/hal/hardware/cpu_device_context.h +20 -11
- mindspore/include/mindspore/ccsrc/plugin/device/cpu/hal/hardware/cpu_somas.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/cpu/hal/hardware/mpi_collective_comm_lib.h +3 -1
- mindspore/include/mindspore/ccsrc/plugin/device/cpu/hal/hardware/ms_collective_comm_lib.h +9 -2
- mindspore/include/mindspore/ccsrc/plugin/device/cpu/kernel/contiguous_cpu_kernel.h +8 -6
- mindspore/include/mindspore/ccsrc/plugin/device/cpu/kernel/cpu_kernel.h +2 -2
- mindspore/include/mindspore/ccsrc/plugin/device/cpu/kernel/custom/custom_kernel_input_info.h +99 -0
- mindspore/include/mindspore/ccsrc/plugin/device/cpu/kernel/custom/{custom_julia_cpu_kernel.h → custom_op_plugin_kernel.h} +23 -11
- mindspore/include/mindspore/ccsrc/plugin/device/gpu/hal/hardware/gpu_device_context.h +18 -10
- mindspore/include/mindspore/ccsrc/plugin/device/gpu/hal/hardware/gpu_somas.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/device/gpu/hal/hardware/nvidia_collective_comm_lib.h +3 -1
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/ascend_device_address/ascend_device_address.h +27 -16
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/ascend_res_manager.h +17 -10
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/collective/ascend_collective_comm_lib.h +6 -5
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/collective/ascend_communication_group.h +27 -12
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/collective/ccool_collective_comm_lib.h +4 -5
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/collective/ccool_communication_group.h +4 -4
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/collective/dummy_ascend_collective_comm_lib.h +3 -5
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/collective/dvm_collective_comm_lib.h +5 -5
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/collective/dvm_communication_group.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/collective/hccl_watch_dog_thread.h +5 -3
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/collective/leaper_trans.h +2 -5
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/collective/lowlatency_collective_comm_lib.h +4 -5
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/collective/multi_ascend_collective_comm_lib.h +6 -6
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/collective/multi_ascend_communication_group.h +4 -2
- mindspore/include/mindspore/ccsrc/plugin/{device/ascend/kernel → res_manager/ascend}/dvm/dvm.h +15 -1
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/hal_manager/ascend_hal_manager.h +1 -0
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/hccl_adapter/hccl_adapter.h +31 -1
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/hccl_adapter/plugin/hccl_plugin.h +8 -0
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/mbuf_manager/mbuf_receive_manager.h +6 -1
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/mem_manager/abstract_ascend_memory_pool_support.h +2 -0
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/mem_manager/ascend_dynamic_mem_adapter.h +3 -2
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/mem_manager/ascend_memory_adapter.h +1 -0
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/mem_manager/ascend_memory_manager.h +1 -0
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/mem_manager/ascend_memory_pool.h +14 -0
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/mem_manager/ascend_two_pointer_mem_adapter.h +1 -0
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/mem_manager/ascend_vmm_adapter.h +3 -1
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/op_adapter/custom_op_proto/cust_array_ops.h +11 -0
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/op_adapter/custom_op_proto/cust_other_ops.h +0 -22
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/op_adapter/op_adapter_base.h +38 -33
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/op_adapter/op_adapter_map.h +3 -2
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/op_adapter/op_adapter_util.h +3 -0
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/op_adapter/op_declare/array_ops_declare.h +3 -0
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/op_adapter/op_declare/transform_fusion_ops_declare.h +0 -6
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/op_adapter/transform_util.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/symbol_interface/acl_rt_symbol.h +6 -0
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/symbol_interface/symbol_utils.h +11 -6
- mindspore/include/mindspore/ccsrc/plugin/res_manager/cpu/cpu_device_address/cpu_device_address.h +17 -12
- mindspore/include/mindspore/ccsrc/plugin/{device/cpu/hal/device → res_manager/cpu/cpu_mem_manager}/cpu_hash_table.h +2 -1
- mindspore/include/mindspore/ccsrc/plugin/{device/cpu/hal/device → res_manager/cpu/cpu_mem_manager}/cpu_hash_table_util.h +1 -1
- mindspore/include/mindspore/ccsrc/plugin/res_manager/cpu/cpu_mem_manager/cpu_memory_manager.h +4 -10
- mindspore/include/mindspore/ccsrc/plugin/res_manager/cpu/cpu_mem_manager/cpu_memory_pool.h +3 -1
- mindspore/include/mindspore/ccsrc/plugin/res_manager/cpu/cpu_res_manager.h +8 -15
- mindspore/include/mindspore/ccsrc/plugin/res_manager/cpu/visible.h +32 -0
- mindspore/include/mindspore/ccsrc/plugin/res_manager/gpu/device/gpu_device_address.h +19 -11
- mindspore/include/mindspore/ccsrc/plugin/res_manager/gpu/gpu_res_manager.h +8 -8
- mindspore/include/mindspore/ccsrc/ps/core/collective_ops_impl.h +31 -6
- mindspore/include/mindspore/ccsrc/ps/core/communicator/http_request_handler.h +0 -1
- mindspore/include/mindspore/ccsrc/ps/core/file_configuration.h +2 -2
- mindspore/include/mindspore/ccsrc/ps/core/node.h +1 -1
- mindspore/include/mindspore/ccsrc/pybind_api/hal/event_py.h +2 -2
- mindspore/include/mindspore/ccsrc/pybind_api/hal/memory_py.h +2 -0
- mindspore/include/mindspore/ccsrc/pybind_api/hal/stream_py.h +3 -4
- mindspore/include/mindspore/ccsrc/pybind_api/ir/tensor_api/auto_generate/tensor_api.h +131 -128
- mindspore/include/mindspore/ccsrc/pybind_api/ir/tensor_index_py.h +10 -107
- mindspore/include/mindspore/ccsrc/pybind_api/ir/tensor_register/auto_generate/tensor_py_gen.h +131 -129
- mindspore/include/mindspore/ccsrc/pybind_api/ir/tensor_register/tensor_func_reg.h +0 -1
- mindspore/include/mindspore/ccsrc/pybind_api/resource/manager.h +2 -2
- mindspore/include/mindspore/ccsrc/pybind_api/storage_py.h +36 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/abs.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/acos_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/acosh_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/adamw.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/adaptive_avg_pool1d.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/adaptive_avg_pool2d_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/adaptive_avg_pool2d_grad_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/adaptive_avg_pool3d_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/adaptive_avg_pool3d_grad_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/adaptive_max_pool1d.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/adaptive_max_pool2d.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/add.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/add_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/add_layer_norm_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/add_layernorm_v2.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/add_rms_norm.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/add_rmsnorm_quant_v2.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/add_scalar.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/addbmm.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/addcdiv_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/addcmul_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/addmm.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/addmv.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/all_finite.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/all_gather_matmul.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/any.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/any_ext.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/apply_rotary_pos_emb.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/arange.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/argmax_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/argmax_with_value.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/argmin_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/argmin_with_value.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/argsort.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/as_strided.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/asin_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/asinh_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/atan2_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/atan_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/atanh.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/avg_pool1d.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/avg_pool2d.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/avg_pool2d_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/avg_pool3d_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/avg_pool3d_grad_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/baddbmm.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/batch_mat_mul.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/batch_norm_elemt.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/batch_norm_elemt_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/batch_norm_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/batch_norm_gather_stats_with_counts.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/batch_norm_grad_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/batch_norm_reduce_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/batch_norm_stats.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/bernoulli_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/binary_cross_entropy.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/binary_cross_entropy_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/binary_cross_entropy_with_logits.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/binary_cross_entropy_with_logits_backward.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/bincount_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/bitwise_and_scalar.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/bitwise_and_tensor.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/bitwise_not.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/bitwise_or_scalar.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/bitwise_or_tensor.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/bitwise_xor_scalar.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/bitwise_xor_tensor.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/bmm_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/broadcast_to.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/broadcast_to_view.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/cast.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/ceil.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/chunk.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/{unstack_ext.h → chunk_view.h} +8 -8
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/clamp_scalar.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/clamp_tensor.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/clone.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/col2im_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/col2im_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/concat.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/constant_pad_nd.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/contiguous.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/conv1d_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/conv1d_padding.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/conv2d_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/conv2d_padding.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/conv3d_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/conv3d_padding.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/conv_transpose2d.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/convolution.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/convolution_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/convolution_str.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/convolution_str_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/copy.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/cos.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/cosh.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/count_nonzero.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/cross.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/cummax.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/cummin_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/cumsum_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/custom_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/dense.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/diag_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/{transpose_ext.h → diagonal_view.h} +8 -8
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/dist_comm_all_gather.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/dist_comm_all_gather_into_tensor.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/dist_comm_all_reduce.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/dist_comm_all_to_all_v.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/dist_comm_all_to_all_v_single.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/dist_comm_barrier.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/dist_comm_batch_isend_irecv.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/dist_comm_broadcast.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/dist_comm_gather.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/dist_comm_gather_into_tensor.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/dist_comm_irecv.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/dist_comm_isend.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/dist_comm_reduce.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/dist_comm_reduce_scatter.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/dist_comm_reduce_scatter_tensor.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/dist_comm_scatter.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/dist_comm_scatter_tensor.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/div.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/divmod.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/divmods.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/divs.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/dot.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/dropout_do_mask_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/dropout_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/dropout_gen_mask_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/dropout_grad_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/dynamic_quant_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/{select_ext.h → einsum_ext.h} +8 -8
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/elu.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/elu_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/elu_grad_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/embedding.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/embedding_dense_backward.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/empty.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/empty_like.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/equal.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/equal_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/erf.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/erfc.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/erfinv.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/exp.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/exp2.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/expand_as.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/expand_dims.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/expand_dims_view.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/expm1.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/eye.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/ffn_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/fill_scalar.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/fill_tensor.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/flash_attention_score.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/flash_attention_score_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/flatten_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/floor.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/floor_div.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/floor_div_scalar.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/fmod_scalar.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/fmod_tensor.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/frac.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/full_like.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/func_max_pool2d.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/fused_infer_attention_score.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/gather_d.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/gather_d_grad_v2.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/gcd.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/gelu.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/gelu_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/gelu_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/gelu_grad_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/generator.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/glu.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/glu_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/gmm.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/gmm_backward.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/gmm_backward_fusion.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/gmm_v2.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/gmm_v2_backward.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/gmm_v2_backward_fusion.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/greater.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/greater_equal.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/greater_equal_scalar.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/grid_sampler_2d.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/grid_sampler_2d_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/grid_sampler_3d.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/grid_sampler_3d_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/group_norm.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/group_norm_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/grouped_matmul.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/grouped_matmul_v2.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/grouped_matmul_v4.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/hardtanh.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/hardtanh_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/histc_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/hshrink.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/hshrink_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/hsigmoid.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/hsigmoid_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/hswish.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/hswish_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/identity.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/im2col_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/incre_flash_attention.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/index.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/index_add_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/index_fill_scalar.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/index_fill_tensor.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/index_select.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inner_comm_all_gather.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inner_comm_all_reduce.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inner_comm_all_to_all_v.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inner_comm_irecv.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inner_comm_isend.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inner_comm_reduce_scatter.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inner_index.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inner_inplace_index_put.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inner_moe_token_unpermute.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inner_non_zero.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_add_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_addmm.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_adds_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_clamp_scalar.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_clamp_tensor.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_copy.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_div.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_divmod.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_divmods.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_divs.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_elu.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_erfinv.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_exp.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_exponential.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_fill_diagonal.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_fill_scalar.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_fill_tensor.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_floor.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_floor_divide.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_floor_divides.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_grouped_matmul_add.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_hardtanh.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_index_add.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_index_put.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_log.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_masked_fill_scalar.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_masked_fill_tensor.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_matmul_add.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_mul.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_muls.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_normal.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_put.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_random.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_relu.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_scatter_add.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_scatter_src.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_scatter_src_reduce.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_scatter_value.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_scatter_value_reduce.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_silu.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_stop_gradient.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_sub_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_sub_scalar.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_tanh.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_threshold.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_uniform.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_zero.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/isclose.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/isfinite.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/isinf.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/isneginf.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/kl_div.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/kl_div_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/kthvalue.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/kv_cache_scatter_update.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/l1_loss_backward_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/l1_loss_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/layer_norm_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/layer_norm_grad_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/leaky_relu_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/leaky_relu_grad_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/lerp.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/lerp_scalar.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/less.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/less_equal.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/lin_space_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/linalg_qr.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/linalg_vector_norm.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/log.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/log10.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/log1p.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/log2.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/log_softmax.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/log_softmax_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/log_softmax_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/logaddexp.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/logaddexp2.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/logical_and.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/logical_not.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/logical_or.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/logical_xor.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/logsigmoid.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/logsigmoid_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/logsumexp.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/masked_fill.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/masked_select.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/masked_select_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/matmul.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/matmul_allreduce_add_rmsnorm.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/matmul_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/matmul_reduce_scatter.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/matrix_inverse_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/max.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/max_dim.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/max_pool_grad_with_indices.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/max_pool_grad_with_mask.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/max_pool_with_indices.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/max_pool_with_mask.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/max_unpool2d_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/maximum.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/mean_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/median_dim.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/median_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/meshgrid.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/min.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/min_dim.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/minimum.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/mish_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/mish_grad_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/mm_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/moe_compute_expert_tokens.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/moe_distribute_combine.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/moe_distribute_dispatch.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/moe_finalize_routing.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/moe_gating_top_k_softmax.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/moe_init_routing.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/moe_init_routing_quant_v2.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/moe_init_routing_v2.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/moe_token_permute.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/moe_token_permute_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/moe_token_unpermute.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/moe_token_unpermute_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/mse_loss_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/mse_loss_grad_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/mul.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/muls.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/multi_scale_deformable_attn.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/multi_scale_deformable_attn_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/multinomial_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/mv.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/nan_to_num.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/nansum.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/narrow.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/narrow_view.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/ne_scalar.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/neg.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/new_empty.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/new_full.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/new_ones.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/new_zeros.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/nllloss.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/nllloss_2d.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/nllloss_2d_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/nllloss_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/non_zero.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/non_zero_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/norm.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/normal_float_float.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/normal_float_tensor.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/normal_tensor_float.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/normal_tensor_tensor.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/not_equal.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/one_hot_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/ones.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/ones_like_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/outer.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/paged_attention.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/pixel_shuffle.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/polar.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/pow.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/pow_scalar_tensor.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/pow_tensor_scalar.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/prelu.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/prelu_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/prod_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/prompt_flash_attention.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/quant_batch_matmul.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/quant_matmul.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/quant_v2.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/rand_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/rand_like_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/randint.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/randint_like.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/randn.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/randn_like.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/randperm_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/reciprocal.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/reduce_all.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/reduce_any.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/reduce_max.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/reduce_min.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/reflection_pad_1d.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/reflection_pad_1d_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/reflection_pad_2d.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/reflection_pad_2d_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/reflection_pad_3d.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/reflection_pad_3d_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/relu.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/relu_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/remainder_scalar_tensor.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/remainder_tensor_scalar.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/remainder_tensor_tensor.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/repeat.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/repeat_interleave_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/repeat_interleave_int.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/repeat_interleave_tensor.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/replication_pad_1d.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/replication_pad_1d_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/replication_pad_2d.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/replication_pad_2d_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/replication_pad_3d.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/replication_pad_3d_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/reshape.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/reshape_and_cache.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/reverse_v2.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/rms_norm.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/rms_norm_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/roll.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/rotary_position_embedding.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/rotary_position_embedding_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/round.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/rsqrt.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/scatter.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/scatter_add_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/scatter_value.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/searchsorted.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/select.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/select_ext_view.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/select_v2.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/selu_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/selu_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/sigmoid.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/sigmoid_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/sign.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/silent_check_v2.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/silent_check_v3.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/silu.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/silu_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/sin.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/sinc.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/sinh.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/slice.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/slice_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/slice_ext_view.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/smooth_l1_loss.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/smooth_l1_loss_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/soft_margin_loss.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/soft_margin_loss_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/softmax.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/softmax_backward.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/softplus_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/softplus_grad_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/softshrink.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/softshrink_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/sort_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/speed_fusion_attention.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/speed_fusion_attention_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/split.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/split_tensor.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/split_tensor_view.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/split_with_size.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/split_with_size_view.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/sqrt.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/square.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/squeeze.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/stack_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/std.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/std_mean.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/sub.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/sub_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/sub_scalar.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/sum_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/swiglu.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/swiglu_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/t_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/take.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/tan.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/tanh.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/tanh_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/tensor_scatter_elements.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/threshold.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/threshold_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/tile.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/topk_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/trace_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/transpose.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/transpose_ext_view.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/transpose_view.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/triangular_solve.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/tril_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/triu.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/trunc.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/type_as.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/uniform_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/unique2.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/unique_consecutive.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/unique_dim.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/unstack_ext_view.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/upsample_bicubic2d.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/upsample_bicubic2d_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/upsample_bilinear2d.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/upsample_bilinear2d_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/upsample_linear1d.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/upsample_linear1d_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/upsample_nearest1d.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/upsample_nearest1d_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/upsample_nearest2d.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/upsample_nearest2d_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/upsample_nearest3d.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/upsample_nearest3d_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/upsample_trilinear3d.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/upsample_trilinear3d_grad.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/var.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/var_mean.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/view.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/view_as.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/weight_quant_batch_matmul.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/xlogy.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/xlogy_scalar_other.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/xlogy_scalar_self.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/zeros.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/zeros_like_ext.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/comm_handle.h +6 -2
- mindspore/include/mindspore/ccsrc/pyboost/customize/any.h +39 -0
- mindspore/include/mindspore/ccsrc/pyboost/customize/divmod.h +3 -3
- mindspore/include/mindspore/ccsrc/pyboost/customize/einsum_ext.h +38 -0
- mindspore/include/mindspore/ccsrc/pyboost/customize/identity.h +2 -2
- mindspore/include/mindspore/ccsrc/pyboost/customize/meshgrid.h +7 -4
- mindspore/include/mindspore/ccsrc/pyboost/customize/op_common.h +12 -15
- mindspore/include/mindspore/ccsrc/pyboost/customize/pixel_shuffle.h +2 -3
- mindspore/include/mindspore/ccsrc/pyboost/customize/reshape.h +4 -3
- mindspore/include/mindspore/ccsrc/pyboost/customize/searchsorted.h +4 -5
- mindspore/include/mindspore/ccsrc/pyboost/functions/auto_generate/auto_grad_op_reg.h +1553 -1388
- mindspore/include/mindspore/ccsrc/pyboost/functions/auto_generate/functions.h +563 -481
- mindspore/include/mindspore/ccsrc/pyboost/grad_functions/pyboost_grad_functions.h +3 -0
- mindspore/include/mindspore/ccsrc/pyboost/grad_functions/value_converter.h +30 -4
- mindspore/include/mindspore/ccsrc/pyboost/op_register.h +52 -0
- mindspore/include/mindspore/ccsrc/pyboost/op_runner.h +45 -19
- mindspore/include/mindspore/ccsrc/pyboost/pyboost_utils.h +106 -62
- mindspore/include/mindspore/ccsrc/pynative/base.h +22 -24
- mindspore/include/mindspore/ccsrc/pynative/forward/do_pyboost_cast.h +70 -67
- mindspore/include/mindspore/ccsrc/pynative/forward/forward.h +14 -12
- mindspore/include/mindspore/ccsrc/pynative/forward/forward_task.h +34 -2
- mindspore/include/mindspore/ccsrc/pynative/grad/custom_function.h +14 -7
- mindspore/include/mindspore/ccsrc/pynative/grad/function/auto_generate/pyboost_native_grad_functions.h +487 -455
- mindspore/include/mindspore/ccsrc/pynative/grad/function/func_builder.h +2 -2
- mindspore/include/mindspore/ccsrc/pynative/grad/function/func_grad.h +280 -96
- mindspore/include/mindspore/ccsrc/pynative/grad/function/func_pass.h +0 -1
- mindspore/include/mindspore/ccsrc/pynative/grad/function.h +28 -23
- mindspore/include/mindspore/ccsrc/pynative/grad/function_py.h +19 -11
- mindspore/include/mindspore/ccsrc/pynative/grad/grad.h +30 -97
- mindspore/include/mindspore/ccsrc/pynative/grad/grad_utils.h +39 -23
- mindspore/include/mindspore/ccsrc/pynative/grad/hook_py.h +20 -21
- mindspore/include/mindspore/ccsrc/pynative/grad/jit/jit_grad.h +2 -26
- mindspore/include/mindspore/ccsrc/pynative/grad/top_cell.h +8 -150
- mindspore/include/mindspore/ccsrc/pynative/op_function/auto_generate/pyboost_api.h +552 -0
- mindspore/include/mindspore/ccsrc/pynative/op_function/auto_generate/pyboost_core.h +552 -0
- mindspore/include/mindspore/ccsrc/pynative/op_function/auto_generate/tensor_func_utils.h +486 -483
- mindspore/include/mindspore/ccsrc/pynative/op_function/comm_handle_py.h +2 -0
- mindspore/include/mindspore/ccsrc/pynative/op_function/converter.h +11 -0
- mindspore/include/mindspore/ccsrc/pynative/op_function/customize/direct_ops.h +2 -12
- mindspore/include/mindspore/ccsrc/pynative/predict_out_type_map.h +3 -0
- mindspore/include/mindspore/ccsrc/pynative/pynative_execute.h +4 -2
- mindspore/include/mindspore/ccsrc/pynative/pynative_utils.h +39 -42
- mindspore/include/mindspore/ccsrc/runtime/collective/collective_communication_lib.h +20 -1
- mindspore/include/mindspore/ccsrc/runtime/collective/communication_group.h +5 -0
- mindspore/include/mindspore/ccsrc/runtime/collective/dummy_collective_communication_lib.h +2 -1
- mindspore/include/mindspore/ccsrc/runtime/device/device_address_utils.h +55 -50
- mindspore/include/mindspore/ccsrc/runtime/device/kernel_runtime.h +2 -2
- mindspore/include/mindspore/ccsrc/runtime/device/move_to.h +4 -1
- mindspore/include/mindspore/ccsrc/runtime/device/res_manager/auto_mem_offload.h +0 -1
- mindspore/include/mindspore/ccsrc/runtime/device/res_manager/hal_res_base.h +15 -11
- mindspore/include/mindspore/ccsrc/runtime/device/res_manager/hal_res_manager.h +7 -3
- mindspore/include/mindspore/ccsrc/runtime/device/res_manager/loadable_device_address.h +1 -1
- mindspore/include/mindspore/ccsrc/runtime/device/res_manager/memory_manager.h +1 -1
- mindspore/include/mindspore/ccsrc/runtime/device/res_manager/swap_manager.h +2 -2
- mindspore/include/mindspore/ccsrc/runtime/device/res_manager/tensor_array.h +1 -1
- mindspore/include/mindspore/ccsrc/runtime/device/res_manager/utils/utils.h +0 -1
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/abstract_actor.h +41 -32
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/actor_common.h +21 -20
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/actor_dump.h +4 -3
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/actor_set.h +0 -4
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/any_type_kernel_actor.h +7 -61
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/control_flow/{condition_gather_actor.h → condition_gather_runner.h} +17 -21
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/control_flow/{condition_switch_actor.h → condition_switch_runner.h} +19 -21
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/control_flow/control_actor.h +37 -41
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/control_flow/entrance_actor.h +11 -11
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/control_flow/exit_actor.h +8 -9
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/control_flow/gather_actor.h +6 -7
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/control_flow/stack_actor.h +11 -11
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/control_flow/switch_actor.h +2 -2
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/copy_actor.h +12 -14
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/data_prepare_actor.h +21 -20
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/data_source_actor.h +10 -58
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/debug_actor.h +11 -10
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/debug_aware_actor.h +2 -2
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/fusion/fusion_actor.h +2 -2
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/kernel_actor.h +90 -83
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/kernel_async_infer_actor.h +3 -1
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/kernel_async_launch_actor.h +11 -1
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/kernel_async_resize_actor.h +3 -1
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/kernel_infer_actor.h +4 -4
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/kernel_resize_actor.h +4 -4
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/kernel_runner.h +400 -0
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/loop_count_actor.h +11 -11
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/memory/memory_alloc_actor.h +3 -3
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/memory/memory_free_actor.h +3 -3
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/memory/memory_swap_actor.h +4 -4
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/memory_aware_actor.h +7 -7
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/memory_manager_actor.h +18 -17
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/output_actor.h +13 -7
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/profiler_actor.h +2 -2
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/recorder_actor.h +2 -2
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/rpc/mux_send_actor.h +1 -1
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/rpc/recv_actor.h +6 -10
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/rpc/rpc_actor.h +2 -14
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/rpc/send_actor.h +4 -4
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/super_kernel_actor.h +60 -57
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/any_type_graph_scheduler.h +0 -33
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/control_node_scheduler.h +0 -2
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/device_tensor_copy_store.h +14 -14
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/device_tensor_store.h +28 -27
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/execution_order_check/comm_execution_order_check.h +17 -7
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/execution_order_check/kernel_cache.h +24 -4
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/graph_compiler.h +3 -69
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/graph_parameter_store.h +85 -141
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/graph_scheduler.h +3 -19
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/parameter_store.h +4 -0
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/pre_launch_comm.h +10 -2
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/rpc_node_scheduler.h +4 -13
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/scheduler_helper.h +12 -1
- mindspore/include/mindspore/ccsrc/runtime/hardware/deprecated_interface.h +0 -2
- mindspore/include/mindspore/ccsrc/runtime/hardware/device_context.h +28 -119
- mindspore/include/mindspore/ccsrc/runtime/hardware/device_context_manager.h +1 -1
- mindspore/include/mindspore/ccsrc/runtime/pipeline/async_rqueue.h +2 -2
- mindspore/include/mindspore/ccsrc/runtime/pipeline/ring_queue.h +1 -1
- mindspore/include/mindspore/ccsrc/runtime/pipeline/task/task.h +1 -1
- mindspore/include/mindspore/ccsrc/runtime/pynative/ir_converter.h +8 -7
- mindspore/include/mindspore/ccsrc/runtime/pynative/op_runner.h +6 -6
- mindspore/include/mindspore/ccsrc/runtime/pynative/op_runtime_info.h +4 -4
- mindspore/include/mindspore/ccsrc/utils/dlopen_macro.h +2 -2
- mindspore/include/mindspore/core/include/abstract/abstract_function.h +54 -13
- mindspore/include/mindspore/core/include/abstract/abstract_value.h +66 -3
- mindspore/include/mindspore/core/include/abstract/ops/primitive_infer_map.h +1 -1
- mindspore/include/mindspore/core/include/abstract/param_validator.h +3 -2
- mindspore/include/mindspore/core/include/base/bfloat16.h +1 -1
- mindspore/include/mindspore/core/include/base/{fp8_e4m3.h → float8_e4m3fn.h} +59 -58
- mindspore/include/mindspore/core/include/base/{fp8_e5m2.h → float8_e5m2.h} +7 -7
- mindspore/include/mindspore/core/include/base/hifloat8.h +36 -37
- mindspore/include/mindspore/core/include/ir/anf.h +7 -4
- mindspore/include/mindspore/core/include/ir/device_sync.h +17 -1
- mindspore/include/mindspore/core/include/ir/dtype/number.h +47 -6
- mindspore/include/mindspore/core/include/ir/dtype/op_dtype.h +48 -0
- mindspore/include/mindspore/core/include/ir/dtype.h +4 -0
- mindspore/include/mindspore/core/include/ir/func_graph.h +1 -0
- mindspore/include/mindspore/core/include/ir/func_graph_cloner.h +2 -0
- mindspore/include/mindspore/core/include/ir/meta_grad_data.h +4 -13
- mindspore/include/mindspore/core/include/ir/primitive.h +34 -2
- mindspore/include/mindspore/core/include/ir/scalar.h +2 -2
- mindspore/include/mindspore/core/include/ir/scope.h +16 -3
- mindspore/include/mindspore/core/include/ir/tensor.h +922 -41
- mindspore/include/mindspore/core/include/ir/tensor_py_wrapperbase.h +11 -11
- mindspore/include/mindspore/core/include/ir/tensor_storage_info.h +1 -0
- mindspore/include/mindspore/core/include/load_mindir/infer_mindir.h +3 -2
- mindspore/include/mindspore/core/include/mindapi/base/macros.h +3 -3
- mindspore/include/mindspore/core/include/mindapi/base/type_id.h +3 -0
- mindspore/include/mindspore/core/include/mindapi/base/types.h +7 -0
- mindspore/include/mindspore/core/include/ops/op_def.h +2 -31
- mindspore/include/mindspore/core/include/utils/callback_handler.h +1 -1
- mindspore/include/mindspore/core/include/utils/compact_set.h +4 -0
- mindspore/include/mindspore/core/include/utils/core_op_utils.h +1 -1
- mindspore/include/mindspore/core/include/utils/device_manager_conf.h +4 -0
- mindspore/include/mindspore/core/include/utils/flags.h +0 -2
- mindspore/include/mindspore/core/include/utils/info.h +7 -0
- mindspore/include/mindspore/core/include/utils/llm_manager.h +2 -0
- mindspore/include/mindspore/core/include/utils/log_adapter.h +9 -1
- mindspore/include/mindspore/core/include/utils/ms_context.h +8 -10
- mindspore/include/mindspore/core/include/utils/ms_exception.h +38 -4
- mindspore/include/mindspore/core/include/utils/ms_utils.h +4 -8
- mindspore/include/mindspore/core/include/utils/ms_utils_secure.h +1 -1
- mindspore/include/mindspore/core/include/utils/phase.h +3 -0
- mindspore/include/mindspore/core/include/utils/system/base.h +1 -1
- mindspore/include/mindspore/core/mindrt/include/actor/op_actor.h +68 -0
- mindspore/include/mindspore/core/mindrt/include/async/async.h +2 -2
- mindspore/include/mindspore/core/mindrt/include/thread/hqueue.h +6 -6
- mindspore/include/mindspore/core/mindrt/include/thread/threadpool.h +4 -2
- mindspore/include/mindspore/ops/grad/grad_utils.h +25 -3
- mindspore/include/mindspore/ops/infer/all_gather_v.h +39 -0
- mindspore/include/mindspore/ops/infer/all_to_all.h +38 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/acosh.h +2 -9
- mindspore/include/mindspore/ops/infer/ops_func_impl/asinh.h +2 -9
- mindspore/include/mindspore/ops/infer/ops_func_impl/atanh.h +4 -9
- mindspore/include/mindspore/ops/infer/ops_func_impl/batch_norm_ext.h +6 -11
- mindspore/include/mindspore/ops/infer/ops_func_impl/batch_norm_grad_ext.h +5 -4
- mindspore/include/mindspore/ops/infer/ops_func_impl/bitwise_and_scalar.h +4 -5
- mindspore/include/mindspore/ops/infer/ops_func_impl/bitwise_and_tensor.h +3 -5
- mindspore/include/mindspore/ops/infer/ops_func_impl/bitwise_or_scalar.h +3 -10
- mindspore/include/mindspore/ops/infer/ops_func_impl/bitwise_or_tensor.h +2 -10
- mindspore/include/mindspore/ops/infer/ops_func_impl/bitwise_xor_scalar.h +3 -10
- mindspore/include/mindspore/ops/infer/ops_func_impl/bitwise_xor_tensor.h +2 -10
- mindspore/include/mindspore/ops/infer/ops_func_impl/broadcast_to.h +0 -1
- mindspore/include/mindspore/ops/{view/select_ext_strides_calc.h → infer/ops_func_impl/broadcast_to_view.h} +9 -7
- mindspore/include/mindspore/ops/infer/ops_func_impl/chunk.h +0 -2
- mindspore/include/mindspore/ops/infer/ops_func_impl/{gmm_backward.h → chunk_view.h} +8 -4
- mindspore/include/mindspore/ops/infer/ops_func_impl/diagonal_view.h +32 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/dump_gradient.h +33 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/dynamic_ntk.h +32 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/eltwise_op.h +4 -10
- mindspore/include/mindspore/ops/infer/ops_func_impl/empty.h +7 -3
- mindspore/include/mindspore/ops/infer/ops_func_impl/empty_like.h +7 -3
- mindspore/include/mindspore/ops/infer/ops_func_impl/exp.h +3 -4
- mindspore/include/mindspore/ops/infer/ops_func_impl/expand_dims.h +1 -2
- mindspore/include/mindspore/ops/infer/ops_func_impl/{gmm_v2_backward.h → expand_dims_view.h} +7 -4
- mindspore/include/mindspore/ops/infer/ops_func_impl/fused_add_topk_div.h +56 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/grouped_matmul.h +13 -4
- mindspore/include/mindspore/ops/infer/ops_func_impl/grouped_matmul_base.h +8 -9
- mindspore/include/mindspore/ops/infer/ops_func_impl/grouped_matmul_v2.h +7 -1
- mindspore/include/mindspore/ops/infer/ops_func_impl/grouped_matmul_v4.h +3 -1
- mindspore/include/mindspore/ops/infer/ops_func_impl/{moe_token_unpermute.h → inner_moe_token_unpermute.h} +4 -4
- mindspore/include/mindspore/ops/infer/ops_func_impl/inplace_matmul_add.h +34 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/inplace_silu.h +35 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/masked_fill.h +4 -3
- mindspore/include/mindspore/ops/infer/ops_func_impl/max_pool_grad_with_indices.h +2 -8
- mindspore/include/mindspore/ops/infer/ops_func_impl/max_pool_grad_with_mask.h +4 -2
- mindspore/include/mindspore/ops/infer/ops_func_impl/max_pool_with_indices.h +6 -4
- mindspore/include/mindspore/ops/infer/ops_func_impl/max_pool_with_mask.h +6 -4
- mindspore/include/mindspore/ops/infer/ops_func_impl/{decoder_k_v_cache.h → moe_distribute_combine.h} +6 -12
- mindspore/include/mindspore/ops/infer/ops_func_impl/moe_distribute_dispatch.h +37 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/moe_init_routing_quant_v2.h +39 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/narrow.h +0 -1
- mindspore/include/mindspore/ops/infer/ops_func_impl/narrow_view.h +29 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/neg.h +1 -6
- mindspore/include/mindspore/ops/infer/ops_func_impl/new_empty.h +7 -3
- mindspore/include/mindspore/ops/infer/ops_func_impl/new_full.h +37 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/ones_like.h +2 -6
- mindspore/include/mindspore/ops/infer/ops_func_impl/ones_like_ext.h +1 -2
- mindspore/include/mindspore/ops/infer/ops_func_impl/quant_matmul.h +32 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/reciprocal.h +4 -9
- mindspore/include/mindspore/ops/infer/ops_func_impl/reduce_any.h +4 -5
- mindspore/include/mindspore/ops/infer/ops_func_impl/reduce_arithmetic.h +2 -1
- mindspore/include/mindspore/ops/infer/ops_func_impl/remainder_tensor_scalar.h +4 -4
- mindspore/include/mindspore/ops/infer/ops_func_impl/{select_ext.h → select_ext_view.h} +4 -5
- mindspore/include/mindspore/ops/infer/ops_func_impl/sigmoid.h +4 -5
- mindspore/include/mindspore/ops/infer/ops_func_impl/slice_ext.h +0 -1
- mindspore/include/mindspore/ops/infer/ops_func_impl/slice_ext_view.h +29 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/split_tensor.h +0 -1
- mindspore/include/mindspore/ops/infer/ops_func_impl/split_tensor_view.h +32 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/split_with_size.h +0 -2
- mindspore/include/mindspore/ops/infer/ops_func_impl/split_with_size_view.h +32 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/square.h +2 -10
- mindspore/include/mindspore/ops/infer/ops_func_impl/swiglu_dynamic_quant.h +32 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/{prompt_k_v_cache.h → topprouter.h} +8 -12
- mindspore/include/mindspore/ops/infer/ops_func_impl/transpose.h +0 -2
- mindspore/include/mindspore/ops/infer/ops_func_impl/{transpose_ext.h → transpose_ext_view.h} +4 -5
- mindspore/include/mindspore/ops/infer/ops_func_impl/transpose_view.h +29 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/{unstack_ext.h → unstack_ext_view.h} +6 -7
- mindspore/include/mindspore/ops/infer/ops_func_impl/zeros_like_ext.h +1 -2
- mindspore/include/mindspore/ops/infer/reduce_scatter.h +3 -1
- mindspore/include/mindspore/ops/infer/reduce_scatter_v.h +38 -0
- mindspore/include/mindspore/ops/kernel/ascend/acl/acl_kernel_mod.h +3 -0
- mindspore/include/mindspore/ops/kernel/ascend/acl_ir/acl_adapter_info.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/acl_ir/custom/custom_aclnn_utils.h +95 -0
- mindspore/include/mindspore/ops/kernel/ascend/acl_ir/custom/custom_op_api_cache.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/acl_ir/custom/custom_op_api_exec.h +84 -0
- mindspore/include/mindspore/ops/kernel/ascend/acl_ir/op_api_cache.h +13 -8
- mindspore/include/mindspore/ops/kernel/ascend/acl_ir/op_api_convert.h +20 -74
- mindspore/include/mindspore/ops/kernel/ascend/acl_ir/op_api_exec.h +23 -12
- mindspore/include/mindspore/ops/kernel/ascend/acl_ir/op_api_util.h +1 -0
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/common/kernel_base.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/common/kernel_log.h +11 -11
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/cpu_kernel/format_transfer/formats_definitions.h +5 -1
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/cpu_kernel/format_transfer/register_format_transfer.h +5 -1
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/cpu_kernel/inc/ms_cpu_kernel.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/cpu_kernel/ms_kernel/concat.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/cpu_kernel/ms_kernel/dct.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/cpu_kernel/ms_kernel/dctn.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/cpu_kernel/ms_kernel/fft_ortho.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/cpu_kernel/ms_kernel/fft_shapecopy.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/cpu_kernel/ms_kernel/fftbase.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/cpu_kernel/ms_kernel/fftfreq.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/cpu_kernel/ms_kernel/fftnbase.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/cpu_kernel/ms_kernel/irfft_double.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/cpu_kernel/ms_kernel/nms_with_mask.h +0 -1
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ragged_tensor_to_sparse.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/cpu_kernel/ms_kernel/random/philox_random_dist.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/cpu_kernel/ms_kernel/random/random_distributions.h +27 -25
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/cpu_kernel/ms_kernel/topprouter.h +64 -0
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/cpu_kernel/utils/fused_sparse_utils.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/cpu_kernel/utils/kernel_util.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/cpu_kernel/utils/philox_random.h +75 -138
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/cpu_kernel/utils/range_sampler.h +7 -3
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/cpu_kernel/utils/sparse_group.h +18 -15
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/cpu_kernel/utils/sparse_tensor.h +18 -15
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/adaptive_avg_pool_3d_grad_op.h +0 -11
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/adaptive_avg_pool_3d_op.h +0 -11
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/adaptive_max_pool3_d_grad_op.h +0 -14
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/adaptive_max_pool3d_op.h +0 -18
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/adaptive_max_pool_2d_grad_op.h +0 -14
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/adjust_contrastv2_op.h +0 -21
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/arg_max_op.h +0 -22
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/bartlett_window_op.h +0 -19
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/cauchy_op.h +0 -11
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/cholesky_solve_op.h +0 -23
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/coalesce_op.h +0 -24
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/csr_sparse_matrix_to_dense_op.h +0 -15
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/dense_to_csr_sparse_matrix_op.h +0 -16
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/eig_op.h +0 -17
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/exp.h +0 -18
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/fractional_max_pool_grad_with_fixed_ksize_op.h +0 -22
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/fractional_max_pool_with_fixed_ksize_op.h +0 -19
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/geqrf_op.h +0 -14
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/glu_grad_op.h +0 -17
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/glu_op.h +0 -20
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/hamming_window_op.h +0 -20
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/index_fill.h +0 -18
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/instance_norm_v2_grad.h +0 -28
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/layer_norm_grad_grad_op.h +0 -17
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/log_normal_reverse.h +0 -15
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/logspace.h +0 -23
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/lstsq_op.h +0 -15
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/matrix_logarithm.h +0 -13
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/matrix_power_op.h +0 -16
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/max_pool_3d_grad_with_argmax_op.h +0 -26
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/maximum_grad_grad.h +0 -19
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/median_grad_op.h +0 -19
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/median_op.h +0 -17
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/minimum_grad_grad.h +0 -19
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/multi_margin_loss_grad_op.h +0 -24
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/multi_margin_loss_op.h +0 -19
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/mvlgamma_grad_op.h +0 -17
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/mvlgamma_op.h +0 -15
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/pdist_grad_op.h +0 -21
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/segment_mean_op.h +0 -18
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/segment_min_op.h +0 -19
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/sparse_addmm.h +0 -16
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/sparse_apply_adagrad_da.h +0 -38
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/sparse_apply_centered_rms_prop.h +0 -47
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/sparse_apply_momentum.h +0 -36
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/sparse_apply_proximal_gradient_descent.h +0 -29
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/sparse_matrix_transpose_op.h +0 -29
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/sparse_segment_mean_with_num_segments_op.h +0 -19
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/sparse_segment_sqrt_n_grad_op.h +0 -21
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/sparse_segment_sqrt_n_op.h +0 -18
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/sparse_segment_sqrt_n_with_num_segments_op.h +0 -20
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/sparse_tensor_to_csr_sparse_matrix_op.h +0 -18
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/sspaddmm_op.h +0 -22
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/inc/triplet_margin_loss_op.h +0 -22
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/utils/axis_util.h +5 -1
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/utils/reduce_infer_util.h +1 -2
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/customize/op_proto/utils/transfer_shape_according_to_format.h +5 -1
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/drop_out_gen_mask_kernels.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/gather_grad_kernels.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_util.h +2 -0
- mindspore/include/mindspore/ops/kernel/ascend/availability/silent_check/ascend_silent_check.h +13 -14
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/addbmm_aclnn_kernel.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/addmm_aclnn_kernel.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/addmv_aclnn_kernel.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/baddbmm_aclnn_kernel.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/batch_norm_ext_aclnn_kernel.h +1 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/batch_norm_grad_ext_aclnn_kernel.h +2 -1
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/bincount_ext_aclnn_kernel.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/chunk_aclnn_kernel.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/{view/contiguous.h → contiguous_aclnn_kernel.h} +4 -5
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/conv1d_ext_aclnn_kernel.h +2 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/conv1d_padding_aclnn_kernel.h +2 -1
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/conv2d_ext_aclnn_kernel.h +1 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/conv2d_padding_aclnn_kernel.h +3 -1
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/conv3d_ext_aclnn_kernel.h +4 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/convolution_str_aclnn_kernel.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/custom_aclnn_kernel.h +5 -1
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/custom_aclnn_utils.h +2 -1
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/custom_v2_aclnn_kernel.h +83 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/dense_aclnn_kernel.h +13 -6
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/dropout_ext_aclnn_kernel.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/empty_aclnn_kernel.h +39 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/empty_like_aclnn_kernel.h +39 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/flash_attention_score_aclnn_kernel.h +1 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/flash_attention_score_grad_aclnn_kernel.h +1 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/gather_d_grad_v2_aclnn_kernel.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/grid_sampler_2d_grad_aclnn_kernel.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/grid_sampler_3d_grad_aclnn_kernel.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/grouped_matmul_v2_aclnn_kernel.h +49 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/grouped_matmul_v4_aclnn_kernel.h +6 -3
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/index_add_ext_aclnn_kernel.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/index_fill_scalar_aclnn_kernel.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/index_fill_tensor_aclnn_kernel.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/inner_inplace_index_put_aclnn_kernel.h +1 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/{moe_token_unpermute_aclnn_kernel.h → inner_moe_token_unpermute_aclnn_kernel.h} +6 -6
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/inplace_clamp_scalar_aclnn_kernel.h +2 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/inplace_divs_aclnn_kernel.h +41 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/inplace_index_add_aclnn_kernel.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/inplace_normal_aclnn_kernel.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/inplace_scatter_add_aclnn_kernel.h +45 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/inplace_silu_aclnn_kernel.h +42 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/inplace_sub_scalar_aclnn_kernel.h +41 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/inplace_uniform_aclnn_kernel.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/isinf_aclnn_kernel.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/linalg_vector_norm_aclnn_kernel.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/matmul_all_reduce_aclnn_kernel.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/matmul_reduce_scatter_aclnn_kernel.h +4 -4
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/moe_distribute_combine_aclnn_kernel.h +56 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/moe_distribute_dispatch_aclnn_kernel.h +55 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/moe_init_routing_quant_v2_aclnn_kernel.h +50 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/mse_loss_ext_aclnn_kernel.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/mse_loss_grad_ext_aclnn_kernel.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/multinomial_ext_aclnn_kernel.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/narrow_aclnn_kernel.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/new_empty_aclnn_kernel.h +39 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/new_full_aclnn_kernel.h +41 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/norm_aclnn_kernel.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/prod_ext_aclnn_kernel.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/quant_batch_matmul_all_reduce_aclnn_kernel.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/rand_ext_aclnn_kernel.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/randint_aclnn_kernel.h +4 -4
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/randn_aclnn_kernel.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/randperm_ext_aclnn_kernel.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/repeat_interleave_grad_aclnn_kernel.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/split_with_size_aclnn_kernel.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/unique2_aclnn_kernel.h +3 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/unique_consecutive_aclnn_kernel.h +2 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/unique_dim_aclnn_kernel.h +3 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/view/broadcast_to_view.h +42 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/view/chunk_view.h +42 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/view/concat_view.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/view/diagonal_view.h +42 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/view/expand_dims_view.h +42 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/view/flatten_view.h +42 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/view/narrow_view.h +43 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/view/reshape_view.h +2 -1
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/view/select_ext_view.h +42 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/view/slice_ext_view.h +42 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/view/split_tensor_view.h +42 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/view/split_view.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/view/split_with_size_view.h +42 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/view/squeeze_view.h +42 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/view/strided_slice_view.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/view/transpose_view.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/view/unstack_ext_view.h +42 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/view/view.h +42 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/view/view_utils.h +0 -1
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn_auto_gen/apply_rotary_pos_emb_aclnn_kernel.h +41 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn_auto_gen/{select_ext_aclnn_kernel.h → paged_attention_aclnn_kernel.h} +6 -6
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn_kernel_mod.h +32 -11
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn_kernel_utils.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/aclnn_utils.h +11 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/atb_runner.h +124 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/atb_runner_base.h +48 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/atb_utils.h +63 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/abs.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/acos_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/acosh_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/adamw.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/adaptive_avg_pool1d.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/adaptive_avg_pool2d_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/adaptive_avg_pool2d_grad_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/adaptive_avg_pool3d_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/adaptive_avg_pool3d_grad_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/adaptive_max_pool1d.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/adaptive_max_pool2d.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/add.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/add_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/add_layer_norm_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/add_layernorm_v2.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/add_rms_norm.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/add_rmsnorm_quant_v2.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/add_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/addbmm.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/addcdiv_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/addcmul_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/addmm.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/addmv.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/all_finite.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/all_gather_matmul.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/any.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/any_ext.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/apply_rotary_pos_emb.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/arange.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/argmax_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/argmax_with_value.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/argmin_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/argmin_with_value.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/argsort.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/as_strided.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/asin_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/asinh_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/atan2_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/atan_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/atanh.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/avg_pool1d.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/avg_pool2d.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/avg_pool2d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/avg_pool3d_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/avg_pool3d_grad_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/baddbmm.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/batch_mat_mul.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/batch_norm_elemt.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/batch_norm_elemt_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/batch_norm_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/batch_norm_gather_stats_with_counts.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/batch_norm_grad_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/batch_norm_reduce_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/batch_norm_stats.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/bernoulli_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/binary_cross_entropy.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/binary_cross_entropy_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/binary_cross_entropy_with_logits.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/binary_cross_entropy_with_logits_backward.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/bincount_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/bitwise_and_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/bitwise_and_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/bitwise_not.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/bitwise_or_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/bitwise_or_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/bitwise_xor_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/bitwise_xor_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/bmm_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/broadcast_to.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/broadcast_to_view.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/cast.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/ceil.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/chunk.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/{select_ext.h → chunk_view.h} +9 -9
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/clamp_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/clamp_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/clone.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/col2im_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/col2im_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/concat.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/constant_pad_nd.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/contiguous.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/conv1d_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/conv1d_padding.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/conv2d_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/conv2d_padding.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/conv3d_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/conv3d_padding.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/conv_transpose2d.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/convolution.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/convolution_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/convolution_str.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/convolution_str_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/copy.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/cos.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/cosh.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/count_nonzero.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/cross.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/cummax.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/cummin_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/cumsum_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/custom_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/dense.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/diag_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/{transpose_ext.h → diagonal_view.h} +9 -9
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/div.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/divmod.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/divmods.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/divs.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/dot.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/dropout_do_mask_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/dropout_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/dropout_gen_mask_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/dropout_grad_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/dynamic_quant_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/einsum_ext.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/elu.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/elu_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/elu_grad_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/embedding.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/embedding_dense_backward.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/empty.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/empty_like.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/equal.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/equal_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/erf.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/erfc.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/erfinv.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/exp.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/exp2.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/expand_as.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/expand_dims.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/expand_dims_view.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/expm1.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/eye.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/ffn_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/fill_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/fill_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/flash_attention_score.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/flash_attention_score_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/flatten_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/floor.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/floor_div.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/floor_div_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/fmod_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/fmod_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/frac.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/full_like.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/func_max_pool2d.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/fused_infer_attention_score.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/gather_d.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/gather_d_grad_v2.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/gcd.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/gelu.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/gelu_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/gelu_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/gelu_grad_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/generator.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/glu.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/glu_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/gmm.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/gmm_backward.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/gmm_backward_fusion.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/gmm_v2.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/gmm_v2_backward.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/gmm_v2_backward_fusion.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/greater.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/greater_equal.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/greater_equal_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/grid_sampler_2d.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/grid_sampler_2d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/grid_sampler_3d.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/grid_sampler_3d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/group_norm.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/group_norm_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/grouped_matmul.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/grouped_matmul_v2.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/grouped_matmul_v4.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/hardtanh.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/hardtanh_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/histc_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/hshrink.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/hshrink_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/hsigmoid.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/hsigmoid_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/hswish.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/hswish_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/identity.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/im2col_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/incre_flash_attention.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/index.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/index_add_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/index_fill_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/index_fill_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/index_select.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inner_index.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inner_inplace_index_put.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inner_moe_token_unpermute.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inner_non_zero.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_add_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_addmm.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_adds_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_clamp_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_clamp_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_copy.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_div.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_divmod.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_divmods.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_divs.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_elu.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_erfinv.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_exp.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_exponential.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_fill_diagonal.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_fill_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_fill_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_floor.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_floor_divide.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_floor_divides.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_grouped_matmul_add.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_hardtanh.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_index_add.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_index_put.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_log.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_masked_fill_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_masked_fill_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_matmul_add.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_mul.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_muls.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_normal.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_put.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_random.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_relu.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_scatter_add.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_scatter_src.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_scatter_src_reduce.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_scatter_value.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_scatter_value_reduce.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_silu.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_stop_gradient.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_sub_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_sub_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_tanh.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_threshold.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_uniform.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_zero.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/isclose.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/isfinite.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/isinf.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/isneginf.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/kl_div.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/kl_div_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/kthvalue.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/kv_cache_scatter_update.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/l1_loss_backward_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/l1_loss_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/layer_norm_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/layer_norm_grad_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/leaky_relu_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/leaky_relu_grad_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/lerp.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/lerp_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/less.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/less_equal.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/lin_space_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/linalg_qr.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/linalg_vector_norm.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/log.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/log10.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/log1p.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/log2.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/log_softmax.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/log_softmax_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/log_softmax_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/logaddexp.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/logaddexp2.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/logical_and.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/logical_not.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/logical_or.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/logical_xor.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/logsigmoid.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/logsigmoid_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/logsumexp.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/masked_fill.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/masked_select.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/masked_select_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/matmul.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/matmul_allreduce_add_rmsnorm.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/matmul_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/matmul_reduce_scatter.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/matrix_inverse_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/max.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/max_dim.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/max_pool_grad_with_indices.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/max_pool_grad_with_mask.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/max_pool_with_indices.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/max_pool_with_mask.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/max_unpool2d_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/maximum.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/mean_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/median_dim.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/median_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/meshgrid.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/min.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/min_dim.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/minimum.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/mish_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/mish_grad_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/mm_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/moe_compute_expert_tokens.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/moe_distribute_combine.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/moe_distribute_dispatch.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/moe_finalize_routing.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/moe_gating_top_k_softmax.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/moe_init_routing.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/moe_init_routing_quant_v2.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/moe_init_routing_v2.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/moe_token_permute.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/moe_token_permute_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/moe_token_unpermute.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/moe_token_unpermute_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/mse_loss_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/mse_loss_grad_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/mul.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/muls.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/multi_scale_deformable_attn.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/multi_scale_deformable_attn_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/multinomial_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/mv.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/nan_to_num.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/nansum.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/narrow.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/{unstack_ext.h → narrow_view.h} +9 -9
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/ne_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/neg.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/new_empty.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/new_full.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/new_ones.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/new_zeros.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/nllloss.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/nllloss_2d.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/nllloss_2d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/nllloss_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/non_zero.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/non_zero_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/norm.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/normal_float_float.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/normal_float_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/normal_tensor_float.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/normal_tensor_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/not_equal.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/one_hot_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/ones.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/ones_like_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/outer.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/paged_attention.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/pixel_shuffle.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/polar.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/pow.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/pow_scalar_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/pow_tensor_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/prelu.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/prelu_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/prod_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/prompt_flash_attention.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/quant_batch_matmul.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/quant_matmul.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/quant_v2.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/rand_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/rand_like_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/randint.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/randint_like.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/randn.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/randn_like.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/randperm_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/reciprocal.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/reduce_all.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/reduce_any.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/reduce_max.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/reduce_min.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/reflection_pad_1d.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/reflection_pad_1d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/reflection_pad_2d.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/reflection_pad_2d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/reflection_pad_3d.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/reflection_pad_3d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/relu.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/relu_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/remainder_scalar_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/remainder_tensor_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/remainder_tensor_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/repeat.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/repeat_interleave_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/repeat_interleave_int.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/repeat_interleave_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/replication_pad_1d.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/replication_pad_1d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/replication_pad_2d.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/replication_pad_2d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/replication_pad_3d.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/replication_pad_3d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/reshape.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/reshape_and_cache.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/reverse_v2.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/rms_norm.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/rms_norm_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/roll.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/rotary_position_embedding.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/rotary_position_embedding_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/round.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/rsqrt.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/scatter.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/scatter_add_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/scatter_value.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/searchsorted.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/select.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/select_ext_view.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/select_v2.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/selu_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/selu_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/sigmoid.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/sigmoid_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/sign.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/silent_check_v2.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/silent_check_v3.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/silu.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/silu_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/sin.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/sinc.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/sinh.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/slice.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/slice_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/slice_ext_view.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/smooth_l1_loss.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/smooth_l1_loss_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/soft_margin_loss.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/soft_margin_loss_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/softmax.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/softmax_backward.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/softplus_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/softplus_grad_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/softshrink.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/softshrink_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/sort_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/speed_fusion_attention.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/speed_fusion_attention_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/split.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/split_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/split_tensor_view.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/split_with_size.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/split_with_size_view.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/sqrt.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/square.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/squeeze.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/stack_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/std.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/std_mean.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/sub.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/sub_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/sub_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/sum_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/swiglu.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/swiglu_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/t_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/take.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/tan.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/tanh.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/tanh_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/tensor_scatter_elements.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/threshold.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/threshold_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/tile.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/topk_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/trace_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/transpose.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/transpose_ext_view.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/transpose_view.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/triangular_solve.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/tril_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/triu.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/trunc.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/type_as.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/uniform_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/unique2.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/unique_consecutive.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/unique_dim.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/unstack_ext_view.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/upsample_bicubic2d.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/upsample_bicubic2d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/upsample_bilinear2d.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/upsample_bilinear2d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/upsample_linear1d.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/upsample_linear1d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/upsample_nearest1d.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/upsample_nearest1d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/upsample_nearest2d.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/upsample_nearest2d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/upsample_nearest3d.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/upsample_nearest3d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/upsample_trilinear3d.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/upsample_trilinear3d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/var.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/var_mean.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/view.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/view_as.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/weight_quant_batch_matmul.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/xlogy.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/xlogy_scalar_other.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/xlogy_scalar_self.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/zeros.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/zeros_like_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/adamw.h +5 -5
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/adaptive_avg_pool1d.h +2 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/adaptive_avg_pool3d_ext.h +2 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/adaptive_max_pool1d.h +3 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/adaptive_max_pool2d.h +3 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/add.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/add_layernorm_v2.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/add_rms_norm.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/add_rmsnorm_quant_v2.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/add_scalar.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/addbmm.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/addmm.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/addmv.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/all_finite.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/all_gather_matmul.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/any.h +34 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/any_ext.h +35 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/arange.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/argmax_ext.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/argmax_with_value.h +4 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/argmin_ext.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/argmin_with_value.h +4 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/argsort.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/avg_pool1d.h +4 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/avg_pool2d.h +5 -5
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/avg_pool2d_grad.h +5 -5
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/avg_pool3d_ext.h +6 -6
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/avg_pool3d_grad_ext.h +6 -6
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/baddbmm.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/batch_mat_mul.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/batch_norm_elemt.h +5 -7
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/batch_norm_ext.h +6 -5
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/batch_norm_gather_stats_with_counts.h +5 -5
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/batch_norm_grad_ext.h +6 -6
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/batch_norm_stats.h +3 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/bernoulli_ext.h +2 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/binary_cross_entropy.h +4 -5
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/binary_cross_entropy_grad.h +5 -6
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/binary_cross_entropy_with_logits.h +6 -6
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/binary_cross_entropy_with_logits_backward.h +4 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/bincount_ext.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/bmm_ext.h +2 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/clone.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/contiguous.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/conv1d_ext.h +4 -5
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/conv1d_padding.h +5 -6
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/conv2d_ext.h +4 -5
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/conv2d_padding.h +5 -6
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/conv3d_ext.h +4 -5
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/conv3d_padding.h +5 -6
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/conv_transpose2d.h +6 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/convolution.h +6 -6
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/convolution_grad.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/convolution_str.h +6 -7
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/copy.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/count_nonzero.h +2 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/cross.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/cummax.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/cummin_ext.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/cumsum_ext.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/custom_ext.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/custom_kernel.h +79 -14
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/custom_launch_aclnn.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/dense.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/diag_ext.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/divmod.h +2 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/divmods.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/divs.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/dropout_do_mask_ext.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/dropout_ext.h +2 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/dropout_gen_mask_ext.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/dropout_grad_ext.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/{unstack_ext.h → einsum_ext.h} +7 -7
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/elu_ext.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/elu_grad_ext.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/embedding.h +4 -5
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/embedding_dense_backward.h +5 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/empty.h +35 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/empty_like.h +36 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/eye.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/ffn_ext.h +8 -8
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/fill_scalar.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/fill_tensor.h +2 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/flash_attention_score.h +4 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/flash_attention_score_grad.h +6 -6
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/flatten_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/floor_div.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/floor_div_scalar.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/full_like.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/func_max_pool2d.h +38 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/fused_infer_attention_score.h +15 -18
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/gather_d_grad_v2.h +2 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/gelu_grad.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/gelu_grad_ext.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/gmm.h +38 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/gmm_backward.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/gmm_backward_fusion.h +37 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/gmm_v2.h +38 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/gmm_v2_backward.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/gmm_v2_backward_fusion.h +36 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/grid_sampler_2d_grad.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/grid_sampler_3d_grad.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/group_norm.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/group_norm_grad.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/grouped_matmul.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/grouped_matmul_v4.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/hshrink.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/hshrink_grad.h +2 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/identity.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/incre_flash_attention.h +11 -11
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/index.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/index_add_ext.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/index_fill_scalar.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/index_fill_tensor.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inner_index.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inner_inplace_index_put.h +4 -5
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inner_moe_token_unpermute.h +39 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inner_non_zero.h +1 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_add_ext.h +2 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_addmm.h +3 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_adds_ext.h +2 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_clamp_scalar.h +3 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_clamp_tensor.h +3 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_copy.h +2 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_div.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_divmod.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_divmods.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_divs.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_elu.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_erfinv.h +1 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_exp.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_exponential.h +36 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_fill_diagonal.h +2 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_fill_scalar.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_fill_tensor.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_floor.h +1 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_floor_divide.h +2 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_floor_divides.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_grouped_matmul_add.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_hardtanh.h +2 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_index_add.h +3 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_index_put.h +3 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_log.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_masked_fill_scalar.h +2 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_masked_fill_tensor.h +2 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_matmul_add.h +34 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_mul.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_muls.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_normal.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_put.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_random.h +3 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_relu.h +1 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_scatter_add.h +3 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_scatter_src.h +3 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_scatter_src_reduce.h +4 -5
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_scatter_value.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_scatter_value_reduce.h +4 -5
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_silu.h +35 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_stop_gradient.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_sub_ext.h +2 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_sub_scalar.h +2 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_tanh.h +1 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_threshold.h +2 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_uniform.h +3 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_zero.h +1 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/isclose.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/isinf.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/kl_div.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/kl_div_grad.h +3 -6
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/kv_cache_scatter_update.h +3 -5
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/l1_loss_backward_ext.h +3 -5
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/l1_loss_ext.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/layer_norm_ext.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/layer_norm_grad_ext.h +4 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/lerp.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/lerp_scalar.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/lin_space_ext.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/linalg_qr.h +1 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/linalg_vector_norm.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/log_softmax_ext.h +3 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/log_softmax_grad.h +2 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/masked_fill.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/masked_select.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/masked_select_grad.h +2 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/matmul.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/matmul_allreduce_add_rmsnorm.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/matmul_ext.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/matmul_reduce_scatter.h +5 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/max_pool_grad_with_indices.h +6 -7
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/max_pool_grad_with_mask.h +6 -6
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/max_pool_with_indices.h +5 -5
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/max_pool_with_mask.h +4 -5
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/max_unpool2d_ext.h +6 -6
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/mean_ext.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/meshgrid.h +4 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/mm_ext.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/moe_distribute_combine.h +45 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/moe_distribute_dispatch.h +41 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/moe_token_permute.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/moe_token_unpermute.h +4 -6
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/moe_token_unpermute_grad.h +3 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/move_to.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/mse_loss_ext.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/mse_loss_grad_ext.h +3 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/muls.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/multi_scale_deformable_attn.h +5 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/multi_scale_deformable_attn_grad.h +4 -6
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/multinomial_ext.h +3 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/mv.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/nan_to_num.h +3 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/nansum.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/ne_scalar.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/new_empty.h +36 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/{customize_copy.h → new_full.h} +6 -7
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/new_ones.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/new_zeros.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/nllloss.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/nllloss_2d.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/nllloss_2d_grad.h +4 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/nllloss_grad.h +4 -6
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/non_zero.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/non_zero_ext.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/norm.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/normal_float_float.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/normal_float_tensor.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/normal_tensor_float.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/normal_tensor_tensor.h +3 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/one_hot_ext.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/ones.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/ones_like_ext.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/outer.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/pixel_shuffle.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/pow_scalar_tensor.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/pow_tensor_scalar.h +2 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/prod_ext.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/prompt_flash_attention.h +7 -7
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/quant_batch_matmul.h +7 -7
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/quant_matmul.h +41 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/quant_v2.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/rand_ext.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/rand_like_ext.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/randint.h +3 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/randint_like.h +3 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/randn.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/randn_like.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/randperm_ext.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/reduce_all.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/relu_grad.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/repeat_interleave_grad.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/repeat_interleave_int.h +3 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/repeat_interleave_tensor.h +4 -5
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/reshape.h +4 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/rms_norm.h +4 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/rotary_position_embedding_grad.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/round.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/scatter_add_ext.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/searchsorted.h +3 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/sigmoid_grad.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/silent_check_v2.h +5 -5
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/silent_check_v3.h +6 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/smooth_l1_loss.h +3 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/smooth_l1_loss_grad.h +4 -5
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/soft_margin_loss.h +2 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/soft_margin_loss_grad.h +3 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/softmax.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/softshrink.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/softshrink_grad.h +2 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/sort_ext.h +5 -5
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/speed_fusion_attention.h +4 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/speed_fusion_attention_grad.h +11 -12
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/square.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/std.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/std_mean.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/sub.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/sub_scalar.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/sum_ext.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/t_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/take.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/tanh_grad.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/threshold.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/threshold_grad.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/tile.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/type_as.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/uniform_ext.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/unique2.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/unique_consecutive.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/unique_dim.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/unstack_ext_view.h +38 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/upsample_bicubic2d.h +4 -5
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/upsample_bicubic2d_grad.h +5 -6
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/upsample_bilinear2d.h +4 -5
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/upsample_bilinear2d_grad.h +6 -6
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/upsample_linear1d.h +4 -5
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/upsample_linear1d_grad.h +5 -6
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/upsample_nearest1d.h +3 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/upsample_nearest1d_grad.h +4 -5
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/upsample_nearest2d.h +3 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/upsample_nearest2d_grad.h +4 -5
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/upsample_nearest3d.h +3 -4
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/upsample_nearest3d_grad.h +4 -5
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/upsample_trilinear3d.h +4 -5
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/upsample_trilinear3d_grad.h +6 -6
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/var.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/var_mean.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/view_as.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/weight_quant_batch_matmul.h +5 -5
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/zeros.h +2 -2
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/zeros_like_ext.h +2 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/internal/auto_generate/apply_rotary_pos_emb.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/internal/auto_generate/flash_attention_score.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/internal/auto_generate/paged_attention.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/internal/auto_generate/reshape_and_cache.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/internal/functions/functions.h +34 -0
- mindspore/include/mindspore/ops/kernel/cpu/empty_cpu_kernel.h +55 -0
- mindspore/include/mindspore/ops/kernel/cpu/empty_like_cpu_kernel.h +55 -0
- mindspore/include/mindspore/ops/kernel/cpu/grid_sampler_2d_grad_cpu_kernel.h +3 -15
- mindspore/include/mindspore/ops/kernel/cpu/map_tensor/map_tensor_get_data_cpu_kernel.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/map_tensor/map_tensor_get_grad_cpu_kernel.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/new_empty_cpu_kernel.h +55 -0
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/winograd_utils_fp16.h +1 -409
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/winograd_utils_fp16_macro.h +437 -0
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/conv_sw.h +0 -1
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/intrinsics/ms_simd_avx512_instructions.h +0 -3
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/intrinsics/ms_simd_avx_instructions.h +0 -3
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/intrinsics/ms_simd_sse_instructions.h +0 -3
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/abs.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/acos_ext.h +38 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/adaptive_avg_pool1d.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/adaptive_avg_pool2d_grad_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/adaptive_avg_pool3d_grad_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/adaptive_max_pool1d.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/adaptive_max_pool2d.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/add.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/add_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/add_layer_norm_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/add_rmsnorm_quant_v2.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/add_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/addcdiv_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/addcmul_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/any.h +38 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/any_ext.h +38 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/apply_rotary_pos_emb.h +38 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/argmax_with_value.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/argmin_with_value.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/argsort.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/as_strided.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/atan_ext.h +38 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/atanh.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/avg_pool1d.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/avg_pool2d.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/avg_pool2d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/avg_pool3d_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/avg_pool3d_grad_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/baddbmm.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/batch_mat_mul.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/batch_norm_elemt.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/batch_norm_elemt_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/batch_norm_gather_stats_with_counts.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/batch_norm_reduce_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/batch_norm_stats.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/bernoulli_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/binary_cross_entropy.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/binary_cross_entropy_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/binary_cross_entropy_with_logits.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/binary_cross_entropy_with_logits_backward.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/bitwise_and_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/bitwise_and_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/bitwise_not.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/bitwise_or_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/bitwise_or_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/bitwise_xor_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/bitwise_xor_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/bmm_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/broadcast_to.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/broadcast_to_view.h +38 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/cast.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/ceil.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/chunk.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/{select_ext.h → chunk_view.h} +9 -9
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/clamp_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/clamp_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/clone.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/col2im_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/col2im_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/concat.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/constant_pad_nd.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/contiguous.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/conv1d_padding.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/conv2d_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/conv2d_padding.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/conv3d_padding.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/conv_transpose2d.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/convolution.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/convolution_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/convolution_str.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/convolution_str_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/copy.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/cos.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/cosh.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/count_nonzero.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/cross.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/cummax.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/dense.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/diag_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/{transpose_ext.h → diagonal_view.h} +9 -9
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/dist_comm_all_gather.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/dist_comm_all_gather_into_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/dist_comm_all_reduce.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/dist_comm_all_to_all_v.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/dist_comm_all_to_all_v_single.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/dist_comm_barrier.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/dist_comm_batch_isend_irecv.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/dist_comm_broadcast.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/dist_comm_gather.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/dist_comm_gather_into_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/dist_comm_irecv.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/dist_comm_isend.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/dist_comm_reduce.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/dist_comm_reduce_scatter.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/dist_comm_reduce_scatter_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/dist_comm_scatter.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/dist_comm_scatter_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/div.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/divmod.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/divmods.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/divs.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/dot.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/dynamic_quant_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/einsum_ext.h +38 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/elu.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/embedding.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/embedding_dense_backward.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/empty.h +38 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/empty_like.h +38 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/equal.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/equal_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/erf.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/erfc.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/erfinv.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/exp.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/exp2.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/expand_dims.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/expand_dims_view.h +38 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/expm1.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/eye.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/ffn_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/floor.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/floor_div.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/floor_div_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/fmod_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/frac.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/full_like.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/fused_infer_attention_score.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/gather_d.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/gather_d_grad_v2.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/gcd.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/gelu.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/gelu_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/gelu_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/gelu_grad_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/generator.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/glu.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/glu_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/greater.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/greater_equal.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/greater_equal_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/grid_sampler_2d.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/grid_sampler_2d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/grid_sampler_3d.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/grid_sampler_3d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/group_norm.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/group_norm_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/grouped_matmul.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/grouped_matmul_v2.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/grouped_matmul_v4.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/hardtanh.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/hardtanh_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/hshrink.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/hshrink_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/hsigmoid.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/hsigmoid_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/hswish.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/hswish_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/identity.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/im2col_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/incre_flash_attention.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/index.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/index_add_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/index_fill_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/index_fill_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inner_comm_all_gather.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inner_comm_all_reduce.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inner_comm_all_to_all_v.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inner_comm_irecv.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inner_comm_isend.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inner_comm_reduce_scatter.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inner_index.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inner_inplace_index_put.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inner_moe_token_unpermute.h +38 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inner_non_zero.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_addmm.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_clamp_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_clamp_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_copy.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_div.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_divmod.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_divmods.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_divs.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_elu.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_erfinv.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_exponential.h +38 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_fill_diagonal.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_fill_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_fill_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_floor.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_floor_divide.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_floor_divides.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_grouped_matmul_add.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_hardtanh.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_index_add.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_index_put.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_log.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_masked_fill_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_masked_fill_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_matmul_add.h +38 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_mul.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_muls.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_normal.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_put.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_random.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_relu.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_scatter_add.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_scatter_src.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_scatter_src_reduce.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_scatter_value.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_scatter_value_reduce.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/{gmm_backward.h → inplace_silu.h} +9 -9
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_sub_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_tanh.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_threshold.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_uniform.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_zero.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/isclose.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/isfinite.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/isinf.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/isneginf.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/kl_div.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/kl_div_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/kthvalue.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/kv_cache_scatter_update.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/layer_norm_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/leaky_relu_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/leaky_relu_grad_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/lerp.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/lerp_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/less.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/less_equal.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/linalg_qr.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/linalg_vector_norm.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/log.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/log1p.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/log_softmax.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/log_softmax_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/logaddexp.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/logaddexp2.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/logical_and.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/logical_not.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/logical_or.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/logical_xor.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/logsigmoid.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/logsigmoid_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/logsumexp.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/masked_fill.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/masked_select.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/masked_select_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/matmul.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/matmul_allreduce_add_rmsnorm.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/matmul_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/max.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/max_dim.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/maximum.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/mean_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/median_dim.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/median_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/meshgrid.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/min.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/min_dim.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/minimum.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/mish_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/mish_grad_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/moe_compute_expert_tokens.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/moe_distribute_combine.h +38 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/moe_distribute_dispatch.h +38 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/moe_finalize_routing.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/moe_gating_top_k_softmax.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/moe_init_routing.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/moe_init_routing_quant_v2.h +38 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/moe_init_routing_v2.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/moe_token_permute.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/moe_token_permute_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/moe_token_unpermute.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/moe_token_unpermute_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/mul.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/muls.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/multi_scale_deformable_attn.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/multi_scale_deformable_attn_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/mv.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/nan_to_num.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/ne_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/neg.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/new_empty.h +38 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/new_full.h +38 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/new_ones.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/new_zeros.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/nllloss.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/nllloss_2d.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/nllloss_2d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/nllloss_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/non_zero.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/norm.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/normal_float_float.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/normal_float_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/normal_tensor_float.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/normal_tensor_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/not_equal.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/ones.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/outer.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/paged_attention.h +38 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/pixel_shuffle.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/polar.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/pow.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/pow_scalar_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/pow_tensor_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/prelu.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/prelu_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/prod_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/quant_batch_matmul.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/quant_matmul.h +38 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/quant_v2.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/randint.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/randint_like.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/randn.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/randn_like.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/reciprocal.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/reduce_all.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/reduce_any.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/reduce_max.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/reduce_min.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/reflection_pad_1d.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/reflection_pad_1d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/reflection_pad_2d.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/reflection_pad_2d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/reflection_pad_3d.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/reflection_pad_3d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/relu.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/relu_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/repeat.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/replication_pad_1d.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/replication_pad_1d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/replication_pad_2d.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/replication_pad_2d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/replication_pad_3d.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/replication_pad_3d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/reshape.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/reshape_and_cache.h +38 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/reverse_v2.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/rms_norm_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/roll.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/rotary_position_embedding.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/rotary_position_embedding_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/round.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/rsqrt.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/scatter.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/scatter_add_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/scatter_value.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/searchsorted.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/select.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/{gmm_v2_backward.h → select_ext_view.h} +9 -9
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/select_v2.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/selu_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/selu_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/sigmoid.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/sigmoid_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/sign.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/silu.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/silu_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/sin.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/sinc.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/sinh.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/slice.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/slice_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/slice_ext_view.h +38 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/smooth_l1_loss.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/smooth_l1_loss_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/soft_margin_loss.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/soft_margin_loss_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/softmax.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/softmax_backward.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/softplus_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/softplus_grad_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/softshrink.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/softshrink_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/speed_fusion_attention.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/speed_fusion_attention_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/split.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/sqrt.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/square.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/squeeze.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/stack_ext.h +38 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/sub.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/sub_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/sub_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/sum_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/take.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/tan.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/tanh.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/tanh_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/tensor_scatter_elements.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/threshold.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/threshold_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/tile.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/transpose.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/transpose_ext_view.h +38 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/transpose_view.h +38 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/triangular_solve.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/triu.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/trunc.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/unique_consecutive.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/upsample_bicubic2d.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/upsample_bicubic2d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/upsample_bilinear2d.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/upsample_bilinear2d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/upsample_linear1d.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/upsample_linear1d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/upsample_nearest1d.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/upsample_nearest1d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/upsample_nearest2d.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/upsample_nearest2d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/upsample_nearest3d.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/upsample_nearest3d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/upsample_trilinear3d.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/upsample_trilinear3d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/view.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/weight_quant_batch_matmul.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/xlogy.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/xlogy_scalar_other.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/xlogy_scalar_self.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/zeros.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/zeros_like_ext.h +38 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/any.h +34 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/any_ext.h +35 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/binary_cross_entropy_with_logits.h +4 -4
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/clamp_scalar.h +2 -2
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/clamp_tensor.h +2 -3
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/contiguous.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/copy.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/dense.h +2 -2
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/dist_comm_all_gather.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/dist_comm_all_reduce.h +36 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/dist_comm_barrier.h +36 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/dist_comm_broadcast.h +2 -2
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/dist_comm_gather.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/dist_comm_irecv.h +36 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/dist_comm_isend.h +36 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/dist_comm_scatter.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/divmod.h +2 -3
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/empty.h +35 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/empty_like.h +36 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/group_norm.h +3 -3
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/grouped_matmul.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/identity.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/inner_comm_all_reduce.h +2 -3
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/inplace_copy.h +1 -2
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/layer_norm_ext.h +3 -3
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/masked_select.h +2 -2
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/matmul_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/max.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/mean_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/meshgrid.h +4 -2
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/min.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/new_empty.h +36 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/new_ones.h +2 -2
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/new_zeros.h +2 -2
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/non_zero.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/pixel_shuffle.h +2 -2
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/pow_scalar_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/pow_tensor_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/prod_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/reshape.h +4 -2
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/round.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/searchsorted.h +3 -4
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/silu.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/silu_grad.h +1 -2
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/sum_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/customize/unique_consecutive.h +2 -2
- mindspore/include/mindspore/ops/kernel/cpu/sequence/sequence_len_cpu_kernel.h +0 -1
- mindspore/include/mindspore/ops/kernel/cpu/sparse_apply_adagrad_cpu_kernel.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/sparse_apply_adagrad_v2_cpu_kernel.h +2 -2
- mindspore/include/mindspore/ops/kernel/cpu/stft_cpu_kernel.h +16 -16
- mindspore/include/mindspore/ops/kernel/cpu/utils/sampling_kernels.h +18 -15
- mindspore/include/mindspore/ops/kernel/gpu/arrays/broadcast_to_gpu_kernel.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/arrays/contiguous_gpu_kernel.h +9 -6
- mindspore/include/mindspore/ops/kernel/gpu/arrays/select_gpu_kernel.h +2 -2
- mindspore/include/mindspore/ops/kernel/gpu/arrays/unique_consecutive_gpu_kernel.h +2 -2
- mindspore/include/mindspore/ops/kernel/gpu/cuda_impl/cuda_class/unique_consecutive_helper.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/dynamic_akg/dynamic_utils.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/gpu_kernel.h +3 -4
- mindspore/include/mindspore/ops/kernel/gpu/math/binary_ext_ops_gpu_kernel.h +2 -2
- mindspore/include/mindspore/ops/kernel/gpu/math/correlate_gpu_kernel.h +15 -11
- mindspore/include/mindspore/ops/kernel/gpu/math/eps_gpu_kernel.h +4 -4
- mindspore/include/mindspore/ops/kernel/gpu/math/tracev2_grad_gpu_kernel.h +0 -1
- mindspore/include/mindspore/ops/kernel/gpu/nccl/nccl_recv_gpu_kernel.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/nccl/nccl_send_gpu_kernel.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/nn/activation_grad_kernel.h +3 -3
- mindspore/include/mindspore/ops/kernel/gpu/nn/adagrad_gpu_kernel.h +4 -2
- mindspore/include/mindspore/ops/kernel/gpu/nn/adam_gpu_kernel.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/nn/batch_norm_grad_gpu_kernel.h +1 -0
- mindspore/include/mindspore/ops/kernel/gpu/nn/kl_div_loss_grad_kernel.h +3 -2
- mindspore/include/mindspore/ops/kernel/gpu/other/dynamic_stitch_gpu_kernel.h +6 -6
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/abs.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/adaptive_avg_pool1d.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/adaptive_avg_pool2d_grad_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/adaptive_avg_pool3d_grad_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/adaptive_max_pool1d.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/adaptive_max_pool2d.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/add.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/add_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/add_layer_norm_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/add_rmsnorm_quant_v2.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/add_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/addcdiv_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/addcmul_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/any.h +38 -0
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/any_ext.h +38 -0
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/apply_rotary_pos_emb.h +38 -0
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/argmax_with_value.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/argmin_with_value.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/argsort.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/as_strided.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/atanh.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/avg_pool1d.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/avg_pool2d.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/avg_pool2d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/avg_pool3d_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/avg_pool3d_grad_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/baddbmm.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/batch_mat_mul.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/batch_norm_elemt.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/batch_norm_elemt_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/batch_norm_gather_stats_with_counts.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/batch_norm_reduce_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/batch_norm_stats.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/bernoulli_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/binary_cross_entropy.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/binary_cross_entropy_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/binary_cross_entropy_with_logits.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/binary_cross_entropy_with_logits_backward.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/bitwise_and_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/bitwise_and_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/bitwise_not.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/bitwise_or_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/bitwise_or_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/bitwise_xor_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/bitwise_xor_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/bmm_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/broadcast_to.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/broadcast_to_view.h +38 -0
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/cast.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/ceil.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/chunk.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/{select_ext.h → chunk_view.h} +9 -9
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/clamp_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/clamp_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/clone.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/col2im_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/col2im_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/concat.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/constant_pad_nd.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/contiguous.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/conv1d_padding.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/conv2d_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/conv2d_padding.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/conv3d_padding.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/conv_transpose2d.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/convolution.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/convolution_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/convolution_str.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/convolution_str_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/copy.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/cos.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/cosh.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/count_nonzero.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/cross.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/cummax.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/dense.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/diag_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/{transpose_ext.h → diagonal_view.h} +9 -9
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/dist_comm_all_gather.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/dist_comm_all_gather_into_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/dist_comm_all_reduce.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/dist_comm_all_to_all_v.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/dist_comm_all_to_all_v_single.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/dist_comm_barrier.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/dist_comm_batch_isend_irecv.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/dist_comm_broadcast.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/dist_comm_gather.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/dist_comm_gather_into_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/dist_comm_irecv.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/dist_comm_isend.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/dist_comm_reduce.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/dist_comm_reduce_scatter.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/dist_comm_reduce_scatter_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/dist_comm_scatter.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/dist_comm_scatter_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/div.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/divmod.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/divmods.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/divs.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/dot.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/dynamic_quant_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/einsum_ext.h +38 -0
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/elu.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/embedding.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/embedding_dense_backward.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/equal.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/equal_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/erf.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/erfc.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/erfinv.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/exp.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/exp2.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/expand_dims.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/expand_dims_view.h +38 -0
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/expm1.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/eye.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/ffn_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/floor.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/floor_div.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/floor_div_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/fmod_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/frac.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/full_like.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/fused_infer_attention_score.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/gather_d.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/gather_d_grad_v2.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/gcd.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/gelu.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/gelu_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/gelu_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/gelu_grad_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/generator.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/glu.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/glu_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/greater.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/greater_equal.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/greater_equal_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/grid_sampler_2d.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/grid_sampler_2d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/grid_sampler_3d.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/grid_sampler_3d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/group_norm.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/group_norm_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/grouped_matmul.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/grouped_matmul_v2.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/grouped_matmul_v4.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/hardtanh.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/hardtanh_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/hshrink.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/hshrink_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/hsigmoid.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/hsigmoid_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/hswish.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/hswish_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/identity.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/im2col_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/incre_flash_attention.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/index.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/index_add_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/index_fill_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/index_fill_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inner_comm_all_gather.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inner_comm_all_reduce.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inner_comm_all_to_all_v.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inner_comm_irecv.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inner_comm_isend.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inner_comm_reduce_scatter.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inner_index.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inner_inplace_index_put.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inner_moe_token_unpermute.h +38 -0
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inner_non_zero.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_addmm.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_clamp_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_clamp_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_copy.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_div.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_divmod.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_divmods.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_divs.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_elu.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_erfinv.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_exponential.h +38 -0
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_fill_diagonal.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_fill_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_fill_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_floor.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_floor_divide.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_floor_divides.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_grouped_matmul_add.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_hardtanh.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_index_add.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_index_put.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_log.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_masked_fill_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_masked_fill_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_matmul_add.h +38 -0
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_mul.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_muls.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_normal.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_put.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_random.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_relu.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_scatter_add.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_scatter_src.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_scatter_src_reduce.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_scatter_value.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_scatter_value_reduce.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/{gmm_backward.h → inplace_silu.h} +9 -9
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_sub_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_tanh.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_threshold.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_uniform.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_zero.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/isclose.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/isfinite.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/isinf.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/isneginf.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/kl_div.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/kl_div_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/kthvalue.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/kv_cache_scatter_update.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/leaky_relu_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/leaky_relu_grad_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/lerp.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/lerp_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/less.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/less_equal.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/linalg_qr.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/linalg_vector_norm.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/log.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/log1p.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/log_softmax.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/log_softmax_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/logaddexp.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/logaddexp2.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/logical_and.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/logical_not.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/logical_or.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/logical_xor.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/logsigmoid.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/logsigmoid_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/logsumexp.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/masked_fill.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/masked_select.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/masked_select_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/matmul.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/matmul_allreduce_add_rmsnorm.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/matmul_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/max.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/max_dim.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/maximum.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/mean_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/median_dim.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/median_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/meshgrid.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/min.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/min_dim.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/minimum.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/mish_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/mish_grad_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/moe_compute_expert_tokens.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/moe_distribute_combine.h +38 -0
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/moe_distribute_dispatch.h +38 -0
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/moe_finalize_routing.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/moe_gating_top_k_softmax.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/moe_init_routing.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/moe_init_routing_quant_v2.h +38 -0
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/moe_init_routing_v2.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/moe_token_permute.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/moe_token_permute_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/moe_token_unpermute.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/moe_token_unpermute_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/mul.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/muls.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/multi_scale_deformable_attn.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/multi_scale_deformable_attn_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/mv.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/nan_to_num.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/ne_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/neg.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/new_full.h +38 -0
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/new_ones.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/new_zeros.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/nllloss.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/nllloss_2d.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/nllloss_2d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/nllloss_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/non_zero.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/norm.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/normal_float_float.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/normal_float_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/normal_tensor_float.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/normal_tensor_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/not_equal.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/ones.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/outer.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/paged_attention.h +38 -0
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/pixel_shuffle.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/polar.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/pow.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/pow_scalar_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/pow_tensor_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/prelu.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/prelu_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/prod_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/quant_batch_matmul.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/quant_matmul.h +38 -0
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/quant_v2.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/randint.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/randint_like.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/randn.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/randn_like.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/reciprocal.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/reduce_all.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/reduce_any.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/reduce_max.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/reduce_min.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/reflection_pad_1d.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/reflection_pad_1d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/reflection_pad_2d.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/reflection_pad_2d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/reflection_pad_3d.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/reflection_pad_3d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/relu.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/relu_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/repeat.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/replication_pad_1d.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/replication_pad_1d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/replication_pad_2d.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/replication_pad_2d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/replication_pad_3d.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/replication_pad_3d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/reshape.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/reshape_and_cache.h +38 -0
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/reverse_v2.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/rms_norm_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/roll.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/rotary_position_embedding.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/rotary_position_embedding_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/round.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/rsqrt.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/scatter.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/scatter_add_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/scatter_value.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/searchsorted.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/select.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/{gmm_v2_backward.h → select_ext_view.h} +9 -9
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/select_v2.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/selu_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/selu_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/sigmoid.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/sigmoid_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/sign.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/silu.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/silu_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/sin.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/sinc.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/sinh.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/slice.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/slice_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/slice_ext_view.h +38 -0
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/smooth_l1_loss.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/smooth_l1_loss_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/soft_margin_loss.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/soft_margin_loss_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/softmax.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/softmax_backward.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/softplus_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/softplus_grad_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/softshrink.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/softshrink_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/speed_fusion_attention.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/speed_fusion_attention_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/split.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/sqrt.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/square.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/squeeze.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/sub.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/sub_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/sub_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/sum_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/take.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/tan.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/tanh.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/tanh_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/tensor_scatter_elements.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/threshold.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/threshold_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/tile.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/transpose.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/transpose_ext_view.h +38 -0
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/transpose_view.h +38 -0
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/triangular_solve.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/triu.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/trunc.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/upsample_bicubic2d.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/upsample_bicubic2d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/upsample_bilinear2d.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/upsample_bilinear2d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/upsample_linear1d.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/upsample_linear1d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/upsample_nearest1d.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/upsample_nearest1d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/upsample_nearest2d.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/upsample_nearest2d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/upsample_nearest3d.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/upsample_nearest3d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/upsample_trilinear3d.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/upsample_trilinear3d_grad.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/view.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/weight_quant_batch_matmul.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/xlogy.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/xlogy_scalar_other.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/xlogy_scalar_self.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/zeros.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/customize/any.h +34 -0
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/customize/any_ext.h +35 -0
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/customize/clamp_scalar.h +2 -2
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/customize/clamp_tensor.h +2 -3
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/customize/contiguous.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/customize/copy.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/customize/dense.h +2 -2
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/customize/divmod.h +2 -3
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/customize/grouped_matmul.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/customize/identity.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/customize/inner_comm_all_gather.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/customize/inner_comm_all_reduce.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/customize/inner_comm_isend.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/customize/inner_comm_reduce_scatter.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/customize/masked_select.h +2 -2
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/customize/matmul_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/customize/max.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/customize/mean_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/customize/meshgrid.h +4 -2
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/customize/min.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/customize/new_ones.h +2 -2
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/customize/new_zeros.h +2 -2
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/customize/non_zero.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/customize/pixel_shuffle.h +2 -2
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/customize/pow_scalar_tensor.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/customize/pow_tensor_scalar.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/customize/prod_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/customize/reshape.h +4 -2
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/customize/searchsorted.h +3 -4
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/customize/sum_ext.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/random/random_categorical_gpu_kernel.h +4 -8
- mindspore/include/mindspore/ops/kernel/gpu/rl/buffer_sample_gpu_kernel.h +8 -8
- mindspore/include/mindspore/ops/kernel/gpu/rl/gru_gpu_kernel.h +1 -0
- mindspore/include/mindspore/ops/kernel/gpu/sparse/csr_sparse_matrix_to_sparse_tensor_gpu_kernel.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/sparse/dense_to_csr_sparse_matrix_gpu_kernel.h +3 -3
- mindspore/include/mindspore/ops/kernel/gpu/sparse/sparse_matrix_sparse_matmul_gpu_kernel.h +2 -2
- mindspore/include/mindspore/ops/kernel/include/common/common_utils.h +9 -0
- mindspore/include/mindspore/ops/kernel/include/common/device_address.h +290 -27
- mindspore/include/mindspore/ops/kernel/include/common/device_type.h +3 -5
- mindspore/include/mindspore/ops/kernel/include/common/kernel.h +26 -1
- mindspore/include/mindspore/ops/kernel/include/common/kernel_tensor.h +135 -306
- mindspore/include/mindspore/ops/op_def/array_op_name.h +0 -1
- mindspore/include/mindspore/ops/op_def/array_ops.h +0 -1
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_lite_ops.h +3088 -2881
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_def.h +793 -756
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_a.h +63 -61
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_b.h +27 -26
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_c.h +34 -33
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_d.h +28 -26
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_e.h +27 -25
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_f.h +20 -18
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_g.h +23 -19
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_h.h +8 -8
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_i.h +70 -66
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_k.h +1 -1
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_l.h +31 -31
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_m.h +43 -40
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_n.h +19 -17
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_o.h +3 -3
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_p.h +4 -5
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_q.h +3 -2
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_r.h +60 -60
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_s.h +64 -60
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_t.h +19 -17
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_u.h +13 -13
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_v.h +2 -2
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_x.h +1 -1
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_a.h +63 -61
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_b.h +27 -26
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_c.h +34 -33
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_d.h +28 -26
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_e.h +27 -25
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_f.h +20 -18
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_g.h +23 -19
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_h.h +8 -8
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_i.h +70 -66
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_k.h +1 -1
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_l.h +31 -31
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_m.h +43 -40
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_n.h +19 -17
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_o.h +3 -3
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_p.h +4 -5
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_q.h +3 -2
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_r.h +60 -60
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_s.h +64 -60
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_t.h +19 -17
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_u.h +13 -13
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_v.h +2 -2
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_x.h +1 -1
- mindspore/include/mindspore/ops/op_def/op_enum.h +2 -0
- mindspore/include/mindspore/ops/op_def/other_op_name.h +6 -0
- mindspore/include/mindspore/ops/op_def/other_ops.h +2 -0
- mindspore/include/mindspore/ops/op_def/structure_ops.h +8 -2
- mindspore/include/mindspore/ops/ops_utils/memory_overlap.h +4 -5
- mindspore/include/mindspore/ops/ops_utils/op_constants.h +12 -0
- mindspore/include/mindspore/ops/ops_utils/op_utils.h +5 -11
- mindspore/include/mindspore/ops/ops_utils/type_dispatch.h +42 -42
- mindspore/include/mindspore/ops/view/as_strided_strides_calc.h +5 -1
- mindspore/include/mindspore/ops/view/broadcast_to_strides_calc.h +7 -2
- mindspore/include/mindspore/ops/view/broadcast_to_view_strides_calc.h +34 -0
- mindspore/include/mindspore/ops/view/chunk_strides_calc.h +5 -1
- mindspore/include/mindspore/ops/view/chunk_view_strides_calc.h +35 -0
- mindspore/include/mindspore/ops/view/diagonal_strides_calc.h +6 -3
- mindspore/include/mindspore/ops/view/{transpose_ext_strides_calc.h → diagonal_view_strides_calc.h} +9 -9
- mindspore/include/mindspore/ops/view/expand_dims_strides_calc.h +5 -1
- mindspore/include/mindspore/ops/view/expand_dims_view_strides_calc.h +34 -0
- mindspore/include/mindspore/ops/view/narrow_strides_calc.h +4 -2
- mindspore/include/mindspore/ops/view/narrow_view_strides_calc.h +33 -0
- mindspore/include/mindspore/ops/view/reshape_strides_calc.h +2 -1
- mindspore/include/mindspore/ops/view/select_ext_view_strides_calc.h +33 -0
- mindspore/include/mindspore/ops/view/slice_ext_strides_calc.h +7 -1
- mindspore/include/mindspore/ops/view/slice_ext_view_strides_calc.h +35 -0
- mindspore/include/mindspore/ops/view/slice_strides_calc.h +4 -1
- mindspore/include/mindspore/ops/view/split_strides_calc.h +3 -1
- mindspore/include/mindspore/ops/view/split_tensor_strides_calc.h +5 -2
- mindspore/include/mindspore/ops/view/split_tensor_view_strides_calc.h +33 -0
- mindspore/include/mindspore/ops/view/split_with_size_strides_calc.h +5 -1
- mindspore/include/mindspore/ops/view/split_with_size_view_strides_calc.h +36 -0
- mindspore/include/mindspore/ops/view/squeeze_strides_calc.h +3 -0
- mindspore/include/mindspore/ops/view/transpose_ext_view_strides_calc.h +33 -0
- mindspore/include/mindspore/ops/view/transpose_strides_calc.h +3 -1
- mindspore/include/mindspore/ops/view/transpose_view_strides_calc.h +32 -0
- mindspore/include/mindspore/ops/view/{unstack_ext_strides_calc.h → unstack_ext_view_strides_calc.h} +5 -5
- mindspore/include/mindspore/ops/view/view_strides_calc.h +5 -2
- mindspore/include/mindspore/ops/view/view_strides_calculator.h +1 -2
- mindspore/include/ms_extension.h +12 -5
- 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/libicuuc.so.74 +0 -0
- mindspore/lib/libmindspore_backend_common.so +0 -0
- mindspore/lib/libmindspore_backend_manager.so +0 -0
- mindspore/lib/libmindspore_common.so +0 -0
- mindspore/lib/libmindspore_core.so +0 -0
- mindspore/lib/libmindspore_dump.so +0 -0
- mindspore/lib/libmindspore_extension.so +0 -0
- mindspore/lib/libmindspore_frontend.so +0 -0
- mindspore/lib/libmindspore_ge_backend.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_memory_pool.so +0 -0
- mindspore/lib/libmindspore_ms_backend.so +0 -0
- mindspore/lib/libmindspore_ops.so +0 -0
- mindspore/lib/libmindspore_ops_kernel_common.so +0 -0
- mindspore/lib/libmindspore_profiler.so +0 -0
- mindspore/lib/libmindspore_pyboost.so +0 -0
- mindspore/lib/libmindspore_pynative.so +0 -0
- mindspore/lib/libmindspore_res_manager.so +0 -0
- mindspore/lib/libmindspore_runtime_pipeline.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/custom_aicore_ops_impl/custom.py +4 -4
- mindspore/lib/plugin/ascend/custom_aicore_ops/op_impl/ai_core/tbe/custom_aicore_ops_impl/matmul_tik.py +1 -0
- mindspore/lib/plugin/ascend/custom_aicore_ops/op_impl/vector_core/tbe/custom_aicore_ops_impl/matmul_tik.py +1 -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_impl/cpu/config/cust_aicpu_kernel.json +96 -72
- mindspore/lib/plugin/ascend/custom_aicpu_ops/op_proto/libcust_op_proto.so +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/framework/plugin/npu_supported_ops.json +6 -0
- 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/ascend310p/aic-ascend310p-ops-info.json +0 -180
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/config/ascend910_93/aic-ascend910_93-ops-info.json +0 -180
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/config/ascend910b/aic-ascend910b-ops-info.json +0 -180
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/custom_ascendc_910b_impl/dynamic/all_finite.py +28 -12
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/all_finite/AllFinite_52f59e2a65d9b1bb002de35c2819754a.json +2 -1
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/all_finite/AllFinite_52f59e2a65d9b1bb002de35c2819754a.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/all_finite/AllFinite_6b5e50e30256d85838d6ce83514df20f.json +2 -1
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/all_finite/AllFinite_6b5e50e30256d85838d6ce83514df20f.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/all_finite/AllFinite_74e4ac02880d452e3308c94af273562e.json +2 -1
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/all_finite/AllFinite_74e4ac02880d452e3308c94af273562e.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/all_finite/AllFinite_52f59e2a65d9b1bb002de35c2819754a.json +2 -1
- 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 +2 -1
- 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 +2 -1
- 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/ascend910b/all_finite/AllFinite_52f59e2a65d9b1bb002de35c2819754a.json +2 -1
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/all_finite/AllFinite_52f59e2a65d9b1bb002de35c2819754a.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/all_finite/AllFinite_6b5e50e30256d85838d6ce83514df20f.json +2 -1
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/all_finite/AllFinite_6b5e50e30256d85838d6ce83514df20f.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/all_finite/AllFinite_74e4ac02880d452e3308c94af273562e.json +2 -1
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/all_finite/AllFinite_74e4ac02880d452e3308c94af273562e.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/config/ascend310p/binary_info_config.json +0 -300
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/config/ascend910_93/binary_info_config.json +0 -300
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/config/ascend910b/binary_info_config.json +0 -300
- 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/inc/op_proto.h +0 -22
- 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/OWNERS +1 -0
- mindspore/lib/plugin/ascend/libakg.so +0 -0
- mindspore/lib/plugin/ascend/libascend_collective.so +0 -0
- mindspore/lib/plugin/ascend/libd_collective.so +0 -0
- mindspore/lib/plugin/ascend/libdvpp_utils.so +0 -0
- mindspore/lib/plugin/ascend/libhccl_plugin.so +0 -0
- mindspore/lib/plugin/ascend/liblowlatency_collective.so +0 -0
- mindspore/lib/plugin/ascend/libmindspore_ascend_res_manager.so +0 -0
- mindspore/lib/plugin/ascend/libmindspore_atb_kernels.so +0 -0
- mindspore/lib/plugin/ascend/libmindspore_cpu_kernels.so +0 -0
- mindspore/lib/plugin/ascend/libmindspore_extension_ascend_atb.a +0 -0
- mindspore/lib/plugin/ascend/libmindspore_graph_ir.so +0 -0
- mindspore/lib/plugin/ascend/libmindspore_internal_kernels.so +0 -0
- mindspore/lib/plugin/ascend/libmindspore_pyboost_atb_kernels.so +0 -0
- mindspore/lib/plugin/ascend/libms_atb_boost.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/asdops/params/faUpdate.h +35 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/asdops/params/fill.h +4 -1
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/asdops/params/logprobs.h +28 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/asdops/params/matmul.h +7 -6
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/asdops/params/norm.h +8 -5
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/asdops/params/params.h +3 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/asdops/params/scatter_elements_v2.h +39 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/atbops/params/fused_add_topk_div.h +42 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/atbops/params/kvcache.h +7 -1
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/atbops/params/mla.h +55 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/atbops/params/mla_preprocess.h +39 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/atbops/params/pagedattention.h +1 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/atbops/params/params.h +7 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/atbops/params/reshape_and_cache.h +2 -1
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/atbops/params/rms_norm_and_rope_and_reshape_and_cache.h +31 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/atbops/params/rope_q_concat.h +26 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/atbops/params/swiglu_quant.h +26 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/atbops/params/toppsample_rand.h +31 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/atbops/params/unpad_flash_attention.h +12 -1
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/lcal/lcal_api.h +1 -1
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/lcal/lcal_comm.h +4 -3
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/lcal/lcal_types.h +2 -1
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/lcal/lccl.h +2 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/lcal/lcoc/lcoc.h +53 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/lcal/lcoc/lcoc_args.h +116 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/lcal/lcoc/lcoc_base.h +57 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/lcal/lcoc/lcoc_func.h +33 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/lcal/lcoc/tiling/tiling.h +86 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/lcal/lcoc/tiling/tiling_91093.h +31 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/lcal/lcoc/tiling/tiling_910B.h +31 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/lcal/lcoc/tiling/tiling_args.h +154 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/lcal/lcoc/tiling/tiling_func.h +50 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/lcal/lcoc.h +5 -35
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/lcal/lcoc_args.h +97 -47
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/lcal/lcoc_func.h +33 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/lcal/tiling/tiling.h +86 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/lcal/tiling/tiling_91093.h +31 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/lcal/tiling/tiling_910B.h +31 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/lcal/tiling/tiling_args.h +154 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/lcal/tiling/tiling_func.h +50 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/lcal/tiling.h +86 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/lcal/tiling_91093.h +11 -9
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/lcal/tiling_910B.h +12 -10
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/lcal/tiling_args.h +38 -69
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/lcal/tiling_func.h +14 -8
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/mki/base/aicpu_kernel_base.h +1 -1
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/mki/bin_handle.h +6 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/mki/tensor.h +5 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/mki/types.h +4 -1
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/mki/utils/bf16/bf16_t.h +20 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/mki/utils/cfg/cfg_core.h +39 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/mki/utils/cfg/cfg_item.h +25 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/mki/utils/file_system/file_system.h +2 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/mki/utils/inifile/ini_file.h +2 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/mki/utils/log/log.h +7 -7
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/mki/utils/log/log_core.h +1 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/mki/utils/log/log_sink_file.h +1 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/include/mki/utils/rt/base/types.h +2 -2
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/lib/libasdops.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/lib/libasdops_static.a +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/lib/libatb_mixops.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/lib/libatb_mixops_static.a +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/lib/libexp_mixops_static.a +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/lib/libexp_ops_static.a +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/lib/liblcal.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/lib/liblcal_static.a +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/lib/libmki.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/asdops/lib/libtbe_adapter.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/base_type.h +7 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/internal_op.h +8 -10
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/op_creator.h +25 -6
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/op_param.h +68 -1
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/tiling_utils.h +1 -100
- 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_dynamic_quant_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/libgroup_topk_op.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libkv_scale_cache_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/libmoe_gating_group_topk_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/libmulti_weight_matmul_kernel_gelu_op.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/lib/libswiglu_dynamic_quant_op.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libtranspose_batch_matmul_transpose_op.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/object_kernels/internal_grouped_matmul_f16_310p/internal_grouped_matmul_f16_310p.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/object_kernels/internal_grouped_matmul_f16_310p/internal_grouped_matmul_f16_310p_0.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/object_kernels/internal_grouped_matmul_i8_310p/internal_grouped_matmul_i8_310p.o +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/object_kernels/internal_grouped_matmul_i8_310p/internal_grouped_matmul_i8_310p_0.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.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/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/libmoe_gating_group_topk_ascend310p.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/so_kernels/libmulti_weight_matmul_kernel_ascend310p.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/so_kernels/libmulti_weight_matmul_kernel_gelu_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/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/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/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/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/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/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/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/op_kernels/ascend910b/so_kernels/libadd_layer_norm_ascend910b.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/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_dynamic_quant_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/op_kernels/ascend910b/so_kernels/libapply_rotary_pos_emb_ascend910b.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/so_kernels/libcast_ascend910b.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/so_kernels/libcompare_ascend910b.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/so_kernels/libgelu_ascend910b.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/so_kernels/libgroup_topk_ascend910b.so +0 -0
- mindspore/lib/plugin/ascend/{custom_ascendc_910/op_api/lib/libcust_opapi.so → ms_kernels_internal/internal_kernel/op_kernels/ascend910b/so_kernels/libkv_scale_cache_ascend910b.so} +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/so_kernels/libmatmul_ascend910b.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/so_kernels/libmoe_gating_group_topk_ascend910b.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/so_kernels/libmulti_weight_matmul_kernel_ascend910b.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/so_kernels/libreshape_and_cache_ascend910b.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/so_kernels/librms_norm_ascend910b.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/so_kernels/libswiglu_dynamic_quant_ascend910b.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend910b/so_kernels/libtranspose_batch_matmul_transpose_ascend910b.so +0 -0
- mindspore/lib/plugin/cpu/libakg.so +0 -0
- mindspore/lib/plugin/cpu/libmindspore_cpu_res_manager.so +0 -0
- mindspore/lib/plugin/libmindspore_ascend.so.2 +0 -0
- mindspore/lib/plugin/libmindspore_ops_ascend.so +0 -0
- mindspore/lib/plugin/libmindspore_ops_host.so +0 -0
- mindspore/mindrecord/tools/cifar10.py +60 -11
- mindspore/mindrecord/tools/cifar10_to_mr.py +5 -0
- mindspore/mint/__init__.py +6 -46
- mindspore/mint/distributed/__init__.py +1 -0
- mindspore/mint/distributed/distributed.py +212 -9
- mindspore/mint/nn/__init__.py +1 -1
- mindspore/mint/nn/functional.py +53 -6
- mindspore/mint/nn/layer/_functions.py +164 -294
- mindspore/mint/nn/layer/activation.py +8 -6
- mindspore/mint/nn/layer/conv.py +137 -101
- mindspore/mint/nn/layer/normalization.py +8 -22
- mindspore/mint/optim/adam.py +19 -18
- mindspore/mint/optim/adamw.py +14 -8
- mindspore/mint/optim/sgd.py +5 -5
- mindspore/nn/cell.py +328 -502
- mindspore/nn/grad/cell_grad.py +11 -12
- mindspore/nn/layer/activation.py +32 -34
- mindspore/nn/layer/basic.py +67 -64
- mindspore/nn/layer/channel_shuffle.py +4 -4
- mindspore/nn/layer/combined.py +4 -2
- mindspore/nn/layer/conv.py +117 -110
- mindspore/nn/layer/dense.py +9 -7
- mindspore/nn/layer/embedding.py +50 -52
- mindspore/nn/layer/image.py +37 -39
- mindspore/nn/layer/math.py +111 -112
- mindspore/nn/layer/normalization.py +56 -44
- mindspore/nn/layer/pooling.py +58 -63
- mindspore/nn/layer/rnn_cells.py +33 -33
- mindspore/nn/layer/rnns.py +56 -56
- mindspore/nn/layer/thor_layer.py +74 -73
- mindspore/nn/layer/transformer.py +11 -1
- mindspore/nn/learning_rate_schedule.py +20 -20
- mindspore/nn/loss/loss.py +79 -81
- mindspore/nn/optim/adam.py +3 -3
- mindspore/nn/optim/adasum.py +2 -2
- mindspore/nn/optim/asgd.py +2 -0
- mindspore/nn/optim/optimizer.py +1 -1
- mindspore/nn/optim/thor.py +2 -2
- mindspore/nn/probability/distribution/exponential.py +2 -1
- mindspore/nn/probability/distribution/poisson.py +2 -1
- mindspore/nn/sparse/sparse.py +3 -3
- mindspore/nn/wrap/cell_wrapper.py +34 -37
- mindspore/nn/wrap/grad_reducer.py +37 -37
- mindspore/nn/wrap/loss_scale.py +72 -74
- mindspore/numpy/array_creations.py +5 -5
- mindspore/numpy/fft.py +1 -1
- mindspore/numpy/math_ops.py +5 -5
- mindspore/ops/_grad_experimental/grad_comm_ops.py +51 -13
- mindspore/ops/_grad_experimental/grad_debug_ops.py +14 -0
- mindspore/ops/_vmap/vmap_array_ops.py +31 -13
- mindspore/ops/_vmap/vmap_nn_ops.py +8 -16
- mindspore/ops/auto_generate/cpp_create_prim_instance_helper.py +449 -418
- mindspore/ops/auto_generate/gen_extend_func.py +1238 -1356
- mindspore/ops/auto_generate/gen_ops_def.py +6490 -6084
- mindspore/ops/auto_generate/gen_ops_prim.py +16386 -15649
- mindspore/ops/auto_generate/pyboost_inner_prim.py +355 -325
- mindspore/ops/composite/__init__.py +10 -0
- mindspore/ops/composite/base.py +8 -4
- mindspore/ops/composite/multitype_ops/__init__.py +12 -1
- mindspore/ops/composite/multitype_ops/_compile_utils.py +133 -109
- mindspore/ops/composite/multitype_ops/add_impl.py +70 -2
- mindspore/ops/composite/multitype_ops/div_impl.py +49 -0
- mindspore/ops/composite/multitype_ops/floordiv_impl.py +29 -0
- mindspore/ops/composite/multitype_ops/getitem_impl.py +11 -0
- mindspore/ops/composite/multitype_ops/mod_impl.py +5 -3
- mindspore/ops/composite/multitype_ops/mul_impl.py +49 -0
- mindspore/ops/composite/multitype_ops/setitem_impl.py +57 -0
- mindspore/ops/composite/multitype_ops/sub_impl.py +34 -0
- mindspore/ops/composite/multitype_ops/zeros_like_impl.py +14 -0
- mindspore/ops/function/__init__.py +3 -1
- mindspore/ops/function/_add_attr_func.py +11 -6
- mindspore/ops/function/array_func.py +9 -96
- mindspore/ops/function/debug_func.py +4 -3
- mindspore/ops/function/grad/grad_func.py +1 -1
- mindspore/ops/function/math_func.py +33 -540
- mindspore/ops/function/nn_func.py +28 -74
- mindspore/ops/function/other_func.py +4 -1
- mindspore/ops/function/random_func.py +44 -5
- mindspore/ops/function/vmap_func.py +2 -1
- mindspore/ops/functional.py +2 -3
- mindspore/ops/functional_overload.py +1278 -713
- mindspore/ops/op_info_register.py +21 -0
- mindspore/ops/operations/__init__.py +16 -11
- mindspore/ops/operations/_custom_ops_utils.py +689 -34
- mindspore/ops/operations/_inner_ops.py +3 -6
- mindspore/ops/operations/_sequence_ops.py +1 -1
- mindspore/ops/operations/array_ops.py +2 -2
- mindspore/ops/operations/comm_ops.py +185 -26
- mindspore/ops/operations/custom_ops.py +294 -174
- mindspore/ops/operations/debug_ops.py +59 -4
- mindspore/ops/operations/image_ops.py +13 -13
- mindspore/ops/operations/manually_defined/ops_def.py +15 -16
- mindspore/ops/operations/math_ops.py +3 -4
- mindspore/ops/operations/nn_ops.py +7 -39
- mindspore/ops/primitive.py +6 -10
- mindspore/ops/tensor_method.py +47 -8
- mindspore/ops_generate/api/cpp_create_prim_instance_helper_generator.py +1 -1
- mindspore/ops_generate/api/functional_map_cpp_generator.py +10 -9
- mindspore/ops_generate/api/functions_cc_generator.py +58 -10
- mindspore/ops_generate/api/tensor_func_reg_cpp_generator.py +1 -1
- mindspore/ops_generate/common/base_generator.py +14 -0
- mindspore/ops_generate/common/gen_constants.py +8 -3
- mindspore/ops_generate/common/gen_utils.py +0 -19
- mindspore/ops_generate/common/op_proto.py +11 -4
- mindspore/ops_generate/common/template.py +88 -11
- mindspore/ops_generate/gen_ops.py +1 -1
- mindspore/ops_generate/op_def/lite_ops_cpp_generator.py +4 -4
- mindspore/ops_generate/op_def/ops_def_cc_generator.py +0 -3
- mindspore/ops_generate/op_def/ops_name_h_generator.py +0 -3
- mindspore/ops_generate/op_def/ops_primitive_h_generator.py +0 -4
- mindspore/ops_generate/op_def_py/op_prim_py_generator.py +5 -2
- mindspore/ops_generate/pyboost/auto_grad_impl_cc_generator.py +49 -8
- mindspore/ops_generate/pyboost/auto_grad_reg_cc_generator.py +2 -2
- mindspore/ops_generate/pyboost/gen_pyboost_func.py +31 -0
- mindspore/ops_generate/pyboost/op_template_parser.py +98 -72
- mindspore/ops_generate/pyboost/pyboost_functions_cpp_generator.py +70 -273
- mindspore/ops_generate/pyboost/pyboost_functions_h_generator.py +14 -6
- mindspore/ops_generate/pyboost/pyboost_functions_impl_cpp_generator.py +316 -0
- mindspore/ops_generate/pyboost/pyboost_functions_py_generator.py +1 -1
- mindspore/ops_generate/pyboost/pyboost_grad_function_cpp_generator.py +5 -3
- mindspore/ops_generate/pyboost/pyboost_inner_prim_generator.py +1 -1
- mindspore/ops_generate/pyboost/pyboost_internal_functions_cpp_generator.py +76 -0
- mindspore/ops_generate/pyboost/pyboost_internal_functions_h_generator.py +76 -0
- mindspore/ops_generate/pyboost/pyboost_internal_kernel_info_adapter_generator.py +125 -0
- mindspore/ops_generate/pyboost/pyboost_native_grad_functions_generator.py +4 -3
- mindspore/ops_generate/pyboost/pyboost_op_cpp_code_generator.py +348 -61
- mindspore/ops_generate/pyboost/pyboost_overload_functions_cpp_generator.py +1 -1
- mindspore/ops_generate/pyboost/pyboost_utils.py +118 -9
- mindspore/ops_generate/tensor_py_cc_generator.py +1 -24
- mindspore/parallel/_auto_parallel_context.py +11 -8
- mindspore/parallel/_cell_wrapper.py +113 -45
- mindspore/parallel/_parallel_serialization.py +1 -1
- mindspore/parallel/_ps_context.py +4 -6
- mindspore/parallel/_tensor.py +167 -12
- mindspore/parallel/_transformer/moe.py +1 -1
- mindspore/parallel/_transformer/transformer.py +13 -8
- mindspore/parallel/auto_parallel.py +14 -7
- mindspore/parallel/checkpoint_convert.py +3 -3
- mindspore/parallel/checkpoint_transform.py +11 -7
- mindspore/parallel/cluster/process_entity/_api.py +84 -48
- mindspore/parallel/cluster/process_entity/_utils.py +95 -7
- mindspore/parallel/cluster/run.py +43 -4
- mindspore/parallel/function/__init__.py +8 -1
- mindspore/parallel/function/reshard_func.py +6 -7
- mindspore/parallel/nn/__init__.py +15 -2
- mindspore/parallel/nn/parallel_cell_wrapper.py +9 -10
- mindspore/parallel/nn/parallel_grad_reducer.py +7 -6
- mindspore/parallel/shard.py +3 -4
- mindspore/parallel/transform_safetensors.py +463 -174
- mindspore/profiler/__init__.py +2 -1
- mindspore/profiler/analysis/parser/timeline_assembly_factory/ascend_timeline_assembler.py +7 -7
- mindspore/profiler/analysis/parser/timeline_assembly_factory/base_timeline_assembler.py +3 -0
- mindspore/profiler/analysis/parser/timeline_assembly_factory/trace_view_container.py +12 -6
- mindspore/profiler/analysis/parser/timeline_creator/cpu_op_timeline_creator.py +3 -3
- mindspore/profiler/analysis/parser/timeline_creator/fwk_timeline_creator.py +3 -3
- mindspore/profiler/analysis/parser/timeline_creator/msprof_timeline_creator.py +4 -4
- mindspore/profiler/analysis/parser/timeline_creator/scope_layer_timeline_creator.py +3 -3
- mindspore/profiler/analysis/parser/timeline_event/fwk_event.py +4 -1
- mindspore/profiler/analysis/parser/timeline_event/timeline_event_pool.py +2 -1
- mindspore/profiler/analysis/task_manager.py +1 -1
- mindspore/profiler/analysis/viewer/ascend_communication_viewer.py +5 -1
- mindspore/profiler/analysis/viewer/ascend_integrate_viewer.py +2 -1
- mindspore/profiler/analysis/viewer/ascend_op_memory_viewer.py +42 -22
- mindspore/profiler/analysis/viewer/ascend_step_trace_time_viewer.py +3 -2
- mindspore/profiler/analysis/viewer/ms_minddata_viewer.py +9 -5
- mindspore/profiler/analysis/viewer/ms_operator_details_viewer.py +132 -0
- mindspore/profiler/common/constant.py +16 -0
- mindspore/profiler/common/profiler_context.py +25 -27
- mindspore/profiler/common/profiler_info.py +0 -16
- mindspore/profiler/common/profiler_op_analyse.py +235 -0
- mindspore/profiler/common/profiler_output_path.py +23 -8
- mindspore/profiler/common/profiler_parameters.py +128 -35
- mindspore/profiler/dynamic_profile/__init__.py +0 -0
- mindspore/profiler/dynamic_profile/dynamic_monitor_proxy.py +39 -0
- mindspore/profiler/dynamic_profile/dynamic_profiler_config_context.py +666 -0
- mindspore/profiler/dynamic_profile/dynamic_profiler_utils.py +62 -0
- mindspore/profiler/dynamic_profiler.py +305 -314
- mindspore/profiler/envprofiler.py +12 -7
- mindspore/profiler/experimental_config.py +96 -6
- mindspore/profiler/mstx.py +33 -12
- mindspore/profiler/platform/__init__.py +2 -3
- mindspore/profiler/platform/npu_profiler.py +29 -19
- mindspore/profiler/profiler.py +35 -19
- mindspore/profiler/profiler_action_controller.py +64 -76
- mindspore/profiler/schedule.py +10 -4
- mindspore/rewrite/common/config.py +1 -0
- mindspore/rewrite/common/namer.py +1 -0
- mindspore/rewrite/common/namespace.py +1 -0
- mindspore/rewrite/node/node.py +31 -11
- mindspore/rewrite/parsers/assign_parser.py +1 -1
- mindspore/rewrite/symbol_tree/symbol_tree.py +1 -1
- mindspore/run_check/_check_version.py +7 -10
- mindspore/runtime/__init__.py +5 -5
- mindspore/runtime/event.py +10 -4
- mindspore/runtime/executor.py +60 -45
- mindspore/runtime/memory.py +30 -32
- mindspore/runtime/thread_bind_core.py +298 -164
- mindspore/safeguard/rewrite_obfuscation.py +12 -13
- mindspore/scipy/linalg.py +2 -2
- mindspore/scipy/utils_const.py +0 -17
- mindspore/train/_utils.py +14 -4
- mindspore/train/amp.py +43 -20
- mindspore/train/callback/__init__.py +5 -5
- mindspore/train/callback/_checkpoint.py +3 -6
- mindspore/train/callback/_flops_collector.py +1 -1
- mindspore/train/callback/_landscape.py +0 -1
- mindspore/train/callback/_train_fault_tolerance.py +97 -16
- mindspore/train/data_sink.py +11 -2
- mindspore/train/dataset_helper.py +9 -0
- mindspore/train/model.py +135 -55
- mindspore/train/serialization.py +133 -111
- mindspore/train/summary/summary_record.py +13 -2
- mindspore/utils/__init__.py +3 -2
- mindspore/utils/bin/dataset-cache +0 -0
- mindspore/utils/bin/dataset-cache-server +0 -0
- mindspore/utils/dryrun.py +0 -6
- mindspore/utils/runtime_execution_order_check.py +163 -77
- mindspore/utils/sdc_detect.py +68 -0
- mindspore/utils/utils.py +6 -9
- mindspore/version.py +1 -1
- {mindspore-2.6.0rc1.dist-info → mindspore-2.7.0rc1.dist-info}/METADATA +5 -4
- {mindspore-2.6.0rc1.dist-info → mindspore-2.7.0rc1.dist-info}/RECORD +4031 -3947
- mindspore/_deprecated/jit.py +0 -198
- mindspore/experimental/es/__init__.py +0 -22
- mindspore/experimental/es/embedding_service.py +0 -891
- mindspore/experimental/es/embedding_service_layer.py +0 -581
- mindspore/include/mindspore/ccsrc/backend/common/graph_kernel/proactive_fallback_expander.h +0 -39
- mindspore/include/mindspore/ccsrc/backend/common/session/session_context.h +0 -47
- mindspore/include/mindspore/ccsrc/backend/ge_backend/pass/matmul_allreduce_add_rmsnorm_fusion.h +0 -67
- mindspore/include/mindspore/ccsrc/backend/graph_compiler/backend.h +0 -124
- mindspore/include/mindspore/ccsrc/backend/graph_compiler/backend_base.h +0 -205
- mindspore/include/mindspore/ccsrc/backend/graph_compiler/ge_backend/ge_backend.h +0 -86
- mindspore/include/mindspore/ccsrc/debug/data_dump/data_dumper.h +0 -56
- mindspore/include/mindspore/ccsrc/debug/hooker/acl_data_adapter.h +0 -51
- mindspore/include/mindspore/ccsrc/debug/hooker/adapter.h +0 -75
- mindspore/include/mindspore/ccsrc/debug/hooker/hook_debugger.h +0 -55
- mindspore/include/mindspore/ccsrc/debug/hooker/hook_dynamic_loader.h +0 -52
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/prompt_k_v_cache_info.h +0 -59
- mindspore/include/mindspore/ccsrc/include/backend/debug/data_dump/overflow_dumper.h +0 -50
- mindspore/include/mindspore/ccsrc/include/backend/device_synchronizer.h +0 -49
- mindspore/include/mindspore/ccsrc/include/backend/distributed/rpc/rdma/constants.h +0 -174
- mindspore/include/mindspore/ccsrc/include/backend/distributed/rpc/rdma/rdma_client.h +0 -83
- mindspore/include/mindspore/ccsrc/include/backend/distributed/rpc/rdma/rdma_server.h +0 -71
- mindspore/include/mindspore/ccsrc/pipeline/jit/pi/utils/ptr_list_ref.h +0 -423
- mindspore/include/mindspore/ccsrc/pipeline/jit/ps/static_analysis/inplace_validation.h +0 -32
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/hal/common/ascend_utils.h +0 -43
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/hal/device/ascend_stream_assign.h +0 -255
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/ge/ge_kernel_mod.h +0 -78
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/ascend_native/boost_model_ascend_native.h +0 -51
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb_speed/base/model_creator.h +0 -27
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb_speed/log/log_core.h +0 -43
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb_speed/log/log_entity.h +0 -44
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb_speed/log/log_sink.h +0 -32
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb_speed/log/log_sink_file.h +0 -39
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/llm_boost/atb/include/atb_speed/log/log_stream.h +0 -51
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/optimizer/format_type/rectify_do_mask_kernel_info.h +0 -44
- mindspore/include/mindspore/ccsrc/plugin/device/cpu/kernel/custom/julia_api.h +0 -443
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/ascend_device_address/ascend_device_synchronizer.h +0 -45
- mindspore/include/mindspore/ccsrc/plugin/res_manager/cpu/cpu_device_address/cpu_device_synchronizer.h +0 -45
- mindspore/include/mindspore/ccsrc/plugin/res_manager/gpu/device/gpu_device_synchronizer.h +0 -44
- mindspore/include/mindspore/ccsrc/ps/core/communicator/ssl_http.h +0 -60
- mindspore/include/mindspore/ccsrc/pynative/grad/auto_grad.h +0 -77
- mindspore/include/mindspore/ccsrc/pynative/grad/ir/bprop_tensor_replace.h +0 -58
- mindspore/include/mindspore/ccsrc/pynative/grad/ir/dynamic_shape.h +0 -204
- mindspore/include/mindspore/ccsrc/pynative/grad/ir/ir_bprop.h +0 -163
- mindspore/include/mindspore/ccsrc/pynative/grad/ir/ir_grad.h +0 -114
- mindspore/include/mindspore/ccsrc/pynative/grad/ir/ir_pass.h +0 -71
- mindspore/include/mindspore/ccsrc/pynative/grad/variable.h +0 -466
- mindspore/include/mindspore/ccsrc/pynative/op_function/auto_generate/pyboost_functions.h +0 -1019
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/custom_actor.h +0 -66
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/inline_control_flow_scheduler.h +0 -81
- mindspore/include/mindspore/core/include/ir/base_tensor.h +0 -1073
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ragged_tensor_to_tensor.h +0 -120
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/cpu_kernel/ms_kernel/sample_distorted_bounding_box_ext2.h +0 -103
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/cpu_kernel/ms_kernel/scale_and_translate.h +0 -77
- mindspore/include/mindspore/ops/kernel/ascend/ascendc/op_host/decoder_kv_cache_tiling.h +0 -40
- mindspore/include/mindspore/ops/kernel/ascend/ascendc/op_host/prompt_kv_cache_tiling.h +0 -39
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/conv3d_padding_aclnn_kernel.h +0 -77
- mindspore/include/third_party/securec/src/secinput.h +0 -181
- mindspore/include/third_party/securec/src/securecutil.h +0 -574
- mindspore/include/third_party/securec/src/secureprintoutput.h +0 -153
- mindspore/lib/plugin/ascend/custom_ascendc_910/framework/npu_supported_ops.json +0 -10
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_api/include/aclnn_decoder_kv_cache.h +0 -59
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_api/include/aclnn_prompt_kv_cache.h +0 -59
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/config/ascend910/aic-ascend910-ops-info.json +0 -182
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/custom_ascendc_910_impl/dynamic/decoder_kv_cache.cpp +0 -192
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/custom_ascendc_910_impl/dynamic/decoder_kv_cache.py +0 -215
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/custom_ascendc_910_impl/dynamic/prompt_kv_cache.cpp +0 -274
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/custom_ascendc_910_impl/dynamic/prompt_kv_cache.py +0 -215
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/decoder_kv_cache/DecoderKvCache_0d5520cc587ad44ce634bf3fbcffc272.json +0 -158
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/decoder_kv_cache/DecoderKvCache_0d5520cc587ad44ce634bf3fbcffc272.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/decoder_kv_cache/DecoderKvCache_20390d30b3c4c0d23167ccca6c030c2b.json +0 -158
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/decoder_kv_cache/DecoderKvCache_20390d30b3c4c0d23167ccca6c030c2b.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/decoder_kv_cache/DecoderKvCache_2d151f0b1d2db51faa2968d5b67544e2.json +0 -158
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/decoder_kv_cache/DecoderKvCache_2d151f0b1d2db51faa2968d5b67544e2.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/decoder_kv_cache/DecoderKvCache_561690ec17cc1def3d2fcf68c1b07b56.json +0 -158
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/decoder_kv_cache/DecoderKvCache_561690ec17cc1def3d2fcf68c1b07b56.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/decoder_kv_cache/DecoderKvCache_570f9aaa99e5e773b3dd0a33784363f4.json +0 -158
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/decoder_kv_cache/DecoderKvCache_570f9aaa99e5e773b3dd0a33784363f4.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/decoder_kv_cache/DecoderKvCache_59668a0f0764afb98fda8ab9e84126f1.json +0 -158
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/decoder_kv_cache/DecoderKvCache_59668a0f0764afb98fda8ab9e84126f1.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/decoder_kv_cache/DecoderKvCache_91d9833e4792b70b670e4e2b916abd86.json +0 -158
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/decoder_kv_cache/DecoderKvCache_91d9833e4792b70b670e4e2b916abd86.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/decoder_kv_cache/DecoderKvCache_c74cdc5fef094383401856f8519504af.json +0 -158
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/decoder_kv_cache/DecoderKvCache_c74cdc5fef094383401856f8519504af.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/prompt_kv_cache/PromptKvCache_0515c7b1a4cd614449e38c5e9a7e3f8d.json +0 -167
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/prompt_kv_cache/PromptKvCache_0515c7b1a4cd614449e38c5e9a7e3f8d.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/prompt_kv_cache/PromptKvCache_09f22d898d6358c91e7c4fc48bac48e7.json +0 -167
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/prompt_kv_cache/PromptKvCache_09f22d898d6358c91e7c4fc48bac48e7.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/prompt_kv_cache/PromptKvCache_0cb9a6f894b925250227136e5aab7061.json +0 -167
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/prompt_kv_cache/PromptKvCache_0cb9a6f894b925250227136e5aab7061.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/prompt_kv_cache/PromptKvCache_2fa8702ffd7ca85e9e194f62644415d5.json +0 -167
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/prompt_kv_cache/PromptKvCache_2fa8702ffd7ca85e9e194f62644415d5.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/prompt_kv_cache/PromptKvCache_570b62f187dfd439b64613d881deedb7.json +0 -167
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/prompt_kv_cache/PromptKvCache_570b62f187dfd439b64613d881deedb7.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/prompt_kv_cache/PromptKvCache_585218c11411ff84709b9e725b66c435.json +0 -167
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/prompt_kv_cache/PromptKvCache_585218c11411ff84709b9e725b66c435.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/prompt_kv_cache/PromptKvCache_5c9365ccde170b358c5b126d69dae13e.json +0 -167
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/prompt_kv_cache/PromptKvCache_5c9365ccde170b358c5b126d69dae13e.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/prompt_kv_cache/PromptKvCache_6d97c45b7c43bc16fcff8baa5dacac4e.json +0 -167
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/ascend910/prompt_kv_cache/PromptKvCache_6d97c45b7c43bc16fcff8baa5dacac4e.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/config/ascend910/binary_info_config.json +0 -302
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/config/ascend910/decoder_kv_cache.json +0 -892
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/kernel/config/ascend910/prompt_kv_cache.json +0 -892
- 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/inc/op_proto.h +0 -33
- 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 +0 -1
- mindspore/lib/plugin/ascend/custom_ascendc_910b/framework/npu_supported_ops.json +0 -14
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_api/include/aclnn_decoder_kv_cache.h +0 -59
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_api/include/aclnn_prompt_kv_cache.h +0 -59
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/custom_ascendc_910b_impl/dynamic/decoder_kv_cache.cpp +0 -192
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/custom_ascendc_910b_impl/dynamic/decoder_kv_cache.py +0 -215
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/custom_ascendc_910b_impl/dynamic/prompt_kv_cache.cpp +0 -274
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/custom_ascendc_910b_impl/dynamic/prompt_kv_cache.py +0 -215
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_0d5520cc587ad44ce634bf3fbcffc272.json +0 -158
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_0d5520cc587ad44ce634bf3fbcffc272.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_20390d30b3c4c0d23167ccca6c030c2b.json +0 -158
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_20390d30b3c4c0d23167ccca6c030c2b.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_2d151f0b1d2db51faa2968d5b67544e2.json +0 -158
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_2d151f0b1d2db51faa2968d5b67544e2.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_561690ec17cc1def3d2fcf68c1b07b56.json +0 -158
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_561690ec17cc1def3d2fcf68c1b07b56.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_570f9aaa99e5e773b3dd0a33784363f4.json +0 -158
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_570f9aaa99e5e773b3dd0a33784363f4.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_59668a0f0764afb98fda8ab9e84126f1.json +0 -158
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_59668a0f0764afb98fda8ab9e84126f1.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_91d9833e4792b70b670e4e2b916abd86.json +0 -158
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_91d9833e4792b70b670e4e2b916abd86.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_c74cdc5fef094383401856f8519504af.json +0 -158
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_c74cdc5fef094383401856f8519504af.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_0515c7b1a4cd614449e38c5e9a7e3f8d.json +0 -167
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_0515c7b1a4cd614449e38c5e9a7e3f8d.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_09f22d898d6358c91e7c4fc48bac48e7.json +0 -167
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_09f22d898d6358c91e7c4fc48bac48e7.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_0cb9a6f894b925250227136e5aab7061.json +0 -167
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_0cb9a6f894b925250227136e5aab7061.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_2fa8702ffd7ca85e9e194f62644415d5.json +0 -167
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_2fa8702ffd7ca85e9e194f62644415d5.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_570b62f187dfd439b64613d881deedb7.json +0 -167
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_570b62f187dfd439b64613d881deedb7.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_585218c11411ff84709b9e725b66c435.json +0 -167
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_585218c11411ff84709b9e725b66c435.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_5c9365ccde170b358c5b126d69dae13e.json +0 -167
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_5c9365ccde170b358c5b126d69dae13e.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_6d97c45b7c43bc16fcff8baa5dacac4e.json +0 -167
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_6d97c45b7c43bc16fcff8baa5dacac4e.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_0d5520cc587ad44ce634bf3fbcffc272.json +0 -156
- 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 +0 -156
- 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 +0 -156
- 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 +0 -156
- 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 +0 -156
- 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 +0 -156
- 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 +0 -156
- 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 +0 -156
- 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 +0 -165
- 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 +0 -165
- 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 +0 -165
- 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 +0 -165
- 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 +0 -165
- 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 +0 -165
- 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 +0 -165
- 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 +0 -165
- 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/ascend910b/decoder_kv_cache/DecoderKvCache_0d5520cc587ad44ce634bf3fbcffc272.json +0 -156
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_0d5520cc587ad44ce634bf3fbcffc272.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_20390d30b3c4c0d23167ccca6c030c2b.json +0 -156
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_20390d30b3c4c0d23167ccca6c030c2b.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_2d151f0b1d2db51faa2968d5b67544e2.json +0 -156
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_2d151f0b1d2db51faa2968d5b67544e2.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_561690ec17cc1def3d2fcf68c1b07b56.json +0 -156
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_561690ec17cc1def3d2fcf68c1b07b56.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_570f9aaa99e5e773b3dd0a33784363f4.json +0 -156
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_570f9aaa99e5e773b3dd0a33784363f4.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_59668a0f0764afb98fda8ab9e84126f1.json +0 -156
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_59668a0f0764afb98fda8ab9e84126f1.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_91d9833e4792b70b670e4e2b916abd86.json +0 -156
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_91d9833e4792b70b670e4e2b916abd86.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_c74cdc5fef094383401856f8519504af.json +0 -156
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_c74cdc5fef094383401856f8519504af.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_0515c7b1a4cd614449e38c5e9a7e3f8d.json +0 -165
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_0515c7b1a4cd614449e38c5e9a7e3f8d.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_09f22d898d6358c91e7c4fc48bac48e7.json +0 -165
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_09f22d898d6358c91e7c4fc48bac48e7.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_0cb9a6f894b925250227136e5aab7061.json +0 -165
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_0cb9a6f894b925250227136e5aab7061.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_2fa8702ffd7ca85e9e194f62644415d5.json +0 -165
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_2fa8702ffd7ca85e9e194f62644415d5.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_570b62f187dfd439b64613d881deedb7.json +0 -165
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_570b62f187dfd439b64613d881deedb7.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_585218c11411ff84709b9e725b66c435.json +0 -165
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_585218c11411ff84709b9e725b66c435.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_5c9365ccde170b358c5b126d69dae13e.json +0 -165
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_5c9365ccde170b358c5b126d69dae13e.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_6d97c45b7c43bc16fcff8baa5dacac4e.json +0 -165
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_6d97c45b7c43bc16fcff8baa5dacac4e.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/config/ascend310p/decoder_kv_cache.json +0 -892
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/config/ascend310p/prompt_kv_cache.json +0 -892
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/config/ascend910_93/decoder_kv_cache.json +0 -892
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/config/ascend910_93/prompt_kv_cache.json +0 -892
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/config/ascend910b/decoder_kv_cache.json +0 -892
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/config/ascend910b/prompt_kv_cache.json +0 -892
- mindspore/lib/plugin/ascend/libms_ascend_native_boost.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/libms_optiling.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 +0 -163
- 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 +0 -163
- 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/so_kernels/libllama_ascend910b.so +0 -0
- mindspore/profiler/parser/__init__.py +0 -14
- mindspore/profiler/parser/aicpu_data_parser.py +0 -272
- mindspore/profiler/parser/ascend_analysis/__init__.py +0 -14
- mindspore/profiler/parser/ascend_analysis/constant.py +0 -71
- mindspore/profiler/parser/ascend_analysis/file_manager.py +0 -180
- mindspore/profiler/parser/ascend_analysis/function_event.py +0 -185
- mindspore/profiler/parser/ascend_analysis/fwk_cann_parser.py +0 -136
- mindspore/profiler/parser/ascend_analysis/fwk_file_parser.py +0 -131
- mindspore/profiler/parser/ascend_analysis/msprof_timeline_parser.py +0 -104
- mindspore/profiler/parser/ascend_analysis/path_manager.py +0 -313
- mindspore/profiler/parser/ascend_analysis/profiler_info_parser.py +0 -123
- mindspore/profiler/parser/ascend_analysis/tlv_decoder.py +0 -86
- mindspore/profiler/parser/ascend_analysis/trace_event_manager.py +0 -75
- mindspore/profiler/parser/ascend_cluster_generator.py +0 -116
- mindspore/profiler/parser/ascend_communicate_generator.py +0 -314
- mindspore/profiler/parser/ascend_flops_generator.py +0 -116
- mindspore/profiler/parser/ascend_fpbp_generator.py +0 -82
- mindspore/profiler/parser/ascend_hccl_generator.py +0 -271
- mindspore/profiler/parser/ascend_integrate_generator.py +0 -42
- mindspore/profiler/parser/ascend_memory_generator.py +0 -185
- mindspore/profiler/parser/ascend_msprof_exporter.py +0 -282
- mindspore/profiler/parser/ascend_msprof_generator.py +0 -187
- mindspore/profiler/parser/ascend_op_generator.py +0 -334
- mindspore/profiler/parser/ascend_steptrace_generator.py +0 -94
- mindspore/profiler/parser/ascend_timeline_generator.py +0 -545
- mindspore/profiler/parser/base_timeline_generator.py +0 -483
- mindspore/profiler/parser/container.py +0 -229
- mindspore/profiler/parser/cpu_gpu_timeline_generator.py +0 -697
- mindspore/profiler/parser/flops_parser.py +0 -531
- mindspore/profiler/parser/framework_enum.py +0 -111
- mindspore/profiler/parser/framework_parser.py +0 -464
- mindspore/profiler/parser/framework_struct.py +0 -61
- mindspore/profiler/parser/gpu_analysis/__init__.py +0 -14
- mindspore/profiler/parser/gpu_analysis/function_event.py +0 -44
- mindspore/profiler/parser/gpu_analysis/fwk_file_parser.py +0 -89
- mindspore/profiler/parser/gpu_analysis/profiler_info_parser.py +0 -72
- mindspore/profiler/parser/hccl_parser.py +0 -573
- mindspore/profiler/parser/hwts_log_parser.py +0 -122
- mindspore/profiler/parser/integrator.py +0 -526
- mindspore/profiler/parser/memory_usage_parser.py +0 -277
- mindspore/profiler/parser/minddata_analyzer.py +0 -800
- mindspore/profiler/parser/minddata_parser.py +0 -186
- mindspore/profiler/parser/minddata_pipeline_parser.py +0 -299
- mindspore/profiler/parser/op_intermediate_parser.py +0 -149
- mindspore/profiler/parser/optime_parser.py +0 -250
- mindspore/profiler/parser/profiler_info.py +0 -213
- mindspore/profiler/parser/step_trace_parser.py +0 -666
- /mindspore/include/mindspore/ops/kernel/ascend/ascendc/{op_host → all_finite/op_host}/all_finite_tiling.h +0 -0
- /mindspore/include/third_party/{securec/include → include}/securec.h +0 -0
- /mindspore/include/third_party/{securec/include → include}/securectype.h +0 -0
- {mindspore-2.6.0rc1.dist-info → mindspore-2.7.0rc1.dist-info}/WHEEL +0 -0
- {mindspore-2.6.0rc1.dist-info → mindspore-2.7.0rc1.dist-info}/entry_points.txt +0 -0
- {mindspore-2.6.0rc1.dist-info → mindspore-2.7.0rc1.dist-info}/top_level.txt +0 -0
|
@@ -17,122 +17,62 @@ from mindspore.common import dtype as mstype
|
|
|
17
17
|
from mindspore.ops.auto_generate.pyboost_inner_prim import *
|
|
18
18
|
|
|
19
19
|
|
|
20
|
-
def
|
|
20
|
+
def adaptive_avg_pool2d_grad(grad_output, x):
|
|
21
21
|
r"""
|
|
22
|
-
|
|
23
|
-
|
|
24
|
-
.. math::
|
|
25
|
-
y_i = \log_2(x_i)
|
|
26
|
-
|
|
27
|
-
.. warning::
|
|
28
|
-
- If the input value of operator Log2 is within the range (0, 0.01] or [0.95, 1.05], the output accuracy
|
|
29
|
-
may be affacted.
|
|
30
|
-
|
|
31
|
-
Args:
|
|
32
|
-
input (Tensor): Input Tensor of any dimension. The value must be greater than 0.
|
|
33
|
-
|
|
34
|
-
Returns:
|
|
35
|
-
Tensor, has the same shape as the `input`. If `input.dtype` is of integer or boolean type, the output dtype
|
|
36
|
-
will be float32. Otherwise, the output dtype will be the same as `input.dtype`.
|
|
37
|
-
|
|
38
|
-
Raises:
|
|
39
|
-
TypeError: If `input` is not a Tensor.
|
|
40
|
-
|
|
41
|
-
Supported Platforms:
|
|
42
|
-
``Ascend``
|
|
43
|
-
|
|
44
|
-
Examples:
|
|
45
|
-
>>> import mindspore
|
|
46
|
-
>>> import numpy as np
|
|
47
|
-
>>> from mindspore import Tensor, mint
|
|
48
|
-
>>> x = Tensor(np.array([3.0, 5.0, 7.0]), mindspore.float32)
|
|
49
|
-
>>> output = mint.log2(x)
|
|
50
|
-
>>> print(output)
|
|
51
|
-
[1.5849625 2.321928 2.807355 ]
|
|
22
|
+
None
|
|
52
23
|
"""
|
|
53
|
-
return
|
|
24
|
+
return adaptive_avg_pool2d_grad_impl(grad_output, x)
|
|
54
25
|
|
|
55
26
|
|
|
56
|
-
def
|
|
27
|
+
def asin(input):
|
|
57
28
|
r"""
|
|
58
|
-
Computes
|
|
59
|
-
of input tensors element-wise.
|
|
60
|
-
|
|
61
|
-
The formula is defined as follows:
|
|
29
|
+
Computes arcsine of input tensors element-wise.
|
|
62
30
|
|
|
63
31
|
.. math::
|
|
64
|
-
\text{mish}(input) = input * \tanh(softplus(\text{input}))
|
|
65
|
-
|
|
66
|
-
See more details in `A Self Regularized Non-Monotonic Neural Activation Function
|
|
67
|
-
<https://arxiv.org/abs/1908.08681>`_.
|
|
68
32
|
|
|
69
|
-
|
|
70
|
-
|
|
71
|
-
.. image:: ../images/Mish.png
|
|
72
|
-
:align: center
|
|
33
|
+
out_i = \sin^{-1}(input_i)
|
|
73
34
|
|
|
74
35
|
Args:
|
|
75
|
-
input (Tensor): The
|
|
76
|
-
|
|
77
|
-
- Ascend: float16, float32.
|
|
36
|
+
input (Tensor): The shape of tensor is
|
|
37
|
+
:math:`(N,*)`, where :math:`*` means any number of additional dimensions.
|
|
78
38
|
|
|
79
39
|
Returns:
|
|
80
|
-
Tensor, has the same
|
|
40
|
+
Tensor, has the same shape as `input`. The dtype of output is float32 when dtype of `input` is in [bool, int8, uint8, int16, int32, int64]. Otherwise output has the same dtype as `input`.
|
|
81
41
|
|
|
82
42
|
Raises:
|
|
83
43
|
TypeError: If `input` is not a Tensor.
|
|
84
|
-
TypeError: If dtype of `input` is not float16 or float32.
|
|
85
44
|
|
|
86
45
|
Supported Platforms:
|
|
87
|
-
``Ascend``
|
|
46
|
+
``Ascend`` ``GPU`` ``CPU``
|
|
88
47
|
|
|
89
48
|
Examples:
|
|
90
49
|
>>> import mindspore
|
|
91
|
-
>>> from mindspore import Tensor, ops
|
|
92
50
|
>>> import numpy as np
|
|
93
|
-
>>>
|
|
94
|
-
>>>
|
|
51
|
+
>>> from mindspore import Tensor, ops
|
|
52
|
+
>>> input = Tensor(np.array([0.74, 0.04, 0.30, 0.56]), mindspore.float32)
|
|
53
|
+
>>> output = ops.asin_ext(input)
|
|
95
54
|
>>> print(output)
|
|
96
|
-
[
|
|
97
|
-
[ 1.9439589e+00 -3.3576239e-02 8.9999990e+00]]
|
|
55
|
+
[0.8330927 0.04001068 0.30469266 0.59438497 ]
|
|
98
56
|
"""
|
|
99
|
-
return
|
|
57
|
+
return asin_impl(input)
|
|
100
58
|
|
|
101
59
|
|
|
102
|
-
def
|
|
60
|
+
def asinh(input):
|
|
103
61
|
r"""
|
|
104
|
-
|
|
105
|
-
|
|
106
|
-
Typically the input is of shape :math:`(N_{in}, C_{in}, L_{in})`, avg_pool1d outputs regional average in the
|
|
107
|
-
:math:`(L_{in})`-dimension. Given kernel size as :math:`ks = l_{ker}` and `stride` as :math:`s = s_0`, the
|
|
108
|
-
operation is as follows.
|
|
62
|
+
Computes inverse hyperbolic sine of the input element-wise.
|
|
109
63
|
|
|
110
64
|
.. math::
|
|
111
|
-
\text{output}(N_i, C_j, l) = \frac{1}{l_{ker}} \sum_{n=0}^{l_{ker}-1}
|
|
112
|
-
\text{input}(N_i, C_j, s_0 \times l + n)
|
|
113
65
|
|
|
114
|
-
|
|
115
|
-
This is an experimental API that is subject to change or deletion.
|
|
66
|
+
out_i = \sinh^{-1}(input_i)
|
|
116
67
|
|
|
117
68
|
Args:
|
|
118
|
-
input (Tensor):
|
|
119
|
-
kernel_size (Union(int, tuple[int])): The size of kernel window used to take the average value.
|
|
120
|
-
stride (Union(int, tuple[int]), optional): The distance of kernel moving. `stride` can either be an int
|
|
121
|
-
number or a tuple of one int number. Default: ``None``, the same value as `kernel_size`.
|
|
122
|
-
padding (Union(int, tuple[int]), optional): The pad length to be filled. `padding` can either be an integer
|
|
123
|
-
or a tuple of one integer. Default: ``0`` .
|
|
124
|
-
ceil_mode (bool, optional): If True, apply ceil instead of floor to compute the output shape. Default: ``False``.
|
|
125
|
-
count_include_pad (bool, optional): If True, include the zero-padding in the averaging calculation. Default: ``True`` .
|
|
69
|
+
input (Tensor): The input tensor of inverse hyperbolic sine function.
|
|
126
70
|
|
|
127
71
|
Returns:
|
|
128
|
-
Tensor
|
|
72
|
+
Tensor, has the same shape as `input`. The dtype of output is float32 when dtype of `input` is in [bool, int8, uint8, int16, int32, int64]. Otherwise output has the same dtype as `input`.
|
|
129
73
|
|
|
130
74
|
Raises:
|
|
131
75
|
TypeError: If `input` is not a Tensor.
|
|
132
|
-
TypeError: If `kernel_size` or `stride` is not an int.
|
|
133
|
-
TypeError: If `ceil_mode` or `count_include_pad` is not a bool.
|
|
134
|
-
ValueError: If `kernel_size` or `stride` is less than `1`.
|
|
135
|
-
ValueError: If `kernel_size` or `stride` or `padding` is not int nor a tuple whose length is greater than `1`.
|
|
136
76
|
|
|
137
77
|
Supported Platforms:
|
|
138
78
|
``Ascend`` ``GPU`` ``CPU``
|
|
@@ -140,260 +80,268 @@ def avg_pool1d(input, kernel_size, stride=None, padding=0, ceil_mode=False, coun
|
|
|
140
80
|
Examples:
|
|
141
81
|
>>> import mindspore
|
|
142
82
|
>>> import numpy as np
|
|
143
|
-
>>> from mindspore import Tensor,
|
|
144
|
-
>>>
|
|
145
|
-
>>> output =
|
|
146
|
-
>>> print(output
|
|
147
|
-
|
|
148
|
-
"""
|
|
149
|
-
return avg_pool1d_impl(input, kernel_size, stride, padding, ceil_mode, count_include_pad)
|
|
150
|
-
|
|
151
|
-
|
|
152
|
-
def inplace_add(input, other, alpha=1):
|
|
153
|
-
r"""
|
|
154
|
-
None
|
|
83
|
+
>>> from mindspore import Tensor, ops
|
|
84
|
+
>>> input = Tensor(np.array([-5.0, 1.5, 3.0, 100.0]), mindspore.float32)
|
|
85
|
+
>>> output = ops.asinh_ext(input)
|
|
86
|
+
>>> print(output)
|
|
87
|
+
[-2.3124385 1.1947632 1.8184465 5.298342 ]
|
|
155
88
|
"""
|
|
156
|
-
return
|
|
89
|
+
return asinh_impl(input)
|
|
157
90
|
|
|
158
91
|
|
|
159
|
-
def
|
|
92
|
+
def logsumexp(input, dim, keepdim=False):
|
|
160
93
|
r"""
|
|
161
|
-
|
|
162
|
-
|
|
163
|
-
Consider a batched input tensor of shape :math:`(N, C, *)`,
|
|
164
|
-
where :math:`N` is the batch dimension, :math:`C` is the channel dimension,
|
|
165
|
-
and :math:`*` represent arbitrary spatial dimensions. This operation flattens
|
|
166
|
-
each sliding `Kernel_size`- sized block within the spatial dimensions
|
|
167
|
-
of `input` into a column (i.e., last dimension) of a 3-D output
|
|
168
|
-
tensor of shape :math:`(N, C \times \prod(\text{kernel_size}), L)`, where
|
|
169
|
-
:math:`C \times \prod(\text{kernel_size})` is the total number of values
|
|
170
|
-
within each block (a block has :math:`\prod(\text{kernel_size})` spatial
|
|
171
|
-
locations each containing a `C`-channeled vector), and :math:`L` is
|
|
172
|
-
the total number of such blocks:
|
|
94
|
+
Computes the logarithm of the sum of exponentiations of all elements along the specified `dim` dimension of the `input` (with numerical stabilization), and retains the dimension based on the `keepdim` parameter.
|
|
173
95
|
|
|
174
96
|
.. math::
|
|
175
|
-
L = \prod_d \left\lfloor\frac{\text{spatial_size}[d] + 2 \times \text{padding}[d] %
|
|
176
|
-
- \text{dilation}[d] \times (\text{kernel_size}[d] - 1) - 1}{\text{stride}[d]} + 1\right\rfloor,
|
|
177
|
-
|
|
178
|
-
where :math:`\text{spatial_size}` is formed by the spatial dimensions
|
|
179
|
-
of `input` (:math:`*` above), and :math:`d` is over all spatial
|
|
180
|
-
dimensions.
|
|
181
97
|
|
|
182
|
-
|
|
183
|
-
gives all values within a certain block.
|
|
184
|
-
|
|
185
|
-
The `dilation`, `padding` and `stride` arguments specify
|
|
186
|
-
how the sliding blocks are retrieved.
|
|
98
|
+
logsumexp(input) = \log(\sum(e^{input-input_{max}})) + input_{max}
|
|
187
99
|
|
|
188
100
|
.. warning::
|
|
189
|
-
|
|
190
|
-
- For Ascend, it is only supported on platforms above Atlas A2.
|
|
101
|
+
This is an experimental API that is subject to change or deletion.
|
|
191
102
|
|
|
192
103
|
Args:
|
|
193
|
-
input (Tensor):
|
|
194
|
-
|
|
195
|
-
|
|
196
|
-
dilation (Union[int, tuple[int], list[int]], optional): The dilation of the window, should be two int
|
|
197
|
-
for height and width. If type is int, it means that height equal with width. Default: ``1`` .
|
|
198
|
-
padding (Union[int, tuple[int], list[int]], optional): The pad of the window, should be two int
|
|
199
|
-
for height and width. If type is int, it means that height equal with width. Default: ``0`` .
|
|
200
|
-
stride (Union[int, tuple[int], list[int]], optional): The stride of the window, should be two int
|
|
201
|
-
for height and width. If type is int, it means that height equal with width. Default: ``1`` .
|
|
104
|
+
input (Tensor): Input Tensor.
|
|
105
|
+
dim (Union[int, tuple(int), list(int)], optional): The dimension to be reduced (the value should be within `[0, len(input.shape) - 1]`), when the `dim` is `()`, all dimensions are reduced.
|
|
106
|
+
keepdim (bool, optional): Whether the output tensor retains the dimension `dim`, default: `False`.
|
|
202
107
|
|
|
203
108
|
Returns:
|
|
204
|
-
|
|
109
|
+
Tensor, the dtype changes according to the `input.dtype`, and the shape changes according to the values of `dim` and `keepdim`.
|
|
205
110
|
|
|
206
|
-
|
|
207
|
-
-
|
|
208
|
-
-
|
|
111
|
+
- If `input.dtype` is in [float16, float32, bfloat16], the output dtype is the same as the `input.dtype`.
|
|
112
|
+
- If `input.dtype` is an integer or boolean type, the output dtype is float32.
|
|
113
|
+
- If `dim` is (), and `keepdim` is False, the output is a 0-D tensor representing the logarithm of the sum of exponentiations of all elements in the `input` tensor.
|
|
114
|
+
- If `dim` is `1`, and `keepdim` is False, the shape of output is :math:`(input.shape[0], input.shape[2], ..., input.shape[n])`.
|
|
115
|
+
- If `dim` is `(1, 2)`, and `keepdim` is False, the shape of output is :math:`(input.shape[0], input.shape[3], ..., input.shape[n])`.
|
|
209
116
|
|
|
210
117
|
Raises:
|
|
211
|
-
TypeError: If
|
|
212
|
-
|
|
213
|
-
|
|
214
|
-
|
|
118
|
+
TypeError: If `input` is not a Tensor.
|
|
119
|
+
TypeError: If dtype of `input` is not one of: bool, int8, int16, int32, int64, uint8, float16, float32, bfloat16.
|
|
120
|
+
TypeError: If `dim` is not an int or tuple(int) or list(list).
|
|
121
|
+
TypeError: If `keepdim` is not a bool.
|
|
122
|
+
ValueError: If the value of any elements of `dim` is not in the range `[0, len(input.shape) - 1]`.
|
|
123
|
+
RuntimeError: If any element of `dim` is repeated.
|
|
215
124
|
|
|
216
125
|
Supported Platforms:
|
|
217
126
|
``Ascend``
|
|
218
127
|
|
|
219
128
|
Examples:
|
|
220
|
-
>>> import mindspore
|
|
221
129
|
>>> import numpy as np
|
|
222
130
|
>>> from mindspore import Tensor, ops
|
|
223
|
-
>>> x = Tensor(np.random.
|
|
224
|
-
>>> output = ops.auto_generate.
|
|
131
|
+
>>> x = Tensor(np.random.randn(3, 4, 5, 6).astype(np.float32))
|
|
132
|
+
>>> output = ops.auto_generate.logsumexp_ext(x, 1, keepdim=True)
|
|
225
133
|
>>> print(output.shape)
|
|
226
|
-
(
|
|
134
|
+
(3, 1, 5, 6)
|
|
227
135
|
"""
|
|
228
|
-
return
|
|
136
|
+
return logsumexp_impl(input, dim, keepdim)
|
|
229
137
|
|
|
230
138
|
|
|
231
|
-
def
|
|
139
|
+
def cummin(input, dim):
|
|
232
140
|
r"""
|
|
233
|
-
|
|
234
|
-
|
|
235
|
-
The activation function is defined as:
|
|
141
|
+
Returns a tuple (values, indices) where `values` is the cumulative minimum value of input Tensor `input`
|
|
142
|
+
along the dimension `dim`, and `indices` is the index location of each minimum value.
|
|
236
143
|
|
|
237
144
|
.. math::
|
|
238
|
-
|
|
239
|
-
|
|
240
|
-
\
|
|
241
|
-
x_{i}, &\text{if } x_{i} \geq 0; \cr
|
|
242
|
-
\text{alpha} * (\exp(x_i) - 1), &\text{otherwise.}
|
|
243
|
-
\end{cases}
|
|
244
|
-
|
|
245
|
-
where :math:`alpha` and :math:`scale` are pre-defined constants(:math:`alpha=1.67326324`
|
|
246
|
-
and :math:`scale=1.05070098`).
|
|
247
|
-
|
|
248
|
-
See more details in `Self-Normalizing Neural Networks <https://arxiv.org/abs/1706.02515>`_.
|
|
249
|
-
|
|
250
|
-
SELU Activation Function Graph:
|
|
145
|
+
\begin{array}{ll} \\
|
|
146
|
+
y_{i} = \min(x_{1}, x_{2}, ... , x_{i})
|
|
147
|
+
\end{array}
|
|
251
148
|
|
|
252
|
-
..
|
|
253
|
-
|
|
149
|
+
.. note::
|
|
150
|
+
GE backend is not supported in Ascend.
|
|
254
151
|
|
|
255
152
|
Args:
|
|
256
|
-
input (Tensor): Tensor
|
|
257
|
-
|
|
153
|
+
input (Tensor): The input Tensor, The dimension must be greater than 0.
|
|
154
|
+
dim (int): Operation dimension. The value of `dim` must be in the range `[-input.ndim, input.ndim - 1]`.
|
|
258
155
|
|
|
259
156
|
Returns:
|
|
260
|
-
Tensor,
|
|
157
|
+
tuple [Tensor], tuple of 2 Tensors, containing the cumulative minimum of elements and the index.
|
|
158
|
+
The shape of each output tensor is the same as that of input `input`.
|
|
261
159
|
|
|
262
160
|
Raises:
|
|
263
|
-
TypeError: If
|
|
161
|
+
TypeError: If `input` is not a Tensor.
|
|
162
|
+
TypeError: If `input` is a Tensor, but the type is complex or bool.
|
|
163
|
+
TypeError: If `dim` is not an int.
|
|
164
|
+
ValueError: If `dim` is out the range of `[-input.ndim, input.ndim - 1]`.
|
|
264
165
|
|
|
265
166
|
Supported Platforms:
|
|
266
167
|
``Ascend``
|
|
267
168
|
|
|
268
169
|
Examples:
|
|
170
|
+
>>> from mindspore import Tensor, ops
|
|
269
171
|
>>> import mindspore
|
|
270
|
-
>>>
|
|
271
|
-
>>>
|
|
272
|
-
>>>
|
|
273
|
-
|
|
274
|
-
>>> print(output)
|
|
275
|
-
[
|
|
276
|
-
[ 2.101402 -1.7462534 9.456309 ]]
|
|
172
|
+
>>> a = Tensor([-0.2284, -0.6628, 0.0975, 0.2680, -1.3298, -0.4220], mindspore.float32)
|
|
173
|
+
>>> output = ops.cummin_ext(a, dim=0)
|
|
174
|
+
>>> print(output[0])
|
|
175
|
+
[-0.2284 -0.6628 -0.6628 -0.6628 -1.3298 -1.3298]
|
|
176
|
+
>>> print(output[1])
|
|
177
|
+
[0 1 1 1 4 4]
|
|
277
178
|
"""
|
|
278
|
-
return
|
|
179
|
+
return cummin_impl(input, dim)
|
|
279
180
|
|
|
280
181
|
|
|
281
|
-
def
|
|
182
|
+
def mean(input, dim=None, keepdim=False, dtype=None):
|
|
282
183
|
r"""
|
|
283
|
-
|
|
184
|
+
Reduces all dimension of a tensor by averaging all elements in the dimension, by default.
|
|
185
|
+
And reduce a dimension of `input` along the specified `dim`. `keepdim`
|
|
186
|
+
determines whether the dimensions of the output and input are the same.
|
|
284
187
|
|
|
285
|
-
|
|
286
|
-
|
|
188
|
+
Note:
|
|
189
|
+
The `dim` with tensor type is only used for compatibility with older versions and is not recommended.
|
|
287
190
|
|
|
288
191
|
Args:
|
|
289
|
-
input (Tensor): The input tensor.
|
|
192
|
+
input (Tensor[Number]): The input tensor. The dtype of the tensor to be reduced is number.
|
|
193
|
+
:math:`(N, *)` where :math:`*` means, any number of additional dimensions.
|
|
194
|
+
dim (Union[int, tuple(int), list(int), Tensor]): The dimensions to reduce. Default: ``None`` ,
|
|
195
|
+
reduce all dimensions. Only constant value is allowed. Assume the rank of `input` is r,
|
|
196
|
+
and the value range is [-r,r).
|
|
197
|
+
keepdim (bool): If ``True`` , keep these reduced dimensions and the length is 1.
|
|
198
|
+
If ``False`` , don't keep these dimensions. Default: ``False`` .
|
|
199
|
+
dtype (:class:`mindspore.dtype`): The desired data type of returned Tensor. Default: ``None`` .
|
|
290
200
|
|
|
291
201
|
Returns:
|
|
292
|
-
Tensor,
|
|
202
|
+
Tensor, has the same data type as input tensor.
|
|
203
|
+
|
|
204
|
+
- If `dim` is ``None`` , and `keepdim` is ``False`` ,
|
|
205
|
+
the output is a 0-D tensor representing the product of all elements in the input tensor.
|
|
206
|
+
- If `dim` is int, set as 1, and `keepdim` is ``False`` ,
|
|
207
|
+
the shape of output is :math:`(x_0, x_2, ..., x_R)`.
|
|
208
|
+
- If `dim` is tuple(int), set as (1, 2), and `keepdim` is ``False`` ,
|
|
209
|
+
the shape of output is :math:`(x_0, x_3, ..., x_R)`.
|
|
210
|
+
- If `dim` is 1-D Tensor, set as [1, 2], and `keepdim` is ``False`` ,
|
|
211
|
+
the shape of output is :math:`(x_0, x_3, ..., x_R)`.
|
|
293
212
|
|
|
294
213
|
Raises:
|
|
295
|
-
|
|
296
|
-
|
|
297
|
-
TypeError: If `
|
|
214
|
+
TypeError: If `x` is not a Tensor.
|
|
215
|
+
TypeError: If `dim` is not one of the following: int, tuple, list or Tensor.
|
|
216
|
+
TypeError: If `keepdim` is not a bool.
|
|
217
|
+
ValueError: If `dim` is out of range.
|
|
298
218
|
|
|
299
219
|
Supported Platforms:
|
|
300
|
-
``Ascend``
|
|
220
|
+
``Ascend`` ``GPU`` ``CPU``
|
|
301
221
|
|
|
302
222
|
Examples:
|
|
303
223
|
>>> import mindspore
|
|
304
224
|
>>> import numpy as np
|
|
305
225
|
>>> from mindspore import Tensor, ops
|
|
306
|
-
>>>
|
|
307
|
-
>>> output = ops.
|
|
226
|
+
>>> x = Tensor(np.random.randn(3, 4, 5, 6).astype(np.float32))
|
|
227
|
+
>>> output = ops.mean_ext(x, 1, keepdim=True)
|
|
228
|
+
>>> result = output.shape
|
|
229
|
+
>>> print(result)
|
|
230
|
+
(3, 1, 5, 6)
|
|
231
|
+
>>> # case 1: Reduces a dimension by averaging all elements in the dimension.
|
|
232
|
+
>>> x = Tensor(np.array([[[2, 2, 2, 2, 2, 2], [2, 2, 2, 2, 2, 2], [2, 2, 2, 2, 2, 2]],
|
|
233
|
+
... [[4, 4, 4, 4, 4, 4], [5, 5, 5, 5, 5, 5], [6, 6, 6, 6, 6, 6]],
|
|
234
|
+
... [[6, 6, 6, 6, 6, 6], [8, 8, 8, 8, 8, 8], [10, 10, 10, 10, 10, 10]]]),
|
|
235
|
+
... mindspore.float32)
|
|
236
|
+
>>> output = ops.mean_ext(x)
|
|
308
237
|
>>> print(output)
|
|
309
|
-
|
|
310
|
-
|
|
311
|
-
|
|
238
|
+
5.0
|
|
239
|
+
>>> print(output.shape)
|
|
240
|
+
()
|
|
241
|
+
>>> # case 2: Reduces a dimension along the dim 0
|
|
242
|
+
>>> output = ops.mean_ext(x, 0, True)
|
|
243
|
+
>>> print(output)
|
|
244
|
+
[[[4. 4. 4. 4. 4. 4.]
|
|
245
|
+
[5. 5. 5. 5. 5. 5.]
|
|
246
|
+
[6. 6. 6. 6. 6. 6.]]]
|
|
247
|
+
>>> # case 3: Reduces a dimension along the dim 1
|
|
248
|
+
>>> output = ops.mean_ext(x, 1, True)
|
|
249
|
+
>>> print(output)
|
|
250
|
+
[[[2. 2. 2. 2. 2. 2.]]
|
|
251
|
+
[[5. 5. 5. 5. 5. 5.]]
|
|
252
|
+
[[8. 8. 8. 8. 8. 8.]]]
|
|
253
|
+
>>> # case 4: Reduces a dimension along the dim 2
|
|
254
|
+
>>> output = ops.mean_ext(x, 2, True)
|
|
255
|
+
>>> print(output)
|
|
256
|
+
[[[ 2.]
|
|
257
|
+
[ 2.]
|
|
258
|
+
[ 2.]]
|
|
259
|
+
[[ 4.]
|
|
260
|
+
[ 5.]
|
|
261
|
+
[ 6.]]
|
|
262
|
+
[[ 6.]
|
|
263
|
+
[ 8.]
|
|
264
|
+
[10.]]]
|
|
312
265
|
"""
|
|
313
|
-
return
|
|
266
|
+
return mean_impl(input, dim, keepdim, dtype)
|
|
314
267
|
|
|
315
268
|
|
|
316
|
-
def
|
|
269
|
+
def cumsum(input, dim, dtype=None):
|
|
317
270
|
r"""
|
|
318
|
-
|
|
271
|
+
Computes the cumulative sum of input Tensor along `dim`.
|
|
319
272
|
|
|
320
|
-
..
|
|
321
|
-
|
|
273
|
+
.. math::
|
|
274
|
+
|
|
275
|
+
y_i = x_1 + x_2 + x_3 + ... + x_i
|
|
322
276
|
|
|
323
277
|
Args:
|
|
324
|
-
input(Tensor): The input
|
|
325
|
-
dim (int
|
|
326
|
-
|
|
327
|
-
|
|
328
|
-
|
|
329
|
-
stable (bool, optional): Whether to use stable sorting algorithm. Default: ``False``.
|
|
278
|
+
input (Tensor): The input Tensor.
|
|
279
|
+
dim (int): Dim along which the cumulative sum is computed.
|
|
280
|
+
dtype (:class:`mindspore.dtype`, optional): The desired dtype of returned Tensor. If specified,
|
|
281
|
+
the input Tensor will be cast to `dtype` before the computation. This is useful for preventing overflows.
|
|
282
|
+
If not specified, stay the same as original Tensor. Default: ``None`` .
|
|
330
283
|
|
|
331
284
|
Returns:
|
|
332
|
-
Tensor, the
|
|
285
|
+
Tensor, the shape of the output Tensor is consistent with the input Tensor's.
|
|
333
286
|
|
|
334
287
|
Raises:
|
|
335
|
-
|
|
336
|
-
|
|
337
|
-
TypeError: If dtype of `descending` is not bool.
|
|
338
|
-
TypeError: If dtype of `stable` is not bool.
|
|
288
|
+
TypeError: If `input` is not a Tensor.
|
|
289
|
+
ValueError: If the `dim` is out of range.
|
|
339
290
|
|
|
340
291
|
Supported Platforms:
|
|
341
292
|
``Ascend``
|
|
342
293
|
|
|
343
294
|
Examples:
|
|
344
|
-
>>> import mindspore
|
|
345
295
|
>>> import numpy as np
|
|
346
296
|
>>> from mindspore import Tensor
|
|
347
|
-
>>> import mindspore.
|
|
348
|
-
>>> x = Tensor(np.array([[
|
|
349
|
-
>>>
|
|
350
|
-
>>>
|
|
351
|
-
|
|
352
|
-
[
|
|
353
|
-
[
|
|
297
|
+
>>> import mindspore.ops as ops
|
|
298
|
+
>>> x = Tensor(np.array([[3, 4, 6, 10], [1, 6, 7, 9], [4, 3, 8, 7], [1, 3, 7, 9]]).astype(np.float32))
|
|
299
|
+
>>> # case 1: along the dim 0
|
|
300
|
+
>>> y = ops.auto_generate.cumsum_ext(x, 0)
|
|
301
|
+
>>> print(y)
|
|
302
|
+
[[ 3. 4. 6. 10.]
|
|
303
|
+
[ 4. 10. 13. 19.]
|
|
304
|
+
[ 8. 13. 21. 26.]
|
|
305
|
+
[ 9. 16. 28. 35.]]
|
|
306
|
+
>>> # case 2: along the dim 1
|
|
307
|
+
>>> y = ops.auto_generate.cumsum_ext(x, 1)
|
|
308
|
+
>>> print(y)
|
|
309
|
+
[[ 3. 7. 13. 23.]
|
|
310
|
+
[ 1. 7. 14. 23.]
|
|
311
|
+
[ 4. 7. 15. 22.]
|
|
312
|
+
[ 1. 4. 11. 20.]]
|
|
354
313
|
"""
|
|
355
|
-
return
|
|
314
|
+
return cumsum_impl(input, dim, dtype)
|
|
356
315
|
|
|
357
316
|
|
|
358
|
-
def
|
|
317
|
+
def isneginf(input):
|
|
359
318
|
r"""
|
|
360
|
-
|
|
361
|
-
This function is useful in statistics where the calculated probabilities of events may be
|
|
362
|
-
so small as to exceed the range of normal floating point numbers.
|
|
363
|
-
|
|
364
|
-
.. math::
|
|
365
|
-
|
|
366
|
-
out_i = \log(exp(input_i) + \exp(other_i))
|
|
319
|
+
Determines which elements are -inf for each position.
|
|
367
320
|
|
|
368
321
|
.. warning::
|
|
369
|
-
This
|
|
322
|
+
- This API can be used only on the Atlas A2 training series.
|
|
370
323
|
|
|
371
324
|
Args:
|
|
372
|
-
input (Tensor): Input Tensor.
|
|
373
|
-
other (Tensor): Input Tensor. The dtype of `other` must be float.
|
|
374
|
-
If the shape of `input` is not equal to the shape of `other`,
|
|
375
|
-
they must be broadcastable to a common shape.
|
|
325
|
+
input (Tensor): Input Tensor.
|
|
376
326
|
|
|
377
327
|
Returns:
|
|
378
|
-
Tensor
|
|
328
|
+
Tensor with the same shape as the input, where elements are `True` if the corresponding element in the `input` is negative infinity, and `False` otherwise.
|
|
379
329
|
|
|
380
330
|
Raises:
|
|
381
|
-
TypeError: If
|
|
382
|
-
TypeError: The dtype of `input` or `other` is not float.
|
|
331
|
+
TypeError: If the input is not a tensor.
|
|
383
332
|
|
|
384
333
|
Supported Platforms:
|
|
385
|
-
``Ascend``
|
|
334
|
+
``Ascend`` ``GPU`` ``CPU``
|
|
386
335
|
|
|
387
336
|
Examples:
|
|
388
|
-
>>> import
|
|
389
|
-
>>> from mindspore import
|
|
390
|
-
>>>
|
|
391
|
-
>>> x2 = Tensor(np.array(2).astype(np.float16))
|
|
392
|
-
>>> output = ops.logaddexp_ext(x1, x2)
|
|
337
|
+
>>> from mindspore import ops, Tensor
|
|
338
|
+
>>> from mindspore import dtype as mstype
|
|
339
|
+
>>> output = ops.isneginf(Tensor([[-float("inf"), float("inf")], [1, -float("inf")]], mstype.float32))
|
|
393
340
|
>>> print(output)
|
|
394
|
-
[
|
|
341
|
+
[[ True False]
|
|
342
|
+
[False True]]
|
|
395
343
|
"""
|
|
396
|
-
return
|
|
344
|
+
return isneginf_impl(input)
|
|
397
345
|
|
|
398
346
|
|
|
399
347
|
def atan2(input, other):
|
|
@@ -432,227 +380,178 @@ def atan2(input, other):
|
|
|
432
380
|
>>> from mindspore import Tensor, ops
|
|
433
381
|
>>> input = Tensor(np.array([0, 1]), mindspore.float32)
|
|
434
382
|
>>> other = Tensor(np.array([1, 1]), mindspore.float32)
|
|
435
|
-
>>> output =
|
|
383
|
+
>>> output = ops.auto_generate.atan2_ext(input, other)
|
|
436
384
|
>>> print(output)
|
|
437
385
|
[0. 0.7853982]
|
|
438
386
|
"""
|
|
439
387
|
return atan2_impl(input, other)
|
|
440
388
|
|
|
441
389
|
|
|
442
|
-
def
|
|
443
|
-
r"""
|
|
444
|
-
Return outer product of `input` and `vec2`. If `input` is a vector of size :math:`n`
|
|
445
|
-
and `vec2` is a vector of size :math:`m` , then output must be a matrix of shape :math:`(n, m)` .
|
|
446
|
-
|
|
447
|
-
.. warning::
|
|
448
|
-
This is an experimental API that is subject to change or deletion.
|
|
449
|
-
|
|
450
|
-
.. note::
|
|
451
|
-
This function does not broadcast.
|
|
452
|
-
|
|
453
|
-
Args:
|
|
454
|
-
input (Tensor): 1-D input vector.
|
|
455
|
-
vec2 (Tensor): 1-D input vector.
|
|
456
|
-
|
|
457
|
-
Returns:
|
|
458
|
-
out, 2-D matrix, the outer product of two vectors.
|
|
459
|
-
|
|
460
|
-
Raises:
|
|
461
|
-
TypeError: If `input` or `vec2` is not a Tensor.
|
|
462
|
-
TypeError: The implicitly converted data types of `input` and `vec2` are not one of float16, float32, float64, bool, uint8, int8, int16, int32, int64, complex64, complex128, bfloat16
|
|
463
|
-
ValueError: If the dimension of `input` or `vec2` is not equal to 1.
|
|
464
|
-
|
|
465
|
-
Supported Platforms:
|
|
466
|
-
``Ascend``
|
|
467
|
-
|
|
468
|
-
Examples:
|
|
469
|
-
>>> import mindspore
|
|
470
|
-
>>> import numpy as np
|
|
471
|
-
>>> from mindspore import Tensor
|
|
472
|
-
>>> from mindspore import ops
|
|
473
|
-
>>> input = Tensor(np.array([7, 8, 9]), mindspore.int32)
|
|
474
|
-
>>> vec2 = Tensor(np.array([7, 10, 11]), mindspore.int32)
|
|
475
|
-
>>> out = ops.outer(input, vec2)
|
|
476
|
-
>>> print(out)
|
|
477
|
-
[[49 70 77]
|
|
478
|
-
[56 80 88]
|
|
479
|
-
[63 90 99]]
|
|
480
|
-
"""
|
|
481
|
-
return outer_impl(input, vec2)
|
|
482
|
-
|
|
483
|
-
|
|
484
|
-
def fold(input, output_size, kernel_size, dilation=1, padding=0, stride=1):
|
|
390
|
+
def unfold(input, kernel_size, dilation=1, padding=0, stride=1):
|
|
485
391
|
r"""
|
|
486
|
-
|
|
392
|
+
Extracts sliding local blocks from a batched input tensor.
|
|
487
393
|
|
|
488
|
-
Consider a batched input tensor of shape :math:`(N, C
|
|
489
|
-
where :math:`N` is the batch dimension, :math:`C
|
|
490
|
-
|
|
491
|
-
|
|
394
|
+
Consider a batched input tensor of shape :math:`(N, C, *)`,
|
|
395
|
+
where :math:`N` is the batch dimension, :math:`C` is the channel dimension,
|
|
396
|
+
and :math:`*` represent arbitrary spatial dimensions. This operation flattens
|
|
397
|
+
each sliding `Kernel_size`- sized block within the spatial dimensions
|
|
398
|
+
of `input` into a column (i.e., last dimension) of a 3-D output
|
|
399
|
+
tensor of shape :math:`(N, C \times \prod(\text{kernel_size}), L)`, where
|
|
400
|
+
:math:`C \times \prod(\text{kernel_size})` is the total number of values
|
|
401
|
+
within each block (a block has :math:`\prod(\text{kernel_size})` spatial
|
|
402
|
+
locations each containing a `C`-channeled vector), and :math:`L` is
|
|
403
|
+
the total number of such blocks:
|
|
492
404
|
|
|
493
405
|
.. math::
|
|
494
|
-
L = \prod_d \left\lfloor\frac{\text{
|
|
406
|
+
L = \prod_d \left\lfloor\frac{\text{spatial_size}[d] + 2 \times \text{padding}[d] %
|
|
495
407
|
- \text{dilation}[d] \times (\text{kernel_size}[d] - 1) - 1}{\text{stride}[d]} + 1\right\rfloor,
|
|
496
408
|
|
|
497
|
-
where :math
|
|
409
|
+
where :math:`\text{spatial_size}` is formed by the spatial dimensions
|
|
410
|
+
of `input` (:math:`*` above), and :math:`d` is over all spatial
|
|
411
|
+
dimensions.
|
|
498
412
|
|
|
499
|
-
Therefore, `
|
|
413
|
+
Therefore, indexing `output` at the last dimension (column dimension)
|
|
414
|
+
gives all values within a certain block.
|
|
500
415
|
|
|
501
|
-
The `dilation`, `padding` and `stride` arguments specify
|
|
416
|
+
The `dilation`, `padding` and `stride` arguments specify
|
|
417
|
+
how the sliding blocks are retrieved.
|
|
502
418
|
|
|
503
419
|
.. warning::
|
|
504
|
-
Currently,
|
|
420
|
+
- Currently, batched(4D) image-like tensors are supported.
|
|
421
|
+
- For Ascend, it is only supported on platforms above Atlas A2.
|
|
505
422
|
|
|
506
423
|
Args:
|
|
507
|
-
input (Tensor):
|
|
508
|
-
output_size (Union[int, tuple[int], list[int]]): The shape of the spatial dimensions of
|
|
509
|
-
the output(i.e., output.shape[2:]).
|
|
424
|
+
input (Tensor): 4-D Tensor.
|
|
510
425
|
kernel_size (Union[int, tuple[int], list[int]]): The size of the kernel, should be two int
|
|
511
426
|
for height and width. If type is int, it means that height equal with width. Must be specified.
|
|
512
|
-
dilation (Union[int, tuple[int], list[int]], optional): The
|
|
427
|
+
dilation (Union[int, tuple[int], list[int]], optional): The dilation of the window, should be two int
|
|
513
428
|
for height and width. If type is int, it means that height equal with width. Default: ``1`` .
|
|
514
|
-
padding (Union[int, tuple[int], list[int]], optional): The
|
|
429
|
+
padding (Union[int, tuple[int], list[int]], optional): The pad of the window, should be two int
|
|
515
430
|
for height and width. If type is int, it means that height equal with width. Default: ``0`` .
|
|
516
|
-
stride (Union[int, tuple[int], list[int]], optional): The
|
|
431
|
+
stride (Union[int, tuple[int], list[int]], optional): The stride of the window, should be two int
|
|
517
432
|
for height and width. If type is int, it means that height equal with width. Default: ``1`` .
|
|
518
433
|
|
|
519
434
|
Returns:
|
|
520
435
|
A Tensor, with same type as `input` .
|
|
521
436
|
|
|
522
437
|
Shape:
|
|
523
|
-
- Input: :math:`(N, C
|
|
524
|
-
|
|
525
|
-
- Output: :math:`(N, C, output\_size[0], output\_size[1], ...)` or
|
|
526
|
-
:math:`(C, output\_size[0], output\_size[1], ...)`
|
|
438
|
+
- Input: :math:`(N, C, *)`
|
|
439
|
+
- Output: :math:`(N, C \times \prod(\text{kernel_size}), L)`
|
|
527
440
|
|
|
528
441
|
Raises:
|
|
529
|
-
TypeError: If
|
|
530
|
-
ValueError: If `
|
|
531
|
-
greater than zero or elements number
|
|
532
|
-
ValueError: If `padding` value is less than zero
|
|
533
|
-
ValueError: If input.shape[-2] can't be divisible by the product of kernel_size.
|
|
534
|
-
ValueError: If `input.shape[-1]` is not equal to the calculated number of sliding blocks `L`.
|
|
442
|
+
TypeError: If any data type of `kernel_size`, `stride`, `dilation`, `padding` is not int, tuple or list.
|
|
443
|
+
ValueError: If `kernel_size`, `dilation`, `stride` value is not
|
|
444
|
+
greater than zero or elements number more than `2`.
|
|
445
|
+
ValueError: If `padding` value is less than zero.
|
|
535
446
|
|
|
536
447
|
Supported Platforms:
|
|
537
448
|
``Ascend``
|
|
538
449
|
|
|
539
450
|
Examples:
|
|
451
|
+
>>> import mindspore
|
|
540
452
|
>>> import numpy as np
|
|
541
453
|
>>> from mindspore import Tensor, ops
|
|
542
|
-
>>> x = Tensor(np.random.rand(
|
|
543
|
-
>>> output = ops.auto_generate.
|
|
454
|
+
>>> x = Tensor(np.random.rand(4, 4, 32, 32), mindspore.float32)
|
|
455
|
+
>>> output = ops.auto_generate.unfold_ext(x, kernel_size=3, dilation=1, stride=1)
|
|
544
456
|
>>> print(output.shape)
|
|
545
|
-
(
|
|
457
|
+
(4, 36, 900)
|
|
546
458
|
"""
|
|
547
|
-
return
|
|
459
|
+
return unfold_impl(input, converted_kernel_size, converted_dilation, converted_padding, converted_stride)
|
|
548
460
|
|
|
549
461
|
|
|
550
|
-
def
|
|
462
|
+
def inplace_add(input, other, alpha=1):
|
|
551
463
|
r"""
|
|
552
|
-
|
|
464
|
+
None
|
|
465
|
+
"""
|
|
466
|
+
return inplace_add_impl(input, other, alpha)
|
|
467
|
+
|
|
468
|
+
|
|
469
|
+
def frac(input):
|
|
470
|
+
r"""
|
|
471
|
+
Calculates the fractional part of each element in the input.
|
|
472
|
+
|
|
473
|
+
.. math::
|
|
474
|
+
out_i = input_i - \lfloor |input_i| \rfloor * sgn(input_i)
|
|
475
|
+
|
|
476
|
+
.. warning::
|
|
477
|
+
This is an experimental API that is subject to change or deletion.
|
|
553
478
|
|
|
554
479
|
Args:
|
|
555
|
-
input (Tensor):
|
|
556
|
-
dim (Union[int, None], optional): Specify the axis for calculation. If `dim` is ``None`` , the indices of the minimum
|
|
557
|
-
value within the flattened input will be returned. Default: ``None`` .
|
|
558
|
-
keepdim (bool, optional): Whether the output tensor retains the specified
|
|
559
|
-
dimension. Ignored if `dim` is None. Default: ``False`` .
|
|
480
|
+
input (Tensor): The input Tensor.
|
|
560
481
|
|
|
561
482
|
Returns:
|
|
562
|
-
Tensor,
|
|
483
|
+
Tensor, has the same shape and type as input.
|
|
563
484
|
|
|
564
485
|
Raises:
|
|
565
|
-
TypeError: If `
|
|
566
|
-
ValueError: If `dim` is out of range.
|
|
486
|
+
TypeError: If `input` is not a Tensor.
|
|
567
487
|
|
|
568
488
|
Supported Platforms:
|
|
569
489
|
``Ascend``
|
|
570
490
|
|
|
571
491
|
Examples:
|
|
492
|
+
>>> import mindspore
|
|
572
493
|
>>> import numpy as np
|
|
573
|
-
>>> from mindspore import Tensor
|
|
574
|
-
>>>
|
|
575
|
-
>>>
|
|
576
|
-
>>> output = mint.argmin(x, dim=-1)
|
|
494
|
+
>>> from mindspore import Tensor, ops
|
|
495
|
+
>>> x = Tensor([2, 4.2, -2.5], mindspore.float16)
|
|
496
|
+
>>> output = ops.frac_ext(x)
|
|
577
497
|
>>> print(output)
|
|
578
|
-
|
|
498
|
+
[ 0. 0.1992 -0.5 ]
|
|
579
499
|
"""
|
|
580
|
-
return
|
|
500
|
+
return frac_impl(input)
|
|
581
501
|
|
|
582
502
|
|
|
583
|
-
def
|
|
503
|
+
def logaddexp(input, other):
|
|
584
504
|
r"""
|
|
585
|
-
|
|
586
|
-
|
|
587
|
-
|
|
588
|
-
both are one-dimensional tensors of length :math:`N`, length :math:`N`, `reduction` is set to ``'none'`` ,
|
|
589
|
-
then calculate the loss of :math:`x` and :math:`y` without dimensionality reduction.
|
|
590
|
-
|
|
591
|
-
The formula is as follows:
|
|
505
|
+
Computes the logarithm of the sum of exponentiations of the inputs.
|
|
506
|
+
This function is useful in statistics where the calculated probabilities of events may be
|
|
507
|
+
so small as to exceed the range of normal floating point numbers.
|
|
592
508
|
|
|
593
509
|
.. math::
|
|
594
|
-
\ell(x, y) = L = \{l_1,\dots,l_N\}^\top, \quad \text{with } l_n = \left| x_n - y_n \right|,
|
|
595
|
-
|
|
596
|
-
where :math:`N` is the batch size.
|
|
597
510
|
|
|
598
|
-
|
|
511
|
+
out_i = \log(exp(input_i) + \exp(other_i))
|
|
599
512
|
|
|
600
|
-
..
|
|
601
|
-
|
|
602
|
-
\begin{cases}
|
|
603
|
-
\operatorname{mean}(L), & \text{if reduction} = \text{'mean';}\\
|
|
604
|
-
\operatorname{sum}(L), & \text{if reduction} = \text{'sum'.}
|
|
605
|
-
\end{cases}
|
|
513
|
+
.. warning::
|
|
514
|
+
This is an experimental API that is subject to change or deletion.
|
|
606
515
|
|
|
607
516
|
Args:
|
|
608
|
-
input (Tensor):
|
|
609
|
-
|
|
610
|
-
If
|
|
611
|
-
|
|
612
|
-
``'sum'`` . Default: ``'mean'`` .
|
|
613
|
-
|
|
614
|
-
- ``'none'``: no reduction will be applied.
|
|
615
|
-
- ``'mean'``: compute and return the mean of elements in the output. Notice: At least one of the input and target is float type when the reduction is ``'mean'`` .
|
|
616
|
-
- ``'sum'``: the output elements will be summed.
|
|
517
|
+
input (Tensor): Input Tensor. The dtype of `input` must be float.
|
|
518
|
+
other (Tensor): Input Tensor. The dtype of `other` must be float.
|
|
519
|
+
If the shape of `input` is not equal to the shape of `other`,
|
|
520
|
+
they must be broadcastable to a common shape.
|
|
617
521
|
|
|
618
522
|
Returns:
|
|
619
|
-
Tensor
|
|
620
|
-
Otherwise, a scalar value will be returned.
|
|
523
|
+
Tensor, with the same dtype as `input` and `other`.
|
|
621
524
|
|
|
622
525
|
Raises:
|
|
623
|
-
TypeError: If `input` is not a Tensor.
|
|
624
|
-
TypeError:
|
|
625
|
-
ValueError: If `reduction` is not one of ``'none'`` , ``'mean'`` or ``'sum'`` .
|
|
526
|
+
TypeError: If `input` or `other` is not a Tensor.
|
|
527
|
+
TypeError: The dtype of `input` or `other` is not float.
|
|
626
528
|
|
|
627
529
|
Supported Platforms:
|
|
628
530
|
``Ascend``
|
|
629
531
|
|
|
630
532
|
Examples:
|
|
533
|
+
>>> import numpy as np
|
|
631
534
|
>>> from mindspore import Tensor, ops
|
|
632
|
-
>>>
|
|
633
|
-
>>>
|
|
634
|
-
>>>
|
|
635
|
-
>>> output = ops.l1_loss_ext(x, target, reduction="mean")
|
|
535
|
+
>>> x1 = Tensor(np.array([1, 2, 3]).astype(np.float16))
|
|
536
|
+
>>> x2 = Tensor(np.array(2).astype(np.float16))
|
|
537
|
+
>>> output = ops.logaddexp_ext(x1, x2)
|
|
636
538
|
>>> print(output)
|
|
637
|
-
3.
|
|
539
|
+
[2.312 2.693 3.312]
|
|
638
540
|
"""
|
|
639
|
-
return
|
|
541
|
+
return logaddexp_impl(input, other)
|
|
640
542
|
|
|
641
543
|
|
|
642
|
-
def
|
|
544
|
+
def acos(input):
|
|
643
545
|
r"""
|
|
644
|
-
Computes
|
|
546
|
+
Computes arccosine of input tensors element-wise.
|
|
645
547
|
|
|
646
548
|
.. math::
|
|
647
549
|
|
|
648
|
-
out_i = \
|
|
649
|
-
|
|
650
|
-
.. note::
|
|
651
|
-
Given an input tensor input, the function computes inverse hyperbolic cosine of every element.
|
|
652
|
-
Input range is [1, inf].
|
|
550
|
+
out_i = \cos^{-1}(input_i)
|
|
653
551
|
|
|
654
552
|
Args:
|
|
655
|
-
input (Tensor): The
|
|
553
|
+
input (Tensor): The shape of tensor is
|
|
554
|
+
:math:`(N,*)`, where :math:`*` means any number of additional dimensions.
|
|
656
555
|
|
|
657
556
|
Returns:
|
|
658
557
|
Tensor, has the same shape as `input`. The dtype of output is float32 when dtype of `input` is in [bool, int8, uint8, int16, int32, int64]. Otherwise output has the same dtype as `input`.
|
|
@@ -667,277 +566,262 @@ def acosh(input):
|
|
|
667
566
|
>>> import mindspore
|
|
668
567
|
>>> import numpy as np
|
|
669
568
|
>>> from mindspore import Tensor, ops
|
|
670
|
-
>>> input = Tensor(np.array([
|
|
671
|
-
>>> output = ops.
|
|
569
|
+
>>> input = Tensor(np.array([0.74, 0.04, 0.30, 0.56]), mindspore.float32)
|
|
570
|
+
>>> output = ops.acos_ext(input)
|
|
672
571
|
>>> print(output)
|
|
673
|
-
[0.
|
|
572
|
+
[0.7377037 1.5307857 1.2661037 0.9764114]
|
|
674
573
|
"""
|
|
675
|
-
return
|
|
574
|
+
return acos_impl(input)
|
|
676
575
|
|
|
677
576
|
|
|
678
|
-
def
|
|
577
|
+
def diag(input, diagonal=0):
|
|
679
578
|
r"""
|
|
680
|
-
|
|
579
|
+
If input is a vector (1-D tensor), then returns a 2-D square tensor with the elements of input as the diagonal.
|
|
681
580
|
|
|
682
|
-
|
|
683
|
-
The activation function is defined as:
|
|
581
|
+
If input is a matrix (2-D tensor), then returns a 1-D tensor with the diagonal elements of input.
|
|
684
582
|
|
|
685
|
-
|
|
583
|
+
The argument diagonal controls which diagonal to consider:
|
|
686
584
|
|
|
687
|
-
|
|
688
|
-
\begin{array}{align}
|
|
689
|
-
\alpha(e^{x} - 1) & \text{if } x \le 0\\
|
|
690
|
-
x & \text{if } x \gt 0\\
|
|
691
|
-
\end{array}\right.
|
|
585
|
+
- If `diagonal` = 0, it is the main diagonal.
|
|
692
586
|
|
|
693
|
-
|
|
694
|
-
it determines the smoothness of ELU.
|
|
587
|
+
- If `diagonal` > 0, it is above the main diagonal.
|
|
695
588
|
|
|
696
|
-
|
|
589
|
+
- If `diagonal` < 0, it is below the main diagonal.
|
|
697
590
|
|
|
698
|
-
..
|
|
699
|
-
|
|
591
|
+
.. warning::
|
|
592
|
+
This is an experimental API that is subject to change or deletion.
|
|
700
593
|
|
|
701
594
|
Args:
|
|
702
|
-
input (Tensor): The input
|
|
703
|
-
|
|
704
|
-
Default: ``1.0`` .
|
|
595
|
+
input (Tensor): The input tensor.
|
|
596
|
+
diagonal (int, optional): the diagonal to consider. Defaults: ``0``.
|
|
705
597
|
|
|
706
598
|
Returns:
|
|
707
|
-
Tensor, has the same shape
|
|
599
|
+
Tensor, has the same dtype as the `input`, its shape is up to `diagonal`.
|
|
600
|
+
|
|
601
|
+
- If `input` shape is :math:`(x_0)` : then output shape is :math:`(x_0 + \left | diagonal \right | , x_0 + \left | diagonal \right | )` 2-D Tensor.
|
|
602
|
+
|
|
603
|
+
- If `input` shape is :math:`(x_0, x_1)` : then output shape is main diagonal to move :math:`(\left | diagonal \right |)` elements remains elements' length 1-D Tensor.
|
|
708
604
|
|
|
709
605
|
Raises:
|
|
710
|
-
TypeError: If `
|
|
606
|
+
TypeError: If `input` is not a Tensor.
|
|
607
|
+
ValueError: If shape of `input` is not 1-D and 2-D.
|
|
711
608
|
|
|
712
609
|
Supported Platforms:
|
|
713
610
|
``Ascend``
|
|
714
611
|
|
|
715
612
|
Examples:
|
|
716
|
-
>>> import mindspore
|
|
717
|
-
>>> import numpy as np
|
|
718
613
|
>>> from mindspore import Tensor, ops
|
|
719
|
-
>>>
|
|
720
|
-
>>> output = ops.auto_generate.
|
|
614
|
+
>>> input = Tensor([1, 2, 3, 4]).astype('int32')
|
|
615
|
+
>>> output = ops.auto_generate.diag_ext(input)
|
|
721
616
|
>>> print(output)
|
|
722
|
-
[[
|
|
723
|
-
[ 2
|
|
617
|
+
[[1 0 0 0]
|
|
618
|
+
[0 2 0 0]
|
|
619
|
+
[0 0 3 0]
|
|
620
|
+
[0 0 0 4]]
|
|
724
621
|
"""
|
|
725
|
-
return
|
|
622
|
+
return diag_impl(input, diagonal)
|
|
726
623
|
|
|
727
624
|
|
|
728
|
-
def
|
|
625
|
+
def argmax(input, dim=None, keepdim=False):
|
|
729
626
|
r"""
|
|
730
|
-
|
|
627
|
+
argmax(input) -> Tensor
|
|
628
|
+
|
|
629
|
+
Return the indices of the maximum values of a tensor.
|
|
731
630
|
|
|
732
631
|
Args:
|
|
733
|
-
input (Tensor):
|
|
734
|
-
the last two dimensions must be the same size.
|
|
632
|
+
input (Tensor): Input tensor.
|
|
735
633
|
|
|
736
634
|
Returns:
|
|
737
|
-
Tensor
|
|
738
|
-
|
|
739
|
-
Raises:
|
|
740
|
-
TypeError: If `input` is not a Tensor.
|
|
741
|
-
ValueError: If the size of the last two dimensions of `input` is not the same.
|
|
742
|
-
ValueError: If the dimension of `input` is 1.
|
|
635
|
+
Tensor.
|
|
743
636
|
|
|
744
637
|
Supported Platforms:
|
|
745
638
|
``Ascend``
|
|
746
639
|
|
|
747
640
|
Examples:
|
|
748
|
-
>>>
|
|
749
|
-
>>> from mindspore import
|
|
750
|
-
>>>
|
|
751
|
-
>>>
|
|
752
|
-
|
|
753
|
-
|
|
754
|
-
|
|
755
|
-
return matrix_inverse_impl(input)
|
|
756
|
-
|
|
757
|
-
|
|
758
|
-
def mean(input, dim=None, keepdim=False, dtype=None):
|
|
759
|
-
r"""
|
|
760
|
-
Reduces all dimension of a tensor by averaging all elements in the dimension, by default.
|
|
761
|
-
And reduce a dimension of `input` along the specified `dim`. `keepdim`
|
|
762
|
-
determines whether the dimensions of the output and input are the same.
|
|
641
|
+
>>> import numpy as np
|
|
642
|
+
>>> from mindspore import Tensor
|
|
643
|
+
>>> from mindspore import ops
|
|
644
|
+
>>> x = Tensor(np.array([[1, 20, 5], [67, 8, 9], [130, 24, 15]]).astype(np.float32))
|
|
645
|
+
>>> output = ops.auto_generate.argmax_ext(x)
|
|
646
|
+
>>> print(output)
|
|
647
|
+
6
|
|
763
648
|
|
|
764
|
-
|
|
765
|
-
|
|
649
|
+
.. function:: argmax(input, dim, keepdim=False) -> Tensor
|
|
650
|
+
:noindex:
|
|
651
|
+
|
|
652
|
+
Return the indices of the maximum values of a tensor across a dimension.
|
|
766
653
|
|
|
767
654
|
Args:
|
|
768
|
-
input (Tensor
|
|
769
|
-
|
|
770
|
-
|
|
771
|
-
|
|
772
|
-
and the value range is [-r,r).
|
|
773
|
-
keepdim (bool): If ``True`` , keep these reduced dimensions and the length is 1.
|
|
774
|
-
If ``False`` , don't keep these dimensions. Default: ``False`` .
|
|
775
|
-
dtype (:class:`mindspore.dtype`): The desired data type of returned Tensor. Default: ``None`` .
|
|
655
|
+
input (Tensor): Input tensor.
|
|
656
|
+
dim (int): The dimension to reduce.
|
|
657
|
+
keepdim (bool, optional): Whether the output tensor retains the specified
|
|
658
|
+
dimension. Default: ``False`` .
|
|
776
659
|
|
|
777
660
|
Returns:
|
|
778
|
-
Tensor,
|
|
779
|
-
|
|
780
|
-
- If `dim` is ``None`` , and `keepdim` is ``False`` ,
|
|
781
|
-
the output is a 0-D tensor representing the product of all elements in the input tensor.
|
|
782
|
-
- If `dim` is int, set as 1, and `keepdim` is ``False`` ,
|
|
783
|
-
the shape of output is :math:`(x_0, x_2, ..., x_R)`.
|
|
784
|
-
- If `dim` is tuple(int), set as (1, 2), and `keepdim` is ``False`` ,
|
|
785
|
-
the shape of output is :math:`(x_0, x_3, ..., x_R)`.
|
|
786
|
-
- If `dim` is 1-D Tensor, set as [1, 2], and `keepdim` is ``False`` ,
|
|
787
|
-
the shape of output is :math:`(x_0, x_3, ..., x_R)`.
|
|
661
|
+
Tensor, indices of the maximum values across a dimension.
|
|
788
662
|
|
|
789
663
|
Raises:
|
|
790
|
-
TypeError: If `
|
|
791
|
-
TypeError: If `dim` is not one of the following: int, tuple, list or Tensor.
|
|
792
|
-
TypeError: If `keepdim` is not a bool.
|
|
664
|
+
TypeError: If `keepdim` is not bool.
|
|
793
665
|
ValueError: If `dim` is out of range.
|
|
794
666
|
|
|
795
667
|
Supported Platforms:
|
|
796
|
-
``Ascend``
|
|
668
|
+
``Ascend``
|
|
797
669
|
|
|
798
670
|
Examples:
|
|
799
|
-
>>> import mindspore
|
|
800
671
|
>>> import numpy as np
|
|
801
|
-
>>> from mindspore import Tensor
|
|
802
|
-
>>>
|
|
803
|
-
>>>
|
|
804
|
-
>>>
|
|
805
|
-
>>> print(result)
|
|
806
|
-
(3, 1, 5, 6)
|
|
807
|
-
>>> # case 1: Reduces a dimension by averaging all elements in the dimension.
|
|
808
|
-
>>> x = Tensor(np.array([[[2, 2, 2, 2, 2, 2], [2, 2, 2, 2, 2, 2], [2, 2, 2, 2, 2, 2]],
|
|
809
|
-
... [[4, 4, 4, 4, 4, 4], [5, 5, 5, 5, 5, 5], [6, 6, 6, 6, 6, 6]],
|
|
810
|
-
... [[6, 6, 6, 6, 6, 6], [8, 8, 8, 8, 8, 8], [10, 10, 10, 10, 10, 10]]]),
|
|
811
|
-
... mindspore.float32)
|
|
812
|
-
>>> output = ops.mean_ext(x)
|
|
813
|
-
>>> print(output)
|
|
814
|
-
5.0
|
|
815
|
-
>>> print(output.shape)
|
|
816
|
-
()
|
|
817
|
-
>>> # case 2: Reduces a dimension along the dim 0
|
|
818
|
-
>>> output = ops.mean_ext(x, 0, True)
|
|
819
|
-
>>> print(output)
|
|
820
|
-
[[[4. 4. 4. 4. 4. 4.]
|
|
821
|
-
[5. 5. 5. 5. 5. 5.]
|
|
822
|
-
[6. 6. 6. 6. 6. 6.]]]
|
|
823
|
-
>>> # case 3: Reduces a dimension along the dim 1
|
|
824
|
-
>>> output = ops.mean_ext(x, 1, True)
|
|
825
|
-
>>> print(output)
|
|
826
|
-
[[[2. 2. 2. 2. 2. 2.]]
|
|
827
|
-
[[5. 5. 5. 5. 5. 5.]]
|
|
828
|
-
[[8. 8. 8. 8. 8. 8.]]]
|
|
829
|
-
>>> # case 4: Reduces a dimension along the dim 2
|
|
830
|
-
>>> output = ops.mean_ext(x, 2, True)
|
|
672
|
+
>>> from mindspore import Tensor
|
|
673
|
+
>>> from mindspore import ops
|
|
674
|
+
>>> x = Tensor(np.array([[1, 20, 5], [67, 8, 9], [130, 24, 15]]).astype(np.float32))
|
|
675
|
+
>>> output = ops.auto_generate.argmax_ext(x, dim=-1)
|
|
831
676
|
>>> print(output)
|
|
832
|
-
[
|
|
833
|
-
[ 2.]
|
|
834
|
-
[ 2.]]
|
|
835
|
-
[[ 4.]
|
|
836
|
-
[ 5.]
|
|
837
|
-
[ 6.]]
|
|
838
|
-
[[ 6.]
|
|
839
|
-
[ 8.]
|
|
840
|
-
[10.]]]
|
|
677
|
+
[1 0 0]
|
|
841
678
|
"""
|
|
842
|
-
return
|
|
679
|
+
return argmax_impl(input, dim, keepdim)
|
|
843
680
|
|
|
844
681
|
|
|
845
|
-
def
|
|
682
|
+
def t(input):
|
|
846
683
|
r"""
|
|
847
|
-
|
|
848
|
-
|
|
849
|
-
.. math::
|
|
684
|
+
Transpose the input tensor.
|
|
850
685
|
|
|
851
|
-
|
|
686
|
+
.. warning::
|
|
687
|
+
This is an experimental API that is subject to change or deletion.
|
|
852
688
|
|
|
853
689
|
Args:
|
|
854
|
-
input (Tensor): The
|
|
855
|
-
:math:`(N,*)` where :math:`*` means, any number of additional dimensions.
|
|
690
|
+
input (Tensor): The input tensor.
|
|
856
691
|
|
|
857
692
|
Returns:
|
|
858
|
-
Tensor,
|
|
693
|
+
Tensor, transpose 2D tensor, return 1D tensor as it is.
|
|
859
694
|
|
|
860
695
|
Raises:
|
|
861
|
-
|
|
696
|
+
ValueError: If the dimension of `input` is greater than 2.
|
|
697
|
+
ValueError: If `input` is empty.
|
|
698
|
+
TypeError: If `input` is not a tensor.
|
|
862
699
|
|
|
863
700
|
Supported Platforms:
|
|
864
|
-
``Ascend``
|
|
701
|
+
``Ascend``
|
|
865
702
|
|
|
866
703
|
Examples:
|
|
867
704
|
>>> import mindspore
|
|
868
705
|
>>> import numpy as np
|
|
869
706
|
>>> from mindspore import Tensor, ops
|
|
870
|
-
>>> input = Tensor(np.array([1
|
|
871
|
-
>>> output = ops.
|
|
707
|
+
>>> input = Tensor(np.array([[1, 2, 3], [4, 5, 6]]), mindspore.float32)
|
|
708
|
+
>>> output = ops.t_ext(input)
|
|
872
709
|
>>> print(output)
|
|
873
|
-
[
|
|
710
|
+
[[ 1. 4.]
|
|
711
|
+
[ 2. 5.]
|
|
712
|
+
[ 3. 6.]]
|
|
874
713
|
"""
|
|
875
|
-
return
|
|
714
|
+
return t_impl(input)
|
|
876
715
|
|
|
877
716
|
|
|
878
|
-
def
|
|
717
|
+
def mse_loss(input, target, reduction='mean'):
|
|
879
718
|
r"""
|
|
880
|
-
|
|
719
|
+
Calculates the mean squared error between the predicted value and the label value.
|
|
881
720
|
|
|
882
|
-
|
|
883
|
-
This is an experimental API that is subject to change or deletion.
|
|
721
|
+
For detailed information, please refer to :class:`mindspore.nn.MSELoss`.
|
|
884
722
|
|
|
885
723
|
Args:
|
|
886
|
-
input (Tensor):
|
|
887
|
-
|
|
888
|
-
|
|
724
|
+
input (Tensor): Tensor of any dimension. The data type needs to be consistent with the `target`.
|
|
725
|
+
It should also be broadcastable with the `target`.
|
|
726
|
+
target (Tensor): The input label. Tensor of any dimension. The data type needs to be consistent with the `input`.
|
|
727
|
+
It should also be broadcastable with the `input`.
|
|
728
|
+
reduction (str, optional): Apply specific reduction method to the output: ``'mean'`` , ``'none'`` ,
|
|
729
|
+
``'sum'`` . Default: ``'mean'`` .
|
|
730
|
+
|
|
731
|
+
- ``'none'``: no reduction will be applied.
|
|
732
|
+
- ``'mean'``: compute and return the mean of elements in the output.
|
|
733
|
+
- ``'sum'``: the output elements will be summed.
|
|
889
734
|
|
|
890
735
|
Returns:
|
|
891
|
-
Tensor.
|
|
736
|
+
- Tensor. If `reduction` is ``'mean'`` or ``'sum'``, the shape of output is `Tensor Scalar`.
|
|
737
|
+
- If reduction is ``'none'``, the shape of output is the broadcasted shape of **input** and **target** .
|
|
892
738
|
|
|
893
739
|
Raises:
|
|
894
|
-
|
|
740
|
+
ValueError: If `reduction` is not one of ``'mean'`` , ``'sum'`` or ``'none'``.
|
|
741
|
+
ValueError: If `input` and `target` are not broadcastable.
|
|
742
|
+
TypeError: If `input` and `target` are in different data type.
|
|
895
743
|
|
|
896
744
|
Supported Platforms:
|
|
897
745
|
``Ascend``
|
|
898
746
|
|
|
899
747
|
Examples:
|
|
900
748
|
>>> import mindspore
|
|
901
|
-
>>>
|
|
902
|
-
>>>
|
|
903
|
-
>>>
|
|
904
|
-
>>>
|
|
905
|
-
|
|
749
|
+
>>> import numpy as np
|
|
750
|
+
>>> from mindspore import Tensor, ops
|
|
751
|
+
>>> logits = Tensor(np.array([1, 2, 3]), mindspore.float32)
|
|
752
|
+
>>> labels = Tensor(np.array([[1, 1, 1], [1, 2, 2]]), mindspore.float32)
|
|
753
|
+
>>> output = ops.mse_loss_ext(logits, labels, reduction='none')
|
|
754
|
+
>>> print(output)
|
|
755
|
+
[[0. 1. 4.]
|
|
756
|
+
[0. 0. 1.]]
|
|
906
757
|
"""
|
|
907
|
-
return
|
|
758
|
+
return mse_loss_impl(input, target, converted_reduction)
|
|
908
759
|
|
|
909
760
|
|
|
910
|
-
def
|
|
761
|
+
def fold(input, output_size, kernel_size, dilation=1, padding=0, stride=1):
|
|
911
762
|
r"""
|
|
912
|
-
|
|
763
|
+
Combines an array of sliding local blocks into a large containing tensor.
|
|
764
|
+
|
|
765
|
+
Consider a batched input tensor of shape :math:`(N, C \times \prod(\text{kernel_size}), L)` ,
|
|
766
|
+
where :math:`N` is the batch dimension, :math:`C \times \prod(\text{kernel_size})` is the
|
|
767
|
+
total number of values within each block (a block has :math:`\prod(\text{kernel_size})` spatial
|
|
768
|
+
locations each containing a `C`-channeled vector), and :math:`L` is the total number of such blocks:
|
|
913
769
|
|
|
914
770
|
.. math::
|
|
771
|
+
L = \prod_d \left\lfloor\frac{\text{output_size}[d] + 2 \times \text{padding}[d] %
|
|
772
|
+
- \text{dilation}[d] \times (\text{kernel_size}[d] - 1) - 1}{\text{stride}[d]} + 1\right\rfloor,
|
|
915
773
|
|
|
916
|
-
|
|
774
|
+
where :math:`d` is over all spatial dimensions.
|
|
775
|
+
|
|
776
|
+
Therefore, `output_size` is the spatial shape of the large containing tensor of the sliding local blocks.
|
|
777
|
+
|
|
778
|
+
The `dilation`, `padding` and `stride` arguments specify how the sliding blocks are retrieved.
|
|
779
|
+
|
|
780
|
+
.. warning::
|
|
781
|
+
Currently, only unbatched(3D) or batched(4D) image-like output tensors are supported.
|
|
917
782
|
|
|
918
783
|
Args:
|
|
919
|
-
input (Tensor):
|
|
920
|
-
|
|
784
|
+
input (Tensor): 2-D or 3-D Tensor.
|
|
785
|
+
output_size (Union[int, tuple[int], list[int]]): The shape of the spatial dimensions of
|
|
786
|
+
the output(i.e., output.shape[2:]).
|
|
787
|
+
kernel_size (Union[int, tuple[int], list[int]]): The size of the kernel, should be two int
|
|
788
|
+
for height and width. If type is int, it means that height equal with width. Must be specified.
|
|
789
|
+
dilation (Union[int, tuple[int], list[int]], optional): The size of the dilation, should be two int
|
|
790
|
+
for height and width. If type is int, it means that height equal with width. Default: ``1`` .
|
|
791
|
+
padding (Union[int, tuple[int], list[int]], optional): The size of the padding, should be two int
|
|
792
|
+
for height and width. If type is int, it means that height equal with width. Default: ``0`` .
|
|
793
|
+
stride (Union[int, tuple[int], list[int]], optional): The size of the stride, should be two int
|
|
794
|
+
for height and width. If type is int, it means that height equal with width. Default: ``1`` .
|
|
921
795
|
|
|
922
796
|
Returns:
|
|
923
|
-
Tensor,
|
|
797
|
+
A Tensor, with same type as `input` .
|
|
798
|
+
|
|
799
|
+
Shape:
|
|
800
|
+
- Input: :math:`(N, C \times \prod(\text{kernel_size}), L)` or
|
|
801
|
+
:math:`(C \times \prod(\text{kernel_size}), L)`
|
|
802
|
+
- Output: :math:`(N, C, output\_size[0], output\_size[1], ...)` or
|
|
803
|
+
:math:`(C, output\_size[0], output\_size[1], ...)`
|
|
924
804
|
|
|
925
805
|
Raises:
|
|
926
|
-
TypeError: If `
|
|
806
|
+
TypeError: If `output_size`, `kernel_size`, `stride`, `dilation`, `padding` data type is not int, tuple or list.
|
|
807
|
+
ValueError: If `output_size`, `kernel_size`, `dilation`, `stride` value is not
|
|
808
|
+
greater than zero or elements number invalid.
|
|
809
|
+
ValueError: If `padding` value is less than zero or elements number invalid.
|
|
810
|
+
ValueError: If input.shape[-2] can't be divisible by the product of kernel_size.
|
|
811
|
+
ValueError: If `input.shape[-1]` is not equal to the calculated number of sliding blocks `L`.
|
|
927
812
|
|
|
928
813
|
Supported Platforms:
|
|
929
|
-
``Ascend``
|
|
814
|
+
``Ascend``
|
|
930
815
|
|
|
931
816
|
Examples:
|
|
932
|
-
>>> import mindspore
|
|
933
817
|
>>> import numpy as np
|
|
934
818
|
>>> from mindspore import Tensor, ops
|
|
935
|
-
>>>
|
|
936
|
-
>>> output = ops.
|
|
937
|
-
>>> print(output)
|
|
938
|
-
|
|
819
|
+
>>> x = Tensor(np.random.rand(16, 64, 25).astype(np.float32))
|
|
820
|
+
>>> output = ops.auto_generate.fold_ext(x, (8, 8), [2, 2], [2, 2], [2, 2], [2, 2])
|
|
821
|
+
>>> print(output.shape)
|
|
822
|
+
(16, 16, 8, 8)
|
|
939
823
|
"""
|
|
940
|
-
return
|
|
824
|
+
return fold_impl(input, converted_output_size, converted_kernel_size, converted_dilation, converted_padding, converted_stride)
|
|
941
825
|
|
|
942
826
|
|
|
943
827
|
def flatten(input, start_dim=0, end_dim=-1):
|
|
@@ -974,102 +858,154 @@ def flatten(input, start_dim=0, end_dim=-1):
|
|
|
974
858
|
return flatten_impl(input, start_dim, end_dim)
|
|
975
859
|
|
|
976
860
|
|
|
977
|
-
def
|
|
861
|
+
def bmm(input, mat2):
|
|
978
862
|
r"""
|
|
979
|
-
|
|
980
|
-
|
|
981
|
-
Stacks the list of input tensors with the same rank `R`, output is a tensor of rank `(R+1)`.
|
|
863
|
+
Performs batch matrix-matrix multiplication of two three-dimensional tensors.
|
|
982
864
|
|
|
983
|
-
|
|
984
|
-
|
|
985
|
-
:math:`(x_1, x_2, ..., x_{dim}, N, x_{dim+1}, ..., x_R)`.
|
|
865
|
+
.. math::
|
|
866
|
+
\text{output}= \text{input} @ \text{mat2}
|
|
986
867
|
|
|
987
868
|
Args:
|
|
988
|
-
|
|
989
|
-
|
|
869
|
+
input (Tensor): The first batch of matrices to be multiplied. Must be a three-dimensional tensor of shape `(b, n, m)`.
|
|
870
|
+
mat2 (Tensor): The second batch of matrices to be multiplied. Must be a three-dimensional tensor of shape `(b, m, p)`.
|
|
990
871
|
|
|
991
872
|
Returns:
|
|
992
|
-
Tensor
|
|
873
|
+
Tensor, the output tensor of shape `(b, n, p)`, where each matrix is the product of the corresponding matrices in the input batches.
|
|
993
874
|
|
|
994
875
|
Raises:
|
|
995
|
-
|
|
996
|
-
ValueError: If `
|
|
997
|
-
|
|
876
|
+
ValueError: If `input` or `mat2` is not three-dimensional tensors.
|
|
877
|
+
ValueError: If the length of the third dimension of `input` is not equal to the length of the second dimension of `mat2`.
|
|
878
|
+
ValueError: If the batch size of the inputs is not equal to the batch size of the mat2.
|
|
998
879
|
|
|
999
880
|
Supported Platforms:
|
|
1000
|
-
``Ascend``
|
|
881
|
+
``Ascend`` ``GPU`` ``CPU``
|
|
1001
882
|
|
|
1002
883
|
Examples:
|
|
1003
884
|
>>> import mindspore
|
|
1004
|
-
>>> from mindspore import Tensor, ops
|
|
1005
885
|
>>> import numpy as np
|
|
1006
|
-
>>>
|
|
1007
|
-
>>>
|
|
1008
|
-
>>>
|
|
886
|
+
>>> from mindspore import Tensor
|
|
887
|
+
>>> from mindspore import ops
|
|
888
|
+
>>> a = Tensor(np.ones(shape=[2, 3, 4]), mindspore.float32)
|
|
889
|
+
>>> b = Tensor(np.ones(shape=[2, 4, 5]), mindspore.float32)
|
|
890
|
+
>>> output = ops.auto_generate.bmm_ext(a, b)
|
|
1009
891
|
>>> print(output)
|
|
1010
|
-
[[
|
|
1011
|
-
|
|
1012
|
-
|
|
1013
|
-
|
|
1014
|
-
|
|
1015
|
-
|
|
1016
|
-
def tril(input, diagonal=0):
|
|
1017
|
-
r"""
|
|
1018
|
-
None
|
|
1019
|
-
"""
|
|
1020
|
-
return tril_impl(input, diagonal)
|
|
1021
|
-
|
|
1022
|
-
|
|
1023
|
-
def matmul(input, other):
|
|
1024
|
-
r"""
|
|
1025
|
-
None
|
|
1026
|
-
"""
|
|
1027
|
-
return matmul_impl(input, other)
|
|
1028
|
-
|
|
1029
|
-
|
|
1030
|
-
def adaptive_avg_pool2d_grad(grad_output, x):
|
|
1031
|
-
r"""
|
|
1032
|
-
None
|
|
892
|
+
[[[4. 4. 4. 4. 4.]
|
|
893
|
+
[4. 4. 4. 4. 4.]
|
|
894
|
+
[4. 4. 4. 4. 4.]]
|
|
895
|
+
[[4. 4. 4. 4. 4.]
|
|
896
|
+
[4. 4. 4. 4. 4.]
|
|
897
|
+
[4. 4. 4. 4. 4.]]]
|
|
1033
898
|
"""
|
|
1034
|
-
return
|
|
899
|
+
return bmm_impl(input, mat2)
|
|
1035
900
|
|
|
1036
901
|
|
|
1037
|
-
def
|
|
902
|
+
def add(input, other, alpha=1):
|
|
1038
903
|
r"""
|
|
1039
|
-
|
|
904
|
+
Adds scaled other value to input Tensor.
|
|
905
|
+
|
|
906
|
+
.. math::
|
|
907
|
+
|
|
908
|
+
out_{i} = input_{i} + alpha \times other_{i}
|
|
909
|
+
|
|
910
|
+
Note:
|
|
911
|
+
- When the two inputs have different shapes,
|
|
912
|
+
they must be able to broadcast to a common shape.
|
|
913
|
+
- The two inputs and alpha comply with the implicit type conversion rules to make the data types
|
|
914
|
+
consistent.
|
|
1040
915
|
|
|
1041
916
|
Args:
|
|
1042
|
-
input (Tensor):
|
|
917
|
+
input (Union[Tensor, number.Number, bool]): The first input is a number.Number or
|
|
918
|
+
a bool or a tensor whose data type is
|
|
919
|
+
`number <https://www.mindspore.cn/docs/en/master/api_python/mindspore/mindspore.dtype.html>`_ or
|
|
920
|
+
`bool_ <https://www.mindspore.cn/docs/en/master/api_python/mindspore/mindspore.dtype.html>`_.
|
|
921
|
+
other (Union[Tensor, number.Number, bool]): The second input, is a number.Number or
|
|
922
|
+
a bool or a tensor whose data type is
|
|
923
|
+
`number <https://www.mindspore.cn/docs/en/master/api_python/mindspore/mindspore.dtype.html>`_ or
|
|
924
|
+
`bool_ <https://www.mindspore.cn/docs/en/master/api_python/mindspore/mindspore.dtype.html>`_.
|
|
925
|
+
alpha (number.Number): A scaling factor applied to `other`, default 1.
|
|
1043
926
|
|
|
1044
927
|
Returns:
|
|
1045
|
-
Tensor
|
|
928
|
+
Tensor with a shape that is the same as the broadcasted shape of the input `input` and `other`,
|
|
929
|
+
and the data type is the one with higher precision or higher digits among the two inputs and alpha.
|
|
1046
930
|
|
|
1047
931
|
Raises:
|
|
1048
|
-
TypeError: If `input` is not
|
|
1049
|
-
|
|
1050
|
-
TypeError:
|
|
932
|
+
TypeError: If the type of `input`, `other`, or `alpha` is not one of the following: Tensor, number.Number, bool.
|
|
933
|
+
TypeError: If `alpha` is of type float but `input` and `other` are not of type float.
|
|
934
|
+
TypeError: If `alpha` is of type bool but `input` and `other` are not of type bool.
|
|
1051
935
|
|
|
1052
936
|
Supported Platforms:
|
|
1053
|
-
``Ascend``
|
|
937
|
+
``Ascend`` ``GPU`` ``CPU``
|
|
1054
938
|
|
|
1055
939
|
Examples:
|
|
1056
|
-
>>> import mindspore
|
|
1057
940
|
>>> import numpy as np
|
|
1058
|
-
>>>
|
|
1059
|
-
>>>
|
|
1060
|
-
>>>
|
|
1061
|
-
>>>
|
|
1062
|
-
|
|
1063
|
-
>>>
|
|
1064
|
-
>>> output = ops.
|
|
1065
|
-
>>> print(output)
|
|
1066
|
-
18.0
|
|
1067
|
-
>>> input = Tensor(np.arange(12, 0, -1).reshape(4, 3), mindspore.float32)
|
|
1068
|
-
>>> output = ops.trace_ext(input)
|
|
941
|
+
>>> import mindspore
|
|
942
|
+
>>> from mindspore import Tensor
|
|
943
|
+
>>> from mindspore import ops
|
|
944
|
+
>>> x = Tensor(1, mindspore.int32)
|
|
945
|
+
>>> y = Tensor(np.array([4, 5, 6]).astype(np.float32))
|
|
946
|
+
>>> alpha = 0.5
|
|
947
|
+
>>> output = ops.auto_generate.add_ext(x, y, alpha)
|
|
1069
948
|
>>> print(output)
|
|
1070
|
-
|
|
949
|
+
[3. 3.5 4.]
|
|
950
|
+
>>> # the data type of x is int32, the data type of y is float32,
|
|
951
|
+
>>> # alpha is a float, and the output is the data format of higher precision float32.
|
|
952
|
+
>>> print(output.dtype)
|
|
953
|
+
Float32
|
|
1071
954
|
"""
|
|
1072
|
-
return
|
|
955
|
+
return add_impl(input, other, alpha)
|
|
956
|
+
|
|
957
|
+
|
|
958
|
+
def adaptive_avg_pool3d(input, output_size):
|
|
959
|
+
r"""
|
|
960
|
+
None
|
|
961
|
+
"""
|
|
962
|
+
return adaptive_avg_pool3d_impl(input, output_size)
|
|
963
|
+
|
|
964
|
+
|
|
965
|
+
def bincount(input, weights=None, minlength=0):
|
|
966
|
+
r"""
|
|
967
|
+
Count the occurrences of each value in the input.
|
|
968
|
+
|
|
969
|
+
If `minlength` is not specified, the length of the output Tensor is the maximum value in the input plus one.
|
|
970
|
+
If `minlength` is specified, the length of the output Tensor is the maximum value between `minlength` or
|
|
971
|
+
the maximum value in the input plus one.
|
|
972
|
+
|
|
973
|
+
Each value in the output Tensor represents the number of occurrences of that index value in the input.
|
|
974
|
+
If `weights` is specified, the output results are weighted,
|
|
975
|
+
i.e., :math:`out[n] += weight[i]` instead of :math:`out[n] += 1`.
|
|
976
|
+
|
|
977
|
+
.. warning::
|
|
978
|
+
This is an experimental API that is subject to change or deletion.
|
|
979
|
+
|
|
980
|
+
Args:
|
|
981
|
+
input (Tensor): A one-dimensional Tensor.
|
|
982
|
+
weights (Tensor, optional): Weights with the same shape as the input. Default: ``None``.
|
|
983
|
+
minlength (int, optional): The minimum length of output Tensor. Should be non-negative. Default: ``0``.
|
|
984
|
+
|
|
985
|
+
Returns:
|
|
986
|
+
Tensor, If input is non-empty, the output shape is :math:`(max(max(input)+1, minlength), )`,
|
|
987
|
+
otherwise the shape is :math:`(0, )`.
|
|
988
|
+
|
|
989
|
+
Raises:
|
|
990
|
+
TypeError: If `input` or `weights` is not a Tensor.
|
|
991
|
+
ValueError: If `input` contains negative values.
|
|
992
|
+
ValueError: If `input` is not one-dimensional or `input` and `weights` do not have the same shape.
|
|
993
|
+
|
|
994
|
+
Supported Platforms:
|
|
995
|
+
``Ascend``
|
|
996
|
+
|
|
997
|
+
Examples:
|
|
998
|
+
>>> from mindspore import ops, Tensor
|
|
999
|
+
>>> print(ops.auto_generate.bincount_ext(Tensor(np.arange(5))))
|
|
1000
|
+
[1 1 1 1 1]
|
|
1001
|
+
>>> print(ops.auto_generate.bincount_ext(Tensor(np.array([0, 1, 1, 3, 2, 1, 7]))))
|
|
1002
|
+
[1 3 1 1 0 0 0 1]
|
|
1003
|
+
>>> w = Tensor(np.array([0.3, 0.5, 0.2, 0.7, 1., -0.6])) # weights
|
|
1004
|
+
>>> x = Tensor(np.array([0, 1, 1, 2, 2, 2]))
|
|
1005
|
+
>>> print(ops.auto_generate.bincount_ext(x, weights=w, minlength=5))
|
|
1006
|
+
[0.3 0.7 1.1 0. 0. ]
|
|
1007
|
+
"""
|
|
1008
|
+
return bincount_impl(input, weights, minlength)
|
|
1073
1009
|
|
|
1074
1010
|
|
|
1075
1011
|
def max_unpool2d(input, indices, kernel_size, stride=None, padding=0, output_size=None):
|
|
@@ -1123,36 +1059,40 @@ def max_unpool2d(input, indices, kernel_size, stride=None, padding=0, output_siz
|
|
|
1123
1059
|
return max_unpool2d_impl(input, indices, kernel_size, stride, padding, output_size)
|
|
1124
1060
|
|
|
1125
1061
|
|
|
1126
|
-
def
|
|
1127
|
-
r"""
|
|
1128
|
-
None
|
|
1129
|
-
"""
|
|
1130
|
-
return sort_impl(input, dim, descending, stable)
|
|
1131
|
-
|
|
1132
|
-
|
|
1133
|
-
def sub_tensor_(input, other, alpha=1):
|
|
1134
|
-
r"""
|
|
1135
|
-
None
|
|
1136
|
-
"""
|
|
1137
|
-
return sub_tensor_impl(input, other, alpha)
|
|
1138
|
-
|
|
1139
|
-
|
|
1140
|
-
def asinh(input):
|
|
1062
|
+
def avg_pool1d(input, kernel_size, stride=None, padding=0, ceil_mode=False, count_include_pad=True):
|
|
1141
1063
|
r"""
|
|
1142
|
-
|
|
1064
|
+
Applies a 1D average pooling over an input Tensor which can be regarded as a composition of 1D input planes.
|
|
1065
|
+
|
|
1066
|
+
Typically the input is of shape :math:`(N_{in}, C_{in}, L_{in})`, avg_pool1d outputs regional average in the
|
|
1067
|
+
:math:`(L_{in})`-dimension. Given kernel size as :math:`ks = l_{ker}` and `stride` as :math:`s = s_0`, the
|
|
1068
|
+
operation is as follows.
|
|
1143
1069
|
|
|
1144
1070
|
.. math::
|
|
1071
|
+
\text{output}(N_i, C_j, l) = \frac{1}{l_{ker}} \sum_{n=0}^{l_{ker}-1}
|
|
1072
|
+
\text{input}(N_i, C_j, s_0 \times l + n)
|
|
1145
1073
|
|
|
1146
|
-
|
|
1074
|
+
.. warning::
|
|
1075
|
+
This is an experimental API that is subject to change or deletion.
|
|
1147
1076
|
|
|
1148
1077
|
Args:
|
|
1149
|
-
input (Tensor):
|
|
1078
|
+
input (Tensor): Tensor of shape :math:`(N, C_{in}, L_{in})`.
|
|
1079
|
+
kernel_size (Union(int, tuple[int])): The size of kernel window used to take the average value.
|
|
1080
|
+
stride (Union(int, tuple[int]), optional): The distance of kernel moving. `stride` can either be an int
|
|
1081
|
+
number or a tuple of one int number. Default: ``None``, the same value as `kernel_size`.
|
|
1082
|
+
padding (Union(int, tuple[int]), optional): The pad length to be filled. `padding` can either be an integer
|
|
1083
|
+
or a tuple of one integer. Default: ``0`` .
|
|
1084
|
+
ceil_mode (bool, optional): If True, apply ceil instead of floor to compute the output shape. Default: ``False``.
|
|
1085
|
+
count_include_pad (bool, optional): If True, include the zero-padding in the averaging calculation. Default: ``True`` .
|
|
1150
1086
|
|
|
1151
1087
|
Returns:
|
|
1152
|
-
Tensor
|
|
1088
|
+
Tensor of shape :math:`(N, C_{in}, L_{out})`.
|
|
1153
1089
|
|
|
1154
1090
|
Raises:
|
|
1155
1091
|
TypeError: If `input` is not a Tensor.
|
|
1092
|
+
TypeError: If `kernel_size` or `stride` is not an int.
|
|
1093
|
+
TypeError: If `ceil_mode` or `count_include_pad` is not a bool.
|
|
1094
|
+
ValueError: If `kernel_size` or `stride` is less than `1`.
|
|
1095
|
+
ValueError: If `kernel_size` or `stride` or `padding` is not int nor a tuple whose length is greater than `1`.
|
|
1156
1096
|
|
|
1157
1097
|
Supported Platforms:
|
|
1158
1098
|
``Ascend`` ``GPU`` ``CPU``
|
|
@@ -1161,198 +1101,204 @@ def asinh(input):
|
|
|
1161
1101
|
>>> import mindspore
|
|
1162
1102
|
>>> import numpy as np
|
|
1163
1103
|
>>> from mindspore import Tensor, ops
|
|
1164
|
-
>>>
|
|
1165
|
-
>>> output = ops.
|
|
1166
|
-
>>> print(output)
|
|
1167
|
-
|
|
1104
|
+
>>> input_x = Tensor(np.random.randint(0, 10, [1, 3, 6]), mindspore.float32)
|
|
1105
|
+
>>> output = ops.auto_generate.avg_pool1d_ext(input_x, kernel_size=6, stride=1)
|
|
1106
|
+
>>> print(output.shape)
|
|
1107
|
+
(1, 3, 1)
|
|
1168
1108
|
"""
|
|
1169
|
-
return
|
|
1109
|
+
return avg_pool1d_impl(input, kernel_size, stride, padding, ceil_mode, count_include_pad)
|
|
1170
1110
|
|
|
1171
1111
|
|
|
1172
|
-
def
|
|
1112
|
+
def trace(input):
|
|
1173
1113
|
r"""
|
|
1174
|
-
|
|
1175
|
-
|
|
1176
|
-
.. math::
|
|
1177
|
-
|
|
1178
|
-
out_{i} = input_{i} - alpha \times other_{i}
|
|
1179
|
-
|
|
1180
|
-
Note:
|
|
1181
|
-
- When the two inputs have different shapes,
|
|
1182
|
-
they must be able to broadcast to a common shape.
|
|
1183
|
-
- The two inputs and alpha comply with the implicit type conversion rules to make the data types
|
|
1184
|
-
consistent.
|
|
1114
|
+
Returns a new tensor that is the sum of the `input` main trace.
|
|
1185
1115
|
|
|
1186
1116
|
Args:
|
|
1187
|
-
input (
|
|
1188
|
-
a bool or a tensor whose data type is
|
|
1189
|
-
`number <https://www.mindspore.cn/docs/en/master/api_python/mindspore/mindspore.dtype.html>`_ or
|
|
1190
|
-
`bool_ <https://www.mindspore.cn/docs/en/master/api_python/mindspore/mindspore.dtype.html>`_.
|
|
1191
|
-
other (Union[Tensor, number.Number, bool]): The second input, is a number.Number or
|
|
1192
|
-
a bool or a tensor whose data type is
|
|
1193
|
-
`number <https://www.mindspore.cn/docs/en/master/api_python/mindspore/mindspore.dtype.html>`_ or
|
|
1194
|
-
`bool_ <https://www.mindspore.cn/docs/en/master/api_python/mindspore/mindspore.dtype.html>`_.
|
|
1195
|
-
alpha (number.Number): A scaling factor applied to `other`, default 1.
|
|
1117
|
+
input (Tensor): 2-D Tensor.
|
|
1196
1118
|
|
|
1197
1119
|
Returns:
|
|
1198
|
-
Tensor
|
|
1199
|
-
and the data type is the one with higher precision or higher digits among the two inputs and alpha.
|
|
1120
|
+
Tensor, when the data type of `input` is integer or bool, its data type is int64, otherwise it is the same as `input`, and size equals to 1.
|
|
1200
1121
|
|
|
1201
1122
|
Raises:
|
|
1202
|
-
TypeError: If
|
|
1203
|
-
|
|
1204
|
-
TypeError:
|
|
1123
|
+
TypeError: If `input` is not a Tensor.
|
|
1124
|
+
ValueError: If the dimension of `input` is not equal to 2.
|
|
1125
|
+
TypeError: If the dtype of `input` is not one of float16, float32, float64, bool, uint8, int8, int16, int32, int64, complex64, complex128, bfloat16.
|
|
1205
1126
|
|
|
1206
1127
|
Supported Platforms:
|
|
1207
|
-
``Ascend``
|
|
1128
|
+
``Ascend``
|
|
1208
1129
|
|
|
1209
1130
|
Examples:
|
|
1210
|
-
>>> import numpy as np
|
|
1211
1131
|
>>> import mindspore
|
|
1212
|
-
>>>
|
|
1213
|
-
>>> from mindspore import ops
|
|
1214
|
-
>>>
|
|
1215
|
-
>>>
|
|
1216
|
-
>>> alpha = 0.5
|
|
1217
|
-
>>> output = ops.auto_generate.sub_ext(x, y, alpha)
|
|
1132
|
+
>>> import numpy as np
|
|
1133
|
+
>>> from mindspore import Tensor, ops
|
|
1134
|
+
>>> input = Tensor(np.array([[10, 11, 12], [13, 14, 15], [16, 17, 18]]), mindspore.float32)
|
|
1135
|
+
>>> output = ops.trace_ext(input)
|
|
1218
1136
|
>>> print(output)
|
|
1219
|
-
|
|
1220
|
-
>>>
|
|
1221
|
-
>>>
|
|
1222
|
-
>>> print(output
|
|
1223
|
-
|
|
1137
|
+
42.0
|
|
1138
|
+
>>> input = Tensor(np.arange(1, 13).reshape(3, 4), mindspore.float32)
|
|
1139
|
+
>>> output = ops.trace_ext(input)
|
|
1140
|
+
>>> print(output)
|
|
1141
|
+
18.0
|
|
1142
|
+
>>> input = Tensor(np.arange(12, 0, -1).reshape(4, 3), mindspore.float32)
|
|
1143
|
+
>>> output = ops.trace_ext(input)
|
|
1144
|
+
>>> print(output)
|
|
1145
|
+
24.0
|
|
1224
1146
|
"""
|
|
1225
|
-
return
|
|
1147
|
+
return trace_impl(input)
|
|
1226
1148
|
|
|
1227
1149
|
|
|
1228
|
-
def
|
|
1150
|
+
def argmin(input, dim=None, keepdim=False):
|
|
1229
1151
|
r"""
|
|
1230
|
-
|
|
1231
|
-
|
|
1232
|
-
.. math::
|
|
1233
|
-
out_i = input_i - \lfloor |input_i| \rfloor * sgn(input_i)
|
|
1234
|
-
|
|
1235
|
-
.. warning::
|
|
1236
|
-
This is an experimental API that is subject to change or deletion.
|
|
1152
|
+
Return the indices of the minimum values of a tensor across a dimension.
|
|
1237
1153
|
|
|
1238
1154
|
Args:
|
|
1239
|
-
input (Tensor):
|
|
1155
|
+
input (Tensor): Input tensor.
|
|
1156
|
+
dim (Union[int, None], optional): Specify the axis for calculation. If `dim` is ``None`` , the indices of the minimum
|
|
1157
|
+
value within the flattened input will be returned. Default: ``None`` .
|
|
1158
|
+
keepdim (bool, optional): Whether the output tensor retains the specified
|
|
1159
|
+
dimension. Ignored if `dim` is None. Default: ``False`` .
|
|
1240
1160
|
|
|
1241
1161
|
Returns:
|
|
1242
|
-
Tensor,
|
|
1162
|
+
Tensor, indices of the minimum values of the input tensor across a dimension.
|
|
1243
1163
|
|
|
1244
1164
|
Raises:
|
|
1245
|
-
TypeError: If `
|
|
1165
|
+
TypeError: If `keepdim` is not bool.
|
|
1166
|
+
ValueError: If `dim` is out of range.
|
|
1246
1167
|
|
|
1247
1168
|
Supported Platforms:
|
|
1248
1169
|
``Ascend``
|
|
1249
1170
|
|
|
1250
1171
|
Examples:
|
|
1251
|
-
>>> import mindspore
|
|
1252
1172
|
>>> import numpy as np
|
|
1253
|
-
>>> from mindspore import Tensor
|
|
1254
|
-
>>>
|
|
1255
|
-
>>>
|
|
1173
|
+
>>> from mindspore import Tensor
|
|
1174
|
+
>>> from mindspore import ops
|
|
1175
|
+
>>> x = Tensor(np.array([[1, 20, 5], [67, 8, 9], [130, 24, 15]]).astype(np.float32))
|
|
1176
|
+
>>> output = ops.auto_generate.argmin_ext(x, dim=-1)
|
|
1256
1177
|
>>> print(output)
|
|
1257
|
-
|
|
1178
|
+
[0 1 2]
|
|
1258
1179
|
"""
|
|
1259
|
-
return
|
|
1180
|
+
return argmin_impl(input, dim, keepdim)
|
|
1260
1181
|
|
|
1261
1182
|
|
|
1262
|
-
def
|
|
1183
|
+
def mm(input, mat2):
|
|
1263
1184
|
r"""
|
|
1264
|
-
|
|
1185
|
+
Returns the matrix product of two arrays.
|
|
1186
|
+
If `input` is a :math:`(n \times m)` Tensor, `mat2` is a
|
|
1187
|
+
:math:`(m \times p)` Tensor, `out` will be a :math:`(n \times p)` Tensor.
|
|
1188
|
+
|
|
1189
|
+
Note:
|
|
1190
|
+
This function cannot support broadcasting.
|
|
1191
|
+
Refer to :func:`mindspore.ops.matmul` instead if you need a broadcastable function.
|
|
1265
1192
|
|
|
1266
1193
|
.. warning::
|
|
1267
|
-
|
|
1194
|
+
This is an experimental API that is subject to change or deletion.
|
|
1268
1195
|
|
|
1269
1196
|
Args:
|
|
1270
|
-
input (Tensor):
|
|
1197
|
+
input (Tensor): The first matrix of matrix multiplication.
|
|
1198
|
+
The last dimension of `input` must be the same size as the first dimension of `mat2`.
|
|
1199
|
+
mat2 (Tensor): The second matrix of matrix multiplication.
|
|
1200
|
+
The last dimension of `input` must be the same size as the first dimension of `mat2`.
|
|
1271
1201
|
|
|
1272
1202
|
Returns:
|
|
1273
|
-
Tensor
|
|
1203
|
+
Tensor, the matrix product of the inputs.
|
|
1274
1204
|
|
|
1275
1205
|
Raises:
|
|
1276
|
-
|
|
1206
|
+
ValueError: If the last dimension of `input` is not the same size as the
|
|
1207
|
+
second-to-last dimension of `mat2`.
|
|
1208
|
+
TypeError: If `input` or `mat2` is not a Tensor.
|
|
1209
|
+
TypeError: If dtype of `input` or `mat2` is not float16, float32 or bfloat16.
|
|
1277
1210
|
|
|
1278
1211
|
Supported Platforms:
|
|
1279
|
-
``Ascend``
|
|
1212
|
+
``Ascend``
|
|
1280
1213
|
|
|
1281
1214
|
Examples:
|
|
1282
|
-
>>>
|
|
1283
|
-
>>> from mindspore import
|
|
1284
|
-
>>>
|
|
1285
|
-
>>>
|
|
1286
|
-
|
|
1287
|
-
|
|
1215
|
+
>>> import mindspore as ms
|
|
1216
|
+
>>> from mindspore import ops
|
|
1217
|
+
>>> import numpy as np
|
|
1218
|
+
>>> x1 = ms.Tensor(np.random.rand(2, 3), ms.float32)
|
|
1219
|
+
>>> x2 = ms.Tensor(np.random.rand(3, 4), ms.float32)
|
|
1220
|
+
>>> out = ops.mm_ext(x1, x2)
|
|
1221
|
+
>>> print(out.shape)
|
|
1222
|
+
(2, 4)
|
|
1288
1223
|
"""
|
|
1289
|
-
return
|
|
1224
|
+
return mm_impl(input, mat2)
|
|
1290
1225
|
|
|
1291
1226
|
|
|
1292
|
-
def
|
|
1227
|
+
def inplace_adds(input, other, alpha=1):
|
|
1293
1228
|
r"""
|
|
1294
1229
|
None
|
|
1295
1230
|
"""
|
|
1296
|
-
return
|
|
1231
|
+
return inplace_adds_impl(input, other, alpha)
|
|
1297
1232
|
|
|
1298
1233
|
|
|
1299
|
-
def
|
|
1234
|
+
def sum(input, dim=None, keepdim=False, dtype=None):
|
|
1300
1235
|
r"""
|
|
1301
|
-
|
|
1302
|
-
along the dimension `dim`, and `indices` is the index location of each minimum value.
|
|
1303
|
-
|
|
1304
|
-
.. math::
|
|
1305
|
-
\begin{array}{ll} \\
|
|
1306
|
-
y_{i} = \min(x_{1}, x_{2}, ... , x_{i})
|
|
1307
|
-
\end{array}
|
|
1236
|
+
Calculate sum of Tensor elements over a given dim.
|
|
1308
1237
|
|
|
1309
|
-
|
|
1310
|
-
|
|
1238
|
+
Note:
|
|
1239
|
+
The `dim` with tensor type is only used for compatibility with older versions and is not recommended.
|
|
1311
1240
|
|
|
1312
1241
|
Args:
|
|
1313
|
-
input (Tensor): The input
|
|
1314
|
-
dim (int)
|
|
1242
|
+
input (Tensor): The input tensor.
|
|
1243
|
+
dim (Union[None, int, tuple(int), list(int), Tensor]): Dimensions along which a sum is performed.
|
|
1244
|
+
If ``None`` , sum all the elements of the input tensor.
|
|
1245
|
+
If the `dim` is a tuple or list of ints, a sum is performed on all the dimensions specified in the tuple.
|
|
1246
|
+
Must be in the range :math:`[-input.ndim, input.ndim)` . Default: ``None`` .
|
|
1247
|
+
keepdim (bool): Whether the output tensor has `dim` retained or not.
|
|
1248
|
+
If ``True`` , keep these reduced dimensions and the length is 1.
|
|
1249
|
+
If ``False`` , don't keep these dimensions. Default: ``False`` .
|
|
1250
|
+
dtype (:class:`mindspore.dtype`): The desired data type of returned Tensor. Default: ``None`` .
|
|
1315
1251
|
|
|
1316
1252
|
Returns:
|
|
1317
|
-
|
|
1318
|
-
The shape of each output tensor is the same as that of input `input`.
|
|
1253
|
+
A Tensor, sum of elements over a given `dim` in `input`.
|
|
1319
1254
|
|
|
1320
1255
|
Raises:
|
|
1321
1256
|
TypeError: If `input` is not a Tensor.
|
|
1322
|
-
TypeError: If `
|
|
1323
|
-
|
|
1324
|
-
|
|
1257
|
+
TypeError: If `dim` is not an int, tulpe(int), list(int), Tensor or None.
|
|
1258
|
+
ValueError: If `dim` is not in the range :math:`[-input.ndim, input.ndim)` .
|
|
1259
|
+
TypeError: If `keepdim` is not a bool.
|
|
1325
1260
|
|
|
1326
1261
|
Supported Platforms:
|
|
1327
|
-
``Ascend``
|
|
1262
|
+
``Ascend`` ``GPU`` ``CPU``
|
|
1328
1263
|
|
|
1329
1264
|
Examples:
|
|
1330
|
-
>>> from mindspore import Tensor, ops
|
|
1331
1265
|
>>> import mindspore
|
|
1332
|
-
>>>
|
|
1333
|
-
>>>
|
|
1334
|
-
>>>
|
|
1335
|
-
|
|
1336
|
-
|
|
1337
|
-
[
|
|
1338
|
-
|
|
1339
|
-
|
|
1340
|
-
|
|
1341
|
-
|
|
1342
|
-
|
|
1343
|
-
|
|
1344
|
-
|
|
1266
|
+
>>> import numpy as np
|
|
1267
|
+
>>> from mindspore import Tensor, ops
|
|
1268
|
+
>>> from mindspore import dtype as mstype
|
|
1269
|
+
>>> x = Tensor(np.array([[[1, 1, 1, 1, 1, 1], [2, 2, 2, 2, 2, 2], [3, 3, 3, 3, 3, 3]],
|
|
1270
|
+
... [[4, 4, 4, 4, 4, 4], [5, 5, 5, 5, 5, 5], [6, 6, 6, 6, 6, 6]],
|
|
1271
|
+
... [[7, 7, 7, 7, 7, 7], [8, 8, 8, 8, 8, 8], [9, 9, 9, 9, 9, 9]]]), mstype.float32)
|
|
1272
|
+
>>> out = ops.sum_ext(x)
|
|
1273
|
+
>>> print(out)
|
|
1274
|
+
270.0
|
|
1275
|
+
>>> out = ops.sum_ext(x, dim=2)
|
|
1276
|
+
>>> print(out)
|
|
1277
|
+
[[ 6. 12. 18.]
|
|
1278
|
+
[24. 30. 36.]
|
|
1279
|
+
[42. 48. 54.]]
|
|
1280
|
+
>>> out = ops.sum_ext(x, dim=2, keepdim=True)
|
|
1281
|
+
>>> print(out)
|
|
1282
|
+
[[[ 6.]
|
|
1283
|
+
[12.]
|
|
1284
|
+
[18.]]
|
|
1285
|
+
[[24.]
|
|
1286
|
+
[30.]
|
|
1287
|
+
[36.]]
|
|
1288
|
+
[[42.]
|
|
1289
|
+
[48.]
|
|
1290
|
+
[54.]]]
|
|
1345
1291
|
"""
|
|
1346
|
-
return
|
|
1292
|
+
return sum_impl(input, dim, keepdim, dtype)
|
|
1347
1293
|
|
|
1348
1294
|
|
|
1349
|
-
def
|
|
1295
|
+
def sub(input, other, alpha=1):
|
|
1350
1296
|
r"""
|
|
1351
|
-
|
|
1297
|
+
Subtracts scaled other value from input Tensor.
|
|
1352
1298
|
|
|
1353
1299
|
.. math::
|
|
1354
1300
|
|
|
1355
|
-
out_{i} = input_{i}
|
|
1301
|
+
out_{i} = input_{i} - alpha \times other_{i}
|
|
1356
1302
|
|
|
1357
1303
|
Note:
|
|
1358
1304
|
- When the two inputs have different shapes,
|
|
@@ -1388,381 +1334,204 @@ def add(input, other, alpha=1):
|
|
|
1388
1334
|
>>> import mindspore
|
|
1389
1335
|
>>> from mindspore import Tensor
|
|
1390
1336
|
>>> from mindspore import ops
|
|
1391
|
-
>>> x = Tensor(
|
|
1392
|
-
>>> y = Tensor(
|
|
1337
|
+
>>> x = Tensor(np.array([4, 5, 6]).astype(np.float32))
|
|
1338
|
+
>>> y = Tensor(1, mindspore.int32)
|
|
1393
1339
|
>>> alpha = 0.5
|
|
1394
|
-
>>> output = ops.auto_generate.
|
|
1340
|
+
>>> output = ops.auto_generate.sub_ext(x, y, alpha)
|
|
1395
1341
|
>>> print(output)
|
|
1396
|
-
[3.
|
|
1397
|
-
>>> # the data type of x is
|
|
1342
|
+
[3.5 4.5 5.5]
|
|
1343
|
+
>>> # the data type of x is float32, the data type of y is int32,
|
|
1398
1344
|
>>> # alpha is a float, and the output is the data format of higher precision float32.
|
|
1399
1345
|
>>> print(output.dtype)
|
|
1400
1346
|
Float32
|
|
1401
1347
|
"""
|
|
1402
|
-
return
|
|
1348
|
+
return sub_impl(input, other, alpha)
|
|
1403
1349
|
|
|
1404
1350
|
|
|
1405
|
-
def
|
|
1351
|
+
def index_select(input, dim, index):
|
|
1406
1352
|
r"""
|
|
1407
|
-
|
|
1408
|
-
|
|
1409
|
-
|
|
1410
|
-
|
|
1411
|
-
the maximum value in the input plus one.
|
|
1412
|
-
|
|
1413
|
-
Each value in the output Tensor represents the number of occurrences of that index value in the input.
|
|
1414
|
-
If `weights` is specified, the output results are weighted,
|
|
1415
|
-
i.e., :math:`out[n] += weight[i]` instead of :math:`out[n] += 1`.
|
|
1353
|
+
Generates a new Tensor that accesses the values of `input` along the specified `dim` dimension
|
|
1354
|
+
using the indices specified in `index`. The new Tensor has the same number of dimensions as `input`,
|
|
1355
|
+
with the size of the `dim` dimension being equal to the length of `index`, and the size of all other
|
|
1356
|
+
dimensions will be unchanged from the original `input` Tensor.
|
|
1416
1357
|
|
|
1417
|
-
..
|
|
1418
|
-
|
|
1358
|
+
.. note::
|
|
1359
|
+
The value of index must be in the range of `[0, input.shape[dim])`, the result is undefined out of range.
|
|
1419
1360
|
|
|
1420
1361
|
Args:
|
|
1421
|
-
input (Tensor):
|
|
1422
|
-
|
|
1423
|
-
|
|
1362
|
+
input (Tensor): The input Tensor.
|
|
1363
|
+
dim (int): The dimension to be indexed.
|
|
1364
|
+
index (Tensor): A 1-D Tensor with the indices.
|
|
1424
1365
|
|
|
1425
1366
|
Returns:
|
|
1426
|
-
Tensor,
|
|
1427
|
-
otherwise the shape is :math:`(0, )`.
|
|
1367
|
+
Tensor, has the same dtype as input Tensor.
|
|
1428
1368
|
|
|
1429
1369
|
Raises:
|
|
1430
|
-
TypeError: If `input` or `
|
|
1431
|
-
|
|
1432
|
-
ValueError: If `
|
|
1370
|
+
TypeError: If `input` or `index` is not a Tensor.
|
|
1371
|
+
TypeError: If `dim` is not int number.
|
|
1372
|
+
ValueError: If the value of `dim` is out the range of `[-input.ndim, input.ndim - 1]`.
|
|
1373
|
+
ValueError: If the dimension of `index` is not equal to 1.
|
|
1433
1374
|
|
|
1434
1375
|
Supported Platforms:
|
|
1435
1376
|
``Ascend``
|
|
1436
1377
|
|
|
1437
|
-
Examples:
|
|
1438
|
-
>>> from mindspore import mint, Tensor
|
|
1439
|
-
>>> print(mint.bincount(Tensor(np.arange(5))))
|
|
1440
|
-
[1 1 1 1 1]
|
|
1441
|
-
>>> print(mint.bincount(Tensor(np.array([0, 1, 1, 3, 2, 1, 7]))))
|
|
1442
|
-
[1 3 1 1 0 0 0 1]
|
|
1443
|
-
>>> w = Tensor(np.array([0.3, 0.5, 0.2, 0.7, 1., -0.6])) # weights
|
|
1444
|
-
>>> x = Tensor(np.array([0, 1, 1, 2, 2, 2]))
|
|
1445
|
-
>>> print(mint.bincount(x, weights=w, minlength=5))
|
|
1446
|
-
[0.3 0.7 1.1 0. 0. ]
|
|
1447
|
-
"""
|
|
1448
|
-
return bincount_impl(input, weights, minlength)
|
|
1449
|
-
|
|
1450
|
-
|
|
1451
|
-
def bmm(input, mat2):
|
|
1452
|
-
r"""
|
|
1453
|
-
Performs batch matrix-matrix multiplication of two three-dimensional tensors.
|
|
1454
|
-
|
|
1455
|
-
.. math::
|
|
1456
|
-
\text{output}= \text{input} @ \text{mat2}
|
|
1457
|
-
|
|
1458
|
-
Args:
|
|
1459
|
-
input (Tensor): The first batch of matrices to be multiplied. Must be a three-dimensional tensor of shape `(b, n, m)`.
|
|
1460
|
-
mat2 (Tensor): The second batch of matrices to be multiplied. Must be a three-dimensional tensor of shape `(b, m, p)`.
|
|
1461
|
-
|
|
1462
|
-
Returns:
|
|
1463
|
-
Tensor, the output tensor of shape `(b, n, p)`, where each matrix is the product of the corresponding matrices in the input batches.
|
|
1464
|
-
|
|
1465
|
-
Raises:
|
|
1466
|
-
ValueError: If `input` or `mat2` is not three-dimensional tensors.
|
|
1467
|
-
ValueError: If the length of the third dimension of `input` is not equal to the length of the second dimension of `mat2`.
|
|
1468
|
-
ValueError: If the batch size of the inputs is not equal to the batch size of the mat2.
|
|
1469
|
-
|
|
1470
|
-
Supported Platforms:
|
|
1471
|
-
``Ascend`` ``GPU`` ``CPU``
|
|
1472
|
-
|
|
1473
1378
|
Examples:
|
|
1474
1379
|
>>> import mindspore
|
|
1380
|
+
>>> from mindspore import Tensor, ops
|
|
1475
1381
|
>>> import numpy as np
|
|
1476
|
-
>>>
|
|
1477
|
-
>>>
|
|
1478
|
-
|
|
1479
|
-
|
|
1480
|
-
|
|
1481
|
-
|
|
1482
|
-
|
|
1483
|
-
|
|
1484
|
-
|
|
1485
|
-
|
|
1486
|
-
|
|
1487
|
-
[4. 4. 4. 4. 4.]]]
|
|
1382
|
+
>>> input = Tensor(np.arange(16).astype(np.float32).reshape(2, 2, 4))
|
|
1383
|
+
>>> print(input)
|
|
1384
|
+
[[[ 0. 1. 2. 3.]
|
|
1385
|
+
[ 4. 5. 6. 7.]]
|
|
1386
|
+
[[ 8. 9. 10. 11.]
|
|
1387
|
+
[12. 13. 14. 15.]]]
|
|
1388
|
+
>>> index = Tensor([0,], mindspore.int32)
|
|
1389
|
+
>>> y = ops.auto_generate.index_select_ext(input, 1, index)
|
|
1390
|
+
>>> print(y)
|
|
1391
|
+
[[[ 0. 1. 2. 3.]]
|
|
1392
|
+
[[ 8. 9. 10. 11.]]]
|
|
1488
1393
|
"""
|
|
1489
|
-
return
|
|
1394
|
+
return index_select_impl(input, dim, index)
|
|
1490
1395
|
|
|
1491
1396
|
|
|
1492
|
-
def
|
|
1397
|
+
def histc(input, bins=100, min=0, max=0):
|
|
1493
1398
|
r"""
|
|
1494
|
-
Computes the
|
|
1399
|
+
Computes the histogram of a tensor.
|
|
1495
1400
|
|
|
1496
|
-
|
|
1401
|
+
The elements are sorted into equal width bins between `min` and `max`.
|
|
1402
|
+
If `min` and `max` are both zero, the minimum and maximum values of the data are used.
|
|
1497
1403
|
|
|
1498
|
-
|
|
1404
|
+
Elements lower than min or higher than max are ignored.
|
|
1499
1405
|
|
|
1500
1406
|
.. warning::
|
|
1501
1407
|
This is an experimental API that is subject to change or deletion.
|
|
1408
|
+
If input is int64, valid values fit within int32; exceeding this may cause precision errors.
|
|
1502
1409
|
|
|
1503
1410
|
Args:
|
|
1504
|
-
input (Tensor):
|
|
1505
|
-
|
|
1506
|
-
|
|
1411
|
+
input (Tensor): the input tensor.
|
|
1412
|
+
bins (int, optional): Number of histogram bins, optional. If specified, must be positive. Default: ``100`` .
|
|
1413
|
+
min (int, float, optional): the lower end of the range (inclusive), optional. Default: ``0`` .
|
|
1414
|
+
max (int, float, optional): the upper end of the range (inclusive), optional. Default: ``0`` .
|
|
1507
1415
|
|
|
1508
1416
|
Returns:
|
|
1509
|
-
Tensor, the
|
|
1510
|
-
|
|
1511
|
-
- If `input.dtype` is in [float16, float32, bfloat16], the output dtype is the same as the `input.dtype`.
|
|
1512
|
-
- If `input.dtype` is an integer or boolean type, the output dtype is float32.
|
|
1513
|
-
- If `dim` is (), and `keepdim` is False, the output is a 0-D tensor representing the logarithm of the sum of exponentiations of all elements in the `input` tensor.
|
|
1514
|
-
- If `dim` is `1`, and `keepdim` is False, the shape of output is :math:`(input.shape[0], input.shape[2], ..., input.shape[n])`.
|
|
1515
|
-
- If `dim` is `(1, 2)`, and `keepdim` is False, the shape of output is :math:`(input.shape[0], input.shape[3], ..., input.shape[n])`.
|
|
1417
|
+
A 1-D Tensor, has the same type as `input` with the shape :math:`(bins, )`.
|
|
1516
1418
|
|
|
1517
1419
|
Raises:
|
|
1518
1420
|
TypeError: If `input` is not a Tensor.
|
|
1519
|
-
TypeError: If
|
|
1520
|
-
TypeError: If `
|
|
1521
|
-
TypeError: If `
|
|
1522
|
-
ValueError: If
|
|
1523
|
-
|
|
1421
|
+
TypeError: If `input` datatype is not in support list.
|
|
1422
|
+
TypeError: If attr `min` or `max` is not float or int.
|
|
1423
|
+
TypeError: If attr `bins` is not int.
|
|
1424
|
+
ValueError: If attr value `min` > `max`.
|
|
1425
|
+
ValueError: If attr `bins` <= 0.
|
|
1524
1426
|
|
|
1525
1427
|
Supported Platforms:
|
|
1526
1428
|
``Ascend``
|
|
1527
1429
|
|
|
1528
1430
|
Examples:
|
|
1529
|
-
>>> import numpy as np
|
|
1530
1431
|
>>> from mindspore import Tensor, ops
|
|
1531
|
-
>>> x = Tensor(
|
|
1532
|
-
>>>
|
|
1533
|
-
>>> print(
|
|
1534
|
-
|
|
1432
|
+
>>> x = Tensor([1., 2, 1])
|
|
1433
|
+
>>> y = ops.histc_ext(x, bins=4, min=0, max=3)
|
|
1434
|
+
>>> print(y)
|
|
1435
|
+
[0 2 1 0]
|
|
1535
1436
|
"""
|
|
1536
|
-
return
|
|
1437
|
+
return histc_impl(input, bins, min, max)
|
|
1537
1438
|
|
|
1538
1439
|
|
|
1539
|
-
def
|
|
1440
|
+
def sort(input, dim=-1, descending=False, stable=False):
|
|
1540
1441
|
r"""
|
|
1541
|
-
|
|
1542
|
-
|
|
1543
|
-
.. math::
|
|
1544
|
-
|
|
1545
|
-
y_i = x_1 + x_2 + x_3 + ... + x_i
|
|
1546
|
-
|
|
1547
|
-
Args:
|
|
1548
|
-
input (Tensor): The input Tensor.
|
|
1549
|
-
dim (int): Dim along which the cumulative sum is computed.
|
|
1550
|
-
dtype (:class:`mindspore.dtype`, optional): The desired dtype of returned Tensor. If specified,
|
|
1551
|
-
the input Tensor will be cast to `dtype` before the computation. This is useful for preventing overflows.
|
|
1552
|
-
If not specified, stay the same as original Tensor. Default: ``None`` .
|
|
1553
|
-
|
|
1554
|
-
Returns:
|
|
1555
|
-
Tensor, the shape of the output Tensor is consistent with the input Tensor's.
|
|
1556
|
-
|
|
1557
|
-
Raises:
|
|
1558
|
-
TypeError: If `input` is not a Tensor.
|
|
1559
|
-
ValueError: If the `dim` is out of range.
|
|
1560
|
-
|
|
1561
|
-
Supported Platforms:
|
|
1562
|
-
``Ascend``
|
|
1563
|
-
|
|
1564
|
-
Examples:
|
|
1565
|
-
>>> import numpy as np
|
|
1566
|
-
>>> from mindspore import Tensor
|
|
1567
|
-
>>> import mindspore.ops as ops
|
|
1568
|
-
>>> x = Tensor(np.array([[3, 4, 6, 10], [1, 6, 7, 9], [4, 3, 8, 7], [1, 3, 7, 9]]).astype(np.float32))
|
|
1569
|
-
>>> # case 1: along the dim 0
|
|
1570
|
-
>>> y = ops.auto_generate.cumsum_ext(x, 0)
|
|
1571
|
-
>>> print(y)
|
|
1572
|
-
[[ 3. 4. 6. 10.]
|
|
1573
|
-
[ 4. 10. 13. 19.]
|
|
1574
|
-
[ 8. 13. 21. 26.]
|
|
1575
|
-
[ 9. 16. 28. 35.]]
|
|
1576
|
-
>>> # case 2: along the dim 1
|
|
1577
|
-
>>> y = ops.auto_generate.cumsum_ext(x, 1)
|
|
1578
|
-
>>> print(y)
|
|
1579
|
-
[[ 3. 7. 13. 23.]
|
|
1580
|
-
[ 1. 7. 14. 23.]
|
|
1581
|
-
[ 4. 7. 15. 22.]
|
|
1582
|
-
[ 1. 4. 11. 20.]]
|
|
1442
|
+
None
|
|
1583
1443
|
"""
|
|
1584
|
-
return
|
|
1444
|
+
return sort_impl(input, dim, descending, stable)
|
|
1585
1445
|
|
|
1586
1446
|
|
|
1587
|
-
def
|
|
1447
|
+
def matmul(input, other):
|
|
1588
1448
|
r"""
|
|
1589
|
-
|
|
1590
|
-
|
|
1591
|
-
.. math::
|
|
1592
|
-
|
|
1593
|
-
out_i = \sin^{-1}(input_i)
|
|
1594
|
-
|
|
1595
|
-
Args:
|
|
1596
|
-
input (Tensor): The shape of tensor is
|
|
1597
|
-
:math:`(N,*)`, where :math:`*` means any number of additional dimensions.
|
|
1598
|
-
|
|
1599
|
-
Returns:
|
|
1600
|
-
Tensor, has the same shape as `input`. The dtype of output is float32 when dtype of `input` is in [bool, int8, uint8, int16, int32, int64]. Otherwise output has the same dtype as `input`.
|
|
1601
|
-
|
|
1602
|
-
Raises:
|
|
1603
|
-
TypeError: If `input` is not a Tensor.
|
|
1604
|
-
|
|
1605
|
-
Supported Platforms:
|
|
1606
|
-
``Ascend`` ``GPU`` ``CPU``
|
|
1607
|
-
|
|
1608
|
-
Examples:
|
|
1609
|
-
>>> import mindspore
|
|
1610
|
-
>>> import numpy as np
|
|
1611
|
-
>>> from mindspore import Tensor, ops
|
|
1612
|
-
>>> input = Tensor(np.array([0.74, 0.04, 0.30, 0.56]), mindspore.float32)
|
|
1613
|
-
>>> output = ops.asin_ext(input)
|
|
1614
|
-
>>> print(output)
|
|
1615
|
-
[0.8330927 0.04001068 0.30469266 0.59438497 ]
|
|
1449
|
+
None
|
|
1616
1450
|
"""
|
|
1617
|
-
return
|
|
1451
|
+
return matmul_impl(input, other)
|
|
1618
1452
|
|
|
1619
1453
|
|
|
1620
|
-
def
|
|
1454
|
+
def l1_loss(input, target, reduction='mean'):
|
|
1621
1455
|
r"""
|
|
1622
|
-
|
|
1623
|
-
|
|
1624
|
-
The activation function is defined as:
|
|
1625
|
-
|
|
1626
|
-
.. math::
|
|
1627
|
-
\text{leaky_relu}(input) = \begin{cases}input, &\text{if } input \geq 0; \cr
|
|
1628
|
-
\text{negative_slope} * input, &\text{otherwise.}\end{cases}
|
|
1629
|
-
|
|
1630
|
-
where :math:`negative\_slope` represents the `negative_slope` parameter.
|
|
1631
|
-
|
|
1632
|
-
For more details, see `Rectifier Nonlinearities Improve Neural Network Acoustic Models
|
|
1633
|
-
<https://ai.stanford.edu/~amaas/papers/relu_hybrid_icml2013_final.pdf>`_.
|
|
1634
|
-
|
|
1635
|
-
LeakyReLU Activation Function Graph:
|
|
1636
|
-
|
|
1637
|
-
.. image:: ../images/LeakyReLU.png
|
|
1638
|
-
:align: center
|
|
1639
|
-
|
|
1640
|
-
Args:
|
|
1641
|
-
input (Tensor): The input of leaky_relu is a Tensor of any dimension.
|
|
1642
|
-
negative_slope (Union[int, float], optional): Slope of the activation function when the element of `input` is less than 0.
|
|
1643
|
-
Default: ``0.01`` .
|
|
1644
|
-
|
|
1645
|
-
Returns:
|
|
1646
|
-
Tensor, has the same type and shape as the `input`.
|
|
1647
|
-
|
|
1648
|
-
Raises:
|
|
1649
|
-
TypeError: If `input` is not a Tensor.
|
|
1650
|
-
TypeError: If `negative_slope` is not a float or an int.
|
|
1456
|
+
Calculate the mean absolute error between the `input` value and the `target` value.
|
|
1651
1457
|
|
|
1652
|
-
|
|
1653
|
-
|
|
1458
|
+
Assuming that the :math:`x` and :math:`y` are the predicted value and target value,
|
|
1459
|
+
both are one-dimensional tensors of length :math:`N`, length :math:`N`, `reduction` is set to ``'none'`` ,
|
|
1460
|
+
then calculate the loss of :math:`x` and :math:`y` without dimensionality reduction.
|
|
1654
1461
|
|
|
1655
|
-
|
|
1656
|
-
>>> import mindspore
|
|
1657
|
-
>>> import numpy as np
|
|
1658
|
-
>>> from mindspore import Tensor, ops
|
|
1659
|
-
>>> input = Tensor(np.array([[-1.0, 4.0, -8.0], [2.0, -5.0, 9.0]]), mindspore.float32)
|
|
1660
|
-
>>> print(ops.extend.leaky_relu_ext(input, negative_slope=0.2))
|
|
1661
|
-
[[-0.2 4. -1.6]
|
|
1662
|
-
[ 2. -1. 9. ]]
|
|
1663
|
-
"""
|
|
1664
|
-
return leaky_relu_impl(input, negative_slope)
|
|
1665
|
-
|
|
1666
|
-
|
|
1667
|
-
def topk(input, k, dim=-1, largest=True, sorted=True):
|
|
1668
|
-
r"""
|
|
1669
|
-
Finds values and indices of the `k` largest or smallest entries along a given dimension.
|
|
1462
|
+
The formula is as follows:
|
|
1670
1463
|
|
|
1671
|
-
..
|
|
1672
|
-
|
|
1673
|
-
the display order of calculation results may be inconsistent when `sorted` is False.
|
|
1464
|
+
.. math::
|
|
1465
|
+
\ell(x, y) = L = \{l_1,\dots,l_N\}^\top, \quad \text{with } l_n = \left| x_n - y_n \right|,
|
|
1674
1466
|
|
|
1675
|
-
|
|
1676
|
-
and outputs its value and index as a Tensor. values[`k`] is the `k` largest item in `input`,
|
|
1677
|
-
and its index is indices [`k`].
|
|
1467
|
+
where :math:`N` is the batch size.
|
|
1678
1468
|
|
|
1679
|
-
|
|
1680
|
-
calculates the first or last `k` entries in a given dimension, therefore:
|
|
1469
|
+
If `reduction` is ``'mean'`` or ``'sum'`` , then:
|
|
1681
1470
|
|
|
1682
1471
|
.. math::
|
|
1683
|
-
|
|
1684
|
-
|
|
1685
|
-
|
|
1686
|
-
|
|
1472
|
+
\ell(x, y) =
|
|
1473
|
+
\begin{cases}
|
|
1474
|
+
\operatorname{mean}(L), & \text{if reduction} = \text{'mean';}\\
|
|
1475
|
+
\operatorname{sum}(L), & \text{if reduction} = \text{'sum'.}
|
|
1476
|
+
\end{cases}
|
|
1687
1477
|
|
|
1688
1478
|
Args:
|
|
1689
|
-
input (Tensor):
|
|
1690
|
-
|
|
1691
|
-
|
|
1692
|
-
|
|
1693
|
-
Default: ``
|
|
1694
|
-
sorted (bool, optional): If ``True`` , the obtained elements will be sorted by the values in descending
|
|
1695
|
-
order or ascending order according to `largest`. If ``False`` , the obtained elements will not be
|
|
1696
|
-
sorted. Default: ``True`` .
|
|
1479
|
+
input (Tensor): Predicted value, Tensor of any dimension.
|
|
1480
|
+
target (Tensor): Target value, usually has the same shape as the `input`.
|
|
1481
|
+
If `input` and `target` have different shapes, make sure they can broadcast to each other.
|
|
1482
|
+
reduction (str, optional): Apply specific reduction method to the output: ``'none'`` , ``'mean'`` ,
|
|
1483
|
+
``'sum'`` . Default: ``'mean'`` .
|
|
1697
1484
|
|
|
1698
|
-
|
|
1699
|
-
|
|
1485
|
+
- ``'none'``: no reduction will be applied.
|
|
1486
|
+
- ``'mean'``: compute and return the mean of elements in the output. Notice: At least one of the input and target is float type when the reduction is ``'mean'`` .
|
|
1487
|
+
- ``'sum'``: the output elements will be summed.
|
|
1700
1488
|
|
|
1701
|
-
|
|
1702
|
-
|
|
1489
|
+
Returns:
|
|
1490
|
+
Tensor or Scalar, if `reduction` is ``'none'`` , return a Tensor with same shape and dtype as `input`.
|
|
1491
|
+
Otherwise, a scalar value will be returned.
|
|
1703
1492
|
|
|
1704
1493
|
Raises:
|
|
1705
|
-
TypeError: If `sorted` is not a bool.
|
|
1706
1494
|
TypeError: If `input` is not a Tensor.
|
|
1707
|
-
TypeError: If `
|
|
1495
|
+
TypeError: If `target` is not a Tensor.
|
|
1496
|
+
ValueError: If `reduction` is not one of ``'none'`` , ``'mean'`` or ``'sum'`` .
|
|
1708
1497
|
|
|
1709
1498
|
Supported Platforms:
|
|
1710
1499
|
``Ascend``
|
|
1711
1500
|
|
|
1712
1501
|
Examples:
|
|
1713
|
-
>>>
|
|
1714
|
-
>>> from mindspore import
|
|
1715
|
-
>>> x =
|
|
1716
|
-
|
|
1717
|
-
|
|
1718
|
-
>>> output = ops.topk_ext(x, 2, dim=1)
|
|
1502
|
+
>>> from mindspore import Tensor, ops
|
|
1503
|
+
>>> from mindspore import dtype as mstype
|
|
1504
|
+
>>> x = Tensor([[1, 2, 3], [4, 5, 6]], mstype.float32)
|
|
1505
|
+
>>> target = Tensor([[6, 5, 4], [3, 2, 1]], mstype.float32)
|
|
1506
|
+
>>> output = ops.l1_loss_ext(x, target, reduction="mean")
|
|
1719
1507
|
>>> print(output)
|
|
1720
|
-
|
|
1721
|
-
[[ 9.67299998e-01, 5.36800027e-01],
|
|
1722
|
-
[ 6.52499974e-01, 4.68499988e-01],
|
|
1723
|
-
[ 9.67499971e-01, 8.23000014e-01]]), Tensor(shape=[3, 2], dtype=Int64, value=
|
|
1724
|
-
[[3, 0],
|
|
1725
|
-
[1, 2],
|
|
1726
|
-
[2, 3]]))
|
|
1727
|
-
>>> output2 = ops.topk_ext(x, 2, dim=1, largest=False)
|
|
1728
|
-
>>> print(output2)
|
|
1729
|
-
(Tensor(shape=[3, 2], dtype=Float32, value=
|
|
1730
|
-
[[ 2.44700000e-01, 4.30200011e-01],
|
|
1731
|
-
[ 1.86800003e-01, 4.38800007e-01],
|
|
1732
|
-
[ 3.56299996e-01, 5.15200019e-01]]), Tensor(shape=[3, 2], dtype=Int64, value=
|
|
1733
|
-
[[1, 2],
|
|
1734
|
-
[3, 0],
|
|
1735
|
-
[0, 1]]))
|
|
1508
|
+
3.0
|
|
1736
1509
|
"""
|
|
1737
|
-
return
|
|
1510
|
+
return l1_loss_impl(input, target, converted_reduction)
|
|
1738
1511
|
|
|
1739
1512
|
|
|
1740
|
-
def
|
|
1513
|
+
def outer(input, vec2):
|
|
1741
1514
|
r"""
|
|
1742
|
-
|
|
1743
|
-
|
|
1744
|
-
the Log Softmax function is shown as follows:
|
|
1515
|
+
Return outer product of `input` and `vec2`. If `input` is a vector of size :math:`n`
|
|
1516
|
+
and `vec2` is a vector of size :math:`m` , then output must be a matrix of shape :math:`(n, m)` .
|
|
1745
1517
|
|
|
1746
|
-
..
|
|
1747
|
-
|
|
1518
|
+
.. warning::
|
|
1519
|
+
This is an experimental API that is subject to change or deletion.
|
|
1748
1520
|
|
|
1749
|
-
|
|
1521
|
+
.. note::
|
|
1522
|
+
This function does not broadcast.
|
|
1750
1523
|
|
|
1751
1524
|
Args:
|
|
1752
|
-
input (Tensor):
|
|
1753
|
-
|
|
1754
|
-
|
|
1755
|
-
Keyword Args:
|
|
1756
|
-
dtype (:class:`mindspore.dtype`, optional): The desired dtype of returned Tensor. If not set to None, the input
|
|
1757
|
-
Tensor will be cast to `dtype` before the operation is performed. This is useful for preventing overflows.
|
|
1758
|
-
If set to None, stay the same as original Tensor. Default: ``None`` . Supported data type is {float16, float32, double, bfloat16}.
|
|
1525
|
+
input (Tensor): 1-D input vector.
|
|
1526
|
+
vec2 (Tensor): 1-D input vector.
|
|
1759
1527
|
|
|
1760
1528
|
Returns:
|
|
1761
|
-
|
|
1529
|
+
out, 2-D matrix, the outer product of two vectors.
|
|
1762
1530
|
|
|
1763
1531
|
Raises:
|
|
1764
|
-
TypeError: If `
|
|
1765
|
-
|
|
1532
|
+
TypeError: If `input` or `vec2` is not a Tensor.
|
|
1533
|
+
TypeError: The implicitly converted data types of `input` and `vec2` are not one of float16, float32, float64, bool, uint8, int8, int16, int32, int64, complex64, complex128, bfloat16
|
|
1534
|
+
ValueError: If the dimension of `input` or `vec2` is not equal to 1.
|
|
1766
1535
|
|
|
1767
1536
|
Supported Platforms:
|
|
1768
1537
|
``Ascend``
|
|
@@ -1770,82 +1539,52 @@ def log_softmax(input, dim=None, dtype=None):
|
|
|
1770
1539
|
Examples:
|
|
1771
1540
|
>>> import mindspore
|
|
1772
1541
|
>>> import numpy as np
|
|
1773
|
-
>>> from mindspore import Tensor
|
|
1774
|
-
>>>
|
|
1775
|
-
>>>
|
|
1776
|
-
>>>
|
|
1777
|
-
|
|
1542
|
+
>>> from mindspore import Tensor
|
|
1543
|
+
>>> from mindspore import ops
|
|
1544
|
+
>>> input = Tensor(np.array([7, 8, 9]), mindspore.int32)
|
|
1545
|
+
>>> vec2 = Tensor(np.array([7, 10, 11]), mindspore.int32)
|
|
1546
|
+
>>> out = ops.outer(input, vec2)
|
|
1547
|
+
>>> print(out)
|
|
1548
|
+
[[49 70 77]
|
|
1549
|
+
[56 80 88]
|
|
1550
|
+
[63 90 99]]
|
|
1778
1551
|
"""
|
|
1779
|
-
return
|
|
1552
|
+
return outer_impl(input, vec2)
|
|
1780
1553
|
|
|
1781
1554
|
|
|
1782
|
-
def
|
|
1555
|
+
def selu(input):
|
|
1783
1556
|
r"""
|
|
1784
|
-
|
|
1785
|
-
If `input` is a :math:`(n \times m)` Tensor, `mat2` is a
|
|
1786
|
-
:math:`(m \times p)` Tensor, `out` will be a :math:`(n \times p)` Tensor.
|
|
1787
|
-
|
|
1788
|
-
Note:
|
|
1789
|
-
This function cannot support broadcasting.
|
|
1790
|
-
Refer to :func:`mindspore.ops.matmul` instead if you need a broadcastable function.
|
|
1791
|
-
|
|
1792
|
-
.. warning::
|
|
1793
|
-
This is an experimental API that is subject to change or deletion.
|
|
1557
|
+
Activation function SELU (Scaled exponential Linear Unit).
|
|
1794
1558
|
|
|
1795
|
-
|
|
1796
|
-
input (Tensor): The first matrix of matrix multiplication.
|
|
1797
|
-
The last dimension of `input` must be the same size as the first dimension of `mat2`.
|
|
1798
|
-
mat2 (Tensor): The second matrix of matrix multiplication.
|
|
1799
|
-
The last dimension of `input` must be the same size as the first dimension of `mat2`.
|
|
1559
|
+
The activation function is defined as:
|
|
1800
1560
|
|
|
1801
|
-
|
|
1802
|
-
|
|
1561
|
+
.. math::
|
|
1562
|
+
E_{i} =
|
|
1563
|
+
scale *
|
|
1564
|
+
\begin{cases}
|
|
1565
|
+
x_{i}, &\text{if } x_{i} \geq 0; \cr
|
|
1566
|
+
\text{alpha} * (\exp(x_i) - 1), &\text{otherwise.}
|
|
1567
|
+
\end{cases}
|
|
1803
1568
|
|
|
1804
|
-
|
|
1805
|
-
|
|
1806
|
-
second-to-last dimension of `mat2`.
|
|
1807
|
-
TypeError: If `input` or `mat2` is not a Tensor.
|
|
1808
|
-
TypeError: If dtype of `input` or `mat2` is not float16, float32 or bfloat16.
|
|
1569
|
+
where :math:`alpha` and :math:`scale` are pre-defined constants(:math:`alpha=1.67326324`
|
|
1570
|
+
and :math:`scale=1.05070098`).
|
|
1809
1571
|
|
|
1810
|
-
|
|
1811
|
-
``Ascend``
|
|
1572
|
+
See more details in `Self-Normalizing Neural Networks <https://arxiv.org/abs/1706.02515>`_.
|
|
1812
1573
|
|
|
1813
|
-
|
|
1814
|
-
>>> import mindspore as ms
|
|
1815
|
-
>>> from mindspore import ops
|
|
1816
|
-
>>> import numpy as np
|
|
1817
|
-
>>> x1 = ms.Tensor(np.random.rand(2, 3), ms.float32)
|
|
1818
|
-
>>> x2 = ms.Tensor(np.random.rand(3, 4), ms.float32)
|
|
1819
|
-
>>> out = ops.mm_ext(x1, x2)
|
|
1820
|
-
>>> print(out.shape)
|
|
1821
|
-
(2, 4)
|
|
1822
|
-
"""
|
|
1823
|
-
return mm_impl(input, mat2)
|
|
1824
|
-
|
|
1825
|
-
|
|
1826
|
-
def index_select(input, dim, index):
|
|
1827
|
-
r"""
|
|
1828
|
-
Generates a new Tensor that accesses the values of `input` along the specified `dim` dimension
|
|
1829
|
-
using the indices specified in `index`. The new Tensor has the same number of dimensions as `input`,
|
|
1830
|
-
with the size of the `dim` dimension being equal to the length of `index`, and the size of all other
|
|
1831
|
-
dimensions will be unchanged from the original `input` Tensor.
|
|
1574
|
+
SELU Activation Function Graph:
|
|
1832
1575
|
|
|
1833
|
-
..
|
|
1834
|
-
|
|
1576
|
+
.. image:: ../images/SeLU.png
|
|
1577
|
+
:align: center
|
|
1835
1578
|
|
|
1836
1579
|
Args:
|
|
1837
|
-
input (Tensor):
|
|
1838
|
-
|
|
1839
|
-
index (Tensor): A 1-D Tensor with the indices.
|
|
1580
|
+
input (Tensor): Tensor of any dimension.
|
|
1581
|
+
The data type is float16, float32, bfloat16.
|
|
1840
1582
|
|
|
1841
1583
|
Returns:
|
|
1842
|
-
Tensor,
|
|
1584
|
+
Tensor, with the same type and shape as the `input`.
|
|
1843
1585
|
|
|
1844
1586
|
Raises:
|
|
1845
|
-
TypeError: If
|
|
1846
|
-
TypeError: If `dim` is not int number.
|
|
1847
|
-
ValueError: If the value of `dim` is out the range of `[-input.ndim, input.ndim - 1]`.
|
|
1848
|
-
ValueError: If the dimension of `index` is not equal to 1.
|
|
1587
|
+
TypeError: If dtype of `input` is not float16, float32, bfloat16.
|
|
1849
1588
|
|
|
1850
1589
|
Supported Platforms:
|
|
1851
1590
|
``Ascend``
|
|
@@ -1854,217 +1593,145 @@ def index_select(input, dim, index):
|
|
|
1854
1593
|
>>> import mindspore
|
|
1855
1594
|
>>> from mindspore import Tensor, ops
|
|
1856
1595
|
>>> import numpy as np
|
|
1857
|
-
>>> input = Tensor(np.
|
|
1858
|
-
>>>
|
|
1859
|
-
|
|
1860
|
-
[ 4.
|
|
1861
|
-
|
|
1862
|
-
[12. 13. 14. 15.]]]
|
|
1863
|
-
>>> index = Tensor([0,], mindspore.int32)
|
|
1864
|
-
>>> y = ops.auto_generate.index_select_ext(input, 1, index)
|
|
1865
|
-
>>> print(y)
|
|
1866
|
-
[[[ 0. 1. 2. 3.]]
|
|
1867
|
-
[[ 8. 9. 10. 11.]]]
|
|
1596
|
+
>>> input = Tensor(np.array([[-1.0, 4.0, -8.0], [2.0, -5.0, 9.0]]), mindspore.float32)
|
|
1597
|
+
>>> output = ops.auto_generate.selu_ext(input)
|
|
1598
|
+
>>> print(output)
|
|
1599
|
+
[[-1.1113307 4.202804 -1.7575096]
|
|
1600
|
+
[ 2.101402 -1.7462534 9.456309 ]]
|
|
1868
1601
|
"""
|
|
1869
|
-
return
|
|
1602
|
+
return selu_impl(input)
|
|
1870
1603
|
|
|
1871
1604
|
|
|
1872
|
-
def
|
|
1605
|
+
def elu(input, alpha=1.0):
|
|
1873
1606
|
r"""
|
|
1874
|
-
|
|
1607
|
+
Exponential Linear Unit activation function.
|
|
1875
1608
|
|
|
1876
|
-
|
|
1609
|
+
Applies the exponential linear unit function element-wise.
|
|
1610
|
+
The activation function is defined as:
|
|
1877
1611
|
|
|
1878
|
-
|
|
1612
|
+
.. math::
|
|
1879
1613
|
|
|
1880
|
-
|
|
1614
|
+
\text{ELU}(x)= \left\{
|
|
1615
|
+
\begin{array}{align}
|
|
1616
|
+
\alpha(e^{x} - 1) & \text{if } x \le 0\\
|
|
1617
|
+
x & \text{if } x \gt 0\\
|
|
1618
|
+
\end{array}\right.
|
|
1881
1619
|
|
|
1882
|
-
|
|
1620
|
+
Where :math:`x` is the element of input Tensor `input`, :math:`\alpha` is param `alpha`,
|
|
1621
|
+
it determines the smoothness of ELU.
|
|
1883
1622
|
|
|
1884
|
-
|
|
1623
|
+
ELU function graph:
|
|
1885
1624
|
|
|
1886
|
-
..
|
|
1887
|
-
|
|
1625
|
+
.. image:: ../images/ELU.png
|
|
1626
|
+
:align: center
|
|
1888
1627
|
|
|
1889
1628
|
Args:
|
|
1890
|
-
input (Tensor): The input
|
|
1891
|
-
|
|
1629
|
+
input (Tensor): The input of ELU is a Tensor of any dimension.
|
|
1630
|
+
alpha (float, optional): The alpha value of ELU, the data type is float.
|
|
1631
|
+
Default: ``1.0`` .
|
|
1892
1632
|
|
|
1893
1633
|
Returns:
|
|
1894
|
-
Tensor, has the same
|
|
1895
|
-
|
|
1896
|
-
- If `input` shape is :math:`(x_0)` : then output shape is :math:`(x_0 + \left | diagonal \right | , x_0 + \left | diagonal \right | )` 2-D Tensor.
|
|
1897
|
-
|
|
1898
|
-
- If `input` shape is :math:`(x_0, x_1)` : then output shape is main diagonal to move :math:`(\left | diagonal \right |)` elements remains elements' length 1-D Tensor.
|
|
1634
|
+
Tensor, has the same shape and data type as `input`.
|
|
1899
1635
|
|
|
1900
1636
|
Raises:
|
|
1901
|
-
TypeError: If `
|
|
1902
|
-
ValueError: If shape of `input` is not 1-D and 2-D.
|
|
1637
|
+
TypeError: If `alpha` is not a float.
|
|
1903
1638
|
|
|
1904
1639
|
Supported Platforms:
|
|
1905
1640
|
``Ascend``
|
|
1906
1641
|
|
|
1907
|
-
Examples:
|
|
1908
|
-
>>> from mindspore import Tensor, mint
|
|
1909
|
-
>>> input = Tensor([1, 2, 3, 4]).astype('int32')
|
|
1910
|
-
>>> output = mint.diag(input)
|
|
1911
|
-
>>> print(output)
|
|
1912
|
-
[[1 0 0 0]
|
|
1913
|
-
[0 2 0 0]
|
|
1914
|
-
[0 0 3 0]
|
|
1915
|
-
[0 0 0 4]]
|
|
1916
|
-
"""
|
|
1917
|
-
return diag_impl(input, diagonal)
|
|
1918
|
-
|
|
1919
|
-
|
|
1920
|
-
def sum(input, dim=None, keepdim=False, dtype=None):
|
|
1921
|
-
r"""
|
|
1922
|
-
Calculate sum of Tensor elements over a given dim.
|
|
1923
|
-
|
|
1924
|
-
Note:
|
|
1925
|
-
The `dim` with tensor type is only used for compatibility with older versions and is not recommended.
|
|
1926
|
-
|
|
1927
|
-
Args:
|
|
1928
|
-
input (Tensor): The input tensor.
|
|
1929
|
-
dim (Union[None, int, tuple(int), list(int), Tensor]): Dimensions along which a sum is performed.
|
|
1930
|
-
If ``None`` , sum all the elements of the input tensor.
|
|
1931
|
-
If the `dim` is a tuple or list of ints, a sum is performed on all the dimensions specified in the tuple.
|
|
1932
|
-
Must be in the range :math:`[-input.ndim, input.ndim)` . Default: ``None`` .
|
|
1933
|
-
keepdim (bool): Whether the output tensor has `dim` retained or not.
|
|
1934
|
-
If ``True`` , keep these reduced dimensions and the length is 1.
|
|
1935
|
-
If ``False`` , don't keep these dimensions. Default: ``False`` .
|
|
1936
|
-
dtype (:class:`mindspore.dtype`): The desired data type of returned Tensor. Default: ``None`` .
|
|
1937
|
-
|
|
1938
|
-
Returns:
|
|
1939
|
-
A Tensor, sum of elements over a given `dim` in `input`.
|
|
1940
|
-
|
|
1941
|
-
Raises:
|
|
1942
|
-
TypeError: If `input` is not a Tensor.
|
|
1943
|
-
TypeError: If `dim` is not an int, tulpe(int), list(int), Tensor or None.
|
|
1944
|
-
ValueError: If `dim` is not in the range :math:`[-input.ndim, input.ndim)` .
|
|
1945
|
-
TypeError: If `keepdim` is not a bool.
|
|
1946
|
-
|
|
1947
|
-
Supported Platforms:
|
|
1948
|
-
``Ascend`` ``GPU`` ``CPU``
|
|
1949
|
-
|
|
1950
1642
|
Examples:
|
|
1951
1643
|
>>> import mindspore
|
|
1952
1644
|
>>> import numpy as np
|
|
1953
1645
|
>>> from mindspore import Tensor, ops
|
|
1954
|
-
>>>
|
|
1955
|
-
>>>
|
|
1956
|
-
|
|
1957
|
-
|
|
1958
|
-
|
|
1959
|
-
>>> print(out)
|
|
1960
|
-
270.0
|
|
1961
|
-
>>> out = ops.sum_ext(x, dim=2)
|
|
1962
|
-
>>> print(out)
|
|
1963
|
-
[[ 6. 12. 18.]
|
|
1964
|
-
[24. 30. 36.]
|
|
1965
|
-
[42. 48. 54.]]
|
|
1966
|
-
>>> out = ops.sum_ext(x, dim=2, keepdim=True)
|
|
1967
|
-
>>> print(out)
|
|
1968
|
-
[[[ 6.]
|
|
1969
|
-
[12.]
|
|
1970
|
-
[18.]]
|
|
1971
|
-
[[24.]
|
|
1972
|
-
[30.]
|
|
1973
|
-
[36.]]
|
|
1974
|
-
[[42.]
|
|
1975
|
-
[48.]
|
|
1976
|
-
[54.]]]
|
|
1977
|
-
"""
|
|
1978
|
-
return sum_impl(input, dim, keepdim, dtype)
|
|
1979
|
-
|
|
1980
|
-
|
|
1981
|
-
def inplace_adds(input, other, alpha=1):
|
|
1982
|
-
r"""
|
|
1983
|
-
None
|
|
1646
|
+
>>> x = Tensor(np.array([[-1.0, 4.0, -8.0], [2.0, -5.0, 9.0]]), mindspore.float32)
|
|
1647
|
+
>>> output = ops.auto_generate.elu_ext(x)
|
|
1648
|
+
>>> print(output)
|
|
1649
|
+
[[-0.63212055 4. -0.99966455]
|
|
1650
|
+
[ 2. -0.99326205 9. ]]
|
|
1984
1651
|
"""
|
|
1985
|
-
return
|
|
1652
|
+
return elu_impl(input, alpha)
|
|
1986
1653
|
|
|
1987
1654
|
|
|
1988
|
-
def
|
|
1655
|
+
def stack(tensors, dim=0):
|
|
1989
1656
|
r"""
|
|
1990
|
-
|
|
1657
|
+
Stacks a list of tensors in specified dim.
|
|
1991
1658
|
|
|
1992
|
-
|
|
1993
|
-
y_i = \log_{10}(x_i)
|
|
1659
|
+
Stacks the list of input tensors with the same rank `R`, output is a tensor of rank `(R+1)`.
|
|
1994
1660
|
|
|
1995
|
-
|
|
1996
|
-
|
|
1997
|
-
|
|
1998
|
-
may be affacted.
|
|
1661
|
+
Given input tensors of shape :math:`(x_1, x_2, ..., x_R)`. Set the number of input tensors as `N`.
|
|
1662
|
+
If :math:`dim \ge 0`, the shape of the output tensor is
|
|
1663
|
+
:math:`(x_1, x_2, ..., x_{dim}, N, x_{dim+1}, ..., x_R)`.
|
|
1999
1664
|
|
|
2000
1665
|
Args:
|
|
2001
|
-
|
|
1666
|
+
tensors (Union[tuple, list]): A Tuple or list of Tensor objects with the same shape and type.
|
|
1667
|
+
dim (int, optional): Dimension to stack. The range is [-(R+1), R+1). Default: ``0`` .
|
|
2002
1668
|
|
|
2003
1669
|
Returns:
|
|
2004
|
-
Tensor
|
|
2005
|
-
|
|
2006
|
-
- if `input.dtype` is in [float16, float32, float64, bfloat16], the output dtype is the same as the `input.dtype`.
|
|
2007
|
-
- if `input.dtype` is integer or boolean type, the output dtype is float32.
|
|
1670
|
+
Tensor. A stacked Tensor with the same type as `tensors`.
|
|
2008
1671
|
|
|
2009
1672
|
Raises:
|
|
2010
|
-
TypeError: If `
|
|
1673
|
+
TypeError: If the data types of elements in `tensors` are not the same.
|
|
1674
|
+
ValueError: If `dim` is out of the range [-(R+1), R+1);
|
|
1675
|
+
or if the shapes of elements in `tensors` are not the same.
|
|
2011
1676
|
|
|
2012
1677
|
Supported Platforms:
|
|
2013
1678
|
``Ascend``
|
|
2014
1679
|
|
|
2015
1680
|
Examples:
|
|
2016
1681
|
>>> import mindspore
|
|
1682
|
+
>>> from mindspore import Tensor, ops
|
|
2017
1683
|
>>> import numpy as np
|
|
2018
|
-
>>>
|
|
2019
|
-
>>>
|
|
2020
|
-
>>> output =
|
|
1684
|
+
>>> data1 = Tensor(np.array([0, 1]).astype(np.float32))
|
|
1685
|
+
>>> data2 = Tensor(np.array([2, 3]).astype(np.float32))
|
|
1686
|
+
>>> output = ops.auto_generate.stack_ext([data1, data2], 0)
|
|
2021
1687
|
>>> print(output)
|
|
2022
|
-
[0.
|
|
1688
|
+
[[0. 1.]
|
|
1689
|
+
[2. 3.]]
|
|
2023
1690
|
"""
|
|
2024
|
-
return
|
|
1691
|
+
return stack_impl(tensors, dim)
|
|
2025
1692
|
|
|
2026
1693
|
|
|
2027
|
-
def
|
|
1694
|
+
def acosh(input):
|
|
2028
1695
|
r"""
|
|
2029
|
-
Computes
|
|
1696
|
+
Computes inverse hyperbolic cosine of the inputs element-wise.
|
|
2030
1697
|
|
|
2031
|
-
|
|
2032
|
-
If `min` and `max` are both zero, the minimum and maximum values of the data are used.
|
|
1698
|
+
.. math::
|
|
2033
1699
|
|
|
2034
|
-
|
|
1700
|
+
out_i = \cosh^{-1}(input_i)
|
|
2035
1701
|
|
|
2036
|
-
..
|
|
2037
|
-
|
|
2038
|
-
|
|
1702
|
+
.. note::
|
|
1703
|
+
Given an input tensor input, the function computes inverse hyperbolic cosine of every element.
|
|
1704
|
+
Input range is [1, inf].
|
|
2039
1705
|
|
|
2040
1706
|
Args:
|
|
2041
|
-
input (Tensor):
|
|
2042
|
-
bins (int, optional): Number of histogram bins, optional. If specified, must be positive. Default: ``100`` .
|
|
2043
|
-
min (int, float, optional): the lower end of the range (inclusive), optional. Default: ``0`` .
|
|
2044
|
-
max (int, float, optional): the upper end of the range (inclusive), optional. Default: ``0`` .
|
|
1707
|
+
input (Tensor): The input tensor of inverse hyperbolic cosine function.
|
|
2045
1708
|
|
|
2046
1709
|
Returns:
|
|
2047
|
-
|
|
1710
|
+
Tensor, has the same shape as `input`. The dtype of output is float32 when dtype of `input` is in [bool, int8, uint8, int16, int32, int64]. Otherwise output has the same dtype as `input`.
|
|
2048
1711
|
|
|
2049
1712
|
Raises:
|
|
2050
1713
|
TypeError: If `input` is not a Tensor.
|
|
2051
|
-
TypeError: If `input` datatype is not in support list.
|
|
2052
|
-
TypeError: If attr `min` or `max` is not float or int.
|
|
2053
|
-
TypeError: If attr `bins` is not int.
|
|
2054
|
-
ValueError: If attr value `min` > `max`.
|
|
2055
|
-
ValueError: If attr `bins` <= 0.
|
|
2056
1714
|
|
|
2057
1715
|
Supported Platforms:
|
|
2058
|
-
``Ascend``
|
|
1716
|
+
``Ascend`` ``GPU`` ``CPU``
|
|
2059
1717
|
|
|
2060
1718
|
Examples:
|
|
1719
|
+
>>> import mindspore
|
|
1720
|
+
>>> import numpy as np
|
|
2061
1721
|
>>> from mindspore import Tensor, ops
|
|
2062
|
-
>>>
|
|
2063
|
-
>>>
|
|
2064
|
-
>>> print(
|
|
2065
|
-
[0
|
|
1722
|
+
>>> input = Tensor(np.array([1.0, 1.5, 3.0, 100.0]), mindspore.float32)
|
|
1723
|
+
>>> output = ops.acosh_ext(input)
|
|
1724
|
+
>>> print(output)
|
|
1725
|
+
[0. 0.9624236 1.7627472 5.298292 ]
|
|
2066
1726
|
"""
|
|
2067
|
-
return
|
|
1727
|
+
return acosh_impl(input)
|
|
1728
|
+
|
|
1729
|
+
|
|
1730
|
+
def sub_tensor_(input, other, alpha=1):
|
|
1731
|
+
r"""
|
|
1732
|
+
None
|
|
1733
|
+
"""
|
|
1734
|
+
return sub_tensor_impl(input, other, alpha)
|
|
2068
1735
|
|
|
2069
1736
|
|
|
2070
1737
|
def prod(input, dim=None, keepdim=False, dtype=None):
|
|
@@ -2142,112 +1809,341 @@ def prod(input, dim=None, keepdim=False, dtype=None):
|
|
|
2142
1809
|
[2.62144e+05]
|
|
2143
1810
|
[5.31441e+05]]]
|
|
2144
1811
|
"""
|
|
2145
|
-
return prod_impl(input, dim, keepdim, dtype)
|
|
1812
|
+
return prod_impl(input, dim, keepdim, dtype)
|
|
1813
|
+
|
|
1814
|
+
|
|
1815
|
+
def log_softmax(input, dim=None, dtype=None):
|
|
1816
|
+
r"""
|
|
1817
|
+
Applies the Log Softmax function to the input tensor on the specified axis.
|
|
1818
|
+
Supposes a slice in the given axis, :math:`x` for each element :math:`x_i`,
|
|
1819
|
+
the Log Softmax function is shown as follows:
|
|
1820
|
+
|
|
1821
|
+
.. math::
|
|
1822
|
+
\text{output}(x_i) = \log \left(\frac{\exp(x_i)} {\sum_{j = 0}^{N-1}\exp(x_j)}\right),
|
|
1823
|
+
|
|
1824
|
+
where :math:`N` is the length of the Tensor.
|
|
1825
|
+
|
|
1826
|
+
Args:
|
|
1827
|
+
input (Tensor): The input Tensor.
|
|
1828
|
+
dim (int, optional): The axis to perform the Log softmax operation. Default: ``None`` .
|
|
1829
|
+
|
|
1830
|
+
Keyword Args:
|
|
1831
|
+
dtype (:class:`mindspore.dtype`, optional): The desired dtype of returned Tensor. If not set to None, the input
|
|
1832
|
+
Tensor will be cast to `dtype` before the operation is performed. This is useful for preventing overflows.
|
|
1833
|
+
If set to None, stay the same as original Tensor. Default: ``None`` . Supported data type is {float16, float32, double, bfloat16}.
|
|
1834
|
+
|
|
1835
|
+
Returns:
|
|
1836
|
+
Tensor, with the same shape as the input.
|
|
1837
|
+
|
|
1838
|
+
Raises:
|
|
1839
|
+
TypeError: If `dim` is not an int.
|
|
1840
|
+
ValueError: If `dim` is not in range [-len(input.shape), len(input.shape)).
|
|
1841
|
+
|
|
1842
|
+
Supported Platforms:
|
|
1843
|
+
``Ascend``
|
|
1844
|
+
|
|
1845
|
+
Examples:
|
|
1846
|
+
>>> import mindspore
|
|
1847
|
+
>>> import numpy as np
|
|
1848
|
+
>>> from mindspore import Tensor, ops
|
|
1849
|
+
>>> logits = Tensor(np.array([1, 2, 3, 4, 5]), mindspore.float32)
|
|
1850
|
+
>>> output = ops.auto_generate.log_softmax(logits, dim=-1)
|
|
1851
|
+
>>> print(output)
|
|
1852
|
+
[-4.4519143 -3.4519143 -2.4519143 -1.4519144 -0.4519144]
|
|
1853
|
+
"""
|
|
1854
|
+
return log_softmax_impl(input, dim, dtype)
|
|
1855
|
+
|
|
1856
|
+
|
|
1857
|
+
def atan(input):
|
|
1858
|
+
r"""
|
|
1859
|
+
Computes the trigonometric inverse tangent of the input element-wise.
|
|
1860
|
+
|
|
1861
|
+
.. math::
|
|
1862
|
+
|
|
1863
|
+
out_i = \tan^{-1}(input_i)
|
|
1864
|
+
|
|
1865
|
+
Args:
|
|
1866
|
+
input (Tensor): The shape of tensor is
|
|
1867
|
+
:math:`(N,*)` where :math:`*` means, any number of additional dimensions.
|
|
1868
|
+
|
|
1869
|
+
Returns:
|
|
1870
|
+
Tensor, has the same shape as `input`. The dtype of output is float32 when dtype of `input` is in [bool, int8, uint8, int16, int32, int64]. Otherwise output has the same dtype as `input`.
|
|
1871
|
+
|
|
1872
|
+
Raises:
|
|
1873
|
+
TypeError: If `input` is not a Tensor.
|
|
1874
|
+
|
|
1875
|
+
Supported Platforms:
|
|
1876
|
+
``Ascend`` ``GPU`` ``CPU``
|
|
1877
|
+
|
|
1878
|
+
Examples:
|
|
1879
|
+
>>> import mindspore
|
|
1880
|
+
>>> import numpy as np
|
|
1881
|
+
>>> from mindspore import Tensor, ops
|
|
1882
|
+
>>> input = Tensor(np.array([1.0, 0.0]), mindspore.float32)
|
|
1883
|
+
>>> output = ops.atan_ext(input)
|
|
1884
|
+
>>> print(output)
|
|
1885
|
+
[0.7853982 0. ]
|
|
1886
|
+
"""
|
|
1887
|
+
return atan_impl(input)
|
|
1888
|
+
|
|
1889
|
+
|
|
1890
|
+
def tril(input, diagonal=0):
|
|
1891
|
+
r"""
|
|
1892
|
+
None
|
|
1893
|
+
"""
|
|
1894
|
+
return tril_impl(input, diagonal)
|
|
1895
|
+
|
|
1896
|
+
|
|
1897
|
+
def argsort(input, dim=-1, descending=False, stable=False):
|
|
1898
|
+
r"""
|
|
1899
|
+
Sorts the input tensor along the given dimension in specified order and return the sorted indices.
|
|
1900
|
+
|
|
1901
|
+
.. warning::
|
|
1902
|
+
This is an experimental optimizer API that is subject to change.
|
|
1903
|
+
|
|
1904
|
+
Args:
|
|
1905
|
+
input(Tensor): The input tensor to sort.
|
|
1906
|
+
dim (int, optional): The dim to sort along. Default: ``-1`` , means the last dimension.
|
|
1907
|
+
The Ascend backend only supports sorting the last dimension.
|
|
1908
|
+
descending (bool, optional): The sort order. If `descending` is ``True`` then the elements
|
|
1909
|
+
are sorted in descending order by value. Otherwise sort in ascending order. Default: ``False`` .
|
|
1910
|
+
stable (bool, optional): Whether to use stable sorting algorithm. Default: ``False``.
|
|
1911
|
+
|
|
1912
|
+
Returns:
|
|
1913
|
+
Tensor, the indices of sorted input tensor. Data type is int64.
|
|
1914
|
+
|
|
1915
|
+
Raises:
|
|
1916
|
+
ValueError: If `dim` is out of range.
|
|
1917
|
+
TypeError: If dtype of `dim` is not int32.
|
|
1918
|
+
TypeError: If dtype of `descending` is not bool.
|
|
1919
|
+
TypeError: If dtype of `stable` is not bool.
|
|
1920
|
+
|
|
1921
|
+
Supported Platforms:
|
|
1922
|
+
``Ascend``
|
|
1923
|
+
|
|
1924
|
+
Examples:
|
|
1925
|
+
>>> import mindspore
|
|
1926
|
+
>>> import numpy as np
|
|
1927
|
+
>>> from mindspore import Tensor, ops
|
|
1928
|
+
>>> x = Tensor(np.array([[8, 2, 1], [5, 9, 3], [4, 6, 7]]), mindspore.float16)
|
|
1929
|
+
>>> sort = ops.auto_generate.argsort_ext(x)
|
|
1930
|
+
>>> print(sort)
|
|
1931
|
+
[[2 1 0]
|
|
1932
|
+
[2 0 1]
|
|
1933
|
+
[0 1 2]]
|
|
1934
|
+
"""
|
|
1935
|
+
return argsort_impl(input, dim, descending, stable)
|
|
1936
|
+
|
|
1937
|
+
|
|
1938
|
+
def log10(input):
|
|
1939
|
+
r"""
|
|
1940
|
+
Returns the logarithm to the base 10 of a tensor element-wise.
|
|
1941
|
+
|
|
1942
|
+
.. math::
|
|
1943
|
+
y_i = \log_{10}(x_i)
|
|
1944
|
+
|
|
1945
|
+
.. warning::
|
|
1946
|
+
- This is an experimental API that is subject to change or deletion.
|
|
1947
|
+
- If the input value of operator Log10 is within the range (0, 0.01] or [0.95, 1.05], the output accuracy
|
|
1948
|
+
may be affacted.
|
|
1949
|
+
|
|
1950
|
+
Args:
|
|
1951
|
+
input (Tensor): Input Tensor of any dimension. The value must be greater than 0.
|
|
1952
|
+
|
|
1953
|
+
Returns:
|
|
1954
|
+
Tensor, has the same shape as the `input`, and the dtype changes according to the `input.dtype`.
|
|
1955
|
+
|
|
1956
|
+
- if `input.dtype` is in [float16, float32, float64, bfloat16], the output dtype is the same as the `input.dtype`.
|
|
1957
|
+
- if `input.dtype` is integer or boolean type, the output dtype is float32.
|
|
1958
|
+
|
|
1959
|
+
Raises:
|
|
1960
|
+
TypeError: If `input` is not a Tensor.
|
|
1961
|
+
|
|
1962
|
+
Supported Platforms:
|
|
1963
|
+
``Ascend``
|
|
1964
|
+
|
|
1965
|
+
Examples:
|
|
1966
|
+
>>> import mindspore
|
|
1967
|
+
>>> import numpy as np
|
|
1968
|
+
>>> from mindspore import Tensor, ops
|
|
1969
|
+
>>> x = Tensor(np.array([3.0, 5.0, 7.0]), mindspore.float32)
|
|
1970
|
+
>>> output = ops.auto_generate.log10_ext(x)
|
|
1971
|
+
>>> print(output)
|
|
1972
|
+
[0.47712136 0.69897 0.845098 ]
|
|
1973
|
+
"""
|
|
1974
|
+
return log10_impl(input)
|
|
1975
|
+
|
|
1976
|
+
|
|
1977
|
+
def matrix_inverse(input):
|
|
1978
|
+
r"""
|
|
1979
|
+
Compute the inverse of the input matrix.
|
|
1980
|
+
|
|
1981
|
+
Args:
|
|
1982
|
+
input (Tensor): A matrix to be calculated. Input `input` must be at least two dimensions, and the size of
|
|
1983
|
+
the last two dimensions must be the same size.
|
|
1984
|
+
|
|
1985
|
+
Returns:
|
|
1986
|
+
Tensor, has the same type and shape as input`.
|
|
1987
|
+
|
|
1988
|
+
Raises:
|
|
1989
|
+
TypeError: If `input` is not a Tensor.
|
|
1990
|
+
ValueError: If the size of the last two dimensions of `input` is not the same.
|
|
1991
|
+
ValueError: If the dimension of `input` is 1.
|
|
1992
|
+
|
|
1993
|
+
Supported Platforms:
|
|
1994
|
+
``Ascend``
|
|
1995
|
+
|
|
1996
|
+
Examples:
|
|
1997
|
+
>>> from mindspore import Tensor, ops
|
|
1998
|
+
>>> from mindspore import dtype as mstype
|
|
1999
|
+
>>> x = Tensor([[1., 2.], [3., 4.]], mstype.float32)
|
|
2000
|
+
>>> print(ops.matrix_inverse_ext(x))
|
|
2001
|
+
[[-2. 1. ]
|
|
2002
|
+
[ 1.5 -0.5]]
|
|
2003
|
+
"""
|
|
2004
|
+
return matrix_inverse_impl(input)
|
|
2146
2005
|
|
|
2147
2006
|
|
|
2148
|
-
def
|
|
2007
|
+
def leaky_relu(input, negative_slope=0.01):
|
|
2149
2008
|
r"""
|
|
2150
|
-
|
|
2009
|
+
leaky_relu activation function. The element of `input` less than 0 times `negative_slope` .
|
|
2151
2010
|
|
|
2152
|
-
The
|
|
2011
|
+
The activation function is defined as:
|
|
2153
2012
|
|
|
2154
2013
|
.. math::
|
|
2014
|
+
\text{leaky_relu}(input) = \begin{cases}input, &\text{if } input \geq 0; \cr
|
|
2015
|
+
\text{negative_slope} * input, &\text{otherwise.}\end{cases}
|
|
2155
2016
|
|
|
2156
|
-
|
|
2017
|
+
where :math:`negative\_slope` represents the `negative_slope` parameter.
|
|
2157
2018
|
|
|
2158
|
-
|
|
2019
|
+
For more details, see `Rectifier Nonlinearities Improve Neural Network Acoustic Models
|
|
2020
|
+
<https://ai.stanford.edu/~amaas/papers/relu_hybrid_icml2013_final.pdf>`_.
|
|
2159
2021
|
|
|
2160
|
-
|
|
2161
|
-
input (Tensor): Tensor of any dimension. Supported dtypes:
|
|
2022
|
+
LeakyReLU Activation Function Graph:
|
|
2162
2023
|
|
|
2163
|
-
|
|
2164
|
-
|
|
2165
|
-
|
|
2166
|
-
|
|
2024
|
+
.. image:: ../images/LeakyReLU.png
|
|
2025
|
+
:align: center
|
|
2026
|
+
|
|
2027
|
+
Args:
|
|
2028
|
+
input (Tensor): The input of leaky_relu is a Tensor of any dimension.
|
|
2029
|
+
negative_slope (Union[int, float], optional): Slope of the activation function when the element of `input` is less than 0.
|
|
2030
|
+
Default: ``0.01`` .
|
|
2167
2031
|
|
|
2168
2032
|
Returns:
|
|
2169
|
-
Tensor,
|
|
2033
|
+
Tensor, has the same type and shape as the `input`.
|
|
2170
2034
|
|
|
2171
2035
|
Raises:
|
|
2172
2036
|
TypeError: If `input` is not a Tensor.
|
|
2173
|
-
TypeError: If
|
|
2037
|
+
TypeError: If `negative_slope` is not a float or an int.
|
|
2174
2038
|
|
|
2175
2039
|
Supported Platforms:
|
|
2176
|
-
``Ascend``
|
|
2040
|
+
``Ascend``
|
|
2177
2041
|
|
|
2178
2042
|
Examples:
|
|
2179
2043
|
>>> import mindspore
|
|
2180
2044
|
>>> import numpy as np
|
|
2181
2045
|
>>> from mindspore import Tensor, ops
|
|
2182
|
-
>>> input = Tensor(np.array([0.
|
|
2183
|
-
>>>
|
|
2184
|
-
|
|
2185
|
-
|
|
2046
|
+
>>> input = Tensor(np.array([[-1.0, 4.0, -8.0], [2.0, -5.0, 9.0]]), mindspore.float32)
|
|
2047
|
+
>>> print(ops.extend.leaky_relu_ext(input, negative_slope=0.2))
|
|
2048
|
+
[[-0.2 4. -1.6]
|
|
2049
|
+
[ 2. -1. 9. ]]
|
|
2186
2050
|
"""
|
|
2187
|
-
return
|
|
2051
|
+
return leaky_relu_impl(input, negative_slope)
|
|
2188
2052
|
|
|
2189
2053
|
|
|
2190
|
-
def
|
|
2054
|
+
def topk(input, k, dim=-1, largest=True, sorted=True):
|
|
2191
2055
|
r"""
|
|
2192
|
-
|
|
2056
|
+
Finds values and indices of the `k` largest or smallest entries along a given dimension.
|
|
2193
2057
|
|
|
2194
2058
|
.. warning::
|
|
2195
|
-
|
|
2059
|
+
- If sorted is set to False, due to different memory layout and traversal methods on different platforms,
|
|
2060
|
+
the display order of calculation results may be inconsistent when `sorted` is False.
|
|
2061
|
+
|
|
2062
|
+
If the `input` is a one-dimensional Tensor, finds the `k` largest or smallest entries in the Tensor,
|
|
2063
|
+
and outputs its value and index as a Tensor. values[`k`] is the `k` largest item in `input`,
|
|
2064
|
+
and its index is indices [`k`].
|
|
2065
|
+
|
|
2066
|
+
For a multi-dimensional matrix,
|
|
2067
|
+
calculates the first or last `k` entries in a given dimension, therefore:
|
|
2068
|
+
|
|
2069
|
+
.. math::
|
|
2070
|
+
|
|
2071
|
+
values.shape = indices.shape
|
|
2072
|
+
|
|
2073
|
+
If the two compared elements are the same, the one with the smaller index value is returned first.
|
|
2196
2074
|
|
|
2197
2075
|
Args:
|
|
2198
|
-
input(Tensor): Input
|
|
2199
|
-
|
|
2200
|
-
|
|
2076
|
+
input (Tensor): Input to be computed.
|
|
2077
|
+
k (int): The number of top or bottom elements to be computed along the last dimension.
|
|
2078
|
+
dim (int, optional): The dimension to sort along. Default: ``-1`` .
|
|
2079
|
+
largest (bool, optional): If largest is ``False`` then the k smallest elements are returned.
|
|
2080
|
+
Default: ``True`` .
|
|
2081
|
+
sorted (bool, optional): If ``True`` , the obtained elements will be sorted by the values in descending
|
|
2082
|
+
order or ascending order according to `largest`. If ``False`` , the obtained elements will not be
|
|
2083
|
+
sorted. Default: ``True`` .
|
|
2201
2084
|
|
|
2202
2085
|
Returns:
|
|
2203
|
-
|
|
2086
|
+
A tuple consisting of `values` and `indices`.
|
|
2087
|
+
|
|
2088
|
+
- values (Tensor) - The `k` largest or smallest elements in each slice of the given dimension.
|
|
2089
|
+
- indices (Tensor) - The indices of values within the last dimension of input.
|
|
2204
2090
|
|
|
2205
2091
|
Raises:
|
|
2206
|
-
TypeError: If
|
|
2207
|
-
TypeError: If `
|
|
2208
|
-
|
|
2092
|
+
TypeError: If `sorted` is not a bool.
|
|
2093
|
+
TypeError: If `input` is not a Tensor.
|
|
2094
|
+
TypeError: If `k` is not an int.
|
|
2209
2095
|
|
|
2210
2096
|
Supported Platforms:
|
|
2211
2097
|
``Ascend``
|
|
2212
2098
|
|
|
2213
2099
|
Examples:
|
|
2214
|
-
>>> import
|
|
2215
|
-
>>> from mindspore import
|
|
2216
|
-
>>>
|
|
2217
|
-
|
|
2218
|
-
|
|
2219
|
-
>>>
|
|
2220
|
-
(
|
|
2100
|
+
>>> import mindspore as ms
|
|
2101
|
+
>>> from mindspore import ops
|
|
2102
|
+
>>> x = ms.Tensor([[0.5368, 0.2447, 0.4302, 0.9673],
|
|
2103
|
+
... [0.4388, 0.6525, 0.4685, 0.1868],
|
|
2104
|
+
... [0.3563, 0.5152, 0.9675, 0.8230]], dtype=ms.float32)
|
|
2105
|
+
>>> output = ops.topk_ext(x, 2, dim=1)
|
|
2106
|
+
>>> print(output)
|
|
2107
|
+
(Tensor(shape=[3, 2], dtype=Float32, value=
|
|
2108
|
+
[[ 9.67299998e-01, 5.36800027e-01],
|
|
2109
|
+
[ 6.52499974e-01, 4.68499988e-01],
|
|
2110
|
+
[ 9.67499971e-01, 8.23000014e-01]]), Tensor(shape=[3, 2], dtype=Int64, value=
|
|
2111
|
+
[[3, 0],
|
|
2112
|
+
[1, 2],
|
|
2113
|
+
[2, 3]]))
|
|
2114
|
+
>>> output2 = ops.topk_ext(x, 2, dim=1, largest=False)
|
|
2115
|
+
>>> print(output2)
|
|
2116
|
+
(Tensor(shape=[3, 2], dtype=Float32, value=
|
|
2117
|
+
[[ 2.44700000e-01, 4.30200011e-01],
|
|
2118
|
+
[ 1.86800003e-01, 4.38800007e-01],
|
|
2119
|
+
[ 3.56299996e-01, 5.15200019e-01]]), Tensor(shape=[3, 2], dtype=Int64, value=
|
|
2120
|
+
[[1, 2],
|
|
2121
|
+
[3, 0],
|
|
2122
|
+
[0, 1]]))
|
|
2221
2123
|
"""
|
|
2222
|
-
return
|
|
2124
|
+
return topk_impl(input, k, dim, largest, sorted)
|
|
2223
2125
|
|
|
2224
2126
|
|
|
2225
|
-
def
|
|
2127
|
+
def log2(input):
|
|
2226
2128
|
r"""
|
|
2227
|
-
|
|
2129
|
+
Returns the logarithm to the base 2 of a tensor element-wise.
|
|
2228
2130
|
|
|
2229
|
-
|
|
2131
|
+
.. math::
|
|
2132
|
+
y_i = \log_2(x_i)
|
|
2230
2133
|
|
|
2231
|
-
|
|
2232
|
-
input
|
|
2233
|
-
|
|
2234
|
-
target (Tensor): The input label. Tensor of any dimension. The data type needs to be consistent with the `input`.
|
|
2235
|
-
It should also be broadcastable with the `input`.
|
|
2236
|
-
reduction (str, optional): Apply specific reduction method to the output: ``'mean'`` , ``'none'`` ,
|
|
2237
|
-
``'sum'`` . Default: ``'mean'`` .
|
|
2134
|
+
.. warning::
|
|
2135
|
+
- If the input value of operator Log2 is within the range (0, 0.01] or [0.95, 1.05], the output accuracy
|
|
2136
|
+
may be affacted.
|
|
2238
2137
|
|
|
2239
|
-
|
|
2240
|
-
|
|
2241
|
-
- ``'sum'``: the output elements will be summed.
|
|
2138
|
+
Args:
|
|
2139
|
+
input (Tensor): Input Tensor of any dimension. The value must be greater than 0.
|
|
2242
2140
|
|
|
2243
2141
|
Returns:
|
|
2244
|
-
|
|
2245
|
-
|
|
2142
|
+
Tensor, has the same shape as the `input`. If `input.dtype` is of integer or boolean type, the output dtype
|
|
2143
|
+
will be float32. Otherwise, the output dtype will be the same as `input.dtype`.
|
|
2246
2144
|
|
|
2247
2145
|
Raises:
|
|
2248
|
-
|
|
2249
|
-
ValueError: If `input` and `target` are not broadcastable.
|
|
2250
|
-
TypeError: If `input` and `target` are in different data type.
|
|
2146
|
+
TypeError: If `input` is not a Tensor.
|
|
2251
2147
|
|
|
2252
2148
|
Supported Platforms:
|
|
2253
2149
|
``Ascend``
|
|
@@ -2256,119 +2152,105 @@ def mse_loss(input, target, reduction='mean'):
|
|
|
2256
2152
|
>>> import mindspore
|
|
2257
2153
|
>>> import numpy as np
|
|
2258
2154
|
>>> from mindspore import Tensor, ops
|
|
2259
|
-
>>>
|
|
2260
|
-
>>>
|
|
2261
|
-
>>> output = ops.mse_loss_ext(logits, labels, reduction='none')
|
|
2155
|
+
>>> x = Tensor(np.array([3.0, 5.0, 7.0]), mindspore.float32)
|
|
2156
|
+
>>> output = ops.auto_generate.log2_ext(x)
|
|
2262
2157
|
>>> print(output)
|
|
2263
|
-
[
|
|
2264
|
-
[0. 0. 1.]]
|
|
2158
|
+
[1.5849625 2.321928 2.807355 ]
|
|
2265
2159
|
"""
|
|
2266
|
-
return
|
|
2160
|
+
return log2_impl(input)
|
|
2267
2161
|
|
|
2268
2162
|
|
|
2269
|
-
def
|
|
2163
|
+
def softplus(input, beta=1, threshold=20):
|
|
2270
2164
|
r"""
|
|
2271
|
-
|
|
2165
|
+
Applies softplus function to `input` element-wise.
|
|
2166
|
+
|
|
2167
|
+
The softplus function is shown as follows, x is the element of `input` :
|
|
2272
2168
|
|
|
2273
2169
|
.. math::
|
|
2274
|
-
|
|
2275
|
-
|
|
2276
|
-
|
|
2277
|
-
|
|
2278
|
-
|
|
2279
|
-
|
|
2280
|
-
.. warning::
|
|
2281
|
-
This is an experimental API that is subject to change or deletion.
|
|
2282
|
-
|
|
2170
|
+
|
|
2171
|
+
\text{output} = \frac{1}{beta}\log(1 + \exp(\text{beta * x}))
|
|
2172
|
+
|
|
2173
|
+
where :math:`input * beta > threshold`, the implementation converts to the linear function to ensure numerical stability.
|
|
2174
|
+
|
|
2283
2175
|
Args:
|
|
2284
|
-
input (Tensor):
|
|
2285
|
-
|
|
2286
|
-
|
|
2287
|
-
|
|
2288
|
-
|
|
2289
|
-
|
|
2176
|
+
input (Tensor): Tensor of any dimension. Supported dtypes:
|
|
2177
|
+
|
|
2178
|
+
- Ascend: float16, float32, bfloat16.
|
|
2179
|
+
beta (number.Number, optional): Scaling parameters in the softplus function. Default: ``1`` .
|
|
2180
|
+
threshold (number.Number, optional): For numerical stability, the softplus function is converted
|
|
2181
|
+
to a threshold parameter of a linear function. Default: ``20`` .
|
|
2182
|
+
|
|
2290
2183
|
Returns:
|
|
2291
|
-
Tensor,
|
|
2184
|
+
Tensor, with the same type and shape as the input.
|
|
2292
2185
|
|
|
2293
2186
|
Raises:
|
|
2294
|
-
TypeError: If
|
|
2295
|
-
|
|
2296
|
-
ValueError: If `index` rank is not the same as `source` rank.
|
|
2297
|
-
ValueError: If shape of `index` is not 1D or size of `index` is not equal to dimension of source[dim].
|
|
2298
|
-
ValueError: If the shape of `source` is not the same as that of `input` except the `dim` axis.
|
|
2187
|
+
TypeError: If `input` is not a Tensor.
|
|
2188
|
+
TypeError: If dtype of `input` is not float16, float32, bfloat16.
|
|
2299
2189
|
|
|
2300
2190
|
Supported Platforms:
|
|
2301
|
-
``Ascend``
|
|
2191
|
+
``Ascend``
|
|
2302
2192
|
|
|
2303
2193
|
Examples:
|
|
2304
|
-
>>> import numpy as np
|
|
2305
2194
|
>>> import mindspore
|
|
2195
|
+
>>> import numpy as np
|
|
2306
2196
|
>>> from mindspore import Tensor, ops
|
|
2307
|
-
>>>
|
|
2308
|
-
>>>
|
|
2309
|
-
>>> y = Tensor(np.array([[0.5, 1.0], [1.0, 1.5], [2.0, 2.5]]), mindspore.float32)
|
|
2310
|
-
>>> output = ops.auto_generate.index_add_ext(x, 1, index, y, alpha=1)
|
|
2197
|
+
>>> input = Tensor(np.array([0.1, 0.2, 30, 25]), mindspore.float32)
|
|
2198
|
+
>>> output = ops.auto_generate.softplus_ext(input)
|
|
2311
2199
|
>>> print(output)
|
|
2312
|
-
[
|
|
2313
|
-
[ 5. 5. 7.5]
|
|
2314
|
-
[ 9. 8. 11.5]]
|
|
2200
|
+
[0.74439657 0.7981388 30. 25.]
|
|
2315
2201
|
"""
|
|
2316
|
-
return
|
|
2202
|
+
return softplus_impl(input, beta, threshold)
|
|
2317
2203
|
|
|
2318
2204
|
|
|
2319
|
-
def
|
|
2205
|
+
def mish(input):
|
|
2320
2206
|
r"""
|
|
2321
|
-
|
|
2322
|
-
|
|
2323
|
-
Return the indices of the maximum values of a tensor.
|
|
2324
|
-
|
|
2325
|
-
Args:
|
|
2326
|
-
input (Tensor): Input tensor.
|
|
2207
|
+
Computes MISH (A Self Regularized Non-Monotonic Neural Activation Function)
|
|
2208
|
+
of input tensors element-wise.
|
|
2327
2209
|
|
|
2328
|
-
|
|
2329
|
-
Tensor.
|
|
2210
|
+
The formula is defined as follows:
|
|
2330
2211
|
|
|
2331
|
-
|
|
2332
|
-
|
|
2212
|
+
.. math::
|
|
2213
|
+
\text{mish}(input) = input * \tanh(softplus(\text{input}))
|
|
2333
2214
|
|
|
2334
|
-
|
|
2335
|
-
|
|
2336
|
-
>>> from mindspore import Tensor
|
|
2337
|
-
>>> from mindspore import ops
|
|
2338
|
-
>>> x = Tensor(np.array([[1, 20, 5], [67, 8, 9], [130, 24, 15]]).astype(np.float32))
|
|
2339
|
-
>>> output = ops.auto_generate.argmax_ext(x)
|
|
2340
|
-
>>> print(output)
|
|
2341
|
-
6
|
|
2215
|
+
See more details in `A Self Regularized Non-Monotonic Neural Activation Function
|
|
2216
|
+
<https://arxiv.org/abs/1908.08681>`_.
|
|
2342
2217
|
|
|
2343
|
-
|
|
2344
|
-
:noindex:
|
|
2218
|
+
Mish Activation Function Graph:
|
|
2345
2219
|
|
|
2346
|
-
|
|
2220
|
+
.. image:: ../images/Mish.png
|
|
2221
|
+
:align: center
|
|
2347
2222
|
|
|
2348
2223
|
Args:
|
|
2349
|
-
input (Tensor):
|
|
2350
|
-
|
|
2351
|
-
|
|
2352
|
-
dimension. Default: ``False`` .
|
|
2224
|
+
input (Tensor): The input of MISH. Supported dtypes:
|
|
2225
|
+
|
|
2226
|
+
- Ascend: float16, float32.
|
|
2353
2227
|
|
|
2354
2228
|
Returns:
|
|
2355
|
-
Tensor,
|
|
2229
|
+
Tensor, has the same type and shape as the `input`.
|
|
2356
2230
|
|
|
2357
2231
|
Raises:
|
|
2358
|
-
TypeError: If `
|
|
2359
|
-
|
|
2232
|
+
TypeError: If `input` is not a Tensor.
|
|
2233
|
+
TypeError: If dtype of `input` is not float16 or float32.
|
|
2360
2234
|
|
|
2361
2235
|
Supported Platforms:
|
|
2362
2236
|
``Ascend``
|
|
2363
2237
|
|
|
2364
2238
|
Examples:
|
|
2239
|
+
>>> import mindspore
|
|
2240
|
+
>>> from mindspore import Tensor, ops
|
|
2365
2241
|
>>> import numpy as np
|
|
2366
|
-
>>>
|
|
2367
|
-
>>>
|
|
2368
|
-
>>> x = Tensor(np.array([[1, 20, 5], [67, 8, 9], [130, 24, 15]]).astype(np.float32))
|
|
2369
|
-
>>> output = ops.auto_generate.argmax_ext(x, dim=-1)
|
|
2242
|
+
>>> x = Tensor(np.array([[-1.1, 4.0, -8.0], [2.0, -5.0, 9.0]]), mindspore.float32)
|
|
2243
|
+
>>> output = ops.mish(x)
|
|
2370
2244
|
>>> print(output)
|
|
2371
|
-
[
|
|
2245
|
+
[[-3.0764845e-01 3.9974124e+00 -2.6832507e-03]
|
|
2246
|
+
[ 1.9439589e+00 -3.3576239e-02 8.9999990e+00]]
|
|
2372
2247
|
"""
|
|
2373
|
-
return
|
|
2248
|
+
return mish_impl(input)
|
|
2249
|
+
|
|
2250
|
+
|
|
2251
|
+
def ffn(x, weight1, weight2, expertTokens=None, bias1=None, bias2=None, scale=None, offset=None, deqScale1=None, deqScale2=None, antiquant_scale1=None, antiquant_scale2=None, antiquant_offset1=None, antiquant_offset2=None, activation='fastgelu', inner_precise=0):
|
|
2252
|
+
r"""
|
|
2253
|
+
None
|
|
2254
|
+
"""
|
|
2255
|
+
return ffn_impl(x, weight1, weight2, expertTokens, bias1, bias2, scale, offset, deqScale1, deqScale2, antiquant_scale1, antiquant_scale2, antiquant_offset1, antiquant_offset2, converted_activation, inner_precise)
|
|
2374
2256
|
|