mindspore 2.6.0rc1__cp39-cp39-manylinux1_x86_64.whl → 2.7.0__cp39-cp39-manylinux1_x86_64.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 +2 -2
- mindspore/_c_dataengine.cpython-39-x86_64-linux-gnu.so +0 -0
- mindspore/_c_expression.cpython-39-x86_64-linux-gnu.so +0 -0
- mindspore/_c_mindrecord.cpython-39-x86_64-linux-gnu.so +0 -0
- mindspore/_checkparam.py +42 -11
- mindspore/_extends/builtin_operations.py +3 -3
- mindspore/_extends/optimize/__init__.py +23 -0
- mindspore/_extends/optimize/cell_utils.py +96 -0
- mindspore/_extends/parallel_compile/akg_compiler/custom.py +1109 -0
- mindspore/_extends/parallel_compile/akg_compiler/gen_custom_op_files.py +1 -1
- mindspore/_extends/parse/__init__.py +3 -3
- mindspore/_extends/parse/compile_config.py +44 -22
- mindspore/_extends/parse/deprecated/deprecated_tensor_method.py +1 -2
- mindspore/_extends/parse/parser.py +65 -84
- mindspore/_extends/parse/resources.py +39 -0
- mindspore/_extends/parse/standard_method.py +58 -14
- mindspore/_extends/parse/trope.py +8 -1
- mindspore/_extends/pijit/__init__.py +1 -2
- mindspore/_extends/pijit/pijit_func_white_list.py +2 -5
- mindspore/amp.py +4 -22
- mindspore/boost/adasum.py +1 -1
- mindspore/boost/boost_cell_wrapper.py +4 -4
- mindspore/common/__init__.py +43 -12
- 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 +3175 -3050
- mindspore/common/_utils.py +9 -1
- mindspore/common/api.py +377 -203
- mindspore/common/dtype.py +108 -57
- mindspore/common/dump.py +11 -16
- mindspore/common/dynamic_shape/__init__.py +0 -0
- mindspore/common/dynamic_shape/auto_dynamic_shape.py +498 -0
- mindspore/common/dynamic_shape/enable_dynamic.py +197 -0
- mindspore/common/file_system.py +59 -9
- mindspore/common/generator.py +5 -3
- mindspore/common/hook_handle.py +33 -5
- mindspore/common/jit_config.py +1 -1
- mindspore/common/jit_trace.py +84 -105
- mindspore/common/np_dtype.py +3 -3
- mindspore/common/parameter.py +27 -29
- mindspore/common/recompute.py +5 -7
- mindspore/common/sparse_tensor.py +0 -3
- mindspore/common/symbol.py +0 -1
- mindspore/common/tensor.py +117 -131
- mindspore/communication/_comm_helper.py +46 -4
- mindspore/communication/management.py +79 -7
- mindspore/context.py +67 -55
- mindspore/dataset/__init__.py +1 -1
- mindspore/dataset/audio/transforms.py +1 -1
- mindspore/dataset/core/config.py +38 -4
- mindspore/dataset/engine/datasets.py +350 -322
- mindspore/dataset/engine/datasets_user_defined.py +70 -24
- 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/c_transforms.py +2 -2
- mindspore/dataset/transforms/py_transforms.py +7 -3
- mindspore/dataset/transforms/transforms.py +10 -6
- mindspore/dataset/vision/__init__.py +1 -1
- mindspore/dataset/vision/py_transforms.py +8 -8
- mindspore/dataset/vision/transforms.py +17 -5
- mindspore/dataset/vision/utils.py +632 -21
- mindspore/dataset/vision/validators.py +1 -0
- mindspore/device_context/ascend/device.py +1 -1
- mindspore/device_context/ascend/op_tuning.py +35 -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 -4
- 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 +65 -5
- mindspore/include/api/cfg.h +24 -7
- mindspore/include/api/context.h +1 -0
- mindspore/include/api/delegate.h +10 -2
- mindspore/include/api/dual_abi_helper.h +100 -19
- mindspore/include/api/graph.h +14 -1
- mindspore/include/api/kernel.h +16 -3
- mindspore/include/api/kernel_api.h +9 -1
- mindspore/include/api/metrics/accuracy.h +9 -0
- mindspore/include/api/model.h +8 -1
- mindspore/include/api/model_group.h +4 -0
- mindspore/include/api/model_parallel_runner.h +2 -0
- mindspore/include/api/status.h +48 -10
- mindspore/include/api/types.h +8 -3
- mindspore/include/c_api/model_c.h +0 -58
- mindspore/include/c_api/tensor_c.h +0 -26
- mindspore/include/dataset/constants.h +9 -0
- 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/backend_manager/backend_jit_config.h +47 -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_expander.h +1 -1
- mindspore/include/mindspore/ccsrc/backend/common/graph_kernel/core/graph_kernel_utils.h +25 -1
- 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 +10 -1
- 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/backend/common/pass/ir_fusion/flash_attention_fusion.h +72 -0
- 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 +8 -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/executor.h +0 -41
- mindspore/include/mindspore/ccsrc/backend/common/session/kernel_graph_mgr.h +14 -10
- mindspore/include/mindspore/ccsrc/backend/common/session/session_basic.h +15 -31
- 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/dump/hook_dynamic_loader.h +5 -0
- 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 -9
- 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 +8 -12
- 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 -24
- 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 +10 -5
- mindspore/include/mindspore/ccsrc/debug/checksum/checksum.h +35 -0
- mindspore/include/mindspore/ccsrc/debug/checksum/checksum_kernel.h +64 -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/overflow_counter.h +6 -1
- 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 -53
- mindspore/include/mindspore/ccsrc/debug/dump/tensordump_control.h +6 -2
- mindspore/include/mindspore/ccsrc/debug/dump/utils.h +30 -0
- mindspore/include/mindspore/ccsrc/debug/profiler/mstx/mstx_impl.h +37 -24
- mindspore/include/mindspore/ccsrc/debug/profiler/mstx/mstx_symbol.h +63 -4
- 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/summary/summary.h +1 -1
- 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 +13 -22
- mindspore/include/mindspore/ccsrc/frontend/np_dtypes/np_dtypes.h +29 -0
- 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 +161 -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_dropout_ext.h +28 -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 +22 -3
- mindspore/include/mindspore/ccsrc/frontend/optimizer/ad/dfunctor.h +21 -10
- mindspore/include/mindspore/ccsrc/frontend/optimizer/ad/grad.h +5 -2
- mindspore/include/mindspore/ccsrc/frontend/optimizer/ad/pynative_jit_grad.h +11 -2
- 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 +3 -2
- mindspore/include/mindspore/ccsrc/frontend/optimizer/inplace_input_replace.h +30 -0
- mindspore/include/mindspore/ccsrc/frontend/optimizer/irpass/add_forward_monad_depend.h +51 -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/const_output_eliminate.h +4 -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/incorporate_call.h +3 -3
- mindspore/include/mindspore/ccsrc/frontend/optimizer/irpass/inline.h +35 -9
- 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 +3 -3
- mindspore/include/mindspore/ccsrc/frontend/optimizer/irpass/partial_eliminate.h +5 -3
- mindspore/include/mindspore/ccsrc/frontend/optimizer/irpass/recompute_prepare.h +1 -0
- mindspore/include/mindspore/ccsrc/frontend/optimizer/irpass/special_op_eliminate.h +9 -3
- mindspore/include/mindspore/ccsrc/frontend/optimizer/irpass/stack_unstack_eliminate.h +13 -6
- 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 +13 -5
- 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 +5 -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 +2 -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 +3 -2
- mindspore/include/mindspore/ccsrc/frontend/parallel/graph_util/node_info.h +2 -0
- mindspore/include/mindspore/ccsrc/frontend/parallel/graph_util/parallel_tensordump.h +50 -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/gather_info.h +16 -0
- 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/index_add_info.h +55 -0
- 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/loss_info.h +34 -0
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/matmul_info.h +14 -1
- 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 +7 -2
- mindspore/include/mindspore/ccsrc/frontend/parallel/ops_info/ops_utils.h +18 -1
- 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/reduce_base_method_info.h +3 -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/topprouter_info.h +55 -0
- 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 +5 -4
- 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/frontend/parallel/pass/overlap_recompute_comm.h +28 -0
- 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/detach_backward.h +70 -0
- 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 +14 -3
- 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/pipeline_transformer/zero_bubble_v.h +124 -0
- 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/step_parallel_utils.h +10 -3
- 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_layout.h +3 -0
- 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 -5
- 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/data_dump/tensor_stat_dump.h +0 -3
- 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 -10
- 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/helper.h +0 -2
- 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/common.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 +3 -2
- mindspore/include/mindspore/ccsrc/include/common/expander/core/node.h +1 -1
- mindspore/include/mindspore/ccsrc/include/common/fallback.h +9 -0
- mindspore/include/mindspore/ccsrc/include/common/pybind_api/api_register.h +7 -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 +11 -7
- mindspore/include/mindspore/ccsrc/include/common/runtime_conf/thread_bind_core.h +12 -10
- mindspore/include/mindspore/ccsrc/include/common/symbol_engine/symbol_engine_impl.h +16 -1
- mindspore/include/mindspore/ccsrc/include/common/utils/anfalgo.h +13 -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 +10 -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 +8 -8
- mindspore/include/mindspore/ccsrc/include/common/visible.h +0 -10
- mindspore/include/mindspore/ccsrc/kernel/environ_manager.h +2 -0
- 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 +26 -6
- mindspore/include/mindspore/ccsrc/minddata/dataset/core/config_manager.h +9 -0
- mindspore/include/mindspore/ccsrc/minddata/dataset/core/cv_tensor.h +1 -1
- mindspore/include/mindspore/ccsrc/minddata/dataset/core/data_type.h +1 -56
- mindspore/include/mindspore/ccsrc/minddata/dataset/core/device_buffer.h +74 -0
- mindspore/include/mindspore/ccsrc/minddata/dataset/core/message_queue.h +29 -20
- mindspore/include/mindspore/ccsrc/minddata/dataset/core/shared_memory_queue.h +47 -2
- 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/batch_info.h +61 -0
- mindspore/include/mindspore/ccsrc/minddata/dataset/engine/datasetops/batch_op.h +28 -39
- mindspore/include/mindspore/ccsrc/minddata/dataset/engine/datasetops/map_op/cpu_map_job.h +3 -3
- mindspore/include/mindspore/ccsrc/minddata/dataset/engine/datasetops/map_op/gpu_map_job.h +3 -3
- mindspore/include/mindspore/ccsrc/minddata/dataset/engine/datasetops/map_op/map_job.h +2 -2
- mindspore/include/mindspore/ccsrc/minddata/dataset/engine/datasetops/map_op/map_op.h +12 -2
- mindspore/include/mindspore/ccsrc/minddata/dataset/engine/datasetops/map_op/npu_map_job.h +3 -3
- mindspore/include/mindspore/ccsrc/minddata/dataset/engine/datasetops/receive_bridge_op.h +4 -0
- mindspore/include/mindspore/ccsrc/minddata/dataset/engine/datasetops/send_bridge_op.h +1 -1
- mindspore/include/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/mindrecord_op.h +0 -1
- mindspore/include/mindspore/ccsrc/minddata/dataset/include/dataset/constants.h +9 -0
- 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/acl_adapter.h +28 -1
- 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/acl_plugin.h +19 -0
- mindspore/include/mindspore/ccsrc/minddata/dataset/kernels/image/dvpp/utils/dvpp_video.h +29 -10
- mindspore/include/mindspore/ccsrc/minddata/dataset/kernels/image/dvpp/utils/dvpp_video_utils.h +80 -0
- mindspore/include/mindspore/ccsrc/minddata/dataset/kernels/image/lite_cv/image_process.h +1 -1
- mindspore/include/mindspore/ccsrc/minddata/dataset/kernels/image/pyav/container.h +100 -0
- mindspore/include/mindspore/ccsrc/minddata/dataset/kernels/image/pyav/context.h +102 -0
- mindspore/include/mindspore/ccsrc/minddata/dataset/kernels/image/pyav/format.h +45 -0
- mindspore/include/mindspore/ccsrc/minddata/dataset/kernels/image/pyav/frame.h +74 -0
- mindspore/include/mindspore/ccsrc/minddata/dataset/kernels/image/pyav/packet.h +59 -0
- mindspore/include/mindspore/ccsrc/minddata/dataset/kernels/image/pyav/stream.h +93 -0
- mindspore/include/mindspore/ccsrc/minddata/dataset/kernels/py_func_op.h +37 -0
- mindspore/include/mindspore/ccsrc/minddata/dataset/util/btree.h +1 -1
- mindspore/include/mindspore/ccsrc/minddata/dataset/util/command.h +29 -0
- mindspore/include/mindspore/ccsrc/minddata/dataset/util/cond_var.h +0 -1
- mindspore/include/mindspore/ccsrc/minddata/dataset/util/ftok_key.h +2 -2
- mindspore/include/mindspore/ccsrc/minddata/dataset/util/json_helper.h +1 -1
- mindspore/include/mindspore/ccsrc/minddata/dataset/util/log_adapter.h +0 -5
- mindspore/include/mindspore/ccsrc/minddata/dataset/util/queue.h +1 -1
- mindspore/include/mindspore/ccsrc/minddata/dataset/util/sig_handler.h +11 -3
- 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/common/log_adapter.h +0 -5
- 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/ccsrc/ms_extension/all.h +46 -0
- mindspore/include/mindspore/ccsrc/ms_extension/api.h +33 -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 +319 -0
- mindspore/include/mindspore/ccsrc/ms_extension/common/tensor_utils.h +83 -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 +1 -7
- mindspore/include/mindspore/ccsrc/pipeline/jit/pi/graph_build/func_graph_builder.h +25 -6
- 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 -26
- 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 +3 -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/executor/jit_executor_py.h +68 -0
- 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 +17 -3
- mindspore/include/mindspore/ccsrc/pipeline/jit/ps/parse/resolve.h +15 -1
- mindspore/include/mindspore/ccsrc/pipeline/jit/ps/pass.h +4 -1
- mindspore/include/mindspore/ccsrc/pipeline/jit/ps/pass_config.h +4 -0
- mindspore/include/mindspore/ccsrc/pipeline/jit/ps/pipeline.h +17 -171
- mindspore/include/mindspore/ccsrc/pipeline/jit/ps/resource.h +8 -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 +14 -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/static_analysis/static_analysis.h +4 -5
- 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_dma_handle.h +0 -2
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/hal/device/kernel_select_ascend.h +4 -5
- 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_device_context.h +69 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/hal/hardware/ascend_device_res_manager.h +16 -13
- 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 +12 -8
- 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 +24 -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 +119 -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_matmul.h +63 -0
- 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_matmul_all_reduce.h +2 -1
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/hccl/hcom_matmul_reduce_scatter.h +61 -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_send.h +1 -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_gather_into_tensor_uneven.h +40 -0
- 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_reduce_scatter_tensor_uneven.h +40 -0
- 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_gather_into_tensor_uneven.h +38 -0
- 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_reduce_scatter_tensor_uneven.h +39 -0
- 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/internal/dynamic_ntk.h +31 -0
- 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_kernel_in_out_map.h +16 -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/kv_scale_cache.h +30 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/internal/mla.h +48 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/internal/mla_preprocess.h +32 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/internal/multi_weight_matmul.h +2 -0
- 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 +95 -0
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/internal/pyboost/auto_gen/kernel_info_adapter.h +78 -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/mla.h +53 -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/utils/TaskQueue.h +40 -0
- 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/file_system.h +35 -0
- 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_matmul_split_fusion.h +31 -8
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion_infer/inference_qbmm_elemwise_fusion.h +46 -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/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_op_plugin_kernel.h +62 -0
- mindspore/include/mindspore/ccsrc/plugin/device/cpu/kernel/pyexecute/joinedstr_cpu_kernel.h +46 -0
- mindspore/include/mindspore/ccsrc/plugin/device/cpu/kernel/pyexecute/py_execute_cpu_kernel.h +1 -6
- mindspore/include/mindspore/ccsrc/plugin/device/gpu/hal/device/gpu_kernel_task.h +1 -0
- mindspore/include/mindspore/ccsrc/plugin/device/gpu/hal/hardware/gpu_device_context.h +18 -18
- 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 +27 -11
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/capture_graph/ascend_capture_graph.h +45 -0
- 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 +20 -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/res_manager/ascend/collective/utils.h +83 -0
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/device_context_conf/op_tuning_conf.h +14 -0
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/dvm/dvm.h +246 -0
- 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 -3
- 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/mbuf_manager/tdt_manager.h +44 -0
- 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 +11 -9
- 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 +5 -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/hcom_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/stream_manager/ascend_stream_manager.h +6 -4
- mindspore/include/mindspore/ccsrc/plugin/res_manager/ascend/symbol_interface/acl_mdl_symbol.h +14 -0
- 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 -8
- mindspore/include/mindspore/ccsrc/plugin/res_manager/cpu/cpu_device_address/cpu_device_address.h +17 -12
- mindspore/include/mindspore/ccsrc/plugin/res_manager/cpu/cpu_mem_manager/cpu_hash_table.h +128 -0
- mindspore/include/mindspore/ccsrc/plugin/res_manager/cpu/cpu_mem_manager/cpu_hash_table_util.h +114 -0
- 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 +9 -9
- 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 +132 -127
- 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 +132 -128
- 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/cell_backward_hook.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/chunk.h +1 -1
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/chunk_view.h +44 -0
- 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/cross_entropy_loss.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/cross_entropy_loss_grad.h +44 -0
- 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/diagonal_view.h +44 -0
- 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_gather_into_tensor_uneven.h +44 -0
- 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_reduce_scatter_tensor_uneven.h +44 -0
- 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/einsum_ext.h +44 -0
- 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_dropout_ext.h +44 -0
- 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_bernoulli_scalar.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_bernoulli_tensor.h +44 -0
- 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_remainder_tensor_scalar.h +44 -0
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/inplace_remainder_tensor_tensor.h +44 -0
- 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_scatter.h +44 -0
- 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/mla.h +44 -0
- 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/ring_attention_update.h +44 -0
- 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/cell_backward_hook.h +27 -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 +1589 -1385
- mindspore/include/mindspore/ccsrc/pyboost/functions/auto_generate/functions.h +578 -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 +71 -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 +497 -453
- mindspore/include/mindspore/ccsrc/pynative/grad/function/func_builder.h +3 -3
- 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 +21 -22
- 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 +564 -0
- mindspore/include/mindspore/ccsrc/pynative/op_function/auto_generate/pyboost_core.h +564 -0
- mindspore/include/mindspore/ccsrc/pynative/op_function/auto_generate/tensor_func_utils.h +493 -478
- 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 +6 -2
- mindspore/include/mindspore/ccsrc/pynative/pynative_utils.h +39 -43
- mindspore/include/mindspore/ccsrc/runtime/collective/collective_communication_lib.h +17 -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/memory_scheduler.h +2 -1
- mindspore/include/mindspore/ccsrc/runtime/device/move_to.h +3 -0
- mindspore/include/mindspore/ccsrc/runtime/device/res_manager/auto_mem_offload.h +0 -1
- mindspore/include/mindspore/ccsrc/runtime/device/res_manager/capture_graph.h +35 -0
- mindspore/include/mindspore/ccsrc/runtime/device/res_manager/hal_res_base.h +20 -17
- 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 +46 -33
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/actor_common.h +30 -19
- 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_runner.h +74 -0
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/control_flow/condition_switch_runner.h +89 -0
- 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 +28 -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 +405 -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 +2 -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 +75 -57
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/any_type_graph_scheduler.h +0 -33
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/control_node_parser.h +13 -1
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/control_node_scheduler.h +1 -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_capture/graph_capture_manager.h +117 -0
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/graph_compiler.h +4 -71
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/graph_parameter_store.h +88 -142
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/graph_scheduler.h +9 -22
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/parameter_store.h +4 -0
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/pipeline/async_lf_queue.h +97 -0
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/pipeline/lf_ring_queue.h +205 -0
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/pipeline/runtime_pipeline.h +71 -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/device_context.h +44 -173
- 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/graph_adapter.h +0 -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/float16.h +4 -3
- mindspore/include/mindspore/core/include/base/float8_e4m3fn.h +264 -0
- mindspore/include/mindspore/core/include/base/float8_e5m2.h +260 -0
- mindspore/include/mindspore/core/include/base/hifloat8.h +54 -58
- mindspore/include/mindspore/core/include/ir/anf.h +37 -8
- mindspore/include/mindspore/core/include/ir/device_sync.h +17 -1
- mindspore/include/mindspore/core/include/ir/dtype/number.h +123 -9
- 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 +2 -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/symbolic_shape/operation_builder.h +1 -1
- mindspore/include/mindspore/core/include/utils/anf_utils.h +2 -0
- 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 +11 -2
- mindspore/include/mindspore/core/include/utils/ms_context.h +13 -11
- mindspore/include/mindspore/core/include/utils/ms_exception.h +42 -5
- 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 +17 -2
- mindspore/include/mindspore/core/include/utils/system/base.h +1 -1
- mindspore/include/mindspore/core/include/utils/tensor_hook_map.h +30 -0
- 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/actor_threadpool.h +4 -0
- mindspore/include/mindspore/core/mindrt/include/thread/core_affinity.h +1 -1
- mindspore/include/mindspore/core/mindrt/include/thread/hqueue.h +6 -6
- mindspore/include/mindspore/core/mindrt/include/thread/threadpool.h +6 -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/dtype.h +12 -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/infer/ops_func_impl/broadcast_to_view.h +32 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/cell_backward_hook.h +32 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/chunk.h +0 -2
- mindspore/include/mindspore/ops/infer/ops_func_impl/chunk_view.h +32 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/cross_entropy_loss.h +36 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/cross_entropy_loss_grad.h +36 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/diagonal_view.h +32 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/dist_comm_all_gather_into_tensor_uneven.h +33 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/dist_comm_reduce_scatter_tensor_uneven.h +33 -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/expand_dims_view.h +31 -0
- 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/inner_moe_token_unpermute.h +36 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/inplace_bernoulli_scalar.h +25 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/inplace_bernoulli_tensor.h +40 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/inplace_matmul_add.h +34 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/inplace_remainder_tensor_scalar.h +35 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/inplace_remainder_tensor_tensor.h +35 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/inplace_silu.h +35 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/kv_scale_cache.h +48 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/masked_fill.h +4 -3
- mindspore/include/mindspore/ops/infer/ops_func_impl/masked_scatter.h +37 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/matmul_fusion_utils.h +6 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/matmul_split_silu_fastgelu_add_mul_out1.h +34 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/matmul_split_silu_mul_out1.h +34 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/matmul_split_silu_out2.h +1 -1
- 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/mla.h +54 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/mla_preprocess.h +75 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/moe_distribute_combine.h +34 -0
- 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/normal_float_float.h +1 -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/q_matmul_split_silu_fastgelu_add_mul_out1.h +34 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/q_matmul_split_silu_mul_out1.h +34 -0
- 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/ring_attention_update.h +36 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/select_ext_view.h +39 -0
- 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/topprouter.h +36 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/transpose.h +0 -2
- mindspore/include/mindspore/ops/infer/ops_func_impl/transpose_ext_view.h +34 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/transpose_view.h +29 -0
- mindspore/include/mindspore/ops/infer/ops_func_impl/unstack_ext_view.h +37 -0
- 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 +18 -8
- mindspore/include/mindspore/ops/kernel/ascend/acl_ir/op_api_convert.h +40 -114
- mindspore/include/mindspore/ops/kernel/ascend/acl_ir/op_api_exec.h +41 -32
- mindspore/include/mindspore/ops/kernel/ascend/acl_ir/op_api_util.h +6 -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/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/eigen_tensor.h +18 -15
- 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/sampling_kernels.h +18 -15
- 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_ops/replay_buffer/replay_buffer_factory.h +2 -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/contiguous_aclnn_kernel.h +40 -0
- 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/cross_entropy_loss_aclnn_kernel.h +48 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/cross_entropy_loss_grad_aclnn_kernel.h +47 -0
- 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/inner_moe_token_unpermute_aclnn_kernel.h +45 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/inplace_bernoulli_scalar_aclnn_kernel.h +47 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/inplace_bernoulli_tensor_aclnn_kernel.h +46 -0
- 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/masked_scatter_aclnn_kernel.h +45 -0
- 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/ring_attention_update_aclnn_kernel.h +41 -0
- 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/mla_aclnn_kernel.h +41 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn_auto_gen/paged_attention_aclnn_kernel.h +41 -0
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn_kernel_mod.h +139 -23
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn_kernel_utils.h +3 -3
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/aclnn_utils.h +42 -17
- 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/cell_backward_hook.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/chunk.h +1 -1
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/chunk_view.h +40 -0
- 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/cross_entropy_loss.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/cross_entropy_loss_grad.h +40 -0
- 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/diagonal_view.h +40 -0
- 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_dropout_ext.h +40 -0
- 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_bernoulli_scalar.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_bernoulli_tensor.h +40 -0
- 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_remainder_tensor_scalar.h +40 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/inplace_remainder_tensor_tensor.h +40 -0
- 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_scatter.h +40 -0
- 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/mla.h +40 -0
- 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/narrow_view.h +40 -0
- 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/ring_attention_update.h +40 -0
- 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/cell_backward_hook.h +27 -0
- 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/cross_entropy_loss.h +37 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/cross_entropy_loss_grad.h +38 -0
- 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/einsum_ext.h +36 -0
- 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_dropout_ext.h +38 -0
- 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_bernoulli_scalar.h +35 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_bernoulli_tensor.h +35 -0
- 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_remainder_tensor_scalar.h +34 -0
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/inplace_remainder_tensor_tensor.h +35 -0
- 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_scatter.h +35 -0
- 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/new_full.h +36 -0
- 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/ring_attention_update.h +38 -0
- 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/mla.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 +35 -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/masked_fill_cpu_kernel.h +1 -0
- mindspore/include/mindspore/ops/kernel/cpu/multi_margin_loss_cpu_kernel.h +6 -6
- mindspore/include/mindspore/ops/kernel/cpu/new_empty_cpu_kernel.h +55 -0
- 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/cell_backward_hook.h +38 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/chunk.h +1 -1
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/chunk_view.h +38 -0
- 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/cross_entropy_loss.h +38 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/cross_entropy_loss_grad.h +38 -0
- 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/diagonal_view.h +38 -0
- 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_gather_into_tensor_uneven.h +38 -0
- 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_reduce_scatter_tensor_uneven.h +38 -0
- 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/func_dropout_ext.h +38 -0
- 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_bernoulli_scalar.h +38 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_bernoulli_tensor.h +38 -0
- 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_remainder_tensor_scalar.h +38 -0
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/inplace_remainder_tensor_tensor.h +38 -0
- 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/inplace_silu.h +38 -0
- 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_scatter.h +38 -0
- 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/mla.h +38 -0
- 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/select_ext_view.h +38 -0
- 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/cell_backward_hook.h +27 -0
- 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 +2 -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/sample_distorted_bounding_box_v2_cpu_kernel.h +4 -0
- mindspore/include/mindspore/ops/kernel/cpu/sequence/bool_binary_arithmetic_cpu_kernel.h +42 -0
- 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/cell_backward_hook.h +38 -0
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/chunk.h +1 -1
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/chunk_view.h +38 -0
- 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/cross_entropy_loss.h +38 -0
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/cross_entropy_loss_grad.h +38 -0
- 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/diagonal_view.h +38 -0
- 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_gather_into_tensor_uneven.h +38 -0
- 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_reduce_scatter_tensor_uneven.h +38 -0
- 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/func_dropout_ext.h +38 -0
- 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_bernoulli_scalar.h +38 -0
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/inplace_bernoulli_tensor.h +38 -0
- 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/inplace_silu.h +38 -0
- 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_scatter.h +38 -0
- 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/mla.h +38 -0
- 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/select_ext_view.h +38 -0
- 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/cell_backward_hook.h +27 -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 -33
- 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 -2
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_lite_ops.h +3234 -2894
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_def.h +816 -761
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_a.h +61 -59
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_b.h +24 -23
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_c.h +37 -33
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_d.h +30 -26
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_e.h +28 -26
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_f.h +21 -18
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_g.h +24 -20
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_h.h +7 -7
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_i.h +72 -64
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_k.h +2 -1
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_l.h +32 -32
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_m.h +45 -37
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_n.h +17 -15
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_o.h +2 -2
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_p.h +5 -6
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_q.h +6 -3
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_r.h +61 -60
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_s.h +65 -61
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_t.h +20 -18
- 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 +1 -1
- 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_name_z.h +1 -1
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_a.h +61 -59
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_b.h +24 -23
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_c.h +37 -33
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_d.h +30 -26
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_e.h +28 -26
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_f.h +21 -18
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_g.h +24 -20
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_h.h +7 -7
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_i.h +72 -64
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_k.h +2 -1
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_l.h +32 -32
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_m.h +45 -37
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_n.h +17 -15
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_o.h +2 -2
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_p.h +5 -6
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_q.h +6 -3
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_r.h +61 -60
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_s.h +65 -61
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_t.h +20 -18
- 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 +1 -1
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_x.h +1 -1
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_z.h +1 -1
- mindspore/include/mindspore/ops/op_def/framework_op_name.h +0 -1
- mindspore/include/mindspore/ops/op_def/framework_ops.h +3 -2
- mindspore/include/mindspore/ops/op_def/nn_op_name.h +4 -0
- mindspore/include/mindspore/ops/op_def/op_enum.h +4 -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 +12 -3
- mindspore/include/mindspore/ops/ops_utils/memory_overlap.h +4 -5
- mindspore/include/mindspore/ops/ops_utils/op_constants.h +13 -0
- mindspore/include/mindspore/ops/ops_utils/op_utils.h +14 -16
- mindspore/include/mindspore/ops/ops_utils/type_dispatch.h +51 -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/diagonal_view_strides_calc.h +32 -0
- 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_view_strides_calc.h +28 -0
- 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/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_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 -152
- 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/x86_64/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/x86_64/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 -3
- 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_ascend_native_boost.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 +9 -4
- 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 +40 -7
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/op_param.h +85 -1
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/include/tiling_utils.h +3 -138
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libadd_layer_norm_op.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libadd_rms_norm_op.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libadd_rms_norm_quant_op.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libapply_rotary_pos_emb_310p_op.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libapply_rotary_pos_emb_op.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libcast_op.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libcompare_op.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libfused_add_topk_div_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/libllama_op.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libmatmul_op.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libmoe_gating_group_topk_op.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libmoe_init_routing_op.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libmoe_token_unpermute_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/libswft_dynamic_quant_op.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libswft_matmul_op.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libswft_moe_init_routing_op.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libswft_paged_attention_op.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libswft_reshape_and_cache_nz_op.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/lib/libswft_transpose_batch_matmul_transpose_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_ascend310p.so +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/libfused_add_topk_div_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/libmoe_init_routing_ascend310p.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/so_kernels/libmoe_token_unpermute_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/ascend310p/so_kernels/libswft_dynamic_quant_ascend310p.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/so_kernels/libswft_matmul_ascend310p.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/so_kernels/libswft_moe_init_routing_ascend310p.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/so_kernels/libswft_paged_attention_ascend310p.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/so_kernels/libswft_reshape_and_cache_nz_ascend310p.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/so_kernels/libswft_transpose_batch_matmul_transpose_ascend310p.so +0 -0
- mindspore/lib/plugin/ascend/ms_kernels_internal/internal_kernel/op_kernels/ascend310p/so_kernels/libswiglu_dynamic_quant_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_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/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/libllama_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/gpu/libcuda_ops.so.11 +0 -0
- mindspore/lib/plugin/gpu/libmindspore_gpu_res_manager.so +0 -0
- mindspore/lib/plugin/gpu11.1/libakg.so +0 -0
- mindspore/lib/plugin/gpu11.1/libnccl.so.2 +0 -0
- mindspore/lib/plugin/gpu11.1/libnvidia_collective.so +0 -0
- mindspore/lib/plugin/gpu11.6/libakg.so +0 -0
- mindspore/lib/plugin/gpu11.6/libnccl.so.2 +0 -0
- mindspore/lib/plugin/gpu11.6/libnvidia_collective.so +0 -0
- mindspore/lib/plugin/libmindspore_ascend.so.2 +0 -0
- mindspore/lib/plugin/libmindspore_gpu.so.11.1 +0 -0
- mindspore/lib/plugin/libmindspore_gpu.so.11.6 +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 +61 -11
- mindspore/mindrecord/tools/cifar10_to_mr.py +5 -0
- mindspore/mint/__init__.py +6 -46
- mindspore/mint/distributed/__init__.py +5 -0
- mindspore/mint/distributed/distributed.py +429 -23
- mindspore/mint/nn/__init__.py +1 -1
- mindspore/mint/nn/functional.py +53 -6
- mindspore/mint/nn/layer/_functions.py +163 -294
- mindspore/mint/nn/layer/activation.py +8 -6
- mindspore/mint/nn/layer/conv.py +140 -104
- mindspore/mint/nn/layer/normalization.py +11 -25
- 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 +491 -623
- mindspore/nn/grad/cell_grad.py +11 -12
- mindspore/nn/layer/activation.py +36 -36
- mindspore/nn/layer/basic.py +74 -77
- 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 +38 -40
- 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 +4 -6
- mindspore/nn/optim/adasum.py +2 -2
- mindspore/nn/optim/asgd.py +2 -0
- mindspore/nn/optim/lamb.py +1 -3
- mindspore/nn/optim/optimizer.py +1 -1
- mindspore/nn/optim/tft_wrapper.py +2 -3
- mindspore/nn/optim/thor.py +2 -2
- mindspore/nn/probability/distribution/_utils/utils.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 +73 -42
- mindspore/nn/wrap/grad_reducer.py +37 -52
- mindspore/nn/wrap/loss_scale.py +72 -74
- mindspore/numpy/array_creations.py +7 -7
- mindspore/numpy/fft.py +1 -1
- mindspore/numpy/math_ops.py +5 -5
- mindspore/numpy/utils_const.py +1 -1
- mindspore/ops/_grad_experimental/grad_comm_ops.py +51 -13
- mindspore/ops/_grad_experimental/grad_debug_ops.py +14 -0
- mindspore/ops/_grad_experimental/grad_inner_ops.py +0 -9
- mindspore/ops/_op_impl/cpu/__init__.py +1 -0
- mindspore/ops/_op_impl/cpu/joinedstr_op.py +28 -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 +463 -422
- mindspore/ops/auto_generate/gen_extend_func.py +1121 -1239
- mindspore/ops/auto_generate/gen_ops_def.py +6949 -6269
- mindspore/ops/auto_generate/gen_ops_prim.py +16864 -15640
- mindspore/ops/auto_generate/pyboost_inner_prim.py +358 -328
- mindspore/ops/composite/__init__.py +10 -0
- mindspore/ops/composite/base.py +9 -5
- mindspore/ops/composite/multitype_ops/__init__.py +12 -1
- mindspore/ops/composite/multitype_ops/_compile_utils.py +133 -109
- mindspore/ops/composite/multitype_ops/_constexpr_utils.py +1 -1
- 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 +4 -1
- mindspore/ops/function/_add_attr_func.py +11 -6
- mindspore/ops/function/array_func.py +19 -102
- mindspore/ops/function/debug_func.py +8 -5
- mindspore/ops/function/grad/grad_func.py +5 -13
- mindspore/ops/function/math_func.py +77 -572
- mindspore/ops/function/nn_func.py +46 -94
- 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 +4 -4
- mindspore/ops/functional_overload.py +1414 -838
- 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 +14 -18
- mindspore/ops/operations/_sequence_ops.py +1 -1
- mindspore/ops/operations/array_ops.py +5 -51
- mindspore/ops/operations/comm_ops.py +186 -41
- mindspore/ops/operations/custom_ops.py +303 -177
- mindspore/ops/operations/debug_ops.py +59 -4
- mindspore/ops/operations/image_ops.py +13 -13
- mindspore/ops/operations/manually_defined/ops_def.py +27 -28
- mindspore/ops/operations/math_ops.py +8 -9
- mindspore/ops/operations/nn_ops.py +8 -40
- mindspore/ops/primitive.py +9 -20
- mindspore/ops/tensor_method.py +63 -15
- 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 -16
- 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 +16 -23
- mindspore/parallel/_cell_wrapper.py +113 -45
- mindspore/parallel/_parallel_serialization.py +4 -3
- 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 +17 -12
- mindspore/parallel/_utils.py +5 -11
- mindspore/parallel/auto_parallel.py +35 -14
- mindspore/parallel/checkpoint_convert.py +3 -3
- mindspore/parallel/checkpoint_transform.py +13 -7
- mindspore/parallel/cluster/process_entity/_api.py +88 -49
- mindspore/parallel/cluster/process_entity/_utils.py +95 -7
- mindspore/parallel/cluster/run.py +48 -7
- mindspore/parallel/function/__init__.py +8 -1
- mindspore/parallel/function/reshard_func.py +12 -12
- mindspore/parallel/nn/__init__.py +15 -2
- mindspore/parallel/nn/parallel_cell_wrapper.py +50 -14
- mindspore/parallel/nn/parallel_grad_reducer.py +7 -14
- mindspore/parallel/shard.py +10 -25
- mindspore/parallel/transform_safetensors.py +469 -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_kernel_details_viewer.py +10 -9
- mindspore/profiler/analysis/viewer/ascend_op_memory_viewer.py +43 -23
- 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/msprof_cmd_tool.py +2 -2
- mindspore/profiler/common/path_manager.py +9 -0
- mindspore/profiler/common/profiler_context.py +50 -29
- mindspore/profiler/common/profiler_info.py +0 -16
- mindspore/profiler/common/profiler_meta_data.py +1 -0
- mindspore/profiler/common/profiler_op_analyse.py +239 -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 +374 -338
- mindspore/profiler/envprofiler.py +42 -12
- mindspore/profiler/experimental_config.py +112 -7
- mindspore/profiler/mstx.py +33 -12
- mindspore/profiler/platform/__init__.py +2 -3
- mindspore/profiler/platform/cpu_profiler.py +10 -4
- mindspore/profiler/platform/npu_profiler.py +30 -20
- mindspore/profiler/profiler.py +218 -154
- mindspore/profiler/profiler_action_controller.py +65 -77
- mindspore/profiler/profiler_interface.py +2 -2
- 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 +2 -2
- mindspore/run_check/_check_version.py +7 -10
- mindspore/runtime/__init__.py +8 -6
- mindspore/runtime/event.py +10 -4
- mindspore/runtime/executor.py +87 -45
- mindspore/runtime/memory.py +31 -32
- mindspore/runtime/thread_bind_core.py +299 -165
- mindspore/safeguard/rewrite_obfuscation.py +12 -13
- mindspore/scipy/linalg.py +2 -2
- mindspore/scipy/utils_const.py +0 -17
- mindspore/train/_utils.py +17 -7
- mindspore/train/amp.py +43 -23
- mindspore/train/callback/__init__.py +5 -5
- mindspore/train/callback/_callback.py +2 -1
- mindspore/train/callback/_checkpoint.py +4 -14
- mindspore/train/callback/_flops_collector.py +11 -7
- mindspore/train/callback/_landscape.py +0 -1
- mindspore/train/callback/_train_fault_tolerance.py +98 -21
- mindspore/train/data_sink.py +15 -6
- mindspore/train/dataset_helper.py +14 -5
- mindspore/train/model.py +133 -69
- mindspore/train/serialization.py +168 -126
- mindspore/train/summary/summary_record.py +13 -2
- mindspore/train/train_thor/model_thor.py +2 -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 +14 -17
- mindspore/version.py +1 -1
- mindspore-2.7.0.dist-info/METADATA +368 -0
- mindspore-2.7.0.dist-info/RECORD +12025 -0
- mindspore-2.7.0.dist-info/WHEEL +5 -0
- mindspore/_deprecated/__init__.py +0 -17
- mindspore/_deprecated/jit.py +0 -198
- mindspore/_extends/remote/kernel_build_server_ascend.py +0 -75
- mindspore/common/auto_dynamic_shape.py +0 -504
- mindspore/communication/_hccl_management.py +0 -297
- 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/deprecated_env.h +0 -27
- 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/decoder_k_v_cache_info.h +0 -58
- 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/include/common/np_dtype/np_dtypes.h +0 -42
- mindspore/include/mindspore/ccsrc/minddata/dataset/kernels/image/lite_image_utils.h +0 -239
- mindspore/include/mindspore/ccsrc/pipeline/jit/pi/utils/ptr_list_ref.h +0 -423
- mindspore/include/mindspore/ccsrc/pipeline/jit/ps/pipeline_jit.h +0 -68
- 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_kernel_runtime.h +0 -88
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/hal/device/ascend_stream_assign.h +0 -255
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/hal/hardware/ascend_deprecated_interface.h +0 -46
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/hal/hardware/ge_device_context.h +0 -81
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/dvm/dvm.h +0 -232
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/ge/ge_kernel_build.h +0 -28
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/kernel/ge/ge_kernel_mod.h +0 -78
- 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_sink_stdout.h +0 -30
- 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/llm_boost/atb/include/atb_speed/utils/filesystem.h +0 -45
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/optimizer/format_type/rectify_do_mask_kernel_info.h +0 -44
- mindspore/include/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/flash_attention_fusion.h +0 -73
- mindspore/include/mindspore/ccsrc/plugin/device/cpu/hal/device/cpu_hash_table.h +0 -127
- mindspore/include/mindspore/ccsrc/plugin/device/cpu/hal/device/cpu_hash_table_util.h +0 -114
- mindspore/include/mindspore/ccsrc/plugin/device/cpu/hal/device/cpu_kernel_runtime.h +0 -82
- mindspore/include/mindspore/ccsrc/plugin/device/cpu/hal/hardware/cpu_session.h +0 -61
- mindspore/include/mindspore/ccsrc/plugin/device/cpu/kernel/custom/custom_julia_cpu_kernel.h +0 -50
- mindspore/include/mindspore/ccsrc/plugin/device/cpu/kernel/custom/julia_api.h +0 -443
- mindspore/include/mindspore/ccsrc/plugin/device/gpu/hal/device/gpu_kernel_runtime.h +0 -145
- mindspore/include/mindspore/ccsrc/plugin/device/gpu/hal/hardware/gpu_deprecated_interface.h +0 -43
- mindspore/include/mindspore/ccsrc/plugin/device/gpu/hal/hardware/gpu_inference_session.h +0 -50
- mindspore/include/mindspore/ccsrc/plugin/device/gpu/hal/hardware/gpu_session.h +0 -93
- 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/pyboost/auto_generate/select_ext.h +0 -44
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/transpose_ext.h +0 -44
- mindspore/include/mindspore/ccsrc/pyboost/auto_generate/unstack_ext.h +0 -44
- 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/jit/jit_dfunctor.h +0 -28
- 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/device/kernel_runtime.h +0 -223
- mindspore/include/mindspore/ccsrc/runtime/device/kernel_runtime_manager.h +0 -71
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/control_flow/condition_gather_actor.h +0 -78
- mindspore/include/mindspore/ccsrc/runtime/graph_scheduler/actor/control_flow/condition_switch_actor.h +0 -91
- 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/ccsrc/runtime/hardware/deprecated_interface.h +0 -47
- mindspore/include/mindspore/core/include/base/fp8_e4m3.h +0 -263
- mindspore/include/mindspore/core/include/base/fp8_e5m2.h +0 -258
- mindspore/include/mindspore/core/include/ir/base_tensor.h +0 -1073
- mindspore/include/mindspore/ops/infer/masked_scatter.h +0 -44
- mindspore/include/mindspore/ops/infer/ops_func_impl/decoder_k_v_cache.h +0 -40
- mindspore/include/mindspore/ops/infer/ops_func_impl/gmm_backward.h +0 -28
- mindspore/include/mindspore/ops/infer/ops_func_impl/gmm_v2_backward.h +0 -28
- mindspore/include/mindspore/ops/infer/ops_func_impl/moe_token_unpermute.h +0 -36
- mindspore/include/mindspore/ops/infer/ops_func_impl/prompt_k_v_cache.h +0 -40
- mindspore/include/mindspore/ops/infer/ops_func_impl/select_ext.h +0 -40
- mindspore/include/mindspore/ops/infer/ops_func_impl/transpose_ext.h +0 -35
- mindspore/include/mindspore/ops/infer/ops_func_impl/unstack_ext.h +0 -38
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/cpu_kernel/ms_kernel/densetodense_set_operation.h +0 -47
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/cpu_kernel/ms_kernel/densetosparsesetoperation.h +0 -74
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/cpu_kernel/ms_kernel/lu.h +0 -35
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ragged_tensor_to_sparse.h +0 -92
- 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/aicpu/aicpu_ops/cpu_kernel/ms_kernel/sparse_cross.h +0 -111
- mindspore/include/mindspore/ops/kernel/ascend/aicpu/aicpu_ops/cpu_kernel/ms_kernel/sparse_sparse_maximum.h +0 -61
- 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/mindspore/ops/kernel/ascend/opapi/aclnn/moe_token_unpermute_aclnn_kernel.h +0 -45
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn/view/contiguous.h +0 -41
- mindspore/include/mindspore/ops/kernel/ascend/opapi/aclnn_auto_gen/select_ext_aclnn_kernel.h +0 -41
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/select_ext.h +0 -40
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/transpose_ext.h +0 -40
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/auto_generate/unstack_ext.h +0 -40
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/customize_copy.h +0 -37
- mindspore/include/mindspore/ops/kernel/ascend/pyboost/customize/unstack_ext.h +0 -36
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/arithmetic_self_parameter.h +0 -30
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/base/batch_to_space_base.h +0 -33
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/base/cast_base.h +0 -74
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/base/concat_base.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/base/conv1x1_base.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/base/conv_common_base.h +0 -41
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/base/crop_base.h +0 -35
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/base/depth_to_space_base.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/base/fill_base.h +0 -33
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/base/format_transpose.h +0 -30
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/base/gather_d_base.h +0 -55
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/base/minimal_filtering_generator.h +0 -58
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/base/scatter_nd_binary.h +0 -37
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/base/space_to_depth_base.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/base/split_with_over_lap_base.h +0 -33
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/base/stack_base.h +0 -30
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/base/transpose_base.h +0 -69
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/batchnorm_parameter.h +0 -29
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/call_parameter.h +0 -28
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/clip_parameter.h +0 -29
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/conv3d_parameter.h +0 -26
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/cumsum_parameter.h +0 -29
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/custom_gru_parameter.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/custom_is_inf_parameter.h +0 -26
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/custom_masked_fill_parameter.h +0 -26
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/custom_parameter.h +0 -30
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fill_parameter.h +0 -25
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/format_transpose_parameter.h +0 -29
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/activation_fp16.h +0 -43
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/arg_min_max_fp16.h +0 -33
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/arithmetic_self_fp16.h +0 -57
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/batchnorm_fp16.h +0 -36
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/cast_fp16.h +0 -94
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/common_func_fp16.h +0 -40
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/constant_of_shape_fp16.h +0 -38
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/conv_depthwise_fp16.h +0 -65
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/conv_fp16.h +0 -60
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/crop_fp16.h +0 -26
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/custom_gru_fp16.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/deconv_fp16.h +0 -36
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/deconv_winograd_fp16.h +0 -48
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/dynamic_quant_fp16.h +0 -35
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/fill_fp16.h +0 -34
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/gru_fp16.h +0 -30
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/instance_norm_fp16.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/layer_norm_fp16.h +0 -33
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/log_softmax_fp16.h +0 -35
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/lstm_fp16.h +0 -54
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/matmul_fp16.h +0 -128
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/matrix_fp16.h +0 -36
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/pack_fp16.h +0 -93
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/pad_fp16.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/power_fp16.h +0 -64
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/prelu_fp16.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/quant_dtype_cast_fp16.h +0 -35
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/range_fp16.h +0 -27
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/reduce_fp16.h +0 -41
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/resize_fp16.h +0 -56
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/scale_fp16.h +0 -38
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/softmax_fp16.h +0 -35
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/sparse_to_dense_fp16.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/splice_fp16.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/topk_fp16.h +0 -35
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/transpose_fp16.h +0 -35
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/unique_fp16.h +0 -29
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/utils_fp16.h +0 -25
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/where_fp16.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/winograd_transform_fp16.h +0 -57
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16/winograd_utils_fp16.h +0 -571
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16_grad/activation_grad_fp16.h +0 -44
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16_grad/arithmetic_grad.h +0 -41
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16_grad/arithmetic_self_grad.h +0 -39
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16_grad/batch_norm.h +0 -40
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16_grad/convolution_grad_filter.h +0 -33
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16_grad/convolution_grad_input.h +0 -33
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16_grad/dropout_grad.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16_grad/gemm_fp16.h +0 -46
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16_grad/layernorm_grad.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16_grad/pack_fp16_ext.h +0 -37
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16_grad/pooling_grad.h +0 -34
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16_grad/resize_grad.h +0 -45
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16_grad/strided_slice_grad.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp16_grad/unsorted_segment_sum.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/adder_fp32.h +0 -47
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/arg_min_max_fp32.h +0 -34
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/arithmetic_compare_fp32.h +0 -77
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/attention_fp32.h +0 -72
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/batchnorm_fp32.h +0 -40
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/bias_add.h +0 -34
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/common_func_fp32.h +0 -106
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/constant_of_shape_fp32.h +0 -52
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/conv_1x1_avx_fp32.h +0 -40
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/conv_1x1_x86_fp32.h +0 -21
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/conv_common_fp32.h +0 -60
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/conv_depthwise_avx_fp32.h +0 -37
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/conv_depthwise_fp32.h +0 -148
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/conv_im2col_avx512_fp32.h +0 -38
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/conv_im2col_fp32.h +0 -33
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/conv_sw.h +0 -132
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/conv_sw_arm64_fp32.h +0 -33
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/conv_sw_avx_fp32.h +0 -42
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/conv_winograd_fp32.h +0 -48
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/crop_fp32.h +0 -34
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/cumsum_fp32.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/custom_gru_fp32.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/deconv_fp32.h +0 -37
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/deconv_winograd_fp32.h +0 -46
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/detection_post_process_fp32.h +0 -60
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/embedding_lookup_fp32.h +0 -43
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/group_norm_fp32.h +0 -35
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/gru_fp32.h +0 -30
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/instance_norm_fp32.h +0 -50
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/invert_permutation_fp32.h +0 -30
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/l2_norm_fp32.h +0 -34
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/layer_norm_fp32.h +0 -33
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/local_response_norm_fp32.h +0 -26
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/log_softmax_fp32.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/matmul_avx512_mask_fp32.h +0 -209
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/non_max_suppression_fp32.h +0 -25
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/online_fusion/cast_gather_reduce_fp32.h +0 -37
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/online_fusion/reduce_concat_fp32.h +0 -34
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/online_fusion/split_reduce_concat_fp32.h +0 -33
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/pad_fp32.h +0 -40
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/prelu_fp32.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/prior_box_fp32.h +0 -41
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/range_fp32.h +0 -34
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/rank_fp32.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/resize_fp32.h +0 -74
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/reverse_fp32.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/reverse_sequence_fp32.h +0 -33
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/scale_fp32.h +0 -35
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/sparse_to_dense_fp32.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/splice_fp32.h +0 -26
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/transpose_fp32.h +0 -35
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/transpose_server_fp32.h +0 -40
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/triu_tril_fp32.h +0 -42
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/unique_fp32.h +0 -36
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/where_fp32.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/winograd_avx.h +0 -299
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/winograd_transform.h +0 -51
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32/winograd_utils.h +0 -373
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32_grad/batch_norm_grad.h +0 -37
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32_grad/batch_norm_parameter.h +0 -28
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32_grad/binary_cross_entropy_grad.h +0 -36
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32_grad/convolution_grad_filter.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32_grad/convolution_grad_input.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32_grad/dropout_parameter.h +0 -27
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32_grad/layernorm_grad.h +0 -29
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32_grad/nllloss_grad_fp32.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32_grad/optimizer.h +0 -40
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32_grad/pack_ext.h +0 -39
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32_grad/pooling_grad.h +0 -34
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32_grad/reduce_grad.h +0 -30
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32_grad/smooth_l1_loss.h +0 -27
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32_grad/softmax_cross_entropy_with_logits.h +0 -33
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32_grad/softmax_grad_utils.h +0 -33
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/fp32_sparse/matmul_sparse_x1_fp32.h +0 -41
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/gather_nd_parameter.h +0 -26
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/gelu_parameter.h +0 -28
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/grid_sampler_parameter.h +0 -28
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/group_norm_parameter.h +0 -41
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/infer/cast_gather_reduce_infer.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/infer/control/tensor_array_infer.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/infer/control/tensor_array_read_infer.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/infer/control/tensor_array_write_infer.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/infer/control/tensorlist_fromtensor_infer.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/infer/control/tensorlist_getitem_infer.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/infer/control/tensorlist_reserve_infer.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/infer/control/tensorlist_setitem_infer.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/infer/control/tensorlist_stack_infer.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/infer/conv3d_infer.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/infer/custom_is_inf_infer.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/infer/custom_masked_fill_infer.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/infer/custom_tensor_scatter_max_infer.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/infer/format_transpose_infer.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/infer/gather_d_infer.h +0 -33
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/infer/grid_sampler_infer.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/infer/group_norm_infer.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/infer/reduce_concat_infer.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/infer/sparse_fill_empty_rows_infer.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/infer/sparse_reshape_infer.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/infer/sparse_segment_sum_infer.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/infer/split_reduce_concat_infer.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/infer/string/custom_extract_features_infer.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/infer/string/custom_normalize_infer.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/infer/string/custom_predict_infer.h +0 -36
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/infer/string/hashtable_lookup_infer.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/infer/string/lsh_projection_infer.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/infer/string/skip_gram_infer.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/infer/triu_tril_infer.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/instance_norm_parameter.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/add_int8.h +0 -70
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/arg_min_max_int8.h +0 -41
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/arithmetic_int8.h +0 -51
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/arithmetic_self_int8.h +0 -59
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/batch_to_space_int8.h +0 -33
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/batchnorm_int8.h +0 -34
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/common_func_int8.h +0 -95
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/concat_int8.h +0 -33
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/conv1x1_int8.h +0 -46
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/conv3x3_int8.h +0 -48
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/conv_depthwise_int8.h +0 -49
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/conv_int8.h +0 -44
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/crop_int8.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/deconv_int8.h +0 -46
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/depth_to_space_int8.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/div_int8.h +0 -37
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/dynamic_gather_int8.h +0 -40
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/dynamic_matmul_int8.h +0 -74
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/dynamic_quant_int8.h +0 -34
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/gatherNd_int8.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/gather_int8.h +0 -35
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/hswish_int8.h +0 -43
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/l2_norm_int8.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/layer_norm_int8.h +0 -35
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/leaky_relu_int8.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/matmul_int8.h +0 -93
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/mul_int8.h +0 -39
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/pack_int8.h +0 -56
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/pad_int8.h +0 -35
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/pooling_int8.h +0 -50
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/power_int8.h +0 -33
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/quant_dtype_cast_int8.h +0 -56
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/reduce_int8.h +0 -70
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/relux_int8.h +0 -43
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/reshape_int8.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/resize_int8.h +0 -50
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/scale_int8.h +0 -35
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/sigmoid_int8.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/slice_int8.h +0 -35
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/softmax_int8.h +0 -35
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/space_to_batch_int8.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/split_int8.h +0 -33
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/squeeze_int8.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/sub_int8.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/tanh_int8.h +0 -43
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/topk_int8.h +0 -36
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/transpose_int8.h +0 -36
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/int8/unsqueeze_int8.h +0 -33
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/intrinsics/avx/common_utils.h +0 -157
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/intrinsics/sse/sse_common.h +0 -390
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/activation.h +0 -25
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/addn.h +0 -35
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/arg_min_max.h +0 -63
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/arithmetic_compare.h +0 -26
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/arithmetic_self.h +0 -48
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/batch_norm.h +0 -38
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/batch_to_space.h +0 -33
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/biasadd.h +0 -25
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/cast.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/clip.h +0 -34
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/concat.h +0 -52
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/convolution_1x1.h +0 -42
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/convolution_base.h +0 -63
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/convolution_delegate.h +0 -39
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/convolution_depthwise.h +0 -36
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/convolution_depthwise_3x3.h +0 -37
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/convolution_depthwise_indirect.h +0 -39
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/convolution_depthwise_sw.h +0 -36
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/convolution_depthwise_sw_avx.h +0 -40
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/convolution_im2col.h +0 -28
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/convolution_im2col_arm32.h +0 -30
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/convolution_im2col_arm64.h +0 -29
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/convolution_im2col_avx.h +0 -29
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/convolution_im2col_avx512.h +0 -29
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/convolution_im2col_base.h +0 -52
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/convolution_im2col_sse.h +0 -29
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/convolution_slidewindow.h +0 -46
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/convolution_sw_1x1.h +0 -36
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/convolution_sw_arm64.h +0 -28
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/convolution_sw_avx.h +0 -28
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/convolution_winograd.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/convolution_winograd_arm32.h +0 -30
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/convolution_winograd_arm64.h +0 -30
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/convolution_winograd_avx.h +0 -30
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/convolution_winograd_base.h +0 -65
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/convolution_winograd_sse.h +0 -30
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/crop.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/crop_and_resize.h +0 -41
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/deconvolution.h +0 -39
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/deconvolution_depthwise.h +0 -34
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/deconvolution_winograd.h +0 -52
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/depth_to_space.h +0 -42
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/f16/arithmetic_compare_f16.h +0 -26
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/f16/arithmetic_f16.h +0 -42
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/f16/concat_f16.h +0 -25
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/f16/reduce_f16.h +0 -27
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/f16/stack_f16.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/fill.h +0 -36
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/fullconnection.h +0 -25
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/fused_batch_norm.h +0 -37
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/gather.h +0 -46
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/gather_d.h +0 -25
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/gather_nd.h +0 -35
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/group_convolution.h +0 -49
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/group_norm.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/init_vs_kernels.h +0 -20
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/layer_norm.h +0 -49
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/local_response_norm.h +0 -30
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/log_softmax.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/matmul.h +0 -25
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/matmul_arm32.h +0 -28
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/matmul_arm64.h +0 -28
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/matmul_avx.h +0 -28
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/matmul_avx512.h +0 -27
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/matmul_base.h +0 -35
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/matmul_create.h +0 -24
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/matmul_sse.h +0 -27
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/non_max_suppression.h +0 -34
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/non_zero.h +0 -30
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/ones_like.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/pad.h +0 -51
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/pow.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/prelu.h +0 -34
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/prior_box.h +0 -36
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/range.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/rank.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/reduce.h +0 -72
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/reverse.h +0 -36
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/scale.h +0 -41
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/shape.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/size.h +0 -30
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/softmax.h +0 -39
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/splice.h +0 -30
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/stack.h +0 -41
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/transpose.h +0 -49
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/tril.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/triu.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/unique.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/where.h +0 -44
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/kernel/zeros_like.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/l2_norm_parameter.h +0 -41
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/local_response_norm_parameter.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/lsh_projection_parameter.h +0 -35
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/mul_parameter.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/non_max_suppression_parameter.h +0 -28
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/pack.h +0 -23
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/partial_fusion_parameter.h +0 -29
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/predict_parameter.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/prelu_parameter.h +0 -26
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/random_parameter.h +0 -34
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/reverse_parameter.h +0 -30
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/reverse_sequence_parameter.h +0 -45
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/scale_parameter.h +0 -39
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/scatter_elements_parameter.h +0 -25
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/scatter_nd_parameter.h +0 -29
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/sigmoid_parameter.h +0 -41
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/skip_gram_parameter.h +0 -30
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/sparse_to_dense_parameter.h +0 -32
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/tensor_array_parameter.h +0 -29
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/triu_tril_parameter.h +0 -31
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/upsample_parameter.h +0 -29
- mindspore/include/mindspore/ops/kernel/cpu/nnacl/where_parameter.h +0 -25
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/gmm_backward.h +0 -38
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/gmm_v2_backward.h +0 -38
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/select_ext.h +0 -38
- mindspore/include/mindspore/ops/kernel/cpu/pyboost/auto_generate/transpose_ext.h +0 -38
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/gmm_backward.h +0 -38
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/gmm_v2_backward.h +0 -38
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/select_ext.h +0 -38
- mindspore/include/mindspore/ops/kernel/gpu/pyboost/auto_generate/transpose_ext.h +0 -38
- mindspore/include/mindspore/ops/ops_utils/ms_extension.h +0 -39
- mindspore/include/mindspore/ops/view/select_ext_strides_calc.h +0 -30
- mindspore/include/mindspore/ops/view/transpose_ext_strides_calc.h +0 -32
- mindspore/include/mindspore/ops/view/unstack_ext_strides_calc.h +0 -28
- 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_aicore_ops/op_impl/ai_core/tbe/config/ascend310/aic-ascend310-ops-info.json +0 -123
- mindspore/lib/plugin/ascend/custom_aicore_ops/op_impl/ai_core/tbe/config/ascend310p/aic-ascend310p-ops-info.json +0 -152
- mindspore/lib/plugin/ascend/custom_aicore_ops/op_impl/ai_core/tbe/config/ascend910/aic-ascend910-ops-info.json +0 -2048
- mindspore/lib/plugin/ascend/custom_aicore_ops/op_impl/ai_core/tbe/config/ascend910_93/aic-ascend910_93-ops-info.json +0 -2048
- mindspore/lib/plugin/ascend/custom_aicore_ops/op_impl/ai_core/tbe/config/ascend910b/aic-ascend910b-ops-info.json +0 -2048
- mindspore/lib/plugin/ascend/custom_aicore_ops/op_impl/ai_core/tbe/custom_aicore_ops_impl/add_dsl.py +0 -46
- mindspore/lib/plugin/ascend/custom_aicore_ops/op_impl/ai_core/tbe/custom_aicore_ops_impl/add_tik.py +0 -51
- mindspore/lib/plugin/ascend/custom_aicore_ops/op_impl/ai_core/tbe/custom_aicore_ops_impl/build_tbe_kernel.py +0 -529
- mindspore/lib/plugin/ascend/custom_aicore_ops/op_impl/ai_core/tbe/custom_aicore_ops_impl/compiler.py +0 -56
- mindspore/lib/plugin/ascend/custom_aicore_ops/op_impl/ai_core/tbe/custom_aicore_ops_impl/custom.py +0 -1109
- mindspore/lib/plugin/ascend/custom_aicore_ops/op_impl/ai_core/tbe/custom_aicore_ops_impl/get_file_path.py +0 -36
- mindspore/lib/plugin/ascend/custom_aicore_ops/op_impl/ai_core/tbe/custom_aicore_ops_impl/kv_cache_mgr.py +0 -241
- mindspore/lib/plugin/ascend/custom_aicore_ops/op_impl/ai_core/tbe/custom_aicore_ops_impl/matmul_tik.py +0 -212
- mindspore/lib/plugin/ascend/custom_aicore_ops/op_impl/ai_core/tbe/custom_aicore_ops_impl/tbe_topi.py +0 -556
- mindspore/lib/plugin/ascend/custom_aicore_ops/op_impl/vector_core/tbe/custom_aicore_ops_impl/add_dsl.py +0 -46
- mindspore/lib/plugin/ascend/custom_aicore_ops/op_impl/vector_core/tbe/custom_aicore_ops_impl/add_tik.py +0 -51
- mindspore/lib/plugin/ascend/custom_aicore_ops/op_impl/vector_core/tbe/custom_aicore_ops_impl/kv_cache_mgr.py +0 -241
- mindspore/lib/plugin/ascend/custom_aicore_ops/op_impl/vector_core/tbe/custom_aicore_ops_impl/matmul_tik.py +0 -212
- mindspore/lib/plugin/ascend/custom_aicore_ops/op_proto/libop_proto.so +0 -0
- 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_api/lib/libcust_opapi.so +0 -0
- 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/x86_64/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/x86_64/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/libmindspore_gpu_res_manager.so +0 -0
- mindspore/profiler/common/validator/__init__.py +0 -14
- mindspore/profiler/common/validator/validate_path.py +0 -84
- 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/utils/hooks.py +0 -81
- mindspore-2.6.0rc1.dist-info/METADATA +0 -367
- mindspore-2.6.0rc1.dist-info/RECORD +0 -12185
- mindspore-2.6.0rc1.dist-info/WHEEL +0 -5
- /mindspore/common/{_auto_dynamic.py → dynamic_shape/_auto_dynamic.py} +0 -0
- /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.0.dist-info}/entry_points.txt +0 -0
- {mindspore-2.6.0rc1.dist-info → mindspore-2.7.0.dist-info}/top_level.txt +0 -0
|
@@ -26,1447 +26,1651 @@ namespace mindspore {
|
|
|
26
26
|
namespace kernel {
|
|
27
27
|
namespace pyboost {
|
|
28
28
|
enum class OpType {
|
|
29
|
-
|
|
30
|
-
|
|
31
|
-
|
|
32
|
-
|
|
33
|
-
|
|
34
|
-
|
|
35
|
-
|
|
36
|
-
|
|
37
|
-
|
|
38
|
-
|
|
39
|
-
|
|
40
|
-
|
|
41
|
-
|
|
42
|
-
|
|
43
|
-
|
|
44
|
-
|
|
45
|
-
|
|
46
|
-
|
|
47
|
-
|
|
48
|
-
|
|
49
|
-
|
|
50
|
-
|
|
51
|
-
|
|
52
|
-
|
|
53
|
-
|
|
54
|
-
|
|
55
|
-
|
|
56
|
-
|
|
57
|
-
|
|
58
|
-
|
|
59
|
-
|
|
60
|
-
|
|
61
|
-
|
|
62
|
-
|
|
63
|
-
|
|
64
|
-
|
|
65
|
-
|
|
66
|
-
|
|
67
|
-
|
|
68
|
-
|
|
69
|
-
|
|
70
|
-
|
|
71
|
-
|
|
72
|
-
|
|
73
|
-
|
|
74
|
-
|
|
75
|
-
|
|
76
|
-
|
|
77
|
-
|
|
78
|
-
|
|
79
|
-
|
|
80
|
-
|
|
81
|
-
|
|
82
|
-
|
|
83
|
-
|
|
84
|
-
|
|
85
|
-
|
|
86
|
-
|
|
87
|
-
|
|
88
|
-
|
|
89
|
-
|
|
90
|
-
|
|
91
|
-
|
|
92
|
-
|
|
93
|
-
|
|
94
|
-
|
|
95
|
-
|
|
96
|
-
|
|
97
|
-
|
|
98
|
-
|
|
99
|
-
|
|
100
|
-
|
|
101
|
-
|
|
102
|
-
|
|
103
|
-
|
|
104
|
-
|
|
105
|
-
|
|
106
|
-
|
|
107
|
-
|
|
108
|
-
|
|
109
|
-
|
|
110
|
-
|
|
111
|
-
|
|
112
|
-
|
|
113
|
-
|
|
114
|
-
|
|
115
|
-
|
|
116
|
-
|
|
117
|
-
|
|
118
|
-
|
|
119
|
-
|
|
120
|
-
|
|
121
|
-
|
|
122
|
-
|
|
123
|
-
|
|
124
|
-
|
|
125
|
-
|
|
126
|
-
|
|
127
|
-
|
|
128
|
-
|
|
129
|
-
|
|
130
|
-
|
|
131
|
-
|
|
132
|
-
|
|
133
|
-
|
|
134
|
-
|
|
135
|
-
|
|
136
|
-
|
|
137
|
-
|
|
138
|
-
|
|
139
|
-
|
|
140
|
-
|
|
141
|
-
|
|
142
|
-
|
|
143
|
-
|
|
144
|
-
|
|
145
|
-
|
|
146
|
-
|
|
147
|
-
|
|
148
|
-
|
|
149
|
-
|
|
150
|
-
|
|
151
|
-
|
|
152
|
-
|
|
153
|
-
|
|
154
|
-
|
|
155
|
-
|
|
156
|
-
|
|
157
|
-
|
|
158
|
-
|
|
159
|
-
|
|
160
|
-
|
|
161
|
-
|
|
162
|
-
|
|
163
|
-
|
|
164
|
-
|
|
165
|
-
|
|
166
|
-
|
|
167
|
-
|
|
168
|
-
|
|
169
|
-
|
|
170
|
-
|
|
171
|
-
|
|
172
|
-
|
|
173
|
-
|
|
174
|
-
|
|
175
|
-
|
|
176
|
-
|
|
177
|
-
|
|
178
|
-
|
|
179
|
-
|
|
180
|
-
|
|
181
|
-
|
|
182
|
-
|
|
183
|
-
|
|
184
|
-
|
|
185
|
-
|
|
186
|
-
|
|
187
|
-
|
|
188
|
-
|
|
189
|
-
|
|
190
|
-
|
|
191
|
-
|
|
192
|
-
|
|
193
|
-
|
|
194
|
-
|
|
195
|
-
|
|
196
|
-
|
|
197
|
-
|
|
198
|
-
|
|
199
|
-
|
|
200
|
-
|
|
201
|
-
|
|
202
|
-
|
|
203
|
-
|
|
204
|
-
|
|
205
|
-
|
|
206
|
-
|
|
207
|
-
|
|
208
|
-
|
|
209
|
-
|
|
210
|
-
|
|
211
|
-
|
|
212
|
-
|
|
213
|
-
|
|
214
|
-
|
|
215
|
-
|
|
216
|
-
|
|
217
|
-
|
|
218
|
-
|
|
219
|
-
|
|
220
|
-
|
|
221
|
-
|
|
222
|
-
|
|
223
|
-
|
|
224
|
-
|
|
225
|
-
|
|
226
|
-
|
|
227
|
-
|
|
228
|
-
|
|
229
|
-
|
|
230
|
-
|
|
231
|
-
|
|
232
|
-
|
|
233
|
-
|
|
29
|
+
kInplaceElu = 0,
|
|
30
|
+
kCumminExt = 1,
|
|
31
|
+
kInnerCommIsend = 2,
|
|
32
|
+
kMultiScaleDeformableAttn = 3,
|
|
33
|
+
kReduceAny = 4,
|
|
34
|
+
kInplaceFloorDivide = 5,
|
|
35
|
+
kMedianDim = 6,
|
|
36
|
+
kDivMods = 7,
|
|
37
|
+
kNLLLoss2d = 8,
|
|
38
|
+
kExpandDimsView = 9,
|
|
39
|
+
kNormalFloatTensor = 10,
|
|
40
|
+
kSeLUExt = 11,
|
|
41
|
+
kEluExt = 12,
|
|
42
|
+
kZeros = 13,
|
|
43
|
+
kExpm1 = 14,
|
|
44
|
+
kGroupNorm = 15,
|
|
45
|
+
kDistCommBroadcast = 16,
|
|
46
|
+
kCopy = 17,
|
|
47
|
+
kUpsampleNearest3D = 18,
|
|
48
|
+
kInplaceAddsExt = 19,
|
|
49
|
+
kNonZero = 20,
|
|
50
|
+
kInplaceDiv = 21,
|
|
51
|
+
kExpandAs = 22,
|
|
52
|
+
kBatchNormReduceGrad = 23,
|
|
53
|
+
kHardtanh = 24,
|
|
54
|
+
kPReLU = 25,
|
|
55
|
+
kCountNonZero = 26,
|
|
56
|
+
kInplaceMul = 27,
|
|
57
|
+
kInnerCommIrecv = 28,
|
|
58
|
+
kMla = 29,
|
|
59
|
+
kRemainderScalarTensor = 30,
|
|
60
|
+
kRepeat = 31,
|
|
61
|
+
kSoftmaxBackward = 32,
|
|
62
|
+
kFloorDivScalar = 33,
|
|
63
|
+
kInplaceZero = 34,
|
|
64
|
+
kNewZeros = 35,
|
|
65
|
+
kGreaterEqual = 36,
|
|
66
|
+
kGreaterEqualScalar = 37,
|
|
67
|
+
kSmoothL1LossGrad = 38,
|
|
68
|
+
kHSwish = 39,
|
|
69
|
+
kRsqrt = 40,
|
|
70
|
+
kLeakyReLUGradExt = 41,
|
|
71
|
+
kInplaceDivMod = 42,
|
|
72
|
+
kIndexSelect = 43,
|
|
73
|
+
kNonZeroExt = 44,
|
|
74
|
+
kStd = 45,
|
|
75
|
+
kRound = 46,
|
|
76
|
+
kBatchNormExt = 47,
|
|
77
|
+
kInplaceAddmm = 48,
|
|
78
|
+
kIm2ColExt = 49,
|
|
79
|
+
kFloor = 50,
|
|
80
|
+
kStackExt = 51,
|
|
81
|
+
kInnerInplaceIndexPut = 52,
|
|
82
|
+
kInplaceSubScalar = 53,
|
|
83
|
+
kGridSampler3D = 54,
|
|
84
|
+
kMaxUnpool2DExt = 55,
|
|
85
|
+
kAbs = 56,
|
|
86
|
+
kSoftplusGradExt = 57,
|
|
87
|
+
kAvgPool1D = 58,
|
|
88
|
+
kLog2 = 59,
|
|
89
|
+
kNormalTensorTensor = 60,
|
|
90
|
+
kReshapeAndCache = 61,
|
|
91
|
+
kRandn = 62,
|
|
92
|
+
kUpsampleBilinear2D = 63,
|
|
93
|
+
kXlogy = 64,
|
|
94
|
+
kInnerCommAllGather = 65,
|
|
95
|
+
kMax = 66,
|
|
96
|
+
kMatmulReduceScatter = 67,
|
|
97
|
+
kAdaptiveAvgPool2DExt = 68,
|
|
98
|
+
kLogSoftmaxExt = 69,
|
|
99
|
+
kInplaceGroupedMatmulAdd = 70,
|
|
100
|
+
kDot = 71,
|
|
101
|
+
kSortExt = 72,
|
|
102
|
+
kRandInt = 73,
|
|
103
|
+
kReflectionPad2D = 74,
|
|
104
|
+
kThresholdGrad = 75,
|
|
105
|
+
kSoftMarginLossGrad = 76,
|
|
106
|
+
kLogSoftmax = 77,
|
|
107
|
+
kNotEqual = 78,
|
|
108
|
+
kFloorDiv = 79,
|
|
109
|
+
kMin = 80,
|
|
110
|
+
kPagedAttention = 81,
|
|
111
|
+
kEmbeddingDenseBackward = 82,
|
|
112
|
+
kScatterAddExt = 83,
|
|
113
|
+
kCol2ImExt = 84,
|
|
114
|
+
kLog1p = 85,
|
|
115
|
+
kConvolution = 86,
|
|
116
|
+
kBatchNormGatherStatsWithCounts = 87,
|
|
117
|
+
kApplyRotaryPosEmb = 88,
|
|
118
|
+
kRmsNormGrad = 89,
|
|
119
|
+
kSilentCheckV2 = 90,
|
|
120
|
+
kReplicationPad3DGrad = 91,
|
|
121
|
+
kInplaceFloor = 92,
|
|
122
|
+
kInnerNonZero = 93,
|
|
123
|
+
kIsInf = 94,
|
|
124
|
+
kRepeatInterleaveInt = 95,
|
|
125
|
+
kAtan2Ext = 96,
|
|
126
|
+
kInplaceFillTensor = 97,
|
|
127
|
+
kScatterValue = 98,
|
|
128
|
+
kSplitWithSizeView = 99,
|
|
129
|
+
kDistCommBarrier = 100,
|
|
130
|
+
kInplaceIndexPut = 101,
|
|
131
|
+
kViewAs = 102,
|
|
132
|
+
kCellBackwardHook = 103,
|
|
133
|
+
kHardtanhGrad = 104,
|
|
134
|
+
kDistCommGather = 105,
|
|
135
|
+
kLerp = 106,
|
|
136
|
+
kTriu = 107,
|
|
137
|
+
kLogAddExp = 108,
|
|
138
|
+
kUpsampleNearest3DGrad = 109,
|
|
139
|
+
kFlashAttentionScore = 110,
|
|
140
|
+
kPowTensorScalar = 111,
|
|
141
|
+
kCosh = 112,
|
|
142
|
+
kTanhGrad = 113,
|
|
143
|
+
kConv1DExt = 114,
|
|
144
|
+
kTrilExt = 115,
|
|
145
|
+
kBatchNormStats = 116,
|
|
146
|
+
kBitwiseAndTensor = 117,
|
|
147
|
+
kDistCommReduceScatterTensorUneven = 118,
|
|
148
|
+
kReplicationPad2D = 119,
|
|
149
|
+
kIdentity = 120,
|
|
150
|
+
kMaxPoolWithIndices = 121,
|
|
151
|
+
kSub = 122,
|
|
152
|
+
kExp = 123,
|
|
153
|
+
kFillScalar = 124,
|
|
154
|
+
kAddcmulExt = 125,
|
|
155
|
+
kClone = 126,
|
|
156
|
+
kAtanExt = 127,
|
|
157
|
+
kInnerCommAllToAllV = 128,
|
|
158
|
+
kAdaptiveMaxPool2D = 129,
|
|
159
|
+
kAddmm = 130,
|
|
160
|
+
kTransposeView = 131,
|
|
161
|
+
kNorm = 132,
|
|
162
|
+
kDivs = 133,
|
|
163
|
+
kInplaceStopGradient = 134,
|
|
164
|
+
kCol2ImGrad = 135,
|
|
165
|
+
kRotaryPositionEmbeddingGrad = 136,
|
|
166
|
+
kConv3DExt = 137,
|
|
167
|
+
kInplaceScatterAdd = 138,
|
|
168
|
+
kInplaceIndexAddExt = 139,
|
|
169
|
+
kSplitTensor = 140,
|
|
170
|
+
kMaxPoolGradWithIndices = 141,
|
|
171
|
+
kEmbedding = 142,
|
|
172
|
+
kDistCommReduceScatter = 143,
|
|
173
|
+
kNewEmpty = 144,
|
|
174
|
+
kSelectExtView = 145,
|
|
175
|
+
kIncreFlashAttention = 146,
|
|
176
|
+
kReverseV2 = 147,
|
|
177
|
+
kLeakyReLUExt = 148,
|
|
178
|
+
kRandnLike = 149,
|
|
179
|
+
kGeLU = 150,
|
|
180
|
+
kExp2 = 151,
|
|
181
|
+
kReshape = 152,
|
|
182
|
+
kThreshold = 153,
|
|
183
|
+
kUpsampleLinear1DGrad = 154,
|
|
184
|
+
kLogicalAnd = 155,
|
|
185
|
+
kInnerMoeTokenUnpermute = 156,
|
|
186
|
+
kAddbmm = 157,
|
|
187
|
+
kXLogYScalarOther = 158,
|
|
188
|
+
kAcoshExt = 159,
|
|
189
|
+
kView = 160,
|
|
190
|
+
kReduceMin = 161,
|
|
191
|
+
kSpeedFusionAttentionGrad = 162,
|
|
192
|
+
kRemainderTensorScalar = 163,
|
|
193
|
+
kSqrt = 164,
|
|
194
|
+
kTan = 165,
|
|
195
|
+
kSilentCheckV3 = 166,
|
|
196
|
+
kAddmv = 167,
|
|
197
|
+
kInplaceReLU = 168,
|
|
198
|
+
kRandIntLike = 169,
|
|
199
|
+
kGcd = 170,
|
|
200
|
+
kSplit = 171,
|
|
201
|
+
kMul = 172,
|
|
202
|
+
kEluGradExt = 173,
|
|
203
|
+
kRandpermExt = 174,
|
|
204
|
+
kDiagonalView = 175,
|
|
205
|
+
kTrunc = 176,
|
|
206
|
+
kInplaceLog = 177,
|
|
207
|
+
kAddcdivExt = 178,
|
|
208
|
+
kTExt = 179,
|
|
209
|
+
kUpsampleTrilinear3DGrad = 180,
|
|
210
|
+
kBitwiseOrScalar = 181,
|
|
211
|
+
kBitwiseOrTensor = 182,
|
|
212
|
+
kReduceAll = 183,
|
|
213
|
+
kBroadcastTo = 184,
|
|
214
|
+
kAdaptiveAvgPool3DGradExt = 185,
|
|
215
|
+
kRepeatInterleaveGrad = 186,
|
|
216
|
+
kGluGrad = 187,
|
|
217
|
+
kCos = 188,
|
|
218
|
+
kGridSampler2D = 189,
|
|
219
|
+
kConstantPadND = 190,
|
|
220
|
+
kMaskedSelect = 191,
|
|
221
|
+
kUpsampleBicubic2D = 192,
|
|
222
|
+
kMaskedFill = 193,
|
|
223
|
+
kEmptyLike = 194,
|
|
224
|
+
kSiLU = 195,
|
|
225
|
+
kNansum = 196,
|
|
226
|
+
kSplitWithSize = 197,
|
|
227
|
+
kDistCommAllGatherIntoTensor = 198,
|
|
228
|
+
kL1LossBackwardExt = 199,
|
|
229
|
+
kAddRmsNorm = 200,
|
|
230
|
+
kInplaceSiLU = 201,
|
|
231
|
+
kReLU = 202,
|
|
232
|
+
kStdMean = 203,
|
|
233
|
+
kAvgPool2DGrad = 204,
|
|
234
234
|
kSoftMarginLoss = 205,
|
|
235
|
-
|
|
236
|
-
|
|
237
|
-
|
|
238
|
-
|
|
239
|
-
|
|
240
|
-
|
|
241
|
-
|
|
242
|
-
|
|
243
|
-
|
|
244
|
-
|
|
245
|
-
|
|
246
|
-
|
|
247
|
-
|
|
248
|
-
|
|
249
|
-
|
|
250
|
-
|
|
251
|
-
|
|
252
|
-
|
|
253
|
-
|
|
254
|
-
|
|
255
|
-
|
|
256
|
-
|
|
257
|
-
|
|
258
|
-
|
|
259
|
-
|
|
260
|
-
|
|
261
|
-
|
|
262
|
-
|
|
263
|
-
|
|
264
|
-
|
|
265
|
-
|
|
266
|
-
|
|
267
|
-
|
|
268
|
-
|
|
269
|
-
|
|
270
|
-
|
|
271
|
-
|
|
272
|
-
|
|
273
|
-
|
|
274
|
-
|
|
275
|
-
|
|
276
|
-
|
|
277
|
-
|
|
278
|
-
|
|
279
|
-
|
|
280
|
-
|
|
281
|
-
|
|
282
|
-
|
|
283
|
-
|
|
284
|
-
|
|
285
|
-
|
|
286
|
-
|
|
287
|
-
|
|
288
|
-
|
|
289
|
-
|
|
290
|
-
|
|
291
|
-
|
|
292
|
-
|
|
293
|
-
|
|
294
|
-
|
|
295
|
-
|
|
296
|
-
|
|
297
|
-
|
|
298
|
-
|
|
299
|
-
|
|
300
|
-
|
|
301
|
-
|
|
302
|
-
|
|
303
|
-
|
|
304
|
-
|
|
305
|
-
|
|
306
|
-
|
|
307
|
-
|
|
308
|
-
|
|
309
|
-
|
|
310
|
-
|
|
311
|
-
|
|
312
|
-
|
|
313
|
-
|
|
314
|
-
|
|
315
|
-
|
|
316
|
-
|
|
317
|
-
|
|
318
|
-
|
|
319
|
-
|
|
320
|
-
|
|
321
|
-
|
|
322
|
-
|
|
323
|
-
|
|
324
|
-
|
|
325
|
-
|
|
326
|
-
|
|
327
|
-
|
|
328
|
-
|
|
329
|
-
|
|
330
|
-
|
|
331
|
-
|
|
332
|
-
|
|
333
|
-
|
|
334
|
-
|
|
335
|
-
|
|
336
|
-
|
|
337
|
-
|
|
338
|
-
|
|
339
|
-
|
|
340
|
-
|
|
341
|
-
|
|
342
|
-
|
|
343
|
-
|
|
344
|
-
|
|
345
|
-
|
|
346
|
-
|
|
347
|
-
|
|
348
|
-
|
|
349
|
-
|
|
350
|
-
|
|
351
|
-
|
|
352
|
-
|
|
353
|
-
|
|
354
|
-
|
|
355
|
-
|
|
356
|
-
|
|
357
|
-
|
|
358
|
-
|
|
359
|
-
|
|
360
|
-
|
|
361
|
-
|
|
362
|
-
|
|
363
|
-
|
|
364
|
-
|
|
365
|
-
|
|
366
|
-
|
|
367
|
-
|
|
368
|
-
|
|
369
|
-
|
|
370
|
-
|
|
371
|
-
|
|
372
|
-
|
|
373
|
-
|
|
374
|
-
|
|
375
|
-
|
|
376
|
-
|
|
377
|
-
|
|
378
|
-
|
|
379
|
-
|
|
380
|
-
|
|
381
|
-
|
|
382
|
-
|
|
383
|
-
|
|
384
|
-
|
|
385
|
-
|
|
386
|
-
|
|
387
|
-
|
|
388
|
-
|
|
389
|
-
|
|
390
|
-
|
|
391
|
-
|
|
392
|
-
|
|
393
|
-
|
|
394
|
-
|
|
395
|
-
|
|
396
|
-
|
|
397
|
-
|
|
398
|
-
|
|
399
|
-
|
|
400
|
-
|
|
401
|
-
|
|
402
|
-
|
|
403
|
-
|
|
404
|
-
|
|
405
|
-
|
|
406
|
-
|
|
407
|
-
|
|
408
|
-
|
|
409
|
-
|
|
410
|
-
|
|
411
|
-
|
|
412
|
-
|
|
413
|
-
|
|
414
|
-
|
|
415
|
-
|
|
416
|
-
|
|
417
|
-
|
|
418
|
-
|
|
419
|
-
|
|
420
|
-
|
|
421
|
-
|
|
422
|
-
|
|
423
|
-
|
|
424
|
-
|
|
425
|
-
|
|
426
|
-
|
|
427
|
-
|
|
428
|
-
|
|
429
|
-
|
|
430
|
-
|
|
431
|
-
|
|
432
|
-
|
|
433
|
-
|
|
434
|
-
|
|
435
|
-
|
|
436
|
-
|
|
437
|
-
|
|
438
|
-
|
|
439
|
-
|
|
440
|
-
|
|
441
|
-
|
|
442
|
-
|
|
443
|
-
|
|
444
|
-
|
|
445
|
-
|
|
446
|
-
|
|
447
|
-
|
|
448
|
-
|
|
449
|
-
|
|
450
|
-
|
|
451
|
-
|
|
452
|
-
|
|
453
|
-
|
|
454
|
-
|
|
455
|
-
|
|
456
|
-
|
|
457
|
-
|
|
458
|
-
|
|
459
|
-
|
|
460
|
-
|
|
461
|
-
|
|
462
|
-
|
|
463
|
-
|
|
464
|
-
|
|
465
|
-
|
|
466
|
-
|
|
467
|
-
|
|
468
|
-
|
|
469
|
-
|
|
470
|
-
|
|
471
|
-
|
|
472
|
-
|
|
473
|
-
|
|
474
|
-
|
|
475
|
-
|
|
476
|
-
|
|
477
|
-
|
|
478
|
-
|
|
479
|
-
|
|
480
|
-
|
|
481
|
-
|
|
482
|
-
|
|
483
|
-
|
|
484
|
-
|
|
485
|
-
|
|
486
|
-
|
|
487
|
-
|
|
488
|
-
|
|
489
|
-
|
|
490
|
-
|
|
491
|
-
|
|
492
|
-
|
|
493
|
-
|
|
494
|
-
|
|
495
|
-
|
|
496
|
-
|
|
497
|
-
|
|
498
|
-
|
|
499
|
-
|
|
500
|
-
|
|
501
|
-
|
|
502
|
-
|
|
503
|
-
|
|
504
|
-
|
|
505
|
-
|
|
506
|
-
|
|
507
|
-
|
|
235
|
+
kLessEqual = 206,
|
|
236
|
+
kConvTranspose2D = 207,
|
|
237
|
+
kMeanExt = 208,
|
|
238
|
+
kMinimum = 209,
|
|
239
|
+
kBatchMatMul = 210,
|
|
240
|
+
kSiLUGrad = 211,
|
|
241
|
+
kFmodScalar = 212,
|
|
242
|
+
kGatherDGradV2 = 213,
|
|
243
|
+
kEqual = 214,
|
|
244
|
+
kMoeDistributeDispatch = 215,
|
|
245
|
+
kInplaceFloorDivides = 216,
|
|
246
|
+
kCumsumExt = 217,
|
|
247
|
+
kLayerNormGradExt = 218,
|
|
248
|
+
kLogSumExp = 219,
|
|
249
|
+
kBinaryCrossEntropy = 220,
|
|
250
|
+
kChunkView = 221,
|
|
251
|
+
kInplaceExp = 222,
|
|
252
|
+
kInplaceFillScalar = 223,
|
|
253
|
+
kAddScalar = 224,
|
|
254
|
+
kSoftplusExt = 225,
|
|
255
|
+
kHSwishGrad = 226,
|
|
256
|
+
kDistCommGatherIntoTensor = 227,
|
|
257
|
+
kDiagExt = 228,
|
|
258
|
+
kLogAddExp2 = 229,
|
|
259
|
+
kNormalFloatFloat = 230,
|
|
260
|
+
kReflectionPad3D = 231,
|
|
261
|
+
kSinh = 232,
|
|
262
|
+
kArgMaxExt = 233,
|
|
263
|
+
kAsStrided = 234,
|
|
264
|
+
kLinalgQr = 235,
|
|
265
|
+
kAdaptiveAvgPool1D = 236,
|
|
266
|
+
kNarrow = 237,
|
|
267
|
+
kMinDim = 238,
|
|
268
|
+
kInplaceFillDiagonal = 239,
|
|
269
|
+
kLogSigmoidGrad = 240,
|
|
270
|
+
kMishGradExt = 241,
|
|
271
|
+
kLess = 242,
|
|
272
|
+
kCross = 243,
|
|
273
|
+
kLogSoftmaxGrad = 244,
|
|
274
|
+
kInplaceMatmulAdd = 245,
|
|
275
|
+
kScatter = 246,
|
|
276
|
+
kAddExt = 247,
|
|
277
|
+
kReplicationPad3D = 248,
|
|
278
|
+
kHShrink = 249,
|
|
279
|
+
kAvgPool2D = 250,
|
|
280
|
+
kGreater = 251,
|
|
281
|
+
kEmpty = 252,
|
|
282
|
+
kFlattenExt = 253,
|
|
283
|
+
kMaxPoolGradWithMask = 254,
|
|
284
|
+
kL1LossExt = 255,
|
|
285
|
+
kGeluExt = 256,
|
|
286
|
+
kSqueeze = 257,
|
|
287
|
+
kDistCommIrecv = 258,
|
|
288
|
+
kLerpScalar = 259,
|
|
289
|
+
kInnerCommAllReduce = 260,
|
|
290
|
+
kSquare = 261,
|
|
291
|
+
kConv2DPadding = 262,
|
|
292
|
+
kArgMinWithValue = 263,
|
|
293
|
+
kSpeedFusionAttention = 264,
|
|
294
|
+
kDropoutGenMaskExt = 265,
|
|
295
|
+
kBinaryCrossEntropyWithLogitsBackward = 266,
|
|
296
|
+
kConvolutionStrGrad = 267,
|
|
297
|
+
kLogSigmoid = 268,
|
|
298
|
+
kLinalgVectorNorm = 269,
|
|
299
|
+
kBroadcastToView = 270,
|
|
300
|
+
kSoftShrink = 271,
|
|
301
|
+
kAddLayerNormGrad = 272,
|
|
302
|
+
kUnstackExtView = 273,
|
|
303
|
+
kDistCommBatchIsendIrecv = 274,
|
|
304
|
+
kGridSampler2DGrad = 275,
|
|
305
|
+
kRotaryPositionEmbedding = 276,
|
|
306
|
+
kVarMean = 277,
|
|
307
|
+
kKthvalue = 278,
|
|
308
|
+
kMv = 279,
|
|
309
|
+
kUniqueDim = 280,
|
|
310
|
+
kInplaceDivMods = 281,
|
|
311
|
+
kFillTensor = 282,
|
|
312
|
+
kExpandDims = 283,
|
|
313
|
+
kInplaceThreshold = 284,
|
|
314
|
+
kInplaceRemainderTensorTensor = 285,
|
|
315
|
+
kErfinv = 286,
|
|
316
|
+
kTriangularSolve = 287,
|
|
317
|
+
kInplaceScatterSrc = 288,
|
|
318
|
+
kRandExt = 289,
|
|
319
|
+
kBCEWithLogitsLoss = 290,
|
|
320
|
+
kFlashAttentionScoreGrad = 291,
|
|
321
|
+
kMaximum = 292,
|
|
322
|
+
kBaddbmm = 293,
|
|
323
|
+
kMoeTokenPermute = 294,
|
|
324
|
+
kInplaceSubExt = 295,
|
|
325
|
+
kInplaceErfinv = 296,
|
|
326
|
+
kMeshgrid = 297,
|
|
327
|
+
kMaxPoolWithMask = 298,
|
|
328
|
+
kSelectV2 = 299,
|
|
329
|
+
kLinSpaceExt = 300,
|
|
330
|
+
kRmsNorm = 301,
|
|
331
|
+
kSumExt = 302,
|
|
332
|
+
kZerosLikeExt = 303,
|
|
333
|
+
kConcat = 304,
|
|
334
|
+
kBitwiseXorScalar = 305,
|
|
335
|
+
kFmodTensor = 306,
|
|
336
|
+
kDistCommReduceScatterTensor = 307,
|
|
337
|
+
kCrossEntropyLossGrad = 308,
|
|
338
|
+
kUpsampleNearest1D = 309,
|
|
339
|
+
kIsFinite = 310,
|
|
340
|
+
kNarrowView = 311,
|
|
341
|
+
kInplaceMaskedFillTensor = 312,
|
|
342
|
+
kTanh = 313,
|
|
343
|
+
kNewFull = 314,
|
|
344
|
+
kMuls = 315,
|
|
345
|
+
kDropoutExt = 316,
|
|
346
|
+
kDistCommScatter = 317,
|
|
347
|
+
kAsinExt = 318,
|
|
348
|
+
kRemainderTensorTensor = 319,
|
|
349
|
+
kSinc = 320,
|
|
350
|
+
kInplaceDivs = 321,
|
|
351
|
+
kConv2DExt = 322,
|
|
352
|
+
kInnerCommReduceScatter = 323,
|
|
353
|
+
kMishExt = 324,
|
|
354
|
+
kAcosExt = 325,
|
|
355
|
+
kTraceExt = 326,
|
|
356
|
+
kInplacePut = 327,
|
|
357
|
+
kGLU = 328,
|
|
358
|
+
kAdd = 329,
|
|
359
|
+
kNLLLossGrad = 330,
|
|
360
|
+
kInplaceMaskedFillScalar = 331,
|
|
361
|
+
kRandLikeExt = 332,
|
|
362
|
+
kPolar = 333,
|
|
363
|
+
kCast = 334,
|
|
364
|
+
kSwiglu = 335,
|
|
365
|
+
kInplaceAddExt = 336,
|
|
366
|
+
kArange = 337,
|
|
367
|
+
kMoeDistributeCombine = 338,
|
|
368
|
+
kLog = 339,
|
|
369
|
+
kDivMod = 340,
|
|
370
|
+
kArgMaxWithValue = 341,
|
|
371
|
+
kRoll = 342,
|
|
372
|
+
kReduceMax = 343,
|
|
373
|
+
kLogicalOr = 344,
|
|
374
|
+
kNLLLoss2dGrad = 345,
|
|
375
|
+
kReflectionPad3DGrad = 346,
|
|
376
|
+
kInplaceClampTensor = 347,
|
|
377
|
+
kAtanh = 348,
|
|
378
|
+
kReplicationPad1D = 349,
|
|
379
|
+
kBatchNormElemt = 350,
|
|
380
|
+
kErfc = 351,
|
|
381
|
+
kHSigmoidGrad = 352,
|
|
382
|
+
kFrac = 353,
|
|
383
|
+
kSigmoidGrad = 354,
|
|
384
|
+
kFullLike = 355,
|
|
385
|
+
kSoftShrinkGrad = 356,
|
|
386
|
+
kBatchMatMulExt = 357,
|
|
387
|
+
kRingAttentionUpdate = 358,
|
|
388
|
+
kDistCommAllToAllV = 359,
|
|
389
|
+
kPReLUGrad = 360,
|
|
390
|
+
kRepeatInterleaveTensor = 361,
|
|
391
|
+
kAddLayerNormV2 = 362,
|
|
392
|
+
kBitwiseNot = 363,
|
|
393
|
+
kMatrixInverseExt = 364,
|
|
394
|
+
kAdaptiveAvgPool2DGradExt = 365,
|
|
395
|
+
kTransposeExtView = 366,
|
|
396
|
+
kLogicalXor = 367,
|
|
397
|
+
kDistCommIsend = 368,
|
|
398
|
+
kIndexAddExt = 369,
|
|
399
|
+
kChunk = 370,
|
|
400
|
+
kInplaceRemainderTensorScalar = 371,
|
|
401
|
+
kPowScalarTensor = 372,
|
|
402
|
+
kDistCommScatterTensor = 373,
|
|
403
|
+
kKLDivGrad = 374,
|
|
404
|
+
kDiv = 375,
|
|
405
|
+
kMedianExt = 376,
|
|
406
|
+
kSliceExtView = 377,
|
|
407
|
+
kCeil = 378,
|
|
408
|
+
kReflectionPad1D = 379,
|
|
409
|
+
kBatchNormGradExt = 380,
|
|
410
|
+
kMSELossGradExt = 381,
|
|
411
|
+
kSign = 382,
|
|
412
|
+
kTranspose = 383,
|
|
413
|
+
kHistcExt = 384,
|
|
414
|
+
kInplaceClampScalar = 385,
|
|
415
|
+
kGridSampler3DGrad = 386,
|
|
416
|
+
kInplaceHardtanh = 387,
|
|
417
|
+
kIsClose = 388,
|
|
418
|
+
kAvgPool3DExt = 389,
|
|
419
|
+
kUpsampleNearest1DGrad = 390,
|
|
420
|
+
kMaxDim = 391,
|
|
421
|
+
kAvgPool3DGradExt = 392,
|
|
422
|
+
kSubScalar = 393,
|
|
423
|
+
kFFNExt = 394,
|
|
424
|
+
kDropoutDoMaskExt = 395,
|
|
425
|
+
kInplaceRandom = 396,
|
|
426
|
+
kEqualExt = 397,
|
|
427
|
+
kMoeTokenPermuteGrad = 398,
|
|
428
|
+
kReflectionPad1DGrad = 399,
|
|
429
|
+
kDense = 400,
|
|
430
|
+
kInplaceScatterValueReduce = 401,
|
|
431
|
+
kClampScalar = 402,
|
|
432
|
+
kAllFinite = 403,
|
|
433
|
+
kInnerIndex = 404,
|
|
434
|
+
kReplicationPad2DGrad = 405,
|
|
435
|
+
kDistCommReduce = 406,
|
|
436
|
+
kLog10 = 407,
|
|
437
|
+
kProdExt = 408,
|
|
438
|
+
kArgMinExt = 409,
|
|
439
|
+
kContiguous = 410,
|
|
440
|
+
kSlice = 411,
|
|
441
|
+
kUniqueConsecutive = 412,
|
|
442
|
+
kConvolutionGrad = 413,
|
|
443
|
+
kDistCommAllGatherIntoTensorUneven = 414,
|
|
444
|
+
kOneHotExt = 415,
|
|
445
|
+
kOuter = 416,
|
|
446
|
+
kMultiScaleDeformableAttnGrad = 417,
|
|
447
|
+
kInplaceMuls = 418,
|
|
448
|
+
kCrossEntropyLoss = 419,
|
|
449
|
+
kReplicationPad1DGrad = 420,
|
|
450
|
+
kUpsampleNearest2D = 421,
|
|
451
|
+
kMatMul = 422,
|
|
452
|
+
kInplaceBernoulliTensor = 423,
|
|
453
|
+
kUnique2 = 424,
|
|
454
|
+
kUpsampleBilinear2DGrad = 425,
|
|
455
|
+
kBincountExt = 426,
|
|
456
|
+
kConvolutionStr = 427,
|
|
457
|
+
kDistCommAllGather = 428,
|
|
458
|
+
kIsNegInf = 429,
|
|
459
|
+
kGeluGradExt = 430,
|
|
460
|
+
kKLDiv = 431,
|
|
461
|
+
kAdaptiveAvgPool3DExt = 432,
|
|
462
|
+
kIndex = 433,
|
|
463
|
+
kSin = 434,
|
|
464
|
+
kBitwiseAndScalar = 435,
|
|
465
|
+
kUpsampleLinear1D = 436,
|
|
466
|
+
kUpsampleNearest2DGrad = 437,
|
|
467
|
+
kLogicalNot = 438,
|
|
468
|
+
kInplaceUniform = 439,
|
|
469
|
+
kHShrinkGrad = 440,
|
|
470
|
+
kPromptFlashAttention = 441,
|
|
471
|
+
kClampTensor = 442,
|
|
472
|
+
kInplaceNormal = 443,
|
|
473
|
+
kAdaptiveMaxPool1D = 444,
|
|
474
|
+
kLayerNormExt = 445,
|
|
475
|
+
kUniformExt = 446,
|
|
476
|
+
kErf = 447,
|
|
477
|
+
kArgSort = 448,
|
|
478
|
+
kDistCommAllToAllVSingle = 449,
|
|
479
|
+
kInplaceScatterValue = 450,
|
|
480
|
+
kDropoutGradExt = 451,
|
|
481
|
+
kGatherD = 452,
|
|
482
|
+
kBinaryCrossEntropyGrad = 453,
|
|
483
|
+
kOnesLikeExt = 454,
|
|
484
|
+
kDistCommAllReduce = 455,
|
|
485
|
+
kMoeTokenUnpermuteGrad = 456,
|
|
486
|
+
kNewOnes = 457,
|
|
487
|
+
kOnes = 458,
|
|
488
|
+
kReflectionPad2DGrad = 459,
|
|
489
|
+
kInplaceTanh = 460,
|
|
490
|
+
kNanToNum = 461,
|
|
491
|
+
kSmoothL1Loss = 462,
|
|
492
|
+
kVar = 463,
|
|
493
|
+
kMatMulExt = 464,
|
|
494
|
+
kSoftmax = 465,
|
|
495
|
+
kMSELossExt = 466,
|
|
496
|
+
kEye = 467,
|
|
497
|
+
kNLLLoss = 468,
|
|
498
|
+
kUpsampleBicubic2DGrad = 469,
|
|
499
|
+
kTake = 470,
|
|
500
|
+
kElu = 471,
|
|
501
|
+
kSigmoid = 472,
|
|
502
|
+
kCummax = 473,
|
|
503
|
+
kReciprocal = 474,
|
|
504
|
+
kSubExt = 475,
|
|
505
|
+
kIndexFillTensor = 476,
|
|
506
|
+
kGenerator = 477,
|
|
507
|
+
kNeg = 478,
|
|
508
|
+
kSliceExt = 479,
|
|
509
|
+
kSeluGrad = 480,
|
|
510
|
+
kMaskedSelectGrad = 481,
|
|
511
|
+
kPow = 482,
|
|
512
|
+
kInplaceCopy = 483,
|
|
513
|
+
kSwigluGrad = 484,
|
|
514
|
+
kNeScalar = 485,
|
|
515
|
+
kConv1DPadding = 486,
|
|
516
|
+
kCustomExt = 487,
|
|
517
|
+
kNormalTensorFloat = 488,
|
|
518
|
+
kAdamW = 489,
|
|
519
|
+
kMaskedScatter = 490,
|
|
520
|
+
kMultinomialExt = 491,
|
|
521
|
+
kSplitTensorView = 492,
|
|
522
|
+
kTensorScatterElements = 493,
|
|
523
|
+
kMm = 494,
|
|
524
|
+
kUpsampleTrilinear3D = 495,
|
|
525
|
+
kBernoulliExt = 496,
|
|
526
|
+
kHSigmoid = 497,
|
|
527
|
+
kTypeAs = 498,
|
|
528
|
+
kReluGrad = 499,
|
|
529
|
+
kBatchNormElemtGrad = 500,
|
|
530
|
+
kConv3DPadding = 501,
|
|
531
|
+
kGroupNormGrad = 502,
|
|
532
|
+
kAllGatherMatmul = 503,
|
|
533
|
+
kBitwiseXorTensor = 504,
|
|
534
|
+
kSearchSorted = 505,
|
|
535
|
+
kXLogYScalarSelf = 506,
|
|
536
|
+
kTopkExt = 507,
|
|
537
|
+
kIndexFillScalar = 508,
|
|
538
|
+
kSelect = 509,
|
|
539
|
+
kInplaceScatterSrcReduce = 510,
|
|
540
|
+
kTile = 511,
|
|
541
|
+
kAsinhExt = 512,
|
|
542
|
+
kInplaceBernoulliScalar = 513,
|
|
543
|
+
kGeLUGrad = 514,
|
|
544
|
+
kGroupedMatmulV2 = 515,
|
|
545
|
+
kGroupedMatmul = 516,
|
|
546
|
+
kQuantV2 = 517,
|
|
547
|
+
kGroupedMatmulV4 = 518,
|
|
548
|
+
kDynamicQuantExt = 519,
|
|
549
|
+
kMoeFinalizeRouting = 520,
|
|
550
|
+
kAddRmsNormQuantV2 = 521,
|
|
551
|
+
kQuantBatchMatmul = 522,
|
|
552
|
+
kKVCacheScatterUpdate = 523,
|
|
553
|
+
kMatmulAllReduceAddRmsNorm = 524,
|
|
554
|
+
kQuantMatmul = 525,
|
|
555
|
+
kMoeGatingTopKSoftmax = 526,
|
|
556
|
+
kWeightQuantBatchMatmul = 527,
|
|
557
|
+
kMoeInitRoutingV2 = 528,
|
|
558
|
+
kMoeComputeExpertTokens = 529,
|
|
559
|
+
kFusedInferAttentionScore = 530,
|
|
560
|
+
kMoeInitRoutingQuantV2 = 531,
|
|
561
|
+
kMoeInitRouting = 532,
|
|
562
|
+
kGmmV2BackwardFusion = 533,
|
|
563
|
+
kAny = 534,
|
|
564
|
+
kPixelShuffle = 535,
|
|
565
|
+
kGmmV2Backward = 536,
|
|
566
|
+
kGmm = 537,
|
|
567
|
+
kEinsumExt = 538,
|
|
568
|
+
kFuncMaxPool2D = 539,
|
|
569
|
+
kMoeTokenUnpermute = 540,
|
|
570
|
+
kGmmV2 = 541,
|
|
571
|
+
kFuncDropoutExt = 542,
|
|
572
|
+
kInplaceExponential = 543,
|
|
573
|
+
kGmmBackward = 544,
|
|
574
|
+
kGmmBackwardFusion = 545,
|
|
575
|
+
kAnyExt = 546,
|
|
508
576
|
};
|
|
509
577
|
|
|
510
|
-
using
|
|
511
|
-
using
|
|
512
|
-
using
|
|
513
|
-
using
|
|
514
|
-
using
|
|
515
|
-
using
|
|
516
|
-
using
|
|
517
|
-
using
|
|
518
|
-
using
|
|
519
|
-
using
|
|
520
|
-
using
|
|
521
|
-
using
|
|
522
|
-
using
|
|
523
|
-
using InplaceExpGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
524
|
-
using ThresholdGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::FP32ImmPtr &)>;
|
|
525
|
-
using IndexAddExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &)>;
|
|
526
|
-
using LerpGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
527
|
-
using RotaryPositionEmbeddingGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
528
|
-
using DenseGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::tensor::BaseTensorPtr> &)>;
|
|
529
|
-
using ArgMaxWithValueGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
530
|
-
using NewZerosGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
531
|
-
using GridSampler3DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
532
|
-
using ReverseV2GradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
533
|
-
using ErfinvGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
534
|
-
using NLLLossGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
535
|
-
using DropoutGenMaskExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &, const mindspore::FP32ImmPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
536
|
-
using AddbmmGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &)>;
|
|
537
|
-
using ZerosLikeExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
538
|
-
using InplaceHardtanhGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &)>;
|
|
539
|
-
using DropoutDoMaskExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::FP32ImmPtr &)>;
|
|
540
|
-
using UpsampleBicubic2DGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::BoolImmPtr &)>;
|
|
541
|
-
using LogicalAndGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
542
|
-
using EmbeddingGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::Int64ImmPtr> &, const std::optional<mindspore::FP32ImmPtr> &, const mindspore::FP32ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
543
|
-
using ConvTranspose2DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &, const mindspore::ValueTuplePtr &)>;
|
|
544
|
-
using HShrinkGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::FP32ImmPtr &)>;
|
|
545
|
-
using HSwishGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
546
|
-
using CumminExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
547
|
-
using MeshgridGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &)>;
|
|
548
|
-
using MaskedSelectGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
549
|
-
using ThresholdGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::FP32ImmPtr &, const mindspore::FP32ImmPtr &)>;
|
|
550
|
-
using ReduceAnyGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &, const mindspore::BoolImmPtr &)>;
|
|
551
|
-
using XlogyGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
552
|
-
using BinaryCrossEntropyGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const mindspore::Int64ImmPtr &)>;
|
|
553
|
-
using ReflectionPad1DGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
554
|
-
using InplaceGroupedMatmulAddGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
555
|
-
using FlashAttentionScoreGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::Int64ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
556
|
-
using IsInfGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
557
|
-
using RmsNormGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
558
|
-
using SwigluGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
559
|
-
using ReflectionPad1DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
560
|
-
using MmGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
561
|
-
using FracGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
562
|
-
using SilentCheckV2GradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
563
|
-
using LogSoftmaxGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
564
|
-
using RandIntGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::ValueTuplePtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
565
|
-
using ExpandDimsGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
566
|
-
using SqrtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
567
|
-
using HardtanhGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &)>;
|
|
568
|
-
using GcdGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
569
|
-
using ClampTensorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &)>;
|
|
570
|
-
using SoftShrinkGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::FP32ImmPtr &)>;
|
|
571
|
-
using ReplicationPad2DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
572
|
-
using SoftShrinkGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::FP32ImmPtr &)>;
|
|
573
|
-
using LayerNormGradExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
574
|
-
using MaxPoolWithMaskGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::BoolImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
575
|
-
using MulsGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &)>;
|
|
576
|
-
using Col2ImGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &)>;
|
|
577
|
-
using UniformExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
578
|
-
using LogSoftmaxGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
579
|
-
using MatMulExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
580
|
-
using SincGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
581
|
-
using ConstantPadNDGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &, const mindspore::ScalarPtr &)>;
|
|
582
|
-
using UpsampleTrilinear3DGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::BoolImmPtr &)>;
|
|
583
|
-
using BCEWithLogitsLossGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const mindspore::Int64ImmPtr &)>;
|
|
584
|
-
using TrilExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
585
|
-
using RsqrtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
586
|
-
using GridSampler3DGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &, const mindspore::ValueTuplePtr &)>;
|
|
587
|
-
using MaxUnpool2DExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &)>;
|
|
588
|
-
using HardtanhGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &)>;
|
|
589
|
-
using IsFiniteGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
590
|
-
using UpsampleBicubic2DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::BoolImmPtr &)>;
|
|
591
|
-
using SiLUGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
592
|
-
using SelectV2GradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
593
|
-
using EluExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::FP32ImmPtr &)>;
|
|
594
|
-
using GeLUGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
595
|
-
using Conv3DPaddingGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &)>;
|
|
596
|
-
using ReciprocalGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
597
|
-
using MvGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
598
|
-
using AddScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &)>;
|
|
599
|
-
using TileGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
600
|
-
using AllGatherMatmulGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::StringImmPtr &, const mindspore::Int64ImmPtr &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
601
|
-
using SubExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &)>;
|
|
602
|
-
using VarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
603
|
-
using InplaceDivModsGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
604
|
-
using InplaceFillScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &)>;
|
|
605
|
-
using CumsumExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
606
|
-
using BatchNormGatherStatsWithCountsGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const mindspore::FP32ImmPtr &, const mindspore::FP32ImmPtr &, const std::optional<mindspore::tensor::BaseTensorPtr> &)>;
|
|
607
|
-
using CustomExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &)>;
|
|
608
|
-
using TanhGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
609
|
-
using InplacePutGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::BoolImmPtr &)>;
|
|
610
|
-
using BatchNormElemtGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
611
|
-
using ChunkGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
612
|
-
using UpsampleLinear1DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::BoolImmPtr &)>;
|
|
613
|
-
using ArgMinExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::Int64ImmPtr> &, const mindspore::BoolImmPtr &)>;
|
|
614
|
-
using IsNegInfGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
615
|
-
using IndexGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
616
|
-
using NormalTensorFloatGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
617
|
-
using ReplicationPad1DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
618
|
-
using RepeatInterleaveTensorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::Int64ImmPtr> &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
619
|
-
using NanToNumGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::FP32ImmPtr> &, const std::optional<mindspore::FP32ImmPtr> &, const std::optional<mindspore::FP32ImmPtr> &)>;
|
|
620
|
-
using BatchMatMulExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
621
|
-
using InplaceFillTensorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
622
|
-
using GLUGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
623
|
-
using LogSoftmaxExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::Int64ImmPtr> &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
624
|
-
using FlattenExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
625
|
-
using ReplicationPad3DGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
626
|
-
using FmodTensorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
627
|
-
using TransposeExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
628
|
-
using MishExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
629
|
-
using UpsampleNearest2DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &)>;
|
|
630
|
-
using ReplicationPad2DGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
631
|
-
using ReflectionPad2DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
632
|
-
using GeluExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
633
|
-
using LogicalXorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
634
|
-
using AvgPool1DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::ValueTuplePtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
635
|
-
using SmoothL1LossGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::FP32ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
636
|
-
using RandnGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
637
|
-
using CloneGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
638
|
-
using KthvalueGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
639
|
-
using InplaceEluGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::FP32ImmPtr &)>;
|
|
640
|
-
using PReLUGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
641
|
-
using LayerNormExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const mindspore::FP32ImmPtr &)>;
|
|
642
|
-
using ReplicationPad1DGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
643
|
-
using BinaryCrossEntropyGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const mindspore::Int64ImmPtr &)>;
|
|
644
|
-
using InplaceAddmmGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &)>;
|
|
645
|
-
using MoeTokenUnpermuteGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const mindspore::BoolImmPtr &, const std::optional<mindspore::ValueTuplePtr> &)>;
|
|
646
|
-
using BinaryCrossEntropyWithLogitsBackwardGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const mindspore::Int64ImmPtr &)>;
|
|
647
|
-
using BatchNormStatsGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::FP32ImmPtr &)>;
|
|
648
|
-
using MaskedFillGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
649
|
-
using Log10GradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
650
|
-
using Atan2ExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
651
|
-
using IndexFillScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &)>;
|
|
652
|
-
using SumExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::BoolImmPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
653
|
-
using FloorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
654
|
-
using IndexFillTensorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
655
|
-
using OuterGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
656
|
-
using InplaceMaskedFillTensorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
657
|
-
using TransposeGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
658
|
-
using MatrixInverseExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
659
|
-
using ConvolutionGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::BoolImmPtr &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &)>;
|
|
660
|
-
using RandpermExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::Int64ImmPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
661
|
-
using UniqueDimGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
662
|
-
using InplaceScatterValueReduceGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &, const mindspore::Int64ImmPtr &)>;
|
|
663
|
-
using DivModsGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
664
|
-
using PowGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
665
|
-
using Unique2GradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
666
|
-
using AvgPool3DExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::ValueTuplePtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
667
|
-
using BitwiseNotGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
668
|
-
using AdaptiveAvgPool3DExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
669
|
-
using LeakyReLUGradExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &, const mindspore::BoolImmPtr &)>;
|
|
670
|
-
using NotEqualGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
671
|
-
using MinGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
672
|
-
using RandExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
673
|
-
using AllFiniteGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &)>;
|
|
674
|
-
using InplaceIndexAddExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &)>;
|
|
675
|
-
using InplaceMulGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
676
|
-
using InplaceSubScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &)>;
|
|
677
|
-
using ExpandAsGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
678
|
-
using AdaptiveAvgPool2DGradExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
679
|
-
using FmodScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &)>;
|
|
680
|
-
using InplaceFloorDividesGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &)>;
|
|
681
|
-
using NansumGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::BoolImmPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
682
|
-
using OneHotExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
683
|
-
using MedianExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
684
|
-
using AsinExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
685
|
-
using AddmmGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &)>;
|
|
686
|
-
using UpsampleLinear1DGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::BoolImmPtr &)>;
|
|
687
|
-
using GreaterGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
688
|
-
using SeLUExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
689
|
-
using BaddbmmGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &)>;
|
|
690
|
-
using InplaceFloorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
691
|
-
using BatchNormExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const mindspore::BoolImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::FP32ImmPtr &)>;
|
|
692
|
-
using GeneratorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::Int64ImmPtr &, const mindspore::ValueTuplePtr &)>;
|
|
693
|
-
using NewOnesGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
694
|
-
using InplaceMaskedFillScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &)>;
|
|
695
|
-
using NLLLossGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
696
|
-
using SilentCheckV3GradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::FP32ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
697
|
-
using SoftplusGradExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &)>;
|
|
698
|
-
using ScatterValueGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &, const mindspore::Int64ImmPtr &)>;
|
|
699
|
-
using TopkExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
700
|
-
using MaximumGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
701
|
-
using BincountExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const mindspore::Int64ImmPtr &)>;
|
|
702
|
-
using GmmV2BackwardGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const mindspore::Int64ImmPtr &)>;
|
|
703
|
-
using MaxPoolGradWithIndicesGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::BoolImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
704
|
-
using MatmulReduceScatterGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::StringImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
705
|
-
using LogGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
706
|
-
using RotaryPositionEmbeddingGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const mindspore::Int64ImmPtr &)>;
|
|
707
|
-
using MeanExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::BoolImmPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
708
|
-
using SubScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &)>;
|
|
709
|
-
using CoshGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
710
|
-
using MaxDimGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
711
|
-
using RemainderTensorTensorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
712
|
-
using BitwiseAndScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &)>;
|
|
713
|
-
using TakeGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
714
|
-
using NormGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::FP32ImmPtr &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::BoolImmPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
715
|
-
using SoftMarginLossGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
716
|
-
using InplaceAddExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &)>;
|
|
717
|
-
using InplaceClampScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::ScalarPtr> &, const std::optional<mindspore::ScalarPtr> &)>;
|
|
718
|
-
using InplaceReLUGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
719
|
-
using NLLLoss2dGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
720
|
-
using MinDimGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
721
|
-
using AdaptiveAvgPool3DGradExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
722
|
-
using AsStridedGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &)>;
|
|
723
|
-
using BitwiseAndTensorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
724
|
-
using MaxPoolWithIndicesGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::BoolImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
725
|
-
using InplaceUniformGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
726
|
-
using UpsampleNearest3DGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &)>;
|
|
727
|
-
using SignGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
728
|
-
using Conv1DPaddingGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &)>;
|
|
729
|
-
using TanGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
730
|
-
using TExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
731
|
-
using BatchNormGradExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const mindspore::BoolImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::ValueTuplePtr &)>;
|
|
732
|
-
using IsCloseGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::FP32ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
733
|
-
using BatchNormReduceGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
734
|
-
using IncreFlashAttentionGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
735
|
-
using ReshapeGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
736
|
-
using CeilGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
578
|
+
using InplaceEluGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::FP32ImmPtr &)>;
|
|
579
|
+
using CumminExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
580
|
+
using InnerCommIsendGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::StringImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
581
|
+
using MultiScaleDeformableAttnGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
582
|
+
using ReduceAnyGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const mindspore::BoolImmPtr &)>;
|
|
583
|
+
using InplaceFloorDivideGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
584
|
+
using MedianDimGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
585
|
+
using DivModsGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
586
|
+
using NLLLoss2dGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
587
|
+
using ExpandDimsViewGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const int64_t &)>;
|
|
588
|
+
using NormalFloatTensorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ScalarPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
589
|
+
using SeLUExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
590
|
+
using EluExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::FP32ImmPtr &)>;
|
|
737
591
|
using ZerosGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
738
|
-
using
|
|
739
|
-
using
|
|
740
|
-
using
|
|
741
|
-
using
|
|
742
|
-
using
|
|
743
|
-
using
|
|
744
|
-
using
|
|
745
|
-
using
|
|
746
|
-
using
|
|
747
|
-
using
|
|
748
|
-
using
|
|
749
|
-
using
|
|
750
|
-
using
|
|
751
|
-
using
|
|
752
|
-
using
|
|
753
|
-
using
|
|
754
|
-
using
|
|
755
|
-
using
|
|
756
|
-
using
|
|
757
|
-
using
|
|
758
|
-
using
|
|
759
|
-
using
|
|
760
|
-
using
|
|
761
|
-
using
|
|
762
|
-
using
|
|
763
|
-
using
|
|
764
|
-
using
|
|
765
|
-
using
|
|
766
|
-
using
|
|
767
|
-
using
|
|
768
|
-
using
|
|
769
|
-
using
|
|
770
|
-
using
|
|
771
|
-
using
|
|
772
|
-
using
|
|
773
|
-
using
|
|
774
|
-
using
|
|
775
|
-
using InplaceErfinvGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
776
|
-
using ErfcGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
777
|
-
using InplaceStopGradientGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
778
|
-
using GridSampler2DGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &, const mindspore::ValueTuplePtr &)>;
|
|
779
|
-
using ReduceMaxGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &, const mindspore::BoolImmPtr &)>;
|
|
780
|
-
using SliceGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &)>;
|
|
781
|
-
using SinGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
782
|
-
using ArgMaxExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::Int64ImmPtr> &, const mindspore::BoolImmPtr &)>;
|
|
783
|
-
using SoftmaxGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
784
|
-
using TensorScatterElementsGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
785
|
-
using HShrinkGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::FP32ImmPtr &)>;
|
|
786
|
-
using Log1pGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
787
|
-
using LinalgVectorNormGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::FP32ImmPtr &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::BoolImmPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
788
|
-
using FillScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &, const mindspore::ScalarPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
789
|
-
using SplitGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
790
|
-
using SplitWithSizeGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &)>;
|
|
791
|
-
using SmoothL1LossGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::FP32ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
792
|
-
using XLogYScalarSelfGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ScalarPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
793
|
-
using ExpGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
794
|
-
using Col2ImExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &)>;
|
|
795
|
-
using SoftMarginLossGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
796
|
-
using AdaptiveAvgPool1DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
797
|
-
using RmsNormGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::FP32ImmPtr &)>;
|
|
798
|
-
using MultiScaleDeformableAttnGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
799
|
-
using ContiguousGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
800
|
-
using InplaceMulsGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &)>;
|
|
801
|
-
using ScatterAddExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
802
|
-
using AsinhExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
803
|
-
using InplaceIndexPutGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::BoolImmPtr &)>;
|
|
804
|
-
using PromptFlashAttentionGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const mindspore::Int64ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
805
|
-
using GeLUGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
806
|
-
using HSigmoidGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
807
|
-
using SliceExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
808
|
-
using LogSumExpGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &, const mindspore::BoolImmPtr &)>;
|
|
809
|
-
using RepeatInterleaveIntGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const std::optional<mindspore::Int64ImmPtr> &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
810
|
-
using InplaceScatterSrcGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
811
|
-
using GreaterEqualScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &)>;
|
|
812
|
-
using PowTensorScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &)>;
|
|
813
|
-
using RepeatInterleaveGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
814
|
-
using OnesGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
815
|
-
using UpsampleBilinear2DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::BoolImmPtr &)>;
|
|
816
|
-
using InplaceCopyGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
817
|
-
using AddLayerNormV2GradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::FP32ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
818
|
-
using AddcmulExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &)>;
|
|
819
|
-
using CopyGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
820
|
-
using SinhGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
821
|
-
using NormalTensorTensorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
822
|
-
using SigmoidGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
823
|
-
using Conv1DExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &)>;
|
|
824
|
-
using InplaceRandomGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const std::optional<mindspore::Int64ImmPtr> &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
825
|
-
using GluGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
826
|
-
using NarrowGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
827
|
-
using GridSampler2DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
828
|
-
using BitwiseOrScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &)>;
|
|
829
|
-
using ReflectionPad2DGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
830
|
-
using FloorDivScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &)>;
|
|
831
|
-
using PolarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
832
|
-
using UpsampleNearest1DGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &)>;
|
|
833
|
-
using LogAddExp2GradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
834
|
-
using FlashAttentionScoreGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::Int64ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
835
|
-
using NormalFloatFloatGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &, const mindspore::ValueTuplePtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
836
|
-
using UnstackExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
837
|
-
using AtanExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
838
|
-
using InplaceDivModGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
839
|
-
using InplaceScatterSrcReduceGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
840
|
-
using SortExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
841
|
-
using HistcExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &)>;
|
|
842
|
-
using InplaceDivsGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &)>;
|
|
843
|
-
using ConcatGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &)>;
|
|
844
|
-
using InplaceAddsExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &)>;
|
|
845
|
-
using HSigmoidGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
846
|
-
using PowScalarTensorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ScalarPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
847
|
-
using MedianDimGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
848
|
-
using ConvolutionStrGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &, const mindspore::ValueTuplePtr &, const mindspore::BoolImmPtr &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &, const mindspore::ValueTuplePtr &)>;
|
|
849
|
-
using AddGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
850
|
-
using TanhGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
851
|
-
using FullLikeGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
852
|
-
using NeScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &)>;
|
|
853
|
-
using ArgSortGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
854
|
-
using SeluGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
855
|
-
using Conv2DPaddingGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::tensor::BaseTensorPtr> &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &)>;
|
|
856
|
-
using HSwishGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
857
|
-
using InplaceFloorDivideGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
858
|
-
using InplaceScatterAddGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
859
|
-
using MSELossGradExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
860
|
-
using MinimumGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
861
|
-
using Exp2GradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
862
|
-
using AddcdivExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &)>;
|
|
863
|
-
using CountNonZeroGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::ValueTuplePtr> &)>;
|
|
592
|
+
using Expm1GradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
593
|
+
using GroupNormGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::FP32ImmPtr &)>;
|
|
594
|
+
using DistCommBroadcastGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::StringImmPtr &)>;
|
|
595
|
+
using CopyGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
596
|
+
using UpsampleNearest3DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &)>;
|
|
597
|
+
using InplaceAddsExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &)>;
|
|
598
|
+
using NonZeroGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
599
|
+
using InplaceDivGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
600
|
+
using ExpandAsGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
601
|
+
using BatchNormReduceGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
602
|
+
using HardtanhGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &)>;
|
|
603
|
+
using PReLUGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
604
|
+
using CountNonZeroGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::ValueTuplePtr> &)>;
|
|
605
|
+
using InplaceMulGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
606
|
+
using InnerCommIrecvGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::ValueTuplePtr &, const mindspore::StringImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
607
|
+
using MlaGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::Int64ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
608
|
+
using RemainderScalarTensorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ScalarPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
609
|
+
using RepeatGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
610
|
+
using SoftmaxBackwardGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
611
|
+
using FloorDivScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &)>;
|
|
612
|
+
using InplaceZeroGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
613
|
+
using NewZerosGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
614
|
+
using GreaterEqualGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
615
|
+
using GreaterEqualScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &)>;
|
|
616
|
+
using SmoothL1LossGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::FP32ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
617
|
+
using HSwishGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
618
|
+
using RsqrtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
619
|
+
using LeakyReLUGradExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &, const mindspore::BoolImmPtr &)>;
|
|
620
|
+
using InplaceDivModGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
621
|
+
using IndexSelectGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
622
|
+
using NonZeroExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
623
|
+
using StdGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
624
|
+
using RoundGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
625
|
+
using BatchNormExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::BoolImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::FP32ImmPtr &)>;
|
|
626
|
+
using InplaceAddmmGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &)>;
|
|
627
|
+
using Im2ColExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &)>;
|
|
628
|
+
using FloorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
864
629
|
using StackExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &)>;
|
|
865
|
-
using
|
|
866
|
-
using
|
|
867
|
-
using
|
|
868
|
-
using
|
|
869
|
-
using
|
|
870
|
-
using
|
|
871
|
-
using
|
|
872
|
-
using
|
|
873
|
-
using
|
|
874
|
-
using
|
|
875
|
-
using
|
|
876
|
-
using
|
|
877
|
-
using
|
|
878
|
-
using
|
|
879
|
-
using
|
|
880
|
-
using
|
|
881
|
-
using
|
|
882
|
-
using
|
|
883
|
-
using
|
|
884
|
-
using
|
|
885
|
-
using
|
|
886
|
-
using
|
|
887
|
-
using
|
|
888
|
-
using
|
|
889
|
-
using
|
|
890
|
-
using
|
|
891
|
-
using
|
|
892
|
-
using
|
|
630
|
+
using InnerInplaceIndexPutGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const mindspore::tensor::TensorPtr &, const mindspore::BoolImmPtr &)>;
|
|
631
|
+
using InplaceSubScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &)>;
|
|
632
|
+
using GridSampler3DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
633
|
+
using MaxUnpool2DExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &)>;
|
|
634
|
+
using AbsGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
635
|
+
using SoftplusGradExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &)>;
|
|
636
|
+
using AvgPool1DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::ValueTuplePtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
637
|
+
using Log2GradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
638
|
+
using NormalTensorTensorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
639
|
+
using ReshapeAndCacheGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &)>;
|
|
640
|
+
using RandnGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
641
|
+
using UpsampleBilinear2DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::BoolImmPtr &)>;
|
|
642
|
+
using XlogyGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
643
|
+
using InnerCommAllGatherGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::StringImmPtr &)>;
|
|
644
|
+
using MaxGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
645
|
+
using MatmulReduceScatterGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::StringImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
646
|
+
using AdaptiveAvgPool2DExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
647
|
+
using LogSoftmaxExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::Int64ImmPtr> &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
648
|
+
using InplaceGroupedMatmulAddGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
649
|
+
using DotGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
650
|
+
using SortExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
651
|
+
using RandIntGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::ValueTuplePtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
652
|
+
using ReflectionPad2DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
653
|
+
using ThresholdGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &)>;
|
|
654
|
+
using SoftMarginLossGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
655
|
+
using LogSoftmaxGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
656
|
+
using NotEqualGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
657
|
+
using FloorDivGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
658
|
+
using MinGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
659
|
+
using PagedAttentionGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::Int64ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
660
|
+
using EmbeddingDenseBackwardGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const std::optional<mindspore::Int64ImmPtr> &, const mindspore::BoolImmPtr &)>;
|
|
661
|
+
using ScatterAddExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
662
|
+
using Col2ImExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &)>;
|
|
663
|
+
using Log1pGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
664
|
+
using ConvolutionGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::BoolImmPtr &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &)>;
|
|
665
|
+
using BatchNormGatherStatsWithCountsGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::FP32ImmPtr &, const mindspore::FP32ImmPtr &, const std::optional<mindspore::tensor::TensorPtr> &)>;
|
|
666
|
+
using ApplyRotaryPosEmbGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
667
|
+
using RmsNormGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
668
|
+
using SilentCheckV2GradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
669
|
+
using ReplicationPad3DGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
670
|
+
using InplaceFloorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
671
|
+
using InnerNonZeroGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
672
|
+
using IsInfGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
673
|
+
using RepeatInterleaveIntGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const std::optional<mindspore::Int64ImmPtr> &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
674
|
+
using Atan2ExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
675
|
+
using InplaceFillTensorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
676
|
+
using ScatterValueGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &, const mindspore::Int64ImmPtr &)>;
|
|
677
|
+
using SplitWithSizeViewGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::vector<int64_t> &, const int64_t &)>;
|
|
678
|
+
using DistCommBarrierGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::StringImmPtr &)>;
|
|
679
|
+
using InplaceIndexPutGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const mindspore::tensor::TensorPtr &, const mindspore::BoolImmPtr &)>;
|
|
680
|
+
using ViewAsGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
681
|
+
using CellBackwardHookGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &)>;
|
|
682
|
+
using HardtanhGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &)>;
|
|
683
|
+
using DistCommGatherGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::StringImmPtr &)>;
|
|
684
|
+
using LerpGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
685
|
+
using TriuGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
686
|
+
using LogAddExpGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
687
|
+
using UpsampleNearest3DGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &)>;
|
|
688
|
+
using FlashAttentionScoreGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::Int64ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
689
|
+
using PowTensorScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &)>;
|
|
690
|
+
using CoshGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
691
|
+
using TanhGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
692
|
+
using Conv1DExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &)>;
|
|
693
|
+
using TrilExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
694
|
+
using BatchNormStatsGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::FP32ImmPtr &)>;
|
|
695
|
+
using BitwiseAndTensorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
696
|
+
using DistCommReduceScatterTensorUnevenGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &, const mindspore::StringImmPtr &, const mindspore::StringImmPtr &)>;
|
|
697
|
+
using ReplicationPad2DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
698
|
+
using IdentityGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
699
|
+
using MaxPoolWithIndicesGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::BoolImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
700
|
+
using SubGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
701
|
+
using ExpGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
702
|
+
using FillScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &, const mindspore::ScalarPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
703
|
+
using AddcmulExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &)>;
|
|
704
|
+
using CloneGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
705
|
+
using AtanExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
706
|
+
using InnerCommAllToAllVGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::StringImmPtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
707
|
+
using AdaptiveMaxPool2DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
708
|
+
using AddmmGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &)>;
|
|
709
|
+
using TransposeViewGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::vector<int64_t> &)>;
|
|
710
|
+
using NormGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::FP32ImmPtr &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::BoolImmPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
711
|
+
using DivsGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &)>;
|
|
712
|
+
using InplaceStopGradientGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
713
|
+
using Col2ImGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &)>;
|
|
714
|
+
using RotaryPositionEmbeddingGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::Int64ImmPtr &)>;
|
|
715
|
+
using Conv3DExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &)>;
|
|
716
|
+
using InplaceScatterAddGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
717
|
+
using InplaceIndexAddExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &)>;
|
|
718
|
+
using SplitTensorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const int64_t &, const int64_t &)>;
|
|
719
|
+
using MaxPoolGradWithIndicesGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::BoolImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
720
|
+
using EmbeddingGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::Int64ImmPtr> &, const std::optional<mindspore::FP32ImmPtr> &, const mindspore::FP32ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
721
|
+
using DistCommReduceScatterGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &, const mindspore::StringImmPtr &, const mindspore::StringImmPtr &)>;
|
|
722
|
+
using NewEmptyGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::Int64ImmPtr> &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
723
|
+
using SelectExtViewGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const int64_t &, const int64_t &)>;
|
|
724
|
+
using IncreFlashAttentionGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
725
|
+
using ReverseV2GradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
726
|
+
using LeakyReLUExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &)>;
|
|
727
|
+
using RandnLikeGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
728
|
+
using GeLUGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
729
|
+
using Exp2GradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
730
|
+
using ReshapeGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::vector<int64_t> &)>;
|
|
731
|
+
using ThresholdGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &)>;
|
|
732
|
+
using UpsampleLinear1DGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::BoolImmPtr &)>;
|
|
733
|
+
using LogicalAndGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
734
|
+
using InnerMoeTokenUnpermuteGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::BoolImmPtr &, const std::optional<mindspore::ValueTuplePtr> &)>;
|
|
735
|
+
using AddbmmGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &)>;
|
|
736
|
+
using XLogYScalarOtherGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &)>;
|
|
737
|
+
using AcoshExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
738
|
+
using ViewGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::vector<int64_t> &)>;
|
|
739
|
+
using ReduceMinGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const mindspore::BoolImmPtr &)>;
|
|
740
|
+
using SpeedFusionAttentionGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::FP32ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &, const mindspore::Int64ImmPtr &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &)>;
|
|
741
|
+
using RemainderTensorScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &)>;
|
|
742
|
+
using SqrtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
743
|
+
using TanGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
744
|
+
using SilentCheckV3GradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::FP32ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
745
|
+
using AddmvGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &)>;
|
|
746
|
+
using InplaceReLUGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
747
|
+
using RandIntLikeGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
748
|
+
using GcdGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
749
|
+
using SplitGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const int64_t &, const int64_t &)>;
|
|
750
|
+
using MulGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
751
|
+
using EluGradExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::FP32ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
752
|
+
using RandpermExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::Int64ImmPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
753
|
+
using DiagonalViewGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const int64_t &, const int64_t &, const int64_t &)>;
|
|
754
|
+
using TruncGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
755
|
+
using InplaceLogGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
756
|
+
using AddcdivExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &)>;
|
|
757
|
+
using TExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
758
|
+
using UpsampleTrilinear3DGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::BoolImmPtr &)>;
|
|
759
|
+
using BitwiseOrScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &)>;
|
|
760
|
+
using BitwiseOrTensorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
761
|
+
using ReduceAllGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::BoolImmPtr &)>;
|
|
762
|
+
using BroadcastToGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::vector<int64_t> &)>;
|
|
763
|
+
using AdaptiveAvgPool3DGradExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
764
|
+
using RepeatInterleaveGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
765
|
+
using GluGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
766
|
+
using CosGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
767
|
+
using GridSampler2DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
768
|
+
using ConstantPadNDGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const mindspore::ScalarPtr &)>;
|
|
769
|
+
using MaskedSelectGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
770
|
+
using UpsampleBicubic2DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::BoolImmPtr &)>;
|
|
771
|
+
using MaskedFillGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
772
|
+
using EmptyLikeGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::Int64ImmPtr> &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
773
|
+
using SiLUGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
774
|
+
using NansumGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::BoolImmPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
775
|
+
using SplitWithSizeGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::vector<int64_t> &, const int64_t &)>;
|
|
776
|
+
using DistCommAllGatherIntoTensorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::StringImmPtr &)>;
|
|
777
|
+
using L1LossBackwardExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
778
|
+
using AddRmsNormGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::FP32ImmPtr &)>;
|
|
779
|
+
using InplaceSiLUGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
780
|
+
using ReLUGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
781
|
+
using StdMeanGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
782
|
+
using AvgPool2DGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
783
|
+
using SoftMarginLossGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
784
|
+
using LessEqualGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
785
|
+
using ConvTranspose2DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &, const mindspore::ValueTuplePtr &)>;
|
|
786
|
+
using MeanExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::BoolImmPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
787
|
+
using MinimumGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
788
|
+
using BatchMatMulGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
789
|
+
using SiLUGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
790
|
+
using FmodScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &)>;
|
|
791
|
+
using GatherDGradV2GradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
792
|
+
using EqualGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
793
|
+
using MoeDistributeDispatchGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::StringImmPtr> &, const std::optional<mindspore::StringImmPtr> &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
794
|
+
using InplaceFloorDividesGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &)>;
|
|
795
|
+
using CumsumExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
796
|
+
using LayerNormGradExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
797
|
+
using LogSumExpGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const mindspore::BoolImmPtr &)>;
|
|
798
|
+
using BinaryCrossEntropyGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::Int64ImmPtr &)>;
|
|
799
|
+
using ChunkViewGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const int64_t &, const int64_t &)>;
|
|
800
|
+
using InplaceExpGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
801
|
+
using InplaceFillScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &)>;
|
|
802
|
+
using AddScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &)>;
|
|
803
|
+
using SoftplusExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &)>;
|
|
804
|
+
using HSwishGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
805
|
+
using DistCommGatherIntoTensorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::StringImmPtr &)>;
|
|
806
|
+
using DiagExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
807
|
+
using LogAddExp2GradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
808
|
+
using NormalFloatFloatGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &, const mindspore::ValueTuplePtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
809
|
+
using ReflectionPad3DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
810
|
+
using SinhGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
811
|
+
using ArgMaxExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::Int64ImmPtr> &, const mindspore::BoolImmPtr &)>;
|
|
812
|
+
using AsStridedGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::vector<int64_t> &, const std::vector<int64_t> &, const int64_t &)>;
|
|
813
|
+
using LinalgQrGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
814
|
+
using AdaptiveAvgPool1DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
815
|
+
using NarrowGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const int64_t &, const int64_t &, const int64_t &)>;
|
|
816
|
+
using MinDimGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
817
|
+
using InplaceFillDiagonalGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &, const mindspore::BoolImmPtr &)>;
|
|
818
|
+
using LogSigmoidGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
819
|
+
using MishGradExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
820
|
+
using LessGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
821
|
+
using CrossGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
822
|
+
using LogSoftmaxGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
823
|
+
using InplaceMatmulAddGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
824
|
+
using ScatterGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
825
|
+
using AddExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &)>;
|
|
826
|
+
using ReplicationPad3DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
827
|
+
using HShrinkGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::FP32ImmPtr &)>;
|
|
828
|
+
using AvgPool2DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
829
|
+
using GreaterGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
830
|
+
using EmptyGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::Int64ImmPtr> &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
831
|
+
using FlattenExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
832
|
+
using MaxPoolGradWithMaskGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::BoolImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
833
|
+
using L1LossExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
834
|
+
using GeluExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
835
|
+
using SqueezeGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::vector<int64_t> &)>;
|
|
836
|
+
using DistCommIrecvGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::StringImmPtr &)>;
|
|
837
|
+
using LerpScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::FP32ImmPtr &)>;
|
|
838
|
+
using InnerCommAllReduceGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::StringImmPtr &, const mindspore::StringImmPtr &)>;
|
|
839
|
+
using SquareGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
840
|
+
using Conv2DPaddingGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &)>;
|
|
841
|
+
using ArgMinWithValueGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
842
|
+
using SpeedFusionAttentionGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::FP32ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &, const mindspore::Int64ImmPtr &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &)>;
|
|
843
|
+
using DropoutGenMaskExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &, const mindspore::FP32ImmPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
844
|
+
using BinaryCrossEntropyWithLogitsBackwardGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::Int64ImmPtr &)>;
|
|
845
|
+
using ConvolutionStrGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &, const mindspore::ValueTuplePtr &, const mindspore::BoolImmPtr &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &, const mindspore::ValueTuplePtr &)>;
|
|
846
|
+
using LogSigmoidGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
847
|
+
using LinalgVectorNormGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::FP32ImmPtr &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::BoolImmPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
848
|
+
using BroadcastToViewGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::vector<int64_t> &)>;
|
|
849
|
+
using SoftShrinkGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::FP32ImmPtr &)>;
|
|
850
|
+
using AddLayerNormGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
851
|
+
using UnstackExtViewGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const int64_t &)>;
|
|
852
|
+
using DistCommBatchIsendIrecvGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &, const mindspore::StringImmPtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &)>;
|
|
853
|
+
using GridSampler2DGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &, const mindspore::ValueTuplePtr &)>;
|
|
854
|
+
using RotaryPositionEmbeddingGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
855
|
+
using VarMeanGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
856
|
+
using KthvalueGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
857
|
+
using MvGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
858
|
+
using UniqueDimGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
859
|
+
using InplaceDivModsGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
860
|
+
using FillTensorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
861
|
+
using ExpandDimsGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const int64_t &)>;
|
|
862
|
+
using InplaceThresholdGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &)>;
|
|
863
|
+
using InplaceRemainderTensorTensorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
864
|
+
using ErfinvGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
865
|
+
using TriangularSolveGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
866
|
+
using InplaceScatterSrcGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
867
|
+
using RandExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
868
|
+
using BCEWithLogitsLossGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::Int64ImmPtr &)>;
|
|
869
|
+
using FlashAttentionScoreGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::Int64ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
870
|
+
using MaximumGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
871
|
+
using BaddbmmGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &)>;
|
|
872
|
+
using MoeTokenPermuteGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::Int64ImmPtr> &, const mindspore::BoolImmPtr &)>;
|
|
873
|
+
using InplaceSubExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &)>;
|
|
874
|
+
using InplaceErfinvGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
875
|
+
using MeshgridGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &, const int64_t &)>;
|
|
876
|
+
using MaxPoolWithMaskGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::BoolImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
877
|
+
using SelectV2GradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
893
878
|
using LinSpaceExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &, const mindspore::Int64ImmPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
894
|
-
using
|
|
895
|
-
using
|
|
896
|
-
using
|
|
897
|
-
using
|
|
898
|
-
using
|
|
899
|
-
using
|
|
900
|
-
using
|
|
901
|
-
using
|
|
902
|
-
using
|
|
903
|
-
using
|
|
904
|
-
using
|
|
905
|
-
using
|
|
906
|
-
using
|
|
907
|
-
using
|
|
908
|
-
using
|
|
909
|
-
using
|
|
910
|
-
using
|
|
911
|
-
using
|
|
912
|
-
using
|
|
913
|
-
using
|
|
914
|
-
using
|
|
915
|
-
using
|
|
916
|
-
using
|
|
917
|
-
using
|
|
918
|
-
using
|
|
919
|
-
using
|
|
920
|
-
using
|
|
921
|
-
using
|
|
922
|
-
using
|
|
923
|
-
using
|
|
924
|
-
using
|
|
925
|
-
using
|
|
926
|
-
using
|
|
927
|
-
using
|
|
928
|
-
using
|
|
929
|
-
using
|
|
930
|
-
using DropoutExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::FP32ImmPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
931
|
-
using BitwiseXorScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &)>;
|
|
932
|
-
using LogSigmoidGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
933
|
-
using DiagExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
934
|
-
using RandIntLikeGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
935
|
-
using LerpScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::FP32ImmPtr &)>;
|
|
936
|
-
using GmmBackwardGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &)>;
|
|
937
|
-
using RemainderScalarTensorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ScalarPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
938
|
-
using LogicalOrGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
939
|
-
using AvgPool2DGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
940
|
-
using MaskedSelectGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
941
|
-
using ReLUGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
942
|
-
using AdaptiveMaxPool2DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
943
|
-
using ReplicationPad3DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
944
|
-
using SwigluGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
945
|
-
using InplaceSubExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::ScalarPtr &)>;
|
|
946
|
-
using BitwiseXorTensorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::BaseTensorPtr &, const mindspore::tensor::BaseTensorPtr &)>;
|
|
879
|
+
using RmsNormGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::FP32ImmPtr &)>;
|
|
880
|
+
using SumExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::BoolImmPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
881
|
+
using ZerosLikeExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
882
|
+
using ConcatGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &)>;
|
|
883
|
+
using BitwiseXorScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &)>;
|
|
884
|
+
using FmodTensorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
885
|
+
using DistCommReduceScatterTensorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::StringImmPtr &, const mindspore::StringImmPtr &)>;
|
|
886
|
+
using CrossEntropyLossGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::FP32ImmPtr &)>;
|
|
887
|
+
using UpsampleNearest1DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &)>;
|
|
888
|
+
using IsFiniteGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
889
|
+
using NarrowViewGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const int64_t &, const int64_t &, const int64_t &)>;
|
|
890
|
+
using InplaceMaskedFillTensorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
891
|
+
using TanhGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
892
|
+
using NewFullGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const mindspore::ScalarPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
893
|
+
using MulsGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &)>;
|
|
894
|
+
using DropoutExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::FP32ImmPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
895
|
+
using DistCommScatterGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::StringImmPtr &)>;
|
|
896
|
+
using AsinExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
897
|
+
using RemainderTensorTensorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
898
|
+
using SincGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
899
|
+
using InplaceDivsGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &)>;
|
|
900
|
+
using Conv2DExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &)>;
|
|
901
|
+
using InnerCommReduceScatterGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::StringImmPtr &, const mindspore::StringImmPtr &)>;
|
|
902
|
+
using MishExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
903
|
+
using AcosExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
904
|
+
using TraceExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
905
|
+
using InplacePutGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::BoolImmPtr &)>;
|
|
906
|
+
using GLUGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
907
|
+
using AddGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
908
|
+
using NLLLossGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
909
|
+
using InplaceMaskedFillScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &)>;
|
|
910
|
+
using RandLikeExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
911
|
+
using PolarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
912
|
+
using CastGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
913
|
+
using SwigluGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
914
|
+
using InplaceAddExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &)>;
|
|
947
915
|
using ArangeGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
948
|
-
using
|
|
949
|
-
using
|
|
950
|
-
using
|
|
951
|
-
using
|
|
952
|
-
using
|
|
953
|
-
using
|
|
954
|
-
using
|
|
955
|
-
using
|
|
956
|
-
using
|
|
916
|
+
using MoeDistributeCombineGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::StringImmPtr> &, const std::optional<mindspore::StringImmPtr> &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
917
|
+
using LogGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
918
|
+
using DivModGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
919
|
+
using ArgMaxWithValueGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
920
|
+
using RollGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &)>;
|
|
921
|
+
using ReduceMaxGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const mindspore::BoolImmPtr &)>;
|
|
922
|
+
using LogicalOrGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
923
|
+
using NLLLoss2dGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
924
|
+
using ReflectionPad3DGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
925
|
+
using InplaceClampTensorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &)>;
|
|
926
|
+
using AtanhGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
927
|
+
using ReplicationPad1DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
928
|
+
using BatchNormElemtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::FP32ImmPtr &)>;
|
|
929
|
+
using ErfcGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
930
|
+
using HSigmoidGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
931
|
+
using FracGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
932
|
+
using SigmoidGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
933
|
+
using FullLikeGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
934
|
+
using SoftShrinkGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::FP32ImmPtr &)>;
|
|
935
|
+
using BatchMatMulExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
936
|
+
using RingAttentionUpdateGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::Int64ImmPtr &)>;
|
|
937
|
+
using DistCommAllToAllVGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &, const mindspore::tensor::TensorPtr &, const mindspore::StringImmPtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &)>;
|
|
938
|
+
using PReLUGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
939
|
+
using RepeatInterleaveTensorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::Int64ImmPtr> &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
940
|
+
using AddLayerNormV2GradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::FP32ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
941
|
+
using BitwiseNotGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
942
|
+
using MatrixInverseExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
943
|
+
using AdaptiveAvgPool2DGradExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
944
|
+
using TransposeExtViewGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const int64_t &, const int64_t &)>;
|
|
945
|
+
using LogicalXorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
946
|
+
using DistCommIsendGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::StringImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
947
|
+
using IndexAddExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &)>;
|
|
948
|
+
using ChunkGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const int64_t &, const int64_t &)>;
|
|
949
|
+
using InplaceRemainderTensorScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &)>;
|
|
950
|
+
using PowScalarTensorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ScalarPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
951
|
+
using DistCommScatterTensorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::StringImmPtr &)>;
|
|
952
|
+
using KLDivGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
953
|
+
using DivGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
954
|
+
using MedianExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
955
|
+
using SliceExtViewGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const int64_t &, const int64_t &, const int64_t &, const int64_t &)>;
|
|
956
|
+
using CeilGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
957
|
+
using ReflectionPad1DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
958
|
+
using BatchNormGradExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::BoolImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::ValueTuplePtr &)>;
|
|
959
|
+
using MSELossGradExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
960
|
+
using SignGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
961
|
+
using TransposeGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::vector<int64_t> &)>;
|
|
962
|
+
using HistcExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &)>;
|
|
963
|
+
using InplaceClampScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::ScalarPtr> &, const std::optional<mindspore::ScalarPtr> &)>;
|
|
964
|
+
using GridSampler3DGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &, const mindspore::ValueTuplePtr &)>;
|
|
965
|
+
using InplaceHardtanhGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &)>;
|
|
966
|
+
using IsCloseGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::FP32ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
967
|
+
using AvgPool3DExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::ValueTuplePtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
968
|
+
using UpsampleNearest1DGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &)>;
|
|
969
|
+
using MaxDimGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
970
|
+
using AvgPool3DGradExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::ValueTuplePtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
971
|
+
using SubScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &)>;
|
|
972
|
+
using FFNExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
973
|
+
using DropoutDoMaskExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::FP32ImmPtr &)>;
|
|
974
|
+
using InplaceRandomGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const std::optional<mindspore::Int64ImmPtr> &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
975
|
+
using EqualExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
976
|
+
using MoeTokenPermuteGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
977
|
+
using ReflectionPad1DGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
978
|
+
using DenseGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &)>;
|
|
979
|
+
using InplaceScatterValueReduceGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &, const mindspore::Int64ImmPtr &)>;
|
|
980
|
+
using ClampScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::ScalarPtr> &, const std::optional<mindspore::ScalarPtr> &)>;
|
|
981
|
+
using AllFiniteGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &)>;
|
|
982
|
+
using InnerIndexGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
983
|
+
using ReplicationPad2DGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
984
|
+
using DistCommReduceGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::StringImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::StringImmPtr &)>;
|
|
985
|
+
using Log10GradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
986
|
+
using ProdExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::Int64ImmPtr> &, const mindspore::BoolImmPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
987
|
+
using ArgMinExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::Int64ImmPtr> &, const mindspore::BoolImmPtr &)>;
|
|
988
|
+
using ContiguousGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
989
|
+
using SliceGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::vector<int64_t> &, const std::vector<int64_t> &)>;
|
|
990
|
+
using UniqueConsecutiveGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
991
|
+
using ConvolutionGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::BoolImmPtr &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &, const mindspore::ValueTuplePtr &)>;
|
|
992
|
+
using DistCommAllGatherIntoTensorUnevenGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &, const mindspore::StringImmPtr &)>;
|
|
993
|
+
using OneHotExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
994
|
+
using OuterGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
995
|
+
using MultiScaleDeformableAttnGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
996
|
+
using InplaceMulsGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &)>;
|
|
997
|
+
using CrossEntropyLossGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
998
|
+
using ReplicationPad1DGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
999
|
+
using UpsampleNearest2DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &)>;
|
|
1000
|
+
using MatMulGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
1001
|
+
using InplaceBernoulliTensorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1002
|
+
using Unique2GradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
1003
|
+
using UpsampleBilinear2DGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::BoolImmPtr &)>;
|
|
1004
|
+
using BincountExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::Int64ImmPtr &)>;
|
|
1005
|
+
using ConvolutionStrGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &, const mindspore::ValueTuplePtr &, const mindspore::BoolImmPtr &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &)>;
|
|
1006
|
+
using DistCommAllGatherGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::StringImmPtr &)>;
|
|
1007
|
+
using IsNegInfGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1008
|
+
using GeluGradExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
1009
|
+
using KLDivGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
1010
|
+
using AdaptiveAvgPool3DExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
1011
|
+
using IndexGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
1012
|
+
using SinGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1013
|
+
using BitwiseAndScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &)>;
|
|
1014
|
+
using UpsampleLinear1DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::BoolImmPtr &)>;
|
|
1015
|
+
using UpsampleNearest2DGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &)>;
|
|
1016
|
+
using LogicalNotGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1017
|
+
using InplaceUniformGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1018
|
+
using HShrinkGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::FP32ImmPtr &)>;
|
|
1019
|
+
using PromptFlashAttentionGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::Int64ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
1020
|
+
using ClampTensorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &)>;
|
|
1021
|
+
using InplaceNormalGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1022
|
+
using AdaptiveMaxPool1DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
1023
|
+
using LayerNormExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::FP32ImmPtr &)>;
|
|
1024
|
+
using UniformExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &, const mindspore::ScalarPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1025
|
+
using ErfGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1026
|
+
using ArgSortGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
1027
|
+
using DistCommAllToAllVSingleGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::StringImmPtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
1028
|
+
using InplaceScatterValueGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &)>;
|
|
1029
|
+
using DropoutGradExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::FP32ImmPtr &)>;
|
|
1030
|
+
using GatherDGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1031
|
+
using BinaryCrossEntropyGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::Int64ImmPtr &)>;
|
|
1032
|
+
using OnesLikeExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
1033
|
+
using DistCommAllReduceGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::StringImmPtr &, const mindspore::StringImmPtr &)>;
|
|
1034
|
+
using MoeTokenUnpermuteGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::BoolImmPtr &, const std::optional<mindspore::ValueTuplePtr> &)>;
|
|
1035
|
+
using NewOnesGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
1036
|
+
using OnesGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::Int64ImmPtr> &)>;
|
|
1037
|
+
using ReflectionPad2DGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
1038
|
+
using InplaceTanhGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1039
|
+
using NanToNumGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::FP32ImmPtr> &, const std::optional<mindspore::FP32ImmPtr> &, const std::optional<mindspore::FP32ImmPtr> &)>;
|
|
1040
|
+
using SmoothL1LossGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::FP32ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
1041
|
+
using VarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
1042
|
+
using MatMulExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1043
|
+
using SoftmaxGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
1044
|
+
using MSELossExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
957
1045
|
using EyeGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
958
|
-
using
|
|
959
|
-
using
|
|
960
|
-
using
|
|
961
|
-
using
|
|
962
|
-
using
|
|
963
|
-
using
|
|
964
|
-
using
|
|
965
|
-
using
|
|
966
|
-
using
|
|
967
|
-
using
|
|
968
|
-
using
|
|
969
|
-
using
|
|
970
|
-
using
|
|
971
|
-
using
|
|
1046
|
+
using NLLLossGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
1047
|
+
using UpsampleBicubic2DGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::BoolImmPtr &)>;
|
|
1048
|
+
using TakeGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1049
|
+
using EluGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::FP32ImmPtr &)>;
|
|
1050
|
+
using SigmoidGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1051
|
+
using CummaxGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
1052
|
+
using ReciprocalGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1053
|
+
using SubExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &)>;
|
|
1054
|
+
using IndexFillTensorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1055
|
+
using GeneratorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::Int64ImmPtr &, const mindspore::ValueTuplePtr &)>;
|
|
1056
|
+
using NegGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1057
|
+
using SliceExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const int64_t &, const int64_t &, const int64_t &, const int64_t &)>;
|
|
1058
|
+
using SeluGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1059
|
+
using MaskedSelectGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1060
|
+
using PowGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1061
|
+
using InplaceCopyGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::BoolImmPtr &)>;
|
|
1062
|
+
using SwigluGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
1063
|
+
using NeScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &)>;
|
|
1064
|
+
using Conv1DPaddingGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &)>;
|
|
1065
|
+
using CustomExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &)>;
|
|
1066
|
+
using NormalTensorFloatGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1067
|
+
using AdamWGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::FP32ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
1068
|
+
using MaskedScatterGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1069
|
+
using MultinomialExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1070
|
+
using SplitTensorViewGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const int64_t &, const int64_t &)>;
|
|
1071
|
+
using TensorScatterElementsGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
1072
|
+
using MmGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1073
|
+
using UpsampleTrilinear3DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::BoolImmPtr &)>;
|
|
1074
|
+
using BernoulliExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1075
|
+
using HSigmoidGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1076
|
+
using TypeAsGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1077
|
+
using ReluGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1078
|
+
using BatchNormElemtGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1079
|
+
using Conv3DPaddingGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &, const mindspore::ValueTuplePtr &, const mindspore::Int64ImmPtr &)>;
|
|
1080
|
+
using GroupNormGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
1081
|
+
using AllGatherMatmulGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::StringImmPtr &, const mindspore::Int64ImmPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
1082
|
+
using BitwiseXorTensorGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1083
|
+
using SearchSortedGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
1084
|
+
using XLogYScalarSelfGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ScalarPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1085
|
+
using TopkExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
1086
|
+
using IndexFillScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &)>;
|
|
1087
|
+
using SelectGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1088
|
+
using InplaceScatterSrcReduceGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
1089
|
+
using TileGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &)>;
|
|
1090
|
+
using AsinhExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1091
|
+
using InplaceBernoulliScalarGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::FP32ImmPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1092
|
+
using GeLUGradGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
972
1093
|
using GroupedMatmulV2GradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
973
|
-
using
|
|
974
|
-
using
|
|
975
|
-
using
|
|
976
|
-
using
|
|
977
|
-
using
|
|
978
|
-
using
|
|
979
|
-
using
|
|
980
|
-
using
|
|
981
|
-
using
|
|
982
|
-
using
|
|
983
|
-
using
|
|
984
|
-
using
|
|
985
|
-
using
|
|
986
|
-
using
|
|
987
|
-
using
|
|
988
|
-
using
|
|
1094
|
+
using GroupedMatmulGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
1095
|
+
using QuantV2GradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::BoolImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
1096
|
+
using GroupedMatmulV4GradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
1097
|
+
using DynamicQuantExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &)>;
|
|
1098
|
+
using MoeFinalizeRoutingGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &)>;
|
|
1099
|
+
using AddRmsNormQuantV2GradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::FP32ImmPtr &)>;
|
|
1100
|
+
using QuantBatchMatmulGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
1101
|
+
using KVCacheScatterUpdateGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
1102
|
+
using MatmulAllReduceAddRmsNormGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::FP32ImmPtr &, const mindspore::StringImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
1103
|
+
using QuantMatmulGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::Int64ImmPtr> &, const std::optional<mindspore::Int64ImmPtr> &, const std::optional<mindspore::Int64ImmPtr> &, const std::optional<mindspore::Int64ImmPtr> &, const std::optional<mindspore::Int64ImmPtr> &, const std::optional<mindspore::ValueTuplePtr> &)>;
|
|
1104
|
+
using MoeGatingTopKSoftmaxGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::Int64ImmPtr &)>;
|
|
1105
|
+
using WeightQuantBatchMatmulGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
1106
|
+
using MoeInitRoutingV2GradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
1107
|
+
using MoeComputeExpertTokensGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
1108
|
+
using FusedInferAttentionScoreGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::Int64ImmPtr &, const mindspore::FP32ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
1109
|
+
using MoeInitRoutingQuantV2GradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &, const mindspore::Int64ImmPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const std::optional<mindspore::tensor::TensorPtr> &)>;
|
|
1110
|
+
using MoeInitRoutingGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
1111
|
+
using GmmV2BackwardFusionGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::Int64ImmPtr &)>;
|
|
1112
|
+
using AnyGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1113
|
+
using PixelShuffleGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &)>;
|
|
1114
|
+
using GmmV2BackwardGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::Int64ImmPtr &)>;
|
|
1115
|
+
using GmmGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
1116
|
+
using EinsumExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::StringImmPtr &, const mindspore::ValueTuplePtr &)>;
|
|
1117
|
+
using FuncMaxPool2DGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
1118
|
+
using MoeTokenUnpermuteGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::BoolImmPtr &, const std::optional<mindspore::ValueTuplePtr> &)>;
|
|
1119
|
+
using GmmV2GradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &, const std::optional<mindspore::tensor::TensorPtr> &, const mindspore::Int64ImmPtr &, const mindspore::Int64ImmPtr &)>;
|
|
1120
|
+
using FuncDropoutExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::FP32ImmPtr &, const mindspore::BoolImmPtr &, const mindspore::BoolImmPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1121
|
+
using InplaceExponentialGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::ScalarPtr &, const mindspore::tensor::TensorPtr &, const mindspore::tensor::TensorPtr &)>;
|
|
1122
|
+
using GmmBackwardGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::Int64ImmPtr &)>;
|
|
1123
|
+
using GmmBackwardFusionGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::ValueTuplePtr &, const mindspore::ValueTuplePtr &, const std::optional<mindspore::ValueTuplePtr> &, const mindspore::Int64ImmPtr &)>;
|
|
1124
|
+
using AnyExtGradFunc = std::function<void(const kernel::pyboost::OpPtr &, const mindspore::tensor::TensorPtr &, const mindspore::Int64ImmPtr &, const mindspore::BoolImmPtr &)>;
|
|
989
1125
|
|
|
990
1126
|
struct OpsAutoGradRegisters {
|
|
991
|
-
|
|
992
|
-
|
|
993
|
-
|
|
994
|
-
|
|
995
|
-
|
|
996
|
-
|
|
997
|
-
|
|
998
|
-
|
|
999
|
-
|
|
1127
|
+
InplaceEluGradFunc InplaceEluGradFuncObj;
|
|
1128
|
+
CumminExtGradFunc CumminExtGradFuncObj;
|
|
1129
|
+
InnerCommIsendGradFunc InnerCommIsendGradFuncObj;
|
|
1130
|
+
MultiScaleDeformableAttnGradFunc MultiScaleDeformableAttnGradFuncObj;
|
|
1131
|
+
ReduceAnyGradFunc ReduceAnyGradFuncObj;
|
|
1132
|
+
InplaceFloorDivideGradFunc InplaceFloorDivideGradFuncObj;
|
|
1133
|
+
MedianDimGradFunc MedianDimGradFuncObj;
|
|
1134
|
+
DivModsGradFunc DivModsGradFuncObj;
|
|
1135
|
+
NLLLoss2dGradFunc NLLLoss2dGradFuncObj;
|
|
1136
|
+
ExpandDimsViewGradFunc ExpandDimsViewGradFuncObj;
|
|
1137
|
+
NormalFloatTensorGradFunc NormalFloatTensorGradFuncObj;
|
|
1138
|
+
SeLUExtGradFunc SeLUExtGradFuncObj;
|
|
1139
|
+
EluExtGradFunc EluExtGradFuncObj;
|
|
1140
|
+
ZerosGradFunc ZerosGradFuncObj;
|
|
1141
|
+
Expm1GradFunc Expm1GradFuncObj;
|
|
1142
|
+
GroupNormGradFunc GroupNormGradFuncObj;
|
|
1143
|
+
DistCommBroadcastGradFunc DistCommBroadcastGradFuncObj;
|
|
1144
|
+
CopyGradFunc CopyGradFuncObj;
|
|
1145
|
+
UpsampleNearest3DGradFunc UpsampleNearest3DGradFuncObj;
|
|
1146
|
+
InplaceAddsExtGradFunc InplaceAddsExtGradFuncObj;
|
|
1000
1147
|
NonZeroGradFunc NonZeroGradFuncObj;
|
|
1001
|
-
|
|
1002
|
-
|
|
1003
|
-
|
|
1004
|
-
|
|
1005
|
-
|
|
1006
|
-
|
|
1007
|
-
|
|
1008
|
-
|
|
1009
|
-
|
|
1010
|
-
|
|
1148
|
+
InplaceDivGradFunc InplaceDivGradFuncObj;
|
|
1149
|
+
ExpandAsGradFunc ExpandAsGradFuncObj;
|
|
1150
|
+
BatchNormReduceGradGradFunc BatchNormReduceGradGradFuncObj;
|
|
1151
|
+
HardtanhGradFunc HardtanhGradFuncObj;
|
|
1152
|
+
PReLUGradFunc PReLUGradFuncObj;
|
|
1153
|
+
CountNonZeroGradFunc CountNonZeroGradFuncObj;
|
|
1154
|
+
InplaceMulGradFunc InplaceMulGradFuncObj;
|
|
1155
|
+
InnerCommIrecvGradFunc InnerCommIrecvGradFuncObj;
|
|
1156
|
+
MlaGradFunc MlaGradFuncObj;
|
|
1157
|
+
RemainderScalarTensorGradFunc RemainderScalarTensorGradFuncObj;
|
|
1158
|
+
RepeatGradFunc RepeatGradFuncObj;
|
|
1159
|
+
SoftmaxBackwardGradFunc SoftmaxBackwardGradFuncObj;
|
|
1160
|
+
FloorDivScalarGradFunc FloorDivScalarGradFuncObj;
|
|
1161
|
+
InplaceZeroGradFunc InplaceZeroGradFuncObj;
|
|
1011
1162
|
NewZerosGradFunc NewZerosGradFuncObj;
|
|
1163
|
+
GreaterEqualGradFunc GreaterEqualGradFuncObj;
|
|
1164
|
+
GreaterEqualScalarGradFunc GreaterEqualScalarGradFuncObj;
|
|
1165
|
+
SmoothL1LossGradGradFunc SmoothL1LossGradGradFuncObj;
|
|
1166
|
+
HSwishGradFunc HSwishGradFuncObj;
|
|
1167
|
+
RsqrtGradFunc RsqrtGradFuncObj;
|
|
1168
|
+
LeakyReLUGradExtGradFunc LeakyReLUGradExtGradFuncObj;
|
|
1169
|
+
InplaceDivModGradFunc InplaceDivModGradFuncObj;
|
|
1170
|
+
IndexSelectGradFunc IndexSelectGradFuncObj;
|
|
1171
|
+
NonZeroExtGradFunc NonZeroExtGradFuncObj;
|
|
1172
|
+
StdGradFunc StdGradFuncObj;
|
|
1173
|
+
RoundGradFunc RoundGradFuncObj;
|
|
1174
|
+
BatchNormExtGradFunc BatchNormExtGradFuncObj;
|
|
1175
|
+
InplaceAddmmGradFunc InplaceAddmmGradFuncObj;
|
|
1176
|
+
Im2ColExtGradFunc Im2ColExtGradFuncObj;
|
|
1177
|
+
FloorGradFunc FloorGradFuncObj;
|
|
1178
|
+
StackExtGradFunc StackExtGradFuncObj;
|
|
1179
|
+
InnerInplaceIndexPutGradFunc InnerInplaceIndexPutGradFuncObj;
|
|
1180
|
+
InplaceSubScalarGradFunc InplaceSubScalarGradFuncObj;
|
|
1012
1181
|
GridSampler3DGradFunc GridSampler3DGradFuncObj;
|
|
1013
|
-
|
|
1014
|
-
|
|
1015
|
-
|
|
1016
|
-
|
|
1017
|
-
|
|
1018
|
-
|
|
1019
|
-
|
|
1020
|
-
|
|
1021
|
-
|
|
1022
|
-
LogicalAndGradFunc LogicalAndGradFuncObj;
|
|
1023
|
-
EmbeddingGradFunc EmbeddingGradFuncObj;
|
|
1024
|
-
ConvTranspose2DGradFunc ConvTranspose2DGradFuncObj;
|
|
1025
|
-
HShrinkGradFunc HShrinkGradFuncObj;
|
|
1026
|
-
HSwishGradGradFunc HSwishGradGradFuncObj;
|
|
1027
|
-
CumminExtGradFunc CumminExtGradFuncObj;
|
|
1028
|
-
MeshgridGradFunc MeshgridGradFuncObj;
|
|
1029
|
-
MaskedSelectGradGradFunc MaskedSelectGradGradFuncObj;
|
|
1030
|
-
ThresholdGradFunc ThresholdGradFuncObj;
|
|
1031
|
-
ReduceAnyGradFunc ReduceAnyGradFuncObj;
|
|
1182
|
+
MaxUnpool2DExtGradFunc MaxUnpool2DExtGradFuncObj;
|
|
1183
|
+
AbsGradFunc AbsGradFuncObj;
|
|
1184
|
+
SoftplusGradExtGradFunc SoftplusGradExtGradFuncObj;
|
|
1185
|
+
AvgPool1DGradFunc AvgPool1DGradFuncObj;
|
|
1186
|
+
Log2GradFunc Log2GradFuncObj;
|
|
1187
|
+
NormalTensorTensorGradFunc NormalTensorTensorGradFuncObj;
|
|
1188
|
+
ReshapeAndCacheGradFunc ReshapeAndCacheGradFuncObj;
|
|
1189
|
+
RandnGradFunc RandnGradFuncObj;
|
|
1190
|
+
UpsampleBilinear2DGradFunc UpsampleBilinear2DGradFuncObj;
|
|
1032
1191
|
XlogyGradFunc XlogyGradFuncObj;
|
|
1033
|
-
|
|
1034
|
-
|
|
1192
|
+
InnerCommAllGatherGradFunc InnerCommAllGatherGradFuncObj;
|
|
1193
|
+
MaxGradFunc MaxGradFuncObj;
|
|
1194
|
+
MatmulReduceScatterGradFunc MatmulReduceScatterGradFuncObj;
|
|
1195
|
+
AdaptiveAvgPool2DExtGradFunc AdaptiveAvgPool2DExtGradFuncObj;
|
|
1196
|
+
LogSoftmaxExtGradFunc LogSoftmaxExtGradFuncObj;
|
|
1035
1197
|
InplaceGroupedMatmulAddGradFunc InplaceGroupedMatmulAddGradFuncObj;
|
|
1036
|
-
|
|
1037
|
-
|
|
1198
|
+
DotGradFunc DotGradFuncObj;
|
|
1199
|
+
SortExtGradFunc SortExtGradFuncObj;
|
|
1200
|
+
RandIntGradFunc RandIntGradFuncObj;
|
|
1201
|
+
ReflectionPad2DGradFunc ReflectionPad2DGradFuncObj;
|
|
1202
|
+
ThresholdGradGradFunc ThresholdGradGradFuncObj;
|
|
1203
|
+
SoftMarginLossGradGradFunc SoftMarginLossGradGradFuncObj;
|
|
1204
|
+
LogSoftmaxGradFunc LogSoftmaxGradFuncObj;
|
|
1205
|
+
NotEqualGradFunc NotEqualGradFuncObj;
|
|
1206
|
+
FloorDivGradFunc FloorDivGradFuncObj;
|
|
1207
|
+
MinGradFunc MinGradFuncObj;
|
|
1208
|
+
PagedAttentionGradFunc PagedAttentionGradFuncObj;
|
|
1209
|
+
EmbeddingDenseBackwardGradFunc EmbeddingDenseBackwardGradFuncObj;
|
|
1210
|
+
ScatterAddExtGradFunc ScatterAddExtGradFuncObj;
|
|
1211
|
+
Col2ImExtGradFunc Col2ImExtGradFuncObj;
|
|
1212
|
+
Log1pGradFunc Log1pGradFuncObj;
|
|
1213
|
+
ConvolutionGradFunc ConvolutionGradFuncObj;
|
|
1214
|
+
BatchNormGatherStatsWithCountsGradFunc BatchNormGatherStatsWithCountsGradFuncObj;
|
|
1215
|
+
ApplyRotaryPosEmbGradFunc ApplyRotaryPosEmbGradFuncObj;
|
|
1038
1216
|
RmsNormGradGradFunc RmsNormGradGradFuncObj;
|
|
1039
|
-
SwigluGradFunc SwigluGradFuncObj;
|
|
1040
|
-
ReflectionPad1DGradFunc ReflectionPad1DGradFuncObj;
|
|
1041
|
-
MmGradFunc MmGradFuncObj;
|
|
1042
|
-
FracGradFunc FracGradFuncObj;
|
|
1043
1217
|
SilentCheckV2GradFunc SilentCheckV2GradFuncObj;
|
|
1044
|
-
|
|
1045
|
-
|
|
1046
|
-
|
|
1047
|
-
|
|
1048
|
-
|
|
1049
|
-
|
|
1050
|
-
|
|
1051
|
-
|
|
1218
|
+
ReplicationPad3DGradGradFunc ReplicationPad3DGradGradFuncObj;
|
|
1219
|
+
InplaceFloorGradFunc InplaceFloorGradFuncObj;
|
|
1220
|
+
InnerNonZeroGradFunc InnerNonZeroGradFuncObj;
|
|
1221
|
+
IsInfGradFunc IsInfGradFuncObj;
|
|
1222
|
+
RepeatInterleaveIntGradFunc RepeatInterleaveIntGradFuncObj;
|
|
1223
|
+
Atan2ExtGradFunc Atan2ExtGradFuncObj;
|
|
1224
|
+
InplaceFillTensorGradFunc InplaceFillTensorGradFuncObj;
|
|
1225
|
+
ScatterValueGradFunc ScatterValueGradFuncObj;
|
|
1226
|
+
SplitWithSizeViewGradFunc SplitWithSizeViewGradFuncObj;
|
|
1227
|
+
DistCommBarrierGradFunc DistCommBarrierGradFuncObj;
|
|
1228
|
+
InplaceIndexPutGradFunc InplaceIndexPutGradFuncObj;
|
|
1229
|
+
ViewAsGradFunc ViewAsGradFuncObj;
|
|
1230
|
+
CellBackwardHookGradFunc CellBackwardHookGradFuncObj;
|
|
1231
|
+
HardtanhGradGradFunc HardtanhGradGradFuncObj;
|
|
1232
|
+
DistCommGatherGradFunc DistCommGatherGradFuncObj;
|
|
1233
|
+
LerpGradFunc LerpGradFuncObj;
|
|
1234
|
+
TriuGradFunc TriuGradFuncObj;
|
|
1235
|
+
LogAddExpGradFunc LogAddExpGradFuncObj;
|
|
1236
|
+
UpsampleNearest3DGradGradFunc UpsampleNearest3DGradGradFuncObj;
|
|
1237
|
+
FlashAttentionScoreGradFunc FlashAttentionScoreGradFuncObj;
|
|
1238
|
+
PowTensorScalarGradFunc PowTensorScalarGradFuncObj;
|
|
1239
|
+
CoshGradFunc CoshGradFuncObj;
|
|
1240
|
+
TanhGradGradFunc TanhGradGradFuncObj;
|
|
1241
|
+
Conv1DExtGradFunc Conv1DExtGradFuncObj;
|
|
1242
|
+
TrilExtGradFunc TrilExtGradFuncObj;
|
|
1243
|
+
BatchNormStatsGradFunc BatchNormStatsGradFuncObj;
|
|
1244
|
+
BitwiseAndTensorGradFunc BitwiseAndTensorGradFuncObj;
|
|
1245
|
+
DistCommReduceScatterTensorUnevenGradFunc DistCommReduceScatterTensorUnevenGradFuncObj;
|
|
1052
1246
|
ReplicationPad2DGradFunc ReplicationPad2DGradFuncObj;
|
|
1053
|
-
|
|
1054
|
-
|
|
1055
|
-
|
|
1056
|
-
|
|
1247
|
+
IdentityGradFunc IdentityGradFuncObj;
|
|
1248
|
+
MaxPoolWithIndicesGradFunc MaxPoolWithIndicesGradFuncObj;
|
|
1249
|
+
SubGradFunc SubGradFuncObj;
|
|
1250
|
+
ExpGradFunc ExpGradFuncObj;
|
|
1251
|
+
FillScalarGradFunc FillScalarGradFuncObj;
|
|
1252
|
+
AddcmulExtGradFunc AddcmulExtGradFuncObj;
|
|
1253
|
+
CloneGradFunc CloneGradFuncObj;
|
|
1254
|
+
AtanExtGradFunc AtanExtGradFuncObj;
|
|
1255
|
+
InnerCommAllToAllVGradFunc InnerCommAllToAllVGradFuncObj;
|
|
1256
|
+
AdaptiveMaxPool2DGradFunc AdaptiveMaxPool2DGradFuncObj;
|
|
1257
|
+
AddmmGradFunc AddmmGradFuncObj;
|
|
1258
|
+
TransposeViewGradFunc TransposeViewGradFuncObj;
|
|
1259
|
+
NormGradFunc NormGradFuncObj;
|
|
1260
|
+
DivsGradFunc DivsGradFuncObj;
|
|
1261
|
+
InplaceStopGradientGradFunc InplaceStopGradientGradFuncObj;
|
|
1057
1262
|
Col2ImGradGradFunc Col2ImGradGradFuncObj;
|
|
1058
|
-
|
|
1059
|
-
|
|
1060
|
-
|
|
1061
|
-
|
|
1062
|
-
|
|
1263
|
+
RotaryPositionEmbeddingGradGradFunc RotaryPositionEmbeddingGradGradFuncObj;
|
|
1264
|
+
Conv3DExtGradFunc Conv3DExtGradFuncObj;
|
|
1265
|
+
InplaceScatterAddGradFunc InplaceScatterAddGradFuncObj;
|
|
1266
|
+
InplaceIndexAddExtGradFunc InplaceIndexAddExtGradFuncObj;
|
|
1267
|
+
SplitTensorGradFunc SplitTensorGradFuncObj;
|
|
1268
|
+
MaxPoolGradWithIndicesGradFunc MaxPoolGradWithIndicesGradFuncObj;
|
|
1269
|
+
EmbeddingGradFunc EmbeddingGradFuncObj;
|
|
1270
|
+
DistCommReduceScatterGradFunc DistCommReduceScatterGradFuncObj;
|
|
1271
|
+
NewEmptyGradFunc NewEmptyGradFuncObj;
|
|
1272
|
+
SelectExtViewGradFunc SelectExtViewGradFuncObj;
|
|
1273
|
+
IncreFlashAttentionGradFunc IncreFlashAttentionGradFuncObj;
|
|
1274
|
+
ReverseV2GradFunc ReverseV2GradFuncObj;
|
|
1275
|
+
LeakyReLUExtGradFunc LeakyReLUExtGradFuncObj;
|
|
1276
|
+
RandnLikeGradFunc RandnLikeGradFuncObj;
|
|
1277
|
+
GeLUGradFunc GeLUGradFuncObj;
|
|
1278
|
+
Exp2GradFunc Exp2GradFuncObj;
|
|
1279
|
+
ReshapeGradFunc ReshapeGradFuncObj;
|
|
1280
|
+
ThresholdGradFunc ThresholdGradFuncObj;
|
|
1281
|
+
UpsampleLinear1DGradGradFunc UpsampleLinear1DGradGradFuncObj;
|
|
1282
|
+
LogicalAndGradFunc LogicalAndGradFuncObj;
|
|
1283
|
+
InnerMoeTokenUnpermuteGradFunc InnerMoeTokenUnpermuteGradFuncObj;
|
|
1284
|
+
AddbmmGradFunc AddbmmGradFuncObj;
|
|
1285
|
+
XLogYScalarOtherGradFunc XLogYScalarOtherGradFuncObj;
|
|
1286
|
+
AcoshExtGradFunc AcoshExtGradFuncObj;
|
|
1287
|
+
ViewGradFunc ViewGradFuncObj;
|
|
1288
|
+
ReduceMinGradFunc ReduceMinGradFuncObj;
|
|
1289
|
+
SpeedFusionAttentionGradGradFunc SpeedFusionAttentionGradGradFuncObj;
|
|
1290
|
+
RemainderTensorScalarGradFunc RemainderTensorScalarGradFuncObj;
|
|
1291
|
+
SqrtGradFunc SqrtGradFuncObj;
|
|
1292
|
+
TanGradFunc TanGradFuncObj;
|
|
1293
|
+
SilentCheckV3GradFunc SilentCheckV3GradFuncObj;
|
|
1294
|
+
AddmvGradFunc AddmvGradFuncObj;
|
|
1295
|
+
InplaceReLUGradFunc InplaceReLUGradFuncObj;
|
|
1296
|
+
RandIntLikeGradFunc RandIntLikeGradFuncObj;
|
|
1297
|
+
GcdGradFunc GcdGradFuncObj;
|
|
1298
|
+
SplitGradFunc SplitGradFuncObj;
|
|
1299
|
+
MulGradFunc MulGradFuncObj;
|
|
1300
|
+
EluGradExtGradFunc EluGradExtGradFuncObj;
|
|
1301
|
+
RandpermExtGradFunc RandpermExtGradFuncObj;
|
|
1302
|
+
DiagonalViewGradFunc DiagonalViewGradFuncObj;
|
|
1303
|
+
TruncGradFunc TruncGradFuncObj;
|
|
1304
|
+
InplaceLogGradFunc InplaceLogGradFuncObj;
|
|
1305
|
+
AddcdivExtGradFunc AddcdivExtGradFuncObj;
|
|
1306
|
+
TExtGradFunc TExtGradFuncObj;
|
|
1063
1307
|
UpsampleTrilinear3DGradGradFunc UpsampleTrilinear3DGradGradFuncObj;
|
|
1064
|
-
|
|
1065
|
-
|
|
1066
|
-
|
|
1067
|
-
|
|
1068
|
-
|
|
1069
|
-
|
|
1070
|
-
|
|
1308
|
+
BitwiseOrScalarGradFunc BitwiseOrScalarGradFuncObj;
|
|
1309
|
+
BitwiseOrTensorGradFunc BitwiseOrTensorGradFuncObj;
|
|
1310
|
+
ReduceAllGradFunc ReduceAllGradFuncObj;
|
|
1311
|
+
BroadcastToGradFunc BroadcastToGradFuncObj;
|
|
1312
|
+
AdaptiveAvgPool3DGradExtGradFunc AdaptiveAvgPool3DGradExtGradFuncObj;
|
|
1313
|
+
RepeatInterleaveGradGradFunc RepeatInterleaveGradGradFuncObj;
|
|
1314
|
+
GluGradGradFunc GluGradGradFuncObj;
|
|
1315
|
+
CosGradFunc CosGradFuncObj;
|
|
1316
|
+
GridSampler2DGradFunc GridSampler2DGradFuncObj;
|
|
1317
|
+
ConstantPadNDGradFunc ConstantPadNDGradFuncObj;
|
|
1318
|
+
MaskedSelectGradFunc MaskedSelectGradFuncObj;
|
|
1071
1319
|
UpsampleBicubic2DGradFunc UpsampleBicubic2DGradFuncObj;
|
|
1072
|
-
|
|
1073
|
-
|
|
1074
|
-
|
|
1075
|
-
|
|
1076
|
-
|
|
1077
|
-
|
|
1078
|
-
|
|
1079
|
-
|
|
1080
|
-
|
|
1081
|
-
|
|
1082
|
-
|
|
1083
|
-
|
|
1084
|
-
|
|
1085
|
-
|
|
1320
|
+
MaskedFillGradFunc MaskedFillGradFuncObj;
|
|
1321
|
+
EmptyLikeGradFunc EmptyLikeGradFuncObj;
|
|
1322
|
+
SiLUGradFunc SiLUGradFuncObj;
|
|
1323
|
+
NansumGradFunc NansumGradFuncObj;
|
|
1324
|
+
SplitWithSizeGradFunc SplitWithSizeGradFuncObj;
|
|
1325
|
+
DistCommAllGatherIntoTensorGradFunc DistCommAllGatherIntoTensorGradFuncObj;
|
|
1326
|
+
L1LossBackwardExtGradFunc L1LossBackwardExtGradFuncObj;
|
|
1327
|
+
AddRmsNormGradFunc AddRmsNormGradFuncObj;
|
|
1328
|
+
InplaceSiLUGradFunc InplaceSiLUGradFuncObj;
|
|
1329
|
+
ReLUGradFunc ReLUGradFuncObj;
|
|
1330
|
+
StdMeanGradFunc StdMeanGradFuncObj;
|
|
1331
|
+
AvgPool2DGradGradFunc AvgPool2DGradGradFuncObj;
|
|
1332
|
+
SoftMarginLossGradFunc SoftMarginLossGradFuncObj;
|
|
1333
|
+
LessEqualGradFunc LessEqualGradFuncObj;
|
|
1334
|
+
ConvTranspose2DGradFunc ConvTranspose2DGradFuncObj;
|
|
1335
|
+
MeanExtGradFunc MeanExtGradFuncObj;
|
|
1336
|
+
MinimumGradFunc MinimumGradFuncObj;
|
|
1337
|
+
BatchMatMulGradFunc BatchMatMulGradFuncObj;
|
|
1338
|
+
SiLUGradGradFunc SiLUGradGradFuncObj;
|
|
1339
|
+
FmodScalarGradFunc FmodScalarGradFuncObj;
|
|
1340
|
+
GatherDGradV2GradFunc GatherDGradV2GradFuncObj;
|
|
1341
|
+
EqualGradFunc EqualGradFuncObj;
|
|
1342
|
+
MoeDistributeDispatchGradFunc MoeDistributeDispatchGradFuncObj;
|
|
1343
|
+
InplaceFloorDividesGradFunc InplaceFloorDividesGradFuncObj;
|
|
1086
1344
|
CumsumExtGradFunc CumsumExtGradFuncObj;
|
|
1087
|
-
|
|
1088
|
-
|
|
1089
|
-
|
|
1090
|
-
|
|
1091
|
-
|
|
1092
|
-
|
|
1093
|
-
|
|
1094
|
-
|
|
1095
|
-
|
|
1096
|
-
|
|
1097
|
-
|
|
1098
|
-
|
|
1099
|
-
|
|
1100
|
-
|
|
1101
|
-
|
|
1102
|
-
|
|
1103
|
-
|
|
1104
|
-
|
|
1345
|
+
LayerNormGradExtGradFunc LayerNormGradExtGradFuncObj;
|
|
1346
|
+
LogSumExpGradFunc LogSumExpGradFuncObj;
|
|
1347
|
+
BinaryCrossEntropyGradFunc BinaryCrossEntropyGradFuncObj;
|
|
1348
|
+
ChunkViewGradFunc ChunkViewGradFuncObj;
|
|
1349
|
+
InplaceExpGradFunc InplaceExpGradFuncObj;
|
|
1350
|
+
InplaceFillScalarGradFunc InplaceFillScalarGradFuncObj;
|
|
1351
|
+
AddScalarGradFunc AddScalarGradFuncObj;
|
|
1352
|
+
SoftplusExtGradFunc SoftplusExtGradFuncObj;
|
|
1353
|
+
HSwishGradGradFunc HSwishGradGradFuncObj;
|
|
1354
|
+
DistCommGatherIntoTensorGradFunc DistCommGatherIntoTensorGradFuncObj;
|
|
1355
|
+
DiagExtGradFunc DiagExtGradFuncObj;
|
|
1356
|
+
LogAddExp2GradFunc LogAddExp2GradFuncObj;
|
|
1357
|
+
NormalFloatFloatGradFunc NormalFloatFloatGradFuncObj;
|
|
1358
|
+
ReflectionPad3DGradFunc ReflectionPad3DGradFuncObj;
|
|
1359
|
+
SinhGradFunc SinhGradFuncObj;
|
|
1360
|
+
ArgMaxExtGradFunc ArgMaxExtGradFuncObj;
|
|
1361
|
+
AsStridedGradFunc AsStridedGradFuncObj;
|
|
1362
|
+
LinalgQrGradFunc LinalgQrGradFuncObj;
|
|
1363
|
+
AdaptiveAvgPool1DGradFunc AdaptiveAvgPool1DGradFuncObj;
|
|
1364
|
+
NarrowGradFunc NarrowGradFuncObj;
|
|
1365
|
+
MinDimGradFunc MinDimGradFuncObj;
|
|
1366
|
+
InplaceFillDiagonalGradFunc InplaceFillDiagonalGradFuncObj;
|
|
1367
|
+
LogSigmoidGradGradFunc LogSigmoidGradGradFuncObj;
|
|
1368
|
+
MishGradExtGradFunc MishGradExtGradFuncObj;
|
|
1369
|
+
LessGradFunc LessGradFuncObj;
|
|
1370
|
+
CrossGradFunc CrossGradFuncObj;
|
|
1371
|
+
LogSoftmaxGradGradFunc LogSoftmaxGradGradFuncObj;
|
|
1372
|
+
InplaceMatmulAddGradFunc InplaceMatmulAddGradFuncObj;
|
|
1373
|
+
ScatterGradFunc ScatterGradFuncObj;
|
|
1374
|
+
AddExtGradFunc AddExtGradFuncObj;
|
|
1375
|
+
ReplicationPad3DGradFunc ReplicationPad3DGradFuncObj;
|
|
1376
|
+
HShrinkGradFunc HShrinkGradFuncObj;
|
|
1377
|
+
AvgPool2DGradFunc AvgPool2DGradFuncObj;
|
|
1378
|
+
GreaterGradFunc GreaterGradFuncObj;
|
|
1379
|
+
EmptyGradFunc EmptyGradFuncObj;
|
|
1105
1380
|
FlattenExtGradFunc FlattenExtGradFuncObj;
|
|
1106
|
-
|
|
1107
|
-
|
|
1108
|
-
TransposeExtGradFunc TransposeExtGradFuncObj;
|
|
1109
|
-
MishExtGradFunc MishExtGradFuncObj;
|
|
1110
|
-
UpsampleNearest2DGradFunc UpsampleNearest2DGradFuncObj;
|
|
1111
|
-
ReplicationPad2DGradGradFunc ReplicationPad2DGradGradFuncObj;
|
|
1112
|
-
ReflectionPad2DGradFunc ReflectionPad2DGradFuncObj;
|
|
1381
|
+
MaxPoolGradWithMaskGradFunc MaxPoolGradWithMaskGradFuncObj;
|
|
1382
|
+
L1LossExtGradFunc L1LossExtGradFuncObj;
|
|
1113
1383
|
GeluExtGradFunc GeluExtGradFuncObj;
|
|
1114
|
-
|
|
1115
|
-
|
|
1116
|
-
|
|
1117
|
-
|
|
1118
|
-
|
|
1119
|
-
|
|
1120
|
-
|
|
1121
|
-
|
|
1122
|
-
|
|
1123
|
-
ReplicationPad1DGradGradFunc ReplicationPad1DGradGradFuncObj;
|
|
1124
|
-
BinaryCrossEntropyGradFunc BinaryCrossEntropyGradFuncObj;
|
|
1125
|
-
InplaceAddmmGradFunc InplaceAddmmGradFuncObj;
|
|
1126
|
-
MoeTokenUnpermuteGradGradFunc MoeTokenUnpermuteGradGradFuncObj;
|
|
1384
|
+
SqueezeGradFunc SqueezeGradFuncObj;
|
|
1385
|
+
DistCommIrecvGradFunc DistCommIrecvGradFuncObj;
|
|
1386
|
+
LerpScalarGradFunc LerpScalarGradFuncObj;
|
|
1387
|
+
InnerCommAllReduceGradFunc InnerCommAllReduceGradFuncObj;
|
|
1388
|
+
SquareGradFunc SquareGradFuncObj;
|
|
1389
|
+
Conv2DPaddingGradFunc Conv2DPaddingGradFuncObj;
|
|
1390
|
+
ArgMinWithValueGradFunc ArgMinWithValueGradFuncObj;
|
|
1391
|
+
SpeedFusionAttentionGradFunc SpeedFusionAttentionGradFuncObj;
|
|
1392
|
+
DropoutGenMaskExtGradFunc DropoutGenMaskExtGradFuncObj;
|
|
1127
1393
|
BinaryCrossEntropyWithLogitsBackwardGradFunc BinaryCrossEntropyWithLogitsBackwardGradFuncObj;
|
|
1128
|
-
|
|
1129
|
-
|
|
1130
|
-
|
|
1131
|
-
|
|
1132
|
-
|
|
1133
|
-
|
|
1134
|
-
|
|
1135
|
-
|
|
1136
|
-
|
|
1137
|
-
|
|
1138
|
-
|
|
1139
|
-
|
|
1140
|
-
|
|
1141
|
-
RandpermExtGradFunc RandpermExtGradFuncObj;
|
|
1394
|
+
ConvolutionStrGradGradFunc ConvolutionStrGradGradFuncObj;
|
|
1395
|
+
LogSigmoidGradFunc LogSigmoidGradFuncObj;
|
|
1396
|
+
LinalgVectorNormGradFunc LinalgVectorNormGradFuncObj;
|
|
1397
|
+
BroadcastToViewGradFunc BroadcastToViewGradFuncObj;
|
|
1398
|
+
SoftShrinkGradFunc SoftShrinkGradFuncObj;
|
|
1399
|
+
AddLayerNormGradGradFunc AddLayerNormGradGradFuncObj;
|
|
1400
|
+
UnstackExtViewGradFunc UnstackExtViewGradFuncObj;
|
|
1401
|
+
DistCommBatchIsendIrecvGradFunc DistCommBatchIsendIrecvGradFuncObj;
|
|
1402
|
+
GridSampler2DGradGradFunc GridSampler2DGradGradFuncObj;
|
|
1403
|
+
RotaryPositionEmbeddingGradFunc RotaryPositionEmbeddingGradFuncObj;
|
|
1404
|
+
VarMeanGradFunc VarMeanGradFuncObj;
|
|
1405
|
+
KthvalueGradFunc KthvalueGradFuncObj;
|
|
1406
|
+
MvGradFunc MvGradFuncObj;
|
|
1142
1407
|
UniqueDimGradFunc UniqueDimGradFuncObj;
|
|
1143
|
-
|
|
1144
|
-
|
|
1145
|
-
|
|
1146
|
-
|
|
1147
|
-
|
|
1148
|
-
|
|
1149
|
-
|
|
1150
|
-
|
|
1151
|
-
NotEqualGradFunc NotEqualGradFuncObj;
|
|
1152
|
-
MinGradFunc MinGradFuncObj;
|
|
1408
|
+
InplaceDivModsGradFunc InplaceDivModsGradFuncObj;
|
|
1409
|
+
FillTensorGradFunc FillTensorGradFuncObj;
|
|
1410
|
+
ExpandDimsGradFunc ExpandDimsGradFuncObj;
|
|
1411
|
+
InplaceThresholdGradFunc InplaceThresholdGradFuncObj;
|
|
1412
|
+
InplaceRemainderTensorTensorGradFunc InplaceRemainderTensorTensorGradFuncObj;
|
|
1413
|
+
ErfinvGradFunc ErfinvGradFuncObj;
|
|
1414
|
+
TriangularSolveGradFunc TriangularSolveGradFuncObj;
|
|
1415
|
+
InplaceScatterSrcGradFunc InplaceScatterSrcGradFuncObj;
|
|
1153
1416
|
RandExtGradFunc RandExtGradFuncObj;
|
|
1154
|
-
|
|
1155
|
-
|
|
1156
|
-
InplaceMulGradFunc InplaceMulGradFuncObj;
|
|
1157
|
-
InplaceSubScalarGradFunc InplaceSubScalarGradFuncObj;
|
|
1158
|
-
ExpandAsGradFunc ExpandAsGradFuncObj;
|
|
1159
|
-
AdaptiveAvgPool2DGradExtGradFunc AdaptiveAvgPool2DGradExtGradFuncObj;
|
|
1160
|
-
FmodScalarGradFunc FmodScalarGradFuncObj;
|
|
1161
|
-
InplaceFloorDividesGradFunc InplaceFloorDividesGradFuncObj;
|
|
1162
|
-
NansumGradFunc NansumGradFuncObj;
|
|
1163
|
-
OneHotExtGradFunc OneHotExtGradFuncObj;
|
|
1164
|
-
MedianExtGradFunc MedianExtGradFuncObj;
|
|
1165
|
-
AsinExtGradFunc AsinExtGradFuncObj;
|
|
1166
|
-
AddmmGradFunc AddmmGradFuncObj;
|
|
1167
|
-
UpsampleLinear1DGradGradFunc UpsampleLinear1DGradGradFuncObj;
|
|
1168
|
-
GreaterGradFunc GreaterGradFuncObj;
|
|
1169
|
-
SeLUExtGradFunc SeLUExtGradFuncObj;
|
|
1170
|
-
BaddbmmGradFunc BaddbmmGradFuncObj;
|
|
1171
|
-
InplaceFloorGradFunc InplaceFloorGradFuncObj;
|
|
1172
|
-
BatchNormExtGradFunc BatchNormExtGradFuncObj;
|
|
1173
|
-
GeneratorGradFunc GeneratorGradFuncObj;
|
|
1174
|
-
NewOnesGradFunc NewOnesGradFuncObj;
|
|
1175
|
-
InplaceMaskedFillScalarGradFunc InplaceMaskedFillScalarGradFuncObj;
|
|
1176
|
-
NLLLossGradGradFunc NLLLossGradGradFuncObj;
|
|
1177
|
-
SilentCheckV3GradFunc SilentCheckV3GradFuncObj;
|
|
1178
|
-
SoftplusGradExtGradFunc SoftplusGradExtGradFuncObj;
|
|
1179
|
-
ScatterValueGradFunc ScatterValueGradFuncObj;
|
|
1180
|
-
TopkExtGradFunc TopkExtGradFuncObj;
|
|
1417
|
+
BCEWithLogitsLossGradFunc BCEWithLogitsLossGradFuncObj;
|
|
1418
|
+
FlashAttentionScoreGradGradFunc FlashAttentionScoreGradGradFuncObj;
|
|
1181
1419
|
MaximumGradFunc MaximumGradFuncObj;
|
|
1182
|
-
|
|
1183
|
-
|
|
1184
|
-
|
|
1185
|
-
|
|
1186
|
-
|
|
1187
|
-
|
|
1188
|
-
|
|
1189
|
-
|
|
1190
|
-
|
|
1191
|
-
|
|
1420
|
+
BaddbmmGradFunc BaddbmmGradFuncObj;
|
|
1421
|
+
MoeTokenPermuteGradFunc MoeTokenPermuteGradFuncObj;
|
|
1422
|
+
InplaceSubExtGradFunc InplaceSubExtGradFuncObj;
|
|
1423
|
+
InplaceErfinvGradFunc InplaceErfinvGradFuncObj;
|
|
1424
|
+
MeshgridGradFunc MeshgridGradFuncObj;
|
|
1425
|
+
MaxPoolWithMaskGradFunc MaxPoolWithMaskGradFuncObj;
|
|
1426
|
+
SelectV2GradFunc SelectV2GradFuncObj;
|
|
1427
|
+
LinSpaceExtGradFunc LinSpaceExtGradFuncObj;
|
|
1428
|
+
RmsNormGradFunc RmsNormGradFuncObj;
|
|
1429
|
+
SumExtGradFunc SumExtGradFuncObj;
|
|
1430
|
+
ZerosLikeExtGradFunc ZerosLikeExtGradFuncObj;
|
|
1431
|
+
ConcatGradFunc ConcatGradFuncObj;
|
|
1432
|
+
BitwiseXorScalarGradFunc BitwiseXorScalarGradFuncObj;
|
|
1433
|
+
FmodTensorGradFunc FmodTensorGradFuncObj;
|
|
1434
|
+
DistCommReduceScatterTensorGradFunc DistCommReduceScatterTensorGradFuncObj;
|
|
1435
|
+
CrossEntropyLossGradGradFunc CrossEntropyLossGradGradFuncObj;
|
|
1436
|
+
UpsampleNearest1DGradFunc UpsampleNearest1DGradFuncObj;
|
|
1437
|
+
IsFiniteGradFunc IsFiniteGradFuncObj;
|
|
1438
|
+
NarrowViewGradFunc NarrowViewGradFuncObj;
|
|
1439
|
+
InplaceMaskedFillTensorGradFunc InplaceMaskedFillTensorGradFuncObj;
|
|
1440
|
+
TanhGradFunc TanhGradFuncObj;
|
|
1441
|
+
NewFullGradFunc NewFullGradFuncObj;
|
|
1442
|
+
MulsGradFunc MulsGradFuncObj;
|
|
1443
|
+
DropoutExtGradFunc DropoutExtGradFuncObj;
|
|
1444
|
+
DistCommScatterGradFunc DistCommScatterGradFuncObj;
|
|
1445
|
+
AsinExtGradFunc AsinExtGradFuncObj;
|
|
1192
1446
|
RemainderTensorTensorGradFunc RemainderTensorTensorGradFuncObj;
|
|
1193
|
-
|
|
1194
|
-
|
|
1195
|
-
|
|
1196
|
-
|
|
1447
|
+
SincGradFunc SincGradFuncObj;
|
|
1448
|
+
InplaceDivsGradFunc InplaceDivsGradFuncObj;
|
|
1449
|
+
Conv2DExtGradFunc Conv2DExtGradFuncObj;
|
|
1450
|
+
InnerCommReduceScatterGradFunc InnerCommReduceScatterGradFuncObj;
|
|
1451
|
+
MishExtGradFunc MishExtGradFuncObj;
|
|
1452
|
+
AcosExtGradFunc AcosExtGradFuncObj;
|
|
1453
|
+
TraceExtGradFunc TraceExtGradFuncObj;
|
|
1454
|
+
InplacePutGradFunc InplacePutGradFuncObj;
|
|
1455
|
+
GLUGradFunc GLUGradFuncObj;
|
|
1456
|
+
AddGradFunc AddGradFuncObj;
|
|
1457
|
+
NLLLossGradGradFunc NLLLossGradGradFuncObj;
|
|
1458
|
+
InplaceMaskedFillScalarGradFunc InplaceMaskedFillScalarGradFuncObj;
|
|
1459
|
+
RandLikeExtGradFunc RandLikeExtGradFuncObj;
|
|
1460
|
+
PolarGradFunc PolarGradFuncObj;
|
|
1461
|
+
CastGradFunc CastGradFuncObj;
|
|
1462
|
+
SwigluGradFunc SwigluGradFuncObj;
|
|
1197
1463
|
InplaceAddExtGradFunc InplaceAddExtGradFuncObj;
|
|
1198
|
-
|
|
1199
|
-
|
|
1464
|
+
ArangeGradFunc ArangeGradFuncObj;
|
|
1465
|
+
MoeDistributeCombineGradFunc MoeDistributeCombineGradFuncObj;
|
|
1466
|
+
LogGradFunc LogGradFuncObj;
|
|
1467
|
+
DivModGradFunc DivModGradFuncObj;
|
|
1468
|
+
ArgMaxWithValueGradFunc ArgMaxWithValueGradFuncObj;
|
|
1469
|
+
RollGradFunc RollGradFuncObj;
|
|
1470
|
+
ReduceMaxGradFunc ReduceMaxGradFuncObj;
|
|
1471
|
+
LogicalOrGradFunc LogicalOrGradFuncObj;
|
|
1200
1472
|
NLLLoss2dGradGradFunc NLLLoss2dGradGradFuncObj;
|
|
1201
|
-
|
|
1202
|
-
AdaptiveAvgPool3DGradExtGradFunc AdaptiveAvgPool3DGradExtGradFuncObj;
|
|
1203
|
-
AsStridedGradFunc AsStridedGradFuncObj;
|
|
1204
|
-
BitwiseAndTensorGradFunc BitwiseAndTensorGradFuncObj;
|
|
1205
|
-
MaxPoolWithIndicesGradFunc MaxPoolWithIndicesGradFuncObj;
|
|
1206
|
-
InplaceUniformGradFunc InplaceUniformGradFuncObj;
|
|
1207
|
-
UpsampleNearest3DGradGradFunc UpsampleNearest3DGradGradFuncObj;
|
|
1208
|
-
SignGradFunc SignGradFuncObj;
|
|
1209
|
-
Conv1DPaddingGradFunc Conv1DPaddingGradFuncObj;
|
|
1210
|
-
TanGradFunc TanGradFuncObj;
|
|
1211
|
-
TExtGradFunc TExtGradFuncObj;
|
|
1212
|
-
BatchNormGradExtGradFunc BatchNormGradExtGradFuncObj;
|
|
1213
|
-
IsCloseGradFunc IsCloseGradFuncObj;
|
|
1214
|
-
BatchNormReduceGradGradFunc BatchNormReduceGradGradFuncObj;
|
|
1215
|
-
IncreFlashAttentionGradFunc IncreFlashAttentionGradFuncObj;
|
|
1216
|
-
ReshapeGradFunc ReshapeGradFuncObj;
|
|
1217
|
-
CeilGradFunc CeilGradFuncObj;
|
|
1218
|
-
ZerosGradFunc ZerosGradFuncObj;
|
|
1219
|
-
DivsGradFunc DivsGradFuncObj;
|
|
1220
|
-
DivGradFunc DivGradFuncObj;
|
|
1221
|
-
ViewGradFunc ViewGradFuncObj;
|
|
1222
|
-
SoftplusExtGradFunc SoftplusExtGradFuncObj;
|
|
1223
|
-
L1LossExtGradFunc L1LossExtGradFuncObj;
|
|
1224
|
-
AddmvGradFunc AddmvGradFuncObj;
|
|
1225
|
-
CrossGradFunc CrossGradFuncObj;
|
|
1226
|
-
NormalFloatTensorGradFunc NormalFloatTensorGradFuncObj;
|
|
1227
|
-
TraceExtGradFunc TraceExtGradFuncObj;
|
|
1228
|
-
SelectExtGradFunc SelectExtGradFuncObj;
|
|
1229
|
-
InplaceDivGradFunc InplaceDivGradFuncObj;
|
|
1230
|
-
SpeedFusionAttentionGradGradFunc SpeedFusionAttentionGradGradFuncObj;
|
|
1231
|
-
AbsGradFunc AbsGradFuncObj;
|
|
1232
|
-
NegGradFunc NegGradFuncObj;
|
|
1233
|
-
ArgMinWithValueGradFunc ArgMinWithValueGradFuncObj;
|
|
1234
|
-
SigmoidGradGradFunc SigmoidGradGradFuncObj;
|
|
1235
|
-
StdGradFunc StdGradFuncObj;
|
|
1236
|
-
RandnLikeGradFunc RandnLikeGradFuncObj;
|
|
1237
|
-
UpsampleNearest1DGradFunc UpsampleNearest1DGradFuncObj;
|
|
1238
|
-
L1LossBackwardExtGradFunc L1LossBackwardExtGradFuncObj;
|
|
1239
|
-
OnesLikeExtGradFunc OnesLikeExtGradFuncObj;
|
|
1240
|
-
InplaceThresholdGradFunc InplaceThresholdGradFuncObj;
|
|
1241
|
-
MSELossExtGradFunc MSELossExtGradFuncObj;
|
|
1242
|
-
LeakyReLUExtGradFunc LeakyReLUExtGradFuncObj;
|
|
1243
|
-
IndexSelectGradFunc IndexSelectGradFuncObj;
|
|
1244
|
-
DotGradFunc DotGradFuncObj;
|
|
1245
|
-
MoeTokenPermuteGradGradFunc MoeTokenPermuteGradGradFuncObj;
|
|
1246
|
-
TriuGradFunc TriuGradFuncObj;
|
|
1247
|
-
Conv3DExtGradFunc Conv3DExtGradFuncObj;
|
|
1248
|
-
LogSigmoidGradFunc LogSigmoidGradFuncObj;
|
|
1249
|
-
InplaceFillDiagonalGradFunc InplaceFillDiagonalGradFuncObj;
|
|
1473
|
+
ReflectionPad3DGradGradFunc ReflectionPad3DGradGradFuncObj;
|
|
1250
1474
|
InplaceClampTensorGradFunc InplaceClampTensorGradFuncObj;
|
|
1251
|
-
|
|
1252
|
-
|
|
1253
|
-
|
|
1254
|
-
ReflectionPad3DGradFunc ReflectionPad3DGradFuncObj;
|
|
1255
|
-
SqueezeGradFunc SqueezeGradFuncObj;
|
|
1256
|
-
InplaceErfinvGradFunc InplaceErfinvGradFuncObj;
|
|
1475
|
+
AtanhGradFunc AtanhGradFuncObj;
|
|
1476
|
+
ReplicationPad1DGradFunc ReplicationPad1DGradFuncObj;
|
|
1477
|
+
BatchNormElemtGradFunc BatchNormElemtGradFuncObj;
|
|
1257
1478
|
ErfcGradFunc ErfcGradFuncObj;
|
|
1258
|
-
InplaceStopGradientGradFunc InplaceStopGradientGradFuncObj;
|
|
1259
|
-
GridSampler2DGradGradFunc GridSampler2DGradGradFuncObj;
|
|
1260
|
-
ReduceMaxGradFunc ReduceMaxGradFuncObj;
|
|
1261
|
-
SliceGradFunc SliceGradFuncObj;
|
|
1262
|
-
SinGradFunc SinGradFuncObj;
|
|
1263
|
-
ArgMaxExtGradFunc ArgMaxExtGradFuncObj;
|
|
1264
|
-
SoftmaxGradFunc SoftmaxGradFuncObj;
|
|
1265
|
-
TensorScatterElementsGradFunc TensorScatterElementsGradFuncObj;
|
|
1266
|
-
HShrinkGradGradFunc HShrinkGradGradFuncObj;
|
|
1267
|
-
Log1pGradFunc Log1pGradFuncObj;
|
|
1268
|
-
LinalgVectorNormGradFunc LinalgVectorNormGradFuncObj;
|
|
1269
|
-
FillScalarGradFunc FillScalarGradFuncObj;
|
|
1270
|
-
SplitGradFunc SplitGradFuncObj;
|
|
1271
|
-
SplitWithSizeGradFunc SplitWithSizeGradFuncObj;
|
|
1272
|
-
SmoothL1LossGradGradFunc SmoothL1LossGradGradFuncObj;
|
|
1273
|
-
XLogYScalarSelfGradFunc XLogYScalarSelfGradFuncObj;
|
|
1274
|
-
ExpGradFunc ExpGradFuncObj;
|
|
1275
|
-
Col2ImExtGradFunc Col2ImExtGradFuncObj;
|
|
1276
|
-
SoftMarginLossGradGradFunc SoftMarginLossGradGradFuncObj;
|
|
1277
|
-
AdaptiveAvgPool1DGradFunc AdaptiveAvgPool1DGradFuncObj;
|
|
1278
|
-
RmsNormGradFunc RmsNormGradFuncObj;
|
|
1279
|
-
MultiScaleDeformableAttnGradFunc MultiScaleDeformableAttnGradFuncObj;
|
|
1280
|
-
ContiguousGradFunc ContiguousGradFuncObj;
|
|
1281
|
-
InplaceMulsGradFunc InplaceMulsGradFuncObj;
|
|
1282
|
-
ScatterAddExtGradFunc ScatterAddExtGradFuncObj;
|
|
1283
|
-
AsinhExtGradFunc AsinhExtGradFuncObj;
|
|
1284
|
-
InplaceIndexPutGradFunc InplaceIndexPutGradFuncObj;
|
|
1285
|
-
PromptFlashAttentionGradFunc PromptFlashAttentionGradFuncObj;
|
|
1286
|
-
GeLUGradFunc GeLUGradFuncObj;
|
|
1287
|
-
HSigmoidGradFunc HSigmoidGradFuncObj;
|
|
1288
|
-
SliceExtGradFunc SliceExtGradFuncObj;
|
|
1289
|
-
LogSumExpGradFunc LogSumExpGradFuncObj;
|
|
1290
|
-
RepeatInterleaveIntGradFunc RepeatInterleaveIntGradFuncObj;
|
|
1291
|
-
InplaceScatterSrcGradFunc InplaceScatterSrcGradFuncObj;
|
|
1292
|
-
GreaterEqualScalarGradFunc GreaterEqualScalarGradFuncObj;
|
|
1293
|
-
PowTensorScalarGradFunc PowTensorScalarGradFuncObj;
|
|
1294
|
-
RepeatInterleaveGradGradFunc RepeatInterleaveGradGradFuncObj;
|
|
1295
|
-
OnesGradFunc OnesGradFuncObj;
|
|
1296
|
-
UpsampleBilinear2DGradFunc UpsampleBilinear2DGradFuncObj;
|
|
1297
|
-
InplaceCopyGradFunc InplaceCopyGradFuncObj;
|
|
1298
|
-
AddLayerNormV2GradFunc AddLayerNormV2GradFuncObj;
|
|
1299
|
-
AddcmulExtGradFunc AddcmulExtGradFuncObj;
|
|
1300
|
-
CopyGradFunc CopyGradFuncObj;
|
|
1301
|
-
SinhGradFunc SinhGradFuncObj;
|
|
1302
|
-
NormalTensorTensorGradFunc NormalTensorTensorGradFuncObj;
|
|
1303
|
-
SigmoidGradFunc SigmoidGradFuncObj;
|
|
1304
|
-
Conv1DExtGradFunc Conv1DExtGradFuncObj;
|
|
1305
|
-
InplaceRandomGradFunc InplaceRandomGradFuncObj;
|
|
1306
|
-
GluGradGradFunc GluGradGradFuncObj;
|
|
1307
|
-
NarrowGradFunc NarrowGradFuncObj;
|
|
1308
|
-
GridSampler2DGradFunc GridSampler2DGradFuncObj;
|
|
1309
|
-
BitwiseOrScalarGradFunc BitwiseOrScalarGradFuncObj;
|
|
1310
|
-
ReflectionPad2DGradGradFunc ReflectionPad2DGradGradFuncObj;
|
|
1311
|
-
FloorDivScalarGradFunc FloorDivScalarGradFuncObj;
|
|
1312
|
-
PolarGradFunc PolarGradFuncObj;
|
|
1313
|
-
UpsampleNearest1DGradGradFunc UpsampleNearest1DGradGradFuncObj;
|
|
1314
|
-
LogAddExp2GradFunc LogAddExp2GradFuncObj;
|
|
1315
|
-
FlashAttentionScoreGradGradFunc FlashAttentionScoreGradGradFuncObj;
|
|
1316
|
-
NormalFloatFloatGradFunc NormalFloatFloatGradFuncObj;
|
|
1317
|
-
UnstackExtGradFunc UnstackExtGradFuncObj;
|
|
1318
|
-
AtanExtGradFunc AtanExtGradFuncObj;
|
|
1319
|
-
InplaceDivModGradFunc InplaceDivModGradFuncObj;
|
|
1320
|
-
InplaceScatterSrcReduceGradFunc InplaceScatterSrcReduceGradFuncObj;
|
|
1321
|
-
SortExtGradFunc SortExtGradFuncObj;
|
|
1322
|
-
HistcExtGradFunc HistcExtGradFuncObj;
|
|
1323
|
-
InplaceDivsGradFunc InplaceDivsGradFuncObj;
|
|
1324
|
-
ConcatGradFunc ConcatGradFuncObj;
|
|
1325
|
-
InplaceAddsExtGradFunc InplaceAddsExtGradFuncObj;
|
|
1326
1479
|
HSigmoidGradGradFunc HSigmoidGradGradFuncObj;
|
|
1480
|
+
FracGradFunc FracGradFuncObj;
|
|
1481
|
+
SigmoidGradGradFunc SigmoidGradGradFuncObj;
|
|
1482
|
+
FullLikeGradFunc FullLikeGradFuncObj;
|
|
1483
|
+
SoftShrinkGradGradFunc SoftShrinkGradGradFuncObj;
|
|
1484
|
+
BatchMatMulExtGradFunc BatchMatMulExtGradFuncObj;
|
|
1485
|
+
RingAttentionUpdateGradFunc RingAttentionUpdateGradFuncObj;
|
|
1486
|
+
DistCommAllToAllVGradFunc DistCommAllToAllVGradFuncObj;
|
|
1487
|
+
PReLUGradGradFunc PReLUGradGradFuncObj;
|
|
1488
|
+
RepeatInterleaveTensorGradFunc RepeatInterleaveTensorGradFuncObj;
|
|
1489
|
+
AddLayerNormV2GradFunc AddLayerNormV2GradFuncObj;
|
|
1490
|
+
BitwiseNotGradFunc BitwiseNotGradFuncObj;
|
|
1491
|
+
MatrixInverseExtGradFunc MatrixInverseExtGradFuncObj;
|
|
1492
|
+
AdaptiveAvgPool2DGradExtGradFunc AdaptiveAvgPool2DGradExtGradFuncObj;
|
|
1493
|
+
TransposeExtViewGradFunc TransposeExtViewGradFuncObj;
|
|
1494
|
+
LogicalXorGradFunc LogicalXorGradFuncObj;
|
|
1495
|
+
DistCommIsendGradFunc DistCommIsendGradFuncObj;
|
|
1496
|
+
IndexAddExtGradFunc IndexAddExtGradFuncObj;
|
|
1497
|
+
ChunkGradFunc ChunkGradFuncObj;
|
|
1498
|
+
InplaceRemainderTensorScalarGradFunc InplaceRemainderTensorScalarGradFuncObj;
|
|
1327
1499
|
PowScalarTensorGradFunc PowScalarTensorGradFuncObj;
|
|
1328
|
-
|
|
1329
|
-
|
|
1330
|
-
|
|
1331
|
-
|
|
1332
|
-
|
|
1333
|
-
|
|
1334
|
-
|
|
1335
|
-
|
|
1336
|
-
Conv2DPaddingGradFunc Conv2DPaddingGradFuncObj;
|
|
1337
|
-
HSwishGradFunc HSwishGradFuncObj;
|
|
1338
|
-
InplaceFloorDivideGradFunc InplaceFloorDivideGradFuncObj;
|
|
1339
|
-
InplaceScatterAddGradFunc InplaceScatterAddGradFuncObj;
|
|
1500
|
+
DistCommScatterTensorGradFunc DistCommScatterTensorGradFuncObj;
|
|
1501
|
+
KLDivGradGradFunc KLDivGradGradFuncObj;
|
|
1502
|
+
DivGradFunc DivGradFuncObj;
|
|
1503
|
+
MedianExtGradFunc MedianExtGradFuncObj;
|
|
1504
|
+
SliceExtViewGradFunc SliceExtViewGradFuncObj;
|
|
1505
|
+
CeilGradFunc CeilGradFuncObj;
|
|
1506
|
+
ReflectionPad1DGradFunc ReflectionPad1DGradFuncObj;
|
|
1507
|
+
BatchNormGradExtGradFunc BatchNormGradExtGradFuncObj;
|
|
1340
1508
|
MSELossGradExtGradFunc MSELossGradExtGradFuncObj;
|
|
1341
|
-
|
|
1342
|
-
|
|
1343
|
-
|
|
1344
|
-
|
|
1345
|
-
|
|
1346
|
-
|
|
1347
|
-
|
|
1348
|
-
|
|
1509
|
+
SignGradFunc SignGradFuncObj;
|
|
1510
|
+
TransposeGradFunc TransposeGradFuncObj;
|
|
1511
|
+
HistcExtGradFunc HistcExtGradFuncObj;
|
|
1512
|
+
InplaceClampScalarGradFunc InplaceClampScalarGradFuncObj;
|
|
1513
|
+
GridSampler3DGradGradFunc GridSampler3DGradGradFuncObj;
|
|
1514
|
+
InplaceHardtanhGradFunc InplaceHardtanhGradFuncObj;
|
|
1515
|
+
IsCloseGradFunc IsCloseGradFuncObj;
|
|
1516
|
+
AvgPool3DExtGradFunc AvgPool3DExtGradFuncObj;
|
|
1517
|
+
UpsampleNearest1DGradGradFunc UpsampleNearest1DGradGradFuncObj;
|
|
1518
|
+
MaxDimGradFunc MaxDimGradFuncObj;
|
|
1519
|
+
AvgPool3DGradExtGradFunc AvgPool3DGradExtGradFuncObj;
|
|
1520
|
+
SubScalarGradFunc SubScalarGradFuncObj;
|
|
1521
|
+
FFNExtGradFunc FFNExtGradFuncObj;
|
|
1522
|
+
DropoutDoMaskExtGradFunc DropoutDoMaskExtGradFuncObj;
|
|
1523
|
+
InplaceRandomGradFunc InplaceRandomGradFuncObj;
|
|
1524
|
+
EqualExtGradFunc EqualExtGradFuncObj;
|
|
1525
|
+
MoeTokenPermuteGradGradFunc MoeTokenPermuteGradGradFuncObj;
|
|
1526
|
+
ReflectionPad1DGradGradFunc ReflectionPad1DGradGradFuncObj;
|
|
1527
|
+
DenseGradFunc DenseGradFuncObj;
|
|
1528
|
+
InplaceScatterValueReduceGradFunc InplaceScatterValueReduceGradFuncObj;
|
|
1529
|
+
ClampScalarGradFunc ClampScalarGradFuncObj;
|
|
1530
|
+
AllFiniteGradFunc AllFiniteGradFuncObj;
|
|
1531
|
+
InnerIndexGradFunc InnerIndexGradFuncObj;
|
|
1532
|
+
ReplicationPad2DGradGradFunc ReplicationPad2DGradGradFuncObj;
|
|
1533
|
+
DistCommReduceGradFunc DistCommReduceGradFuncObj;
|
|
1534
|
+
Log10GradFunc Log10GradFuncObj;
|
|
1535
|
+
ProdExtGradFunc ProdExtGradFuncObj;
|
|
1536
|
+
ArgMinExtGradFunc ArgMinExtGradFuncObj;
|
|
1537
|
+
ContiguousGradFunc ContiguousGradFuncObj;
|
|
1538
|
+
SliceGradFunc SliceGradFuncObj;
|
|
1539
|
+
UniqueConsecutiveGradFunc UniqueConsecutiveGradFuncObj;
|
|
1540
|
+
ConvolutionGradGradFunc ConvolutionGradGradFuncObj;
|
|
1541
|
+
DistCommAllGatherIntoTensorUnevenGradFunc DistCommAllGatherIntoTensorUnevenGradFuncObj;
|
|
1542
|
+
OneHotExtGradFunc OneHotExtGradFuncObj;
|
|
1543
|
+
OuterGradFunc OuterGradFuncObj;
|
|
1349
1544
|
MultiScaleDeformableAttnGradGradFunc MultiScaleDeformableAttnGradGradFuncObj;
|
|
1545
|
+
InplaceMulsGradFunc InplaceMulsGradFuncObj;
|
|
1546
|
+
CrossEntropyLossGradFunc CrossEntropyLossGradFuncObj;
|
|
1547
|
+
ReplicationPad1DGradGradFunc ReplicationPad1DGradGradFuncObj;
|
|
1548
|
+
UpsampleNearest2DGradFunc UpsampleNearest2DGradFuncObj;
|
|
1549
|
+
MatMulGradFunc MatMulGradFuncObj;
|
|
1550
|
+
InplaceBernoulliTensorGradFunc InplaceBernoulliTensorGradFuncObj;
|
|
1551
|
+
Unique2GradFunc Unique2GradFuncObj;
|
|
1552
|
+
UpsampleBilinear2DGradGradFunc UpsampleBilinear2DGradGradFuncObj;
|
|
1553
|
+
BincountExtGradFunc BincountExtGradFuncObj;
|
|
1554
|
+
ConvolutionStrGradFunc ConvolutionStrGradFuncObj;
|
|
1555
|
+
DistCommAllGatherGradFunc DistCommAllGatherGradFuncObj;
|
|
1556
|
+
IsNegInfGradFunc IsNegInfGradFuncObj;
|
|
1350
1557
|
GeluGradExtGradFunc GeluGradExtGradFuncObj;
|
|
1351
|
-
SearchSortedGradFunc SearchSortedGradFuncObj;
|
|
1352
|
-
ProdExtGradFunc ProdExtGradFuncObj;
|
|
1353
|
-
EqualExtGradFunc EqualExtGradFuncObj;
|
|
1354
|
-
MulGradFunc MulGradFuncObj;
|
|
1355
|
-
RepeatGradFunc RepeatGradFuncObj;
|
|
1356
1558
|
KLDivGradFunc KLDivGradFuncObj;
|
|
1357
|
-
|
|
1358
|
-
|
|
1359
|
-
|
|
1360
|
-
|
|
1361
|
-
|
|
1362
|
-
|
|
1363
|
-
XLogYScalarOtherGradFunc XLogYScalarOtherGradFuncObj;
|
|
1364
|
-
UpsampleNearest3DGradFunc UpsampleNearest3DGradFuncObj;
|
|
1365
|
-
NLLLoss2dGradFunc NLLLoss2dGradFuncObj;
|
|
1366
|
-
SquareGradFunc SquareGradFuncObj;
|
|
1367
|
-
IdentityGradFunc IdentityGradFuncObj;
|
|
1368
|
-
GatherDGradV2GradFunc GatherDGradV2GradFuncObj;
|
|
1369
|
-
LessGradFunc LessGradFuncObj;
|
|
1370
|
-
EluGradFunc EluGradFuncObj;
|
|
1371
|
-
GatherDGradFunc GatherDGradFuncObj;
|
|
1372
|
-
SplitTensorGradFunc SplitTensorGradFuncObj;
|
|
1373
|
-
AvgPool2DGradFunc AvgPool2DGradFuncObj;
|
|
1374
|
-
LinSpaceExtGradFunc LinSpaceExtGradFuncObj;
|
|
1375
|
-
ReduceAllGradFunc ReduceAllGradFuncObj;
|
|
1376
|
-
RoundGradFunc RoundGradFuncObj;
|
|
1377
|
-
FFNExtGradFunc FFNExtGradFuncObj;
|
|
1378
|
-
MultinomialExtGradFunc MultinomialExtGradFuncObj;
|
|
1379
|
-
AtanhGradFunc AtanhGradFuncObj;
|
|
1380
|
-
SubGradFunc SubGradFuncObj;
|
|
1381
|
-
InplaceZeroGradFunc InplaceZeroGradFuncObj;
|
|
1382
|
-
InplaceLogGradFunc InplaceLogGradFuncObj;
|
|
1383
|
-
ReflectionPad3DGradGradFunc ReflectionPad3DGradGradFuncObj;
|
|
1384
|
-
AvgPool3DGradExtGradFunc AvgPool3DGradExtGradFuncObj;
|
|
1385
|
-
TriangularSolveGradFunc TriangularSolveGradFuncObj;
|
|
1386
|
-
CosGradFunc CosGradFuncObj;
|
|
1387
|
-
LogAddExpGradFunc LogAddExpGradFuncObj;
|
|
1388
|
-
AcoshExtGradFunc AcoshExtGradFuncObj;
|
|
1559
|
+
AdaptiveAvgPool3DExtGradFunc AdaptiveAvgPool3DExtGradFuncObj;
|
|
1560
|
+
IndexGradFunc IndexGradFuncObj;
|
|
1561
|
+
SinGradFunc SinGradFuncObj;
|
|
1562
|
+
BitwiseAndScalarGradFunc BitwiseAndScalarGradFuncObj;
|
|
1563
|
+
UpsampleLinear1DGradFunc UpsampleLinear1DGradFuncObj;
|
|
1564
|
+
UpsampleNearest2DGradGradFunc UpsampleNearest2DGradGradFuncObj;
|
|
1389
1565
|
LogicalNotGradFunc LogicalNotGradFuncObj;
|
|
1390
|
-
|
|
1566
|
+
InplaceUniformGradFunc InplaceUniformGradFuncObj;
|
|
1567
|
+
HShrinkGradGradFunc HShrinkGradGradFuncObj;
|
|
1568
|
+
PromptFlashAttentionGradFunc PromptFlashAttentionGradFuncObj;
|
|
1569
|
+
ClampTensorGradFunc ClampTensorGradFuncObj;
|
|
1570
|
+
InplaceNormalGradFunc InplaceNormalGradFuncObj;
|
|
1571
|
+
AdaptiveMaxPool1DGradFunc AdaptiveMaxPool1DGradFuncObj;
|
|
1572
|
+
LayerNormExtGradFunc LayerNormExtGradFuncObj;
|
|
1573
|
+
UniformExtGradFunc UniformExtGradFuncObj;
|
|
1574
|
+
ErfGradFunc ErfGradFuncObj;
|
|
1575
|
+
ArgSortGradFunc ArgSortGradFuncObj;
|
|
1576
|
+
DistCommAllToAllVSingleGradFunc DistCommAllToAllVSingleGradFuncObj;
|
|
1577
|
+
InplaceScatterValueGradFunc InplaceScatterValueGradFuncObj;
|
|
1391
1578
|
DropoutGradExtGradFunc DropoutGradExtGradFuncObj;
|
|
1392
|
-
|
|
1579
|
+
GatherDGradFunc GatherDGradFuncObj;
|
|
1580
|
+
BinaryCrossEntropyGradGradFunc BinaryCrossEntropyGradGradFuncObj;
|
|
1581
|
+
OnesLikeExtGradFunc OnesLikeExtGradFuncObj;
|
|
1582
|
+
DistCommAllReduceGradFunc DistCommAllReduceGradFuncObj;
|
|
1583
|
+
MoeTokenUnpermuteGradGradFunc MoeTokenUnpermuteGradGradFuncObj;
|
|
1584
|
+
NewOnesGradFunc NewOnesGradFuncObj;
|
|
1585
|
+
OnesGradFunc OnesGradFuncObj;
|
|
1586
|
+
ReflectionPad2DGradGradFunc ReflectionPad2DGradGradFuncObj;
|
|
1393
1587
|
InplaceTanhGradFunc InplaceTanhGradFuncObj;
|
|
1394
|
-
|
|
1588
|
+
NanToNumGradFunc NanToNumGradFuncObj;
|
|
1589
|
+
SmoothL1LossGradFunc SmoothL1LossGradFuncObj;
|
|
1590
|
+
VarGradFunc VarGradFuncObj;
|
|
1591
|
+
MatMulExtGradFunc MatMulExtGradFuncObj;
|
|
1592
|
+
SoftmaxGradFunc SoftmaxGradFuncObj;
|
|
1593
|
+
MSELossExtGradFunc MSELossExtGradFuncObj;
|
|
1594
|
+
EyeGradFunc EyeGradFuncObj;
|
|
1595
|
+
NLLLossGradFunc NLLLossGradFuncObj;
|
|
1596
|
+
UpsampleBicubic2DGradGradFunc UpsampleBicubic2DGradGradFuncObj;
|
|
1597
|
+
TakeGradFunc TakeGradFuncObj;
|
|
1598
|
+
EluGradFunc EluGradFuncObj;
|
|
1599
|
+
SigmoidGradFunc SigmoidGradFuncObj;
|
|
1600
|
+
CummaxGradFunc CummaxGradFuncObj;
|
|
1601
|
+
ReciprocalGradFunc ReciprocalGradFuncObj;
|
|
1602
|
+
SubExtGradFunc SubExtGradFuncObj;
|
|
1603
|
+
IndexFillTensorGradFunc IndexFillTensorGradFuncObj;
|
|
1604
|
+
GeneratorGradFunc GeneratorGradFuncObj;
|
|
1605
|
+
NegGradFunc NegGradFuncObj;
|
|
1606
|
+
SliceExtGradFunc SliceExtGradFuncObj;
|
|
1607
|
+
SeluGradGradFunc SeluGradGradFuncObj;
|
|
1608
|
+
MaskedSelectGradGradFunc MaskedSelectGradGradFuncObj;
|
|
1609
|
+
PowGradFunc PowGradFuncObj;
|
|
1610
|
+
InplaceCopyGradFunc InplaceCopyGradFuncObj;
|
|
1611
|
+
SwigluGradGradFunc SwigluGradGradFuncObj;
|
|
1612
|
+
NeScalarGradFunc NeScalarGradFuncObj;
|
|
1613
|
+
Conv1DPaddingGradFunc Conv1DPaddingGradFuncObj;
|
|
1614
|
+
CustomExtGradFunc CustomExtGradFuncObj;
|
|
1615
|
+
NormalTensorFloatGradFunc NormalTensorFloatGradFuncObj;
|
|
1616
|
+
AdamWGradFunc AdamWGradFuncObj;
|
|
1617
|
+
MaskedScatterGradFunc MaskedScatterGradFuncObj;
|
|
1618
|
+
MultinomialExtGradFunc MultinomialExtGradFuncObj;
|
|
1619
|
+
SplitTensorViewGradFunc SplitTensorViewGradFuncObj;
|
|
1620
|
+
TensorScatterElementsGradFunc TensorScatterElementsGradFuncObj;
|
|
1621
|
+
MmGradFunc MmGradFuncObj;
|
|
1622
|
+
UpsampleTrilinear3DGradFunc UpsampleTrilinear3DGradFuncObj;
|
|
1395
1623
|
BernoulliExtGradFunc BernoulliExtGradFuncObj;
|
|
1624
|
+
HSigmoidGradFunc HSigmoidGradFuncObj;
|
|
1625
|
+
TypeAsGradFunc TypeAsGradFuncObj;
|
|
1396
1626
|
ReluGradGradFunc ReluGradGradFuncObj;
|
|
1397
|
-
|
|
1398
|
-
|
|
1399
|
-
BroadcastToGradFunc BroadcastToGradFuncObj;
|
|
1400
|
-
RandLikeExtGradFunc RandLikeExtGradFuncObj;
|
|
1401
|
-
Expm1GradFunc Expm1GradFuncObj;
|
|
1402
|
-
InplaceNormalGradFunc InplaceNormalGradFuncObj;
|
|
1403
|
-
BitwiseOrTensorGradFunc BitwiseOrTensorGradFuncObj;
|
|
1404
|
-
AcosExtGradFunc AcosExtGradFuncObj;
|
|
1405
|
-
GreaterEqualGradFunc GreaterEqualGradFuncObj;
|
|
1406
|
-
TruncGradFunc TruncGradFuncObj;
|
|
1407
|
-
ReduceMinGradFunc ReduceMinGradFuncObj;
|
|
1627
|
+
BatchNormElemtGradGradFunc BatchNormElemtGradGradFuncObj;
|
|
1628
|
+
Conv3DPaddingGradFunc Conv3DPaddingGradFuncObj;
|
|
1408
1629
|
GroupNormGradGradFunc GroupNormGradGradFuncObj;
|
|
1409
|
-
|
|
1410
|
-
UpsampleBilinear2DGradGradFunc UpsampleBilinear2DGradGradFuncObj;
|
|
1411
|
-
DropoutExtGradFunc DropoutExtGradFuncObj;
|
|
1412
|
-
BitwiseXorScalarGradFunc BitwiseXorScalarGradFuncObj;
|
|
1413
|
-
LogSigmoidGradGradFunc LogSigmoidGradGradFuncObj;
|
|
1414
|
-
DiagExtGradFunc DiagExtGradFuncObj;
|
|
1415
|
-
RandIntLikeGradFunc RandIntLikeGradFuncObj;
|
|
1416
|
-
LerpScalarGradFunc LerpScalarGradFuncObj;
|
|
1417
|
-
GmmBackwardGradFunc GmmBackwardGradFuncObj;
|
|
1418
|
-
RemainderScalarTensorGradFunc RemainderScalarTensorGradFuncObj;
|
|
1419
|
-
LogicalOrGradFunc LogicalOrGradFuncObj;
|
|
1420
|
-
AvgPool2DGradGradFunc AvgPool2DGradGradFuncObj;
|
|
1421
|
-
MaskedSelectGradFunc MaskedSelectGradFuncObj;
|
|
1422
|
-
ReLUGradFunc ReLUGradFuncObj;
|
|
1423
|
-
AdaptiveMaxPool2DGradFunc AdaptiveMaxPool2DGradFuncObj;
|
|
1424
|
-
ReplicationPad3DGradFunc ReplicationPad3DGradFuncObj;
|
|
1425
|
-
SwigluGradGradFunc SwigluGradGradFuncObj;
|
|
1426
|
-
InplaceSubExtGradFunc InplaceSubExtGradFuncObj;
|
|
1630
|
+
AllGatherMatmulGradFunc AllGatherMatmulGradFuncObj;
|
|
1427
1631
|
BitwiseXorTensorGradFunc BitwiseXorTensorGradFuncObj;
|
|
1428
|
-
|
|
1429
|
-
|
|
1430
|
-
|
|
1431
|
-
|
|
1432
|
-
MoeTokenPermuteGradFunc MoeTokenPermuteGradFuncObj;
|
|
1433
|
-
MishGradExtGradFunc MishGradExtGradFuncObj;
|
|
1434
|
-
AddRmsNormGradFunc AddRmsNormGradFuncObj;
|
|
1435
|
-
MatMulGradFunc MatMulGradFuncObj;
|
|
1436
|
-
UniqueConsecutiveGradFunc UniqueConsecutiveGradFuncObj;
|
|
1437
|
-
AddLayerNormGradGradFunc AddLayerNormGradGradFuncObj;
|
|
1438
|
-
EyeGradFunc EyeGradFuncObj;
|
|
1439
|
-
PReLUGradGradFunc PReLUGradGradFuncObj;
|
|
1440
|
-
StdMeanGradFunc StdMeanGradFuncObj;
|
|
1441
|
-
GroupNormGradFunc GroupNormGradFuncObj;
|
|
1442
|
-
FillTensorGradFunc FillTensorGradFuncObj;
|
|
1443
|
-
Log2GradFunc Log2GradFuncObj;
|
|
1444
|
-
ScatterGradFunc ScatterGradFuncObj;
|
|
1445
|
-
CastGradFunc CastGradFuncObj;
|
|
1446
|
-
BatchMatMulGradFunc BatchMatMulGradFuncObj;
|
|
1447
|
-
NonZeroExtGradFunc NonZeroExtGradFuncObj;
|
|
1448
|
-
EqualGradFunc EqualGradFuncObj;
|
|
1449
|
-
InplaceScatterValueGradFunc InplaceScatterValueGradFuncObj;
|
|
1632
|
+
SearchSortedGradFunc SearchSortedGradFuncObj;
|
|
1633
|
+
XLogYScalarSelfGradFunc XLogYScalarSelfGradFuncObj;
|
|
1634
|
+
TopkExtGradFunc TopkExtGradFuncObj;
|
|
1635
|
+
IndexFillScalarGradFunc IndexFillScalarGradFuncObj;
|
|
1450
1636
|
SelectGradFunc SelectGradFuncObj;
|
|
1451
|
-
|
|
1452
|
-
|
|
1637
|
+
InplaceScatterSrcReduceGradFunc InplaceScatterSrcReduceGradFuncObj;
|
|
1638
|
+
TileGradFunc TileGradFuncObj;
|
|
1639
|
+
AsinhExtGradFunc AsinhExtGradFuncObj;
|
|
1640
|
+
InplaceBernoulliScalarGradFunc InplaceBernoulliScalarGradFuncObj;
|
|
1641
|
+
GeLUGradGradFunc GeLUGradGradFuncObj;
|
|
1453
1642
|
GroupedMatmulV2GradFunc GroupedMatmulV2GradFuncObj;
|
|
1454
|
-
|
|
1643
|
+
GroupedMatmulGradFunc GroupedMatmulGradFuncObj;
|
|
1644
|
+
QuantV2GradFunc QuantV2GradFuncObj;
|
|
1455
1645
|
GroupedMatmulV4GradFunc GroupedMatmulV4GradFuncObj;
|
|
1456
|
-
|
|
1646
|
+
DynamicQuantExtGradFunc DynamicQuantExtGradFuncObj;
|
|
1457
1647
|
MoeFinalizeRoutingGradFunc MoeFinalizeRoutingGradFuncObj;
|
|
1648
|
+
AddRmsNormQuantV2GradFunc AddRmsNormQuantV2GradFuncObj;
|
|
1649
|
+
QuantBatchMatmulGradFunc QuantBatchMatmulGradFuncObj;
|
|
1650
|
+
KVCacheScatterUpdateGradFunc KVCacheScatterUpdateGradFuncObj;
|
|
1458
1651
|
MatmulAllReduceAddRmsNormGradFunc MatmulAllReduceAddRmsNormGradFuncObj;
|
|
1459
|
-
|
|
1652
|
+
QuantMatmulGradFunc QuantMatmulGradFuncObj;
|
|
1460
1653
|
MoeGatingTopKSoftmaxGradFunc MoeGatingTopKSoftmaxGradFuncObj;
|
|
1461
|
-
|
|
1654
|
+
WeightQuantBatchMatmulGradFunc WeightQuantBatchMatmulGradFuncObj;
|
|
1655
|
+
MoeInitRoutingV2GradFunc MoeInitRoutingV2GradFuncObj;
|
|
1462
1656
|
MoeComputeExpertTokensGradFunc MoeComputeExpertTokensGradFuncObj;
|
|
1463
|
-
QuantBatchMatmulGradFunc QuantBatchMatmulGradFuncObj;
|
|
1464
|
-
MoeInitRoutingGradFunc MoeInitRoutingGradFuncObj;
|
|
1465
1657
|
FusedInferAttentionScoreGradFunc FusedInferAttentionScoreGradFuncObj;
|
|
1466
|
-
|
|
1467
|
-
|
|
1468
|
-
|
|
1658
|
+
MoeInitRoutingQuantV2GradFunc MoeInitRoutingQuantV2GradFuncObj;
|
|
1659
|
+
MoeInitRoutingGradFunc MoeInitRoutingGradFuncObj;
|
|
1660
|
+
GmmV2BackwardFusionGradFunc GmmV2BackwardFusionGradFuncObj;
|
|
1661
|
+
AnyGradFunc AnyGradFuncObj;
|
|
1469
1662
|
PixelShuffleGradFunc PixelShuffleGradFuncObj;
|
|
1663
|
+
GmmV2BackwardGradFunc GmmV2BackwardGradFuncObj;
|
|
1664
|
+
GmmGradFunc GmmGradFuncObj;
|
|
1665
|
+
EinsumExtGradFunc EinsumExtGradFuncObj;
|
|
1666
|
+
FuncMaxPool2DGradFunc FuncMaxPool2DGradFuncObj;
|
|
1667
|
+
MoeTokenUnpermuteGradFunc MoeTokenUnpermuteGradFuncObj;
|
|
1668
|
+
GmmV2GradFunc GmmV2GradFuncObj;
|
|
1669
|
+
FuncDropoutExtGradFunc FuncDropoutExtGradFuncObj;
|
|
1670
|
+
InplaceExponentialGradFunc InplaceExponentialGradFuncObj;
|
|
1671
|
+
GmmBackwardGradFunc GmmBackwardGradFuncObj;
|
|
1672
|
+
GmmBackwardFusionGradFunc GmmBackwardFusionGradFuncObj;
|
|
1673
|
+
AnyExtGradFunc AnyExtGradFuncObj;
|
|
1470
1674
|
};
|
|
1471
1675
|
} // namespace pyboost
|
|
1472
1676
|
} // namespace kernel
|