mindspore 2.6.0rc1__cp39-none-any.whl → 2.7.0__cp39-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of mindspore might be problematic. Click here for more details.
- mindspore/.commit_id +1 -1
- mindspore/Third_Party_Open_Source_Software_Notice +1290 -0
- mindspore/__init__.py +2 -2
- mindspore/_c_dataengine.cpython-39-aarch64-linux-gnu.so +0 -0
- mindspore/_c_expression.cpython-39-aarch64-linux-gnu.so +0 -0
- mindspore/_c_mindrecord.cpython-39-aarch64-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 +3227 -3102
- 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 +135 -130
- 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 +135 -131
- 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 +1588 -1384
- 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 +501 -457
- 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 +498 -483
- mindspore/include/mindspore/ccsrc/pynative/op_function/comm_handle_py.h +2 -0
- mindspore/include/mindspore/ccsrc/pynative/op_function/converter.h +11 -0
- mindspore/include/mindspore/ccsrc/pynative/op_function/customize/direct_ops.h +2 -12
- mindspore/include/mindspore/ccsrc/pynative/predict_out_type_map.h +3 -0
- mindspore/include/mindspore/ccsrc/pynative/pynative_execute.h +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 +3255 -2915
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_def.h +815 -760
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_a.h +60 -58
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_b.h +26 -25
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_c.h +35 -31
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_d.h +29 -25
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_e.h +26 -24
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_f.h +22 -19
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_g.h +23 -19
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_h.h +7 -7
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_i.h +73 -65
- 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 +31 -31
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_m.h +47 -39
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_n.h +18 -16
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_o.h +3 -3
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_p.h +6 -7
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_q.h +5 -2
- 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 +67 -63
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_t.h +22 -20
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_name_u.h +14 -14
- 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 +60 -58
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_b.h +26 -25
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_c.h +35 -31
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_d.h +29 -25
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_e.h +26 -24
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_f.h +22 -19
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_g.h +23 -19
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_h.h +7 -7
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_i.h +73 -65
- 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 +31 -31
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_m.h +47 -39
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_n.h +18 -16
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_o.h +3 -3
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_p.h +6 -7
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_q.h +5 -2
- 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 +67 -63
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_t.h +22 -20
- mindspore/include/mindspore/ops/op_def/auto_generate/gen_ops_primitive_u.h +14 -14
- 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/libicuuc.so.74 +0 -0
- mindspore/lib/libmindspore_backend_common.so +0 -0
- mindspore/lib/libmindspore_backend_manager.so +0 -0
- mindspore/lib/libmindspore_common.so +0 -0
- mindspore/lib/libmindspore_core.so +0 -0
- mindspore/lib/libmindspore_dump.so +0 -0
- mindspore/lib/libmindspore_extension.so +0 -0
- mindspore/lib/libmindspore_frontend.so +0 -0
- mindspore/lib/libmindspore_ge_backend.so +0 -0
- mindspore/lib/libmindspore_glog.so.0 +0 -0
- mindspore/lib/libmindspore_gpr.so.15 +0 -0
- mindspore/lib/libmindspore_grpc++.so.1 +0 -0
- mindspore/lib/libmindspore_grpc.so.15 +0 -0
- mindspore/lib/libmindspore_memory_pool.so +0 -0
- mindspore/lib/libmindspore_ms_backend.so +0 -0
- mindspore/lib/libmindspore_ops.so +0 -0
- mindspore/lib/libmindspore_ops_kernel_common.so +0 -0
- mindspore/lib/libmindspore_profiler.so +0 -0
- mindspore/lib/libmindspore_pyboost.so +0 -0
- mindspore/lib/libmindspore_pynative.so +0 -0
- mindspore/lib/libmindspore_res_manager.so +0 -0
- mindspore/lib/libmindspore_runtime_pipeline.so +0 -0
- mindspore/lib/libmpi_adapter.so +0 -0
- mindspore/lib/libmpi_collective.so +0 -0
- mindspore/lib/libnnacl.so +0 -0
- mindspore/lib/libopencv_core.so.4.5 +0 -0
- mindspore/lib/libopencv_imgcodecs.so.4.5 +0 -0
- mindspore/lib/libopencv_imgproc.so.4.5 +0 -0
- mindspore/lib/libps_cache.so +0 -0
- mindspore/lib/libswresample.so.4 +0 -0
- mindspore/lib/libswscale.so.6 +0 -0
- mindspore/lib/plugin/ascend/custom_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/aarch64/libcust_opmaster_rt2.0.so +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/op_tiling/liboptiling.so +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_proto/inc/op_proto.h +0 -22
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_proto/lib/linux/aarch64/libcust_opsproto_rt2.0.so +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/version.info +1 -1
- mindspore/lib/plugin/ascend/custom_compiler/OWNERS +1 -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/libmindspore_ascend.so.2 +0 -0
- mindspore/lib/plugin/libmindspore_ops_ascend.so +0 -0
- mindspore/lib/plugin/libmindspore_ops_host.so +0 -0
- mindspore/mindrecord/tools/cifar10.py +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 +461 -420
- mindspore/ops/auto_generate/gen_extend_func.py +1250 -1368
- mindspore/ops/auto_generate/gen_ops_def.py +6777 -6097
- mindspore/ops/auto_generate/gen_ops_prim.py +16713 -15489
- mindspore/ops/auto_generate/pyboost_inner_prim.py +365 -335
- 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 +1276 -700
- 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 +12015 -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/aarch64/libcust_opmaster_rt2.0.so +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_impl/ai_core/tbe/op_tiling/liboptiling.so +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_proto/inc/op_proto.h +0 -33
- mindspore/lib/plugin/ascend/custom_ascendc_910/op_proto/lib/linux/aarch64/libcust_opsproto_rt2.0.so +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910/version.info +0 -1
- mindspore/lib/plugin/ascend/custom_ascendc_910b/framework/npu_supported_ops.json +0 -14
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_api/include/aclnn_decoder_kv_cache.h +0 -59
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_api/include/aclnn_prompt_kv_cache.h +0 -59
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/custom_ascendc_910b_impl/dynamic/decoder_kv_cache.cpp +0 -192
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/custom_ascendc_910b_impl/dynamic/decoder_kv_cache.py +0 -215
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/custom_ascendc_910b_impl/dynamic/prompt_kv_cache.cpp +0 -274
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/custom_ascendc_910b_impl/dynamic/prompt_kv_cache.py +0 -215
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_0d5520cc587ad44ce634bf3fbcffc272.json +0 -158
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_0d5520cc587ad44ce634bf3fbcffc272.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_20390d30b3c4c0d23167ccca6c030c2b.json +0 -158
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_20390d30b3c4c0d23167ccca6c030c2b.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_2d151f0b1d2db51faa2968d5b67544e2.json +0 -158
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_2d151f0b1d2db51faa2968d5b67544e2.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_561690ec17cc1def3d2fcf68c1b07b56.json +0 -158
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_561690ec17cc1def3d2fcf68c1b07b56.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_570f9aaa99e5e773b3dd0a33784363f4.json +0 -158
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_570f9aaa99e5e773b3dd0a33784363f4.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_59668a0f0764afb98fda8ab9e84126f1.json +0 -158
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_59668a0f0764afb98fda8ab9e84126f1.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_91d9833e4792b70b670e4e2b916abd86.json +0 -158
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_91d9833e4792b70b670e4e2b916abd86.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_c74cdc5fef094383401856f8519504af.json +0 -158
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/decoder_kv_cache/DecoderKvCache_c74cdc5fef094383401856f8519504af.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_0515c7b1a4cd614449e38c5e9a7e3f8d.json +0 -167
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_0515c7b1a4cd614449e38c5e9a7e3f8d.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_09f22d898d6358c91e7c4fc48bac48e7.json +0 -167
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_09f22d898d6358c91e7c4fc48bac48e7.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_0cb9a6f894b925250227136e5aab7061.json +0 -167
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_0cb9a6f894b925250227136e5aab7061.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_2fa8702ffd7ca85e9e194f62644415d5.json +0 -167
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_2fa8702ffd7ca85e9e194f62644415d5.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_570b62f187dfd439b64613d881deedb7.json +0 -167
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_570b62f187dfd439b64613d881deedb7.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_585218c11411ff84709b9e725b66c435.json +0 -167
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_585218c11411ff84709b9e725b66c435.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_5c9365ccde170b358c5b126d69dae13e.json +0 -167
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_5c9365ccde170b358c5b126d69dae13e.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_6d97c45b7c43bc16fcff8baa5dacac4e.json +0 -167
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend310p/prompt_kv_cache/PromptKvCache_6d97c45b7c43bc16fcff8baa5dacac4e.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_0d5520cc587ad44ce634bf3fbcffc272.json +0 -156
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_0d5520cc587ad44ce634bf3fbcffc272.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_20390d30b3c4c0d23167ccca6c030c2b.json +0 -156
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_20390d30b3c4c0d23167ccca6c030c2b.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_2d151f0b1d2db51faa2968d5b67544e2.json +0 -156
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_2d151f0b1d2db51faa2968d5b67544e2.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_561690ec17cc1def3d2fcf68c1b07b56.json +0 -156
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_561690ec17cc1def3d2fcf68c1b07b56.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_570f9aaa99e5e773b3dd0a33784363f4.json +0 -156
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_570f9aaa99e5e773b3dd0a33784363f4.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_59668a0f0764afb98fda8ab9e84126f1.json +0 -156
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_59668a0f0764afb98fda8ab9e84126f1.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_91d9833e4792b70b670e4e2b916abd86.json +0 -156
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_91d9833e4792b70b670e4e2b916abd86.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_c74cdc5fef094383401856f8519504af.json +0 -156
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/decoder_kv_cache/DecoderKvCache_c74cdc5fef094383401856f8519504af.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_0515c7b1a4cd614449e38c5e9a7e3f8d.json +0 -165
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_0515c7b1a4cd614449e38c5e9a7e3f8d.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_09f22d898d6358c91e7c4fc48bac48e7.json +0 -165
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_09f22d898d6358c91e7c4fc48bac48e7.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_0cb9a6f894b925250227136e5aab7061.json +0 -165
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_0cb9a6f894b925250227136e5aab7061.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_2fa8702ffd7ca85e9e194f62644415d5.json +0 -165
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_2fa8702ffd7ca85e9e194f62644415d5.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_570b62f187dfd439b64613d881deedb7.json +0 -165
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_570b62f187dfd439b64613d881deedb7.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_585218c11411ff84709b9e725b66c435.json +0 -165
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_585218c11411ff84709b9e725b66c435.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_5c9365ccde170b358c5b126d69dae13e.json +0 -165
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_5c9365ccde170b358c5b126d69dae13e.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_6d97c45b7c43bc16fcff8baa5dacac4e.json +0 -165
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910_93/prompt_kv_cache/PromptKvCache_6d97c45b7c43bc16fcff8baa5dacac4e.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_0d5520cc587ad44ce634bf3fbcffc272.json +0 -156
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_0d5520cc587ad44ce634bf3fbcffc272.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_20390d30b3c4c0d23167ccca6c030c2b.json +0 -156
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_20390d30b3c4c0d23167ccca6c030c2b.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_2d151f0b1d2db51faa2968d5b67544e2.json +0 -156
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_2d151f0b1d2db51faa2968d5b67544e2.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_561690ec17cc1def3d2fcf68c1b07b56.json +0 -156
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_561690ec17cc1def3d2fcf68c1b07b56.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_570f9aaa99e5e773b3dd0a33784363f4.json +0 -156
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_570f9aaa99e5e773b3dd0a33784363f4.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_59668a0f0764afb98fda8ab9e84126f1.json +0 -156
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_59668a0f0764afb98fda8ab9e84126f1.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_91d9833e4792b70b670e4e2b916abd86.json +0 -156
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_91d9833e4792b70b670e4e2b916abd86.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_c74cdc5fef094383401856f8519504af.json +0 -156
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/decoder_kv_cache/DecoderKvCache_c74cdc5fef094383401856f8519504af.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_0515c7b1a4cd614449e38c5e9a7e3f8d.json +0 -165
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_0515c7b1a4cd614449e38c5e9a7e3f8d.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_09f22d898d6358c91e7c4fc48bac48e7.json +0 -165
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_09f22d898d6358c91e7c4fc48bac48e7.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_0cb9a6f894b925250227136e5aab7061.json +0 -165
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_0cb9a6f894b925250227136e5aab7061.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_2fa8702ffd7ca85e9e194f62644415d5.json +0 -165
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_2fa8702ffd7ca85e9e194f62644415d5.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_570b62f187dfd439b64613d881deedb7.json +0 -165
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_570b62f187dfd439b64613d881deedb7.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_585218c11411ff84709b9e725b66c435.json +0 -165
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_585218c11411ff84709b9e725b66c435.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_5c9365ccde170b358c5b126d69dae13e.json +0 -165
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_5c9365ccde170b358c5b126d69dae13e.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_6d97c45b7c43bc16fcff8baa5dacac4e.json +0 -165
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/ascend910b/prompt_kv_cache/PromptKvCache_6d97c45b7c43bc16fcff8baa5dacac4e.o +0 -0
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/config/ascend310p/decoder_kv_cache.json +0 -892
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/config/ascend310p/prompt_kv_cache.json +0 -892
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/config/ascend910_93/decoder_kv_cache.json +0 -892
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/config/ascend910_93/prompt_kv_cache.json +0 -892
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/config/ascend910b/decoder_kv_cache.json +0 -892
- mindspore/lib/plugin/ascend/custom_ascendc_910b/op_impl/ai_core/tbe/kernel/config/ascend910b/prompt_kv_cache.json +0 -892
- mindspore/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 -12175
- 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
|
@@ -1,1019 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* Copyright 2024 Huawei Technologies Co., Ltd
|
|
3
|
-
*
|
|
4
|
-
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
5
|
-
* you may not use this file except in compliance with the License.
|
|
6
|
-
* You may obtain a copy of the License at
|
|
7
|
-
*
|
|
8
|
-
* http://www.apache.org/licenses/LICENSE-2.0
|
|
9
|
-
*
|
|
10
|
-
* Unless required by applicable law or agreed to in writing, software
|
|
11
|
-
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
12
|
-
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
13
|
-
* See the License for the specific language governing permissions and
|
|
14
|
-
* limitations under the License.
|
|
15
|
-
*/
|
|
16
|
-
|
|
17
|
-
#include "pynative/pynative_utils.h"
|
|
18
|
-
#include "pybind11/pybind11.h"
|
|
19
|
-
|
|
20
|
-
namespace py = pybind11;
|
|
21
|
-
namespace mindspore::pynative {
|
|
22
|
-
//primitive to pyboost op
|
|
23
|
-
py::object PYNATIVE_EXPORT Pyboost_GroupNorm_Base(const PrimitivePtr &prim, const py::list &args);
|
|
24
|
-
py::object PYNATIVE_EXPORT Pyboost_GreaterEqualScalar_Base(const PrimitivePtr &prim, const py::list &args);
|
|
25
|
-
py::object PYNATIVE_EXPORT Pyboost_DistCommAllGatherIntoTensor_Base(const PrimitivePtr &prim, const py::list &args);
|
|
26
|
-
py::object PYNATIVE_EXPORT Pyboost_BatchNormGatherStatsWithCounts_Base(const PrimitivePtr &prim, const py::list &args);
|
|
27
|
-
py::object PYNATIVE_EXPORT Pyboost_ProdExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
28
|
-
py::object PYNATIVE_EXPORT Pyboost_NormalTensorFloat_Base(const PrimitivePtr &prim, const py::list &args);
|
|
29
|
-
py::object PYNATIVE_EXPORT Pyboost_Scatter_Base(const PrimitivePtr &prim, const py::list &args);
|
|
30
|
-
py::object PYNATIVE_EXPORT Pyboost_ReduceAny_Base(const PrimitivePtr &prim, const py::list &args);
|
|
31
|
-
py::object PYNATIVE_EXPORT Pyboost_Neg_Base(const PrimitivePtr &prim, const py::list &args);
|
|
32
|
-
py::object PYNATIVE_EXPORT Pyboost_Dense_Base(const PrimitivePtr &prim, const py::list &args);
|
|
33
|
-
py::object PYNATIVE_EXPORT Pyboost_UpsampleBicubic2D_Base(const PrimitivePtr &prim, const py::list &args);
|
|
34
|
-
py::object PYNATIVE_EXPORT Pyboost_RmsNormGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
35
|
-
py::object PYNATIVE_EXPORT Pyboost_Sqrt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
36
|
-
py::object PYNATIVE_EXPORT Pyboost_Tan_Base(const PrimitivePtr &prim, const py::list &args);
|
|
37
|
-
py::object PYNATIVE_EXPORT Pyboost_RandnLike_Base(const PrimitivePtr &prim, const py::list &args);
|
|
38
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceScatterSrc_Base(const PrimitivePtr &prim, const py::list &args);
|
|
39
|
-
py::object PYNATIVE_EXPORT Pyboost_BitwiseOrTensor_Base(const PrimitivePtr &prim, const py::list &args);
|
|
40
|
-
py::object PYNATIVE_EXPORT Pyboost_ReflectionPad3D_Base(const PrimitivePtr &prim, const py::list &args);
|
|
41
|
-
py::object PYNATIVE_EXPORT Pyboost_BernoulliExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
42
|
-
py::object PYNATIVE_EXPORT Pyboost_Convolution_Base(const PrimitivePtr &prim, const py::list &args);
|
|
43
|
-
py::object PYNATIVE_EXPORT Pyboost_Cummax_Base(const PrimitivePtr &prim, const py::list &args);
|
|
44
|
-
py::object PYNATIVE_EXPORT Pyboost_ConvolutionStr_Base(const PrimitivePtr &prim, const py::list &args);
|
|
45
|
-
py::object PYNATIVE_EXPORT Pyboost_GeLUGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
46
|
-
py::object PYNATIVE_EXPORT Pyboost_TypeAs_Base(const PrimitivePtr &prim, const py::list &args);
|
|
47
|
-
py::object PYNATIVE_EXPORT Pyboost_MoeTokenPermuteGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
48
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceAddsExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
49
|
-
py::object PYNATIVE_EXPORT Pyboost_UpsampleBicubic2DGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
50
|
-
py::object PYNATIVE_EXPORT Pyboost_ZerosLikeExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
51
|
-
py::object PYNATIVE_EXPORT Pyboost_ReduceAll_Base(const PrimitivePtr &prim, const py::list &args);
|
|
52
|
-
py::object PYNATIVE_EXPORT Pyboost_Round_Base(const PrimitivePtr &prim, const py::list &args);
|
|
53
|
-
py::object PYNATIVE_EXPORT Pyboost_SilentCheckV3_Base(const PrimitivePtr &prim, const py::list &args);
|
|
54
|
-
py::object PYNATIVE_EXPORT Pyboost_Mm_Base(const PrimitivePtr &prim, const py::list &args);
|
|
55
|
-
py::object PYNATIVE_EXPORT Pyboost_AdaptiveMaxPool2D_Base(const PrimitivePtr &prim, const py::list &args);
|
|
56
|
-
py::object PYNATIVE_EXPORT Pyboost_DistCommScatter_Base(const PrimitivePtr &prim, const py::list &args);
|
|
57
|
-
py::object PYNATIVE_EXPORT Pyboost_DropoutExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
58
|
-
py::object PYNATIVE_EXPORT Pyboost_MinDim_Base(const PrimitivePtr &prim, const py::list &args);
|
|
59
|
-
py::object PYNATIVE_EXPORT Pyboost_NanToNum_Base(const PrimitivePtr &prim, const py::list &args);
|
|
60
|
-
py::object PYNATIVE_EXPORT Pyboost_DistCommBatchIsendIrecv_Base(const PrimitivePtr &prim, const py::list &args);
|
|
61
|
-
py::object PYNATIVE_EXPORT Pyboost_RotaryPositionEmbeddingGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
62
|
-
py::object PYNATIVE_EXPORT Pyboost_HShrink_Base(const PrimitivePtr &prim, const py::list &args);
|
|
63
|
-
py::object PYNATIVE_EXPORT Pyboost_DistCommAllGather_Base(const PrimitivePtr &prim, const py::list &args);
|
|
64
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceScatterAdd_Base(const PrimitivePtr &prim, const py::list &args);
|
|
65
|
-
py::object PYNATIVE_EXPORT Pyboost_AddcdivExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
66
|
-
py::object PYNATIVE_EXPORT Pyboost_MaxDim_Base(const PrimitivePtr &prim, const py::list &args);
|
|
67
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceDivMods_Base(const PrimitivePtr &prim, const py::list &args);
|
|
68
|
-
py::object PYNATIVE_EXPORT Pyboost_LinSpaceExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
69
|
-
py::object PYNATIVE_EXPORT Pyboost_Addbmm_Base(const PrimitivePtr &prim, const py::list &args);
|
|
70
|
-
py::object PYNATIVE_EXPORT Pyboost_Embedding_Base(const PrimitivePtr &prim, const py::list &args);
|
|
71
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceMaskedFillScalar_Base(const PrimitivePtr &prim, const py::list &args);
|
|
72
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceLog_Base(const PrimitivePtr &prim, const py::list &args);
|
|
73
|
-
py::object PYNATIVE_EXPORT Pyboost_MaxPoolWithMask_Base(const PrimitivePtr &prim, const py::list &args);
|
|
74
|
-
py::object PYNATIVE_EXPORT Pyboost_DistCommScatterTensor_Base(const PrimitivePtr &prim, const py::list &args);
|
|
75
|
-
py::object PYNATIVE_EXPORT Pyboost_BatchMatMul_Base(const PrimitivePtr &prim, const py::list &args);
|
|
76
|
-
py::object PYNATIVE_EXPORT Pyboost_SpeedFusionAttention_Base(const PrimitivePtr &prim, const py::list &args);
|
|
77
|
-
py::object PYNATIVE_EXPORT Pyboost_InnerNonZero_Base(const PrimitivePtr &prim, const py::list &args);
|
|
78
|
-
py::object PYNATIVE_EXPORT Pyboost_Mv_Base(const PrimitivePtr &prim, const py::list &args);
|
|
79
|
-
py::object PYNATIVE_EXPORT Pyboost_UniqueConsecutive_Base(const PrimitivePtr &prim, const py::list &args);
|
|
80
|
-
py::object PYNATIVE_EXPORT Pyboost_ReluGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
81
|
-
py::object PYNATIVE_EXPORT Pyboost_LogicalAnd_Base(const PrimitivePtr &prim, const py::list &args);
|
|
82
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceScatterValueReduce_Base(const PrimitivePtr &prim, const py::list &args);
|
|
83
|
-
py::object PYNATIVE_EXPORT Pyboost_PowScalarTensor_Base(const PrimitivePtr &prim, const py::list &args);
|
|
84
|
-
py::object PYNATIVE_EXPORT Pyboost_OneHotExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
85
|
-
py::object PYNATIVE_EXPORT Pyboost_BatchNormElemtGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
86
|
-
py::object PYNATIVE_EXPORT Pyboost_IsNegInf_Base(const PrimitivePtr &prim, const py::list &args);
|
|
87
|
-
py::object PYNATIVE_EXPORT Pyboost_FloorDiv_Base(const PrimitivePtr &prim, const py::list &args);
|
|
88
|
-
py::object PYNATIVE_EXPORT Pyboost_Contiguous_Base(const PrimitivePtr &prim, const py::list &args);
|
|
89
|
-
py::object PYNATIVE_EXPORT Pyboost_SubExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
90
|
-
py::object PYNATIVE_EXPORT Pyboost_SplitWithSize_Base(const PrimitivePtr &prim, const py::list &args);
|
|
91
|
-
py::object PYNATIVE_EXPORT Pyboost_Erfc_Base(const PrimitivePtr &prim, const py::list &args);
|
|
92
|
-
py::object PYNATIVE_EXPORT Pyboost_IsInf_Base(const PrimitivePtr &prim, const py::list &args);
|
|
93
|
-
py::object PYNATIVE_EXPORT Pyboost_LogicalNot_Base(const PrimitivePtr &prim, const py::list &args);
|
|
94
|
-
py::object PYNATIVE_EXPORT Pyboost_AddRmsNorm_Base(const PrimitivePtr &prim, const py::list &args);
|
|
95
|
-
py::object PYNATIVE_EXPORT Pyboost_ArgMinExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
96
|
-
py::object PYNATIVE_EXPORT Pyboost_UpsampleNearest1DGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
97
|
-
py::object PYNATIVE_EXPORT Pyboost_Slice_Base(const PrimitivePtr &prim, const py::list &args);
|
|
98
|
-
py::object PYNATIVE_EXPORT Pyboost_Arange_Base(const PrimitivePtr &prim, const py::list &args);
|
|
99
|
-
py::object PYNATIVE_EXPORT Pyboost_RepeatInterleaveGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
100
|
-
py::object PYNATIVE_EXPORT Pyboost_Pow_Base(const PrimitivePtr &prim, const py::list &args);
|
|
101
|
-
py::object PYNATIVE_EXPORT Pyboost_AllGatherMatmul_Base(const PrimitivePtr &prim, const py::list &args);
|
|
102
|
-
py::object PYNATIVE_EXPORT Pyboost_Less_Base(const PrimitivePtr &prim, const py::list &args);
|
|
103
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceFloorDivides_Base(const PrimitivePtr &prim, const py::list &args);
|
|
104
|
-
py::object PYNATIVE_EXPORT Pyboost_SoftShrink_Base(const PrimitivePtr &prim, const py::list &args);
|
|
105
|
-
py::object PYNATIVE_EXPORT Pyboost_ConvolutionStrGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
106
|
-
py::object PYNATIVE_EXPORT Pyboost_ViewAs_Base(const PrimitivePtr &prim, const py::list &args);
|
|
107
|
-
py::object PYNATIVE_EXPORT Pyboost_FFNExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
108
|
-
py::object PYNATIVE_EXPORT Pyboost_LinalgQr_Base(const PrimitivePtr &prim, const py::list &args);
|
|
109
|
-
py::object PYNATIVE_EXPORT Pyboost_BinaryCrossEntropyWithLogitsBackward_Base(const PrimitivePtr &prim, const py::list &args);
|
|
110
|
-
py::object PYNATIVE_EXPORT Pyboost_EluExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
111
|
-
py::object PYNATIVE_EXPORT Pyboost_Maximum_Base(const PrimitivePtr &prim, const py::list &args);
|
|
112
|
-
py::object PYNATIVE_EXPORT Pyboost_GridSampler3D_Base(const PrimitivePtr &prim, const py::list &args);
|
|
113
|
-
py::object PYNATIVE_EXPORT Pyboost_MeanExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
114
|
-
py::object PYNATIVE_EXPORT Pyboost_Atanh_Base(const PrimitivePtr &prim, const py::list &args);
|
|
115
|
-
py::object PYNATIVE_EXPORT Pyboost_AvgPool2D_Base(const PrimitivePtr &prim, const py::list &args);
|
|
116
|
-
py::object PYNATIVE_EXPORT Pyboost_MultiScaleDeformableAttn_Base(const PrimitivePtr &prim, const py::list &args);
|
|
117
|
-
py::object PYNATIVE_EXPORT Pyboost_AddLayerNormGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
118
|
-
py::object PYNATIVE_EXPORT Pyboost_IncreFlashAttention_Base(const PrimitivePtr &prim, const py::list &args);
|
|
119
|
-
py::object PYNATIVE_EXPORT Pyboost_BitwiseAndScalar_Base(const PrimitivePtr &prim, const py::list &args);
|
|
120
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceReLU_Base(const PrimitivePtr &prim, const py::list &args);
|
|
121
|
-
py::object PYNATIVE_EXPORT Pyboost_Expm1_Base(const PrimitivePtr &prim, const py::list &args);
|
|
122
|
-
py::object PYNATIVE_EXPORT Pyboost_FmodScalar_Base(const PrimitivePtr &prim, const py::list &args);
|
|
123
|
-
py::object PYNATIVE_EXPORT Pyboost_Repeat_Base(const PrimitivePtr &prim, const py::list &args);
|
|
124
|
-
py::object PYNATIVE_EXPORT Pyboost_NewZeros_Base(const PrimitivePtr &prim, const py::list &args);
|
|
125
|
-
py::object PYNATIVE_EXPORT Pyboost_SigmoidGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
126
|
-
py::object PYNATIVE_EXPORT Pyboost_Baddbmm_Base(const PrimitivePtr &prim, const py::list &args);
|
|
127
|
-
py::object PYNATIVE_EXPORT Pyboost_GatherDGradV2_Base(const PrimitivePtr &prim, const py::list &args);
|
|
128
|
-
py::object PYNATIVE_EXPORT Pyboost_GmmV2Backward_Base(const PrimitivePtr &prim, const py::list &args);
|
|
129
|
-
py::object PYNATIVE_EXPORT Pyboost_RmsNorm_Base(const PrimitivePtr &prim, const py::list &args);
|
|
130
|
-
py::object PYNATIVE_EXPORT Pyboost_MaxPoolGradWithMask_Base(const PrimitivePtr &prim, const py::list &args);
|
|
131
|
-
py::object PYNATIVE_EXPORT Pyboost_SiLUGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
132
|
-
py::object PYNATIVE_EXPORT Pyboost_Xlogy_Base(const PrimitivePtr &prim, const py::list &args);
|
|
133
|
-
py::object PYNATIVE_EXPORT Pyboost_Reciprocal_Base(const PrimitivePtr &prim, const py::list &args);
|
|
134
|
-
py::object PYNATIVE_EXPORT Pyboost_AdaptiveMaxPool1D_Base(const PrimitivePtr &prim, const py::list &args);
|
|
135
|
-
py::object PYNATIVE_EXPORT Pyboost_StackExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
136
|
-
py::object PYNATIVE_EXPORT Pyboost_AdamW_Base(const PrimitivePtr &prim, const py::list &args);
|
|
137
|
-
py::object PYNATIVE_EXPORT Pyboost_BitwiseNot_Base(const PrimitivePtr &prim, const py::list &args);
|
|
138
|
-
py::object PYNATIVE_EXPORT Pyboost_RepeatInterleaveTensor_Base(const PrimitivePtr &prim, const py::list &args);
|
|
139
|
-
py::object PYNATIVE_EXPORT Pyboost_Lerp_Base(const PrimitivePtr &prim, const py::list &args);
|
|
140
|
-
py::object PYNATIVE_EXPORT Pyboost_XLogYScalarOther_Base(const PrimitivePtr &prim, const py::list &args);
|
|
141
|
-
py::object PYNATIVE_EXPORT Pyboost_NLLLoss2dGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
142
|
-
py::object PYNATIVE_EXPORT Pyboost_IndexFillScalar_Base(const PrimitivePtr &prim, const py::list &args);
|
|
143
|
-
py::object PYNATIVE_EXPORT Pyboost_CumminExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
144
|
-
py::object PYNATIVE_EXPORT Pyboost_KLDiv_Base(const PrimitivePtr &prim, const py::list &args);
|
|
145
|
-
py::object PYNATIVE_EXPORT Pyboost_GmmBackward_Base(const PrimitivePtr &prim, const py::list &args);
|
|
146
|
-
py::object PYNATIVE_EXPORT Pyboost_LogicalXor_Base(const PrimitivePtr &prim, const py::list &args);
|
|
147
|
-
py::object PYNATIVE_EXPORT Pyboost_MaskedSelectGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
148
|
-
py::object PYNATIVE_EXPORT Pyboost_BatchNormGradExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
149
|
-
py::object PYNATIVE_EXPORT Pyboost_FlattenExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
150
|
-
py::object PYNATIVE_EXPORT Pyboost_UpsampleNearest3D_Base(const PrimitivePtr &prim, const py::list &args);
|
|
151
|
-
py::object PYNATIVE_EXPORT Pyboost_Exp_Base(const PrimitivePtr &prim, const py::list &args);
|
|
152
|
-
py::object PYNATIVE_EXPORT Pyboost_Sin_Base(const PrimitivePtr &prim, const py::list &args);
|
|
153
|
-
py::object PYNATIVE_EXPORT Pyboost_Atan2Ext_Base(const PrimitivePtr &prim, const py::list &args);
|
|
154
|
-
py::object PYNATIVE_EXPORT Pyboost_RemainderTensorScalar_Base(const PrimitivePtr &prim, const py::list &args);
|
|
155
|
-
py::object PYNATIVE_EXPORT Pyboost_InnerCommAllReduce_Base(const PrimitivePtr &prim, const py::list &args);
|
|
156
|
-
py::object PYNATIVE_EXPORT Pyboost_FillTensor_Base(const PrimitivePtr &prim, const py::list &args);
|
|
157
|
-
py::object PYNATIVE_EXPORT Pyboost_IndexFillTensor_Base(const PrimitivePtr &prim, const py::list &args);
|
|
158
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceMaskedFillTensor_Base(const PrimitivePtr &prim, const py::list &args);
|
|
159
|
-
py::object PYNATIVE_EXPORT Pyboost_RepeatInterleaveInt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
160
|
-
py::object PYNATIVE_EXPORT Pyboost_Ceil_Base(const PrimitivePtr &prim, const py::list &args);
|
|
161
|
-
py::object PYNATIVE_EXPORT Pyboost_Roll_Base(const PrimitivePtr &prim, const py::list &args);
|
|
162
|
-
py::object PYNATIVE_EXPORT Pyboost_SortExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
163
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceGroupedMatmulAdd_Base(const PrimitivePtr &prim, const py::list &args);
|
|
164
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceTanh_Base(const PrimitivePtr &prim, const py::list &args);
|
|
165
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceStopGradient_Base(const PrimitivePtr &prim, const py::list &args);
|
|
166
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceHardtanh_Base(const PrimitivePtr &prim, const py::list &args);
|
|
167
|
-
py::object PYNATIVE_EXPORT Pyboost_DivMods_Base(const PrimitivePtr &prim, const py::list &args);
|
|
168
|
-
py::object PYNATIVE_EXPORT Pyboost_RandLikeExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
169
|
-
py::object PYNATIVE_EXPORT Pyboost_LogAddExp2_Base(const PrimitivePtr &prim, const py::list &args);
|
|
170
|
-
py::object PYNATIVE_EXPORT Pyboost_ArgMinWithValue_Base(const PrimitivePtr &prim, const py::list &args);
|
|
171
|
-
py::object PYNATIVE_EXPORT Pyboost_Im2ColExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
172
|
-
py::object PYNATIVE_EXPORT Pyboost_Squeeze_Base(const PrimitivePtr &prim, const py::list &args);
|
|
173
|
-
py::object PYNATIVE_EXPORT Pyboost_Copy_Base(const PrimitivePtr &prim, const py::list &args);
|
|
174
|
-
py::object PYNATIVE_EXPORT Pyboost_LinalgVectorNorm_Base(const PrimitivePtr &prim, const py::list &args);
|
|
175
|
-
py::object PYNATIVE_EXPORT Pyboost_Chunk_Base(const PrimitivePtr &prim, const py::list &args);
|
|
176
|
-
py::object PYNATIVE_EXPORT Pyboost_RandInt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
177
|
-
py::object PYNATIVE_EXPORT Pyboost_Frac_Base(const PrimitivePtr &prim, const py::list &args);
|
|
178
|
-
py::object PYNATIVE_EXPORT Pyboost_BinaryCrossEntropyGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
179
|
-
py::object PYNATIVE_EXPORT Pyboost_DropoutGenMaskExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
180
|
-
py::object PYNATIVE_EXPORT Pyboost_AllFinite_Base(const PrimitivePtr &prim, const py::list &args);
|
|
181
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceZero_Base(const PrimitivePtr &prim, const py::list &args);
|
|
182
|
-
py::object PYNATIVE_EXPORT Pyboost_LeakyReLUGradExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
183
|
-
py::object PYNATIVE_EXPORT Pyboost_SwigluGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
184
|
-
py::object PYNATIVE_EXPORT Pyboost_Polar_Base(const PrimitivePtr &prim, const py::list &args);
|
|
185
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceRandom_Base(const PrimitivePtr &prim, const py::list &args);
|
|
186
|
-
py::object PYNATIVE_EXPORT Pyboost_Take_Base(const PrimitivePtr &prim, const py::list &args);
|
|
187
|
-
py::object PYNATIVE_EXPORT Pyboost_Cosh_Base(const PrimitivePtr &prim, const py::list &args);
|
|
188
|
-
py::object PYNATIVE_EXPORT Pyboost_SoftShrinkGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
189
|
-
py::object PYNATIVE_EXPORT Pyboost_KLDivGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
190
|
-
py::object PYNATIVE_EXPORT Pyboost_Clone_Base(const PrimitivePtr &prim, const py::list &args);
|
|
191
|
-
py::object PYNATIVE_EXPORT Pyboost_SpeedFusionAttentionGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
192
|
-
py::object PYNATIVE_EXPORT Pyboost_Add_Base(const PrimitivePtr &prim, const py::list &args);
|
|
193
|
-
py::object PYNATIVE_EXPORT Pyboost_AddLayerNormV2_Base(const PrimitivePtr &prim, const py::list &args);
|
|
194
|
-
py::object PYNATIVE_EXPORT Pyboost_ReflectionPad1D_Base(const PrimitivePtr &prim, const py::list &args);
|
|
195
|
-
py::object PYNATIVE_EXPORT Pyboost_UniqueDim_Base(const PrimitivePtr &prim, const py::list &args);
|
|
196
|
-
py::object PYNATIVE_EXPORT Pyboost_Abs_Base(const PrimitivePtr &prim, const py::list &args);
|
|
197
|
-
py::object PYNATIVE_EXPORT Pyboost_BatchNormElemt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
198
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceFillScalar_Base(const PrimitivePtr &prim, const py::list &args);
|
|
199
|
-
py::object PYNATIVE_EXPORT Pyboost_SubScalar_Base(const PrimitivePtr &prim, const py::list &args);
|
|
200
|
-
py::object PYNATIVE_EXPORT Pyboost_FlashAttentionScoreGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
201
|
-
py::object PYNATIVE_EXPORT Pyboost_NonZeroExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
202
|
-
py::object PYNATIVE_EXPORT Pyboost_NormalFloatFloat_Base(const PrimitivePtr &prim, const py::list &args);
|
|
203
|
-
py::object PYNATIVE_EXPORT Pyboost_MoeTokenUnpermuteGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
204
|
-
py::object PYNATIVE_EXPORT Pyboost_MaxPoolWithIndices_Base(const PrimitivePtr &prim, const py::list &args);
|
|
205
|
-
py::object PYNATIVE_EXPORT Pyboost_EqualExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
206
|
-
py::object PYNATIVE_EXPORT Pyboost_Sigmoid_Base(const PrimitivePtr &prim, const py::list &args);
|
|
207
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceMul_Base(const PrimitivePtr &prim, const py::list &args);
|
|
208
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceElu_Base(const PrimitivePtr &prim, const py::list &args);
|
|
209
|
-
py::object PYNATIVE_EXPORT Pyboost_TriangularSolve_Base(const PrimitivePtr &prim, const py::list &args);
|
|
210
|
-
py::object PYNATIVE_EXPORT Pyboost_Unique2_Base(const PrimitivePtr &prim, const py::list &args);
|
|
211
|
-
py::object PYNATIVE_EXPORT Pyboost_GridSampler2D_Base(const PrimitivePtr &prim, const py::list &args);
|
|
212
|
-
py::object PYNATIVE_EXPORT Pyboost_ReplicationPad1D_Base(const PrimitivePtr &prim, const py::list &args);
|
|
213
|
-
py::object PYNATIVE_EXPORT Pyboost_InnerCommIsend_Base(const PrimitivePtr &prim, const py::list &args);
|
|
214
|
-
py::object PYNATIVE_EXPORT Pyboost_RemainderScalarTensor_Base(const PrimitivePtr &prim, const py::list &args);
|
|
215
|
-
py::object PYNATIVE_EXPORT Pyboost_MatmulReduceScatter_Base(const PrimitivePtr &prim, const py::list &args);
|
|
216
|
-
py::object PYNATIVE_EXPORT Pyboost_SelectExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
217
|
-
py::object PYNATIVE_EXPORT Pyboost_ReflectionPad1DGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
218
|
-
py::object PYNATIVE_EXPORT Pyboost_DistCommBroadcast_Base(const PrimitivePtr &prim, const py::list &args);
|
|
219
|
-
py::object PYNATIVE_EXPORT Pyboost_BitwiseXorTensor_Base(const PrimitivePtr &prim, const py::list &args);
|
|
220
|
-
py::object PYNATIVE_EXPORT Pyboost_NotEqual_Base(const PrimitivePtr &prim, const py::list &args);
|
|
221
|
-
py::object PYNATIVE_EXPORT Pyboost_Threshold_Base(const PrimitivePtr &prim, const py::list &args);
|
|
222
|
-
py::object PYNATIVE_EXPORT Pyboost_MishExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
223
|
-
py::object PYNATIVE_EXPORT Pyboost_DistCommReduce_Base(const PrimitivePtr &prim, const py::list &args);
|
|
224
|
-
py::object PYNATIVE_EXPORT Pyboost_DistCommIrecv_Base(const PrimitivePtr &prim, const py::list &args);
|
|
225
|
-
py::object PYNATIVE_EXPORT Pyboost_MaskedSelect_Base(const PrimitivePtr &prim, const py::list &args);
|
|
226
|
-
py::object PYNATIVE_EXPORT Pyboost_MedianDim_Base(const PrimitivePtr &prim, const py::list &args);
|
|
227
|
-
py::object PYNATIVE_EXPORT Pyboost_HShrinkGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
228
|
-
py::object PYNATIVE_EXPORT Pyboost_LogAddExp_Base(const PrimitivePtr &prim, const py::list &args);
|
|
229
|
-
py::object PYNATIVE_EXPORT Pyboost_DistCommGatherIntoTensor_Base(const PrimitivePtr &prim, const py::list &args);
|
|
230
|
-
py::object PYNATIVE_EXPORT Pyboost_ScatterAddExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
231
|
-
py::object PYNATIVE_EXPORT Pyboost_Equal_Base(const PrimitivePtr &prim, const py::list &args);
|
|
232
|
-
py::object PYNATIVE_EXPORT Pyboost_DistCommReduceScatterTensor_Base(const PrimitivePtr &prim, const py::list &args);
|
|
233
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceAddmm_Base(const PrimitivePtr &prim, const py::list &args);
|
|
234
|
-
py::object PYNATIVE_EXPORT Pyboost_FloorDivScalar_Base(const PrimitivePtr &prim, const py::list &args);
|
|
235
|
-
py::object PYNATIVE_EXPORT Pyboost_BatchNormReduceGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
236
|
-
py::object PYNATIVE_EXPORT Pyboost_Conv1DExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
237
|
-
py::object PYNATIVE_EXPORT Pyboost_BatchNormStats_Base(const PrimitivePtr &prim, const py::list &args);
|
|
238
|
-
py::object PYNATIVE_EXPORT Pyboost_SumExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
239
|
-
py::object PYNATIVE_EXPORT Pyboost_DistCommReduceScatter_Base(const PrimitivePtr &prim, const py::list &args);
|
|
240
|
-
py::object PYNATIVE_EXPORT Pyboost_SelectV2_Base(const PrimitivePtr &prim, const py::list &args);
|
|
241
|
-
py::object PYNATIVE_EXPORT Pyboost_Swiglu_Base(const PrimitivePtr &prim, const py::list &args);
|
|
242
|
-
py::object PYNATIVE_EXPORT Pyboost_NLLLossGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
243
|
-
py::object PYNATIVE_EXPORT Pyboost_GLU_Base(const PrimitivePtr &prim, const py::list &args);
|
|
244
|
-
py::object PYNATIVE_EXPORT Pyboost_ReduceMin_Base(const PrimitivePtr &prim, const py::list &args);
|
|
245
|
-
py::object PYNATIVE_EXPORT Pyboost_BitwiseXorScalar_Base(const PrimitivePtr &prim, const py::list &args);
|
|
246
|
-
py::object PYNATIVE_EXPORT Pyboost_MedianExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
247
|
-
py::object PYNATIVE_EXPORT Pyboost_Erf_Base(const PrimitivePtr &prim, const py::list &args);
|
|
248
|
-
py::object PYNATIVE_EXPORT Pyboost_DistCommAllReduce_Base(const PrimitivePtr &prim, const py::list &args);
|
|
249
|
-
py::object PYNATIVE_EXPORT Pyboost_UpsampleBilinear2D_Base(const PrimitivePtr &prim, const py::list &args);
|
|
250
|
-
py::object PYNATIVE_EXPORT Pyboost_Sign_Base(const PrimitivePtr &prim, const py::list &args);
|
|
251
|
-
py::object PYNATIVE_EXPORT Pyboost_SoftMarginLossGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
252
|
-
py::object PYNATIVE_EXPORT Pyboost_InnerCommReduceScatter_Base(const PrimitivePtr &prim, const py::list &args);
|
|
253
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceUniform_Base(const PrimitivePtr &prim, const py::list &args);
|
|
254
|
-
py::object PYNATIVE_EXPORT Pyboost_ClampTensor_Base(const PrimitivePtr &prim, const py::list &args);
|
|
255
|
-
py::object PYNATIVE_EXPORT Pyboost_LogSumExp_Base(const PrimitivePtr &prim, const py::list &args);
|
|
256
|
-
py::object PYNATIVE_EXPORT Pyboost_Split_Base(const PrimitivePtr &prim, const py::list &args);
|
|
257
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceDivMod_Base(const PrimitivePtr &prim, const py::list &args);
|
|
258
|
-
py::object PYNATIVE_EXPORT Pyboost_ReLU_Base(const PrimitivePtr &prim, const py::list &args);
|
|
259
|
-
py::object PYNATIVE_EXPORT Pyboost_UpsampleNearest3DGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
260
|
-
py::object PYNATIVE_EXPORT Pyboost_OnesLikeExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
261
|
-
py::object PYNATIVE_EXPORT Pyboost_BitwiseAndTensor_Base(const PrimitivePtr &prim, const py::list &args);
|
|
262
|
-
py::object PYNATIVE_EXPORT Pyboost_AddScalar_Base(const PrimitivePtr &prim, const py::list &args);
|
|
263
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceFillTensor_Base(const PrimitivePtr &prim, const py::list &args);
|
|
264
|
-
py::object PYNATIVE_EXPORT Pyboost_TraceExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
265
|
-
py::object PYNATIVE_EXPORT Pyboost_SeLUExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
266
|
-
py::object PYNATIVE_EXPORT Pyboost_LogSigmoid_Base(const PrimitivePtr &prim, const py::list &args);
|
|
267
|
-
py::object PYNATIVE_EXPORT Pyboost_ExpandDims_Base(const PrimitivePtr &prim, const py::list &args);
|
|
268
|
-
py::object PYNATIVE_EXPORT Pyboost_PromptFlashAttention_Base(const PrimitivePtr &prim, const py::list &args);
|
|
269
|
-
py::object PYNATIVE_EXPORT Pyboost_BatchMatMulExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
270
|
-
py::object PYNATIVE_EXPORT Pyboost_LogSigmoidGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
271
|
-
py::object PYNATIVE_EXPORT Pyboost_SliceExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
272
|
-
py::object PYNATIVE_EXPORT Pyboost_MatrixInverseExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
273
|
-
py::object PYNATIVE_EXPORT Pyboost_BitwiseOrScalar_Base(const PrimitivePtr &prim, const py::list &args);
|
|
274
|
-
py::object PYNATIVE_EXPORT Pyboost_ReduceMax_Base(const PrimitivePtr &prim, const py::list &args);
|
|
275
|
-
py::object PYNATIVE_EXPORT Pyboost_InnerIndex_Base(const PrimitivePtr &prim, const py::list &args);
|
|
276
|
-
py::object PYNATIVE_EXPORT Pyboost_ReflectionPad2DGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
277
|
-
py::object PYNATIVE_EXPORT Pyboost_Log_Base(const PrimitivePtr &prim, const py::list &args);
|
|
278
|
-
py::object PYNATIVE_EXPORT Pyboost_AdaptiveAvgPool3DExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
279
|
-
py::object PYNATIVE_EXPORT Pyboost_MaskedFill_Base(const PrimitivePtr &prim, const py::list &args);
|
|
280
|
-
py::object PYNATIVE_EXPORT Pyboost_DropoutGradExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
281
|
-
py::object PYNATIVE_EXPORT Pyboost_RandpermExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
282
|
-
py::object PYNATIVE_EXPORT Pyboost_NeScalar_Base(const PrimitivePtr &prim, const py::list &args);
|
|
283
|
-
py::object PYNATIVE_EXPORT Pyboost_Col2ImGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
284
|
-
py::object PYNATIVE_EXPORT Pyboost_HistcExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
285
|
-
py::object PYNATIVE_EXPORT Pyboost_ThresholdGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
286
|
-
py::object PYNATIVE_EXPORT Pyboost_AddcmulExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
287
|
-
py::object PYNATIVE_EXPORT Pyboost_UpsampleNearest2DGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
288
|
-
py::object PYNATIVE_EXPORT Pyboost_AcoshExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
289
|
-
py::object PYNATIVE_EXPORT Pyboost_AvgPool2DGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
290
|
-
py::object PYNATIVE_EXPORT Pyboost_Greater_Base(const PrimitivePtr &prim, const py::list &args);
|
|
291
|
-
py::object PYNATIVE_EXPORT Pyboost_InnerInplaceIndexPut_Base(const PrimitivePtr &prim, const py::list &args);
|
|
292
|
-
py::object PYNATIVE_EXPORT Pyboost_Mul_Base(const PrimitivePtr &prim, const py::list &args);
|
|
293
|
-
py::object PYNATIVE_EXPORT Pyboost_Nansum_Base(const PrimitivePtr &prim, const py::list &args);
|
|
294
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceFloorDivide_Base(const PrimitivePtr &prim, const py::list &args);
|
|
295
|
-
py::object PYNATIVE_EXPORT Pyboost_Index_Base(const PrimitivePtr &prim, const py::list &args);
|
|
296
|
-
py::object PYNATIVE_EXPORT Pyboost_MSELossGradExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
297
|
-
py::object PYNATIVE_EXPORT Pyboost_Var_Base(const PrimitivePtr &prim, const py::list &args);
|
|
298
|
-
py::object PYNATIVE_EXPORT Pyboost_Tile_Base(const PrimitivePtr &prim, const py::list &args);
|
|
299
|
-
py::object PYNATIVE_EXPORT Pyboost_VarMean_Base(const PrimitivePtr &prim, const py::list &args);
|
|
300
|
-
py::object PYNATIVE_EXPORT Pyboost_Divs_Base(const PrimitivePtr &prim, const py::list &args);
|
|
301
|
-
py::object PYNATIVE_EXPORT Pyboost_Conv2DExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
302
|
-
py::object PYNATIVE_EXPORT Pyboost_BincountExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
303
|
-
py::object PYNATIVE_EXPORT Pyboost_Tanh_Base(const PrimitivePtr &prim, const py::list &args);
|
|
304
|
-
py::object PYNATIVE_EXPORT Pyboost_UpsampleLinear1D_Base(const PrimitivePtr &prim, const py::list &args);
|
|
305
|
-
py::object PYNATIVE_EXPORT Pyboost_DistCommGather_Base(const PrimitivePtr &prim, const py::list &args);
|
|
306
|
-
py::object PYNATIVE_EXPORT Pyboost_SoftplusGradExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
307
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceIndexAddExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
308
|
-
py::object PYNATIVE_EXPORT Pyboost_DistCommAllToAllVSingle_Base(const PrimitivePtr &prim, const py::list &args);
|
|
309
|
-
py::object PYNATIVE_EXPORT Pyboost_ReflectionPad3DGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
310
|
-
py::object PYNATIVE_EXPORT Pyboost_Addmm_Base(const PrimitivePtr &prim, const py::list &args);
|
|
311
|
-
py::object PYNATIVE_EXPORT Pyboost_Min_Base(const PrimitivePtr &prim, const py::list &args);
|
|
312
|
-
py::object PYNATIVE_EXPORT Pyboost_ClampScalar_Base(const PrimitivePtr &prim, const py::list &args);
|
|
313
|
-
py::object PYNATIVE_EXPORT Pyboost_Log1p_Base(const PrimitivePtr &prim, const py::list &args);
|
|
314
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceScatterSrcReduce_Base(const PrimitivePtr &prim, const py::list &args);
|
|
315
|
-
py::object PYNATIVE_EXPORT Pyboost_ReflectionPad2D_Base(const PrimitivePtr &prim, const py::list &args);
|
|
316
|
-
py::object PYNATIVE_EXPORT Pyboost_ExpandAs_Base(const PrimitivePtr &prim, const py::list &args);
|
|
317
|
-
py::object PYNATIVE_EXPORT Pyboost_AsinhExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
318
|
-
py::object PYNATIVE_EXPORT Pyboost_Randn_Base(const PrimitivePtr &prim, const py::list &args);
|
|
319
|
-
py::object PYNATIVE_EXPORT Pyboost_FullLike_Base(const PrimitivePtr &prim, const py::list &args);
|
|
320
|
-
py::object PYNATIVE_EXPORT Pyboost_ConstantPadND_Base(const PrimitivePtr &prim, const py::list &args);
|
|
321
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceIndexPut_Base(const PrimitivePtr &prim, const py::list &args);
|
|
322
|
-
py::object PYNATIVE_EXPORT Pyboost_TExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
323
|
-
py::object PYNATIVE_EXPORT Pyboost_PowTensorScalar_Base(const PrimitivePtr &prim, const py::list &args);
|
|
324
|
-
py::object PYNATIVE_EXPORT Pyboost_Max_Base(const PrimitivePtr &prim, const py::list &args);
|
|
325
|
-
py::object PYNATIVE_EXPORT Pyboost_SplitTensor_Base(const PrimitivePtr &prim, const py::list &args);
|
|
326
|
-
py::object PYNATIVE_EXPORT Pyboost_GeLU_Base(const PrimitivePtr &prim, const py::list &args);
|
|
327
|
-
py::object PYNATIVE_EXPORT Pyboost_NLLLoss_Base(const PrimitivePtr &prim, const py::list &args);
|
|
328
|
-
py::object PYNATIVE_EXPORT Pyboost_LessEqual_Base(const PrimitivePtr &prim, const py::list &args);
|
|
329
|
-
py::object PYNATIVE_EXPORT Pyboost_Log2_Base(const PrimitivePtr &prim, const py::list &args);
|
|
330
|
-
py::object PYNATIVE_EXPORT Pyboost_HSigmoidGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
331
|
-
py::object PYNATIVE_EXPORT Pyboost_Transpose_Base(const PrimitivePtr &prim, const py::list &args);
|
|
332
|
-
py::object PYNATIVE_EXPORT Pyboost_UpsampleLinear1DGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
333
|
-
py::object PYNATIVE_EXPORT Pyboost_UpsampleTrilinear3D_Base(const PrimitivePtr &prim, const py::list &args);
|
|
334
|
-
py::object PYNATIVE_EXPORT Pyboost_GluGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
335
|
-
py::object PYNATIVE_EXPORT Pyboost_GeluExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
336
|
-
py::object PYNATIVE_EXPORT Pyboost_MishGradExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
337
|
-
py::object PYNATIVE_EXPORT Pyboost_AdaptiveAvgPool2DExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
338
|
-
py::object PYNATIVE_EXPORT Pyboost_Floor_Base(const PrimitivePtr &prim, const py::list &args);
|
|
339
|
-
py::object PYNATIVE_EXPORT Pyboost_LerpScalar_Base(const PrimitivePtr &prim, const py::list &args);
|
|
340
|
-
py::object PYNATIVE_EXPORT Pyboost_GreaterEqual_Base(const PrimitivePtr &prim, const py::list &args);
|
|
341
|
-
py::object PYNATIVE_EXPORT Pyboost_Conv3DExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
342
|
-
py::object PYNATIVE_EXPORT Pyboost_HSwish_Base(const PrimitivePtr &prim, const py::list &args);
|
|
343
|
-
py::object PYNATIVE_EXPORT Pyboost_Muls_Base(const PrimitivePtr &prim, const py::list &args);
|
|
344
|
-
py::object PYNATIVE_EXPORT Pyboost_DistCommBarrier_Base(const PrimitivePtr &prim, const py::list &args);
|
|
345
|
-
py::object PYNATIVE_EXPORT Pyboost_EmbeddingDenseBackward_Base(const PrimitivePtr &prim, const py::list &args);
|
|
346
|
-
py::object PYNATIVE_EXPORT Pyboost_CountNonZero_Base(const PrimitivePtr &prim, const py::list &args);
|
|
347
|
-
py::object PYNATIVE_EXPORT Pyboost_UpsampleTrilinear3DGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
348
|
-
py::object PYNATIVE_EXPORT Pyboost_NormalFloatTensor_Base(const PrimitivePtr &prim, const py::list &args);
|
|
349
|
-
py::object PYNATIVE_EXPORT Pyboost_AdaptiveAvgPool1D_Base(const PrimitivePtr &prim, const py::list &args);
|
|
350
|
-
py::object PYNATIVE_EXPORT Pyboost_Addmv_Base(const PrimitivePtr &prim, const py::list &args);
|
|
351
|
-
py::object PYNATIVE_EXPORT Pyboost_Conv2DPadding_Base(const PrimitivePtr &prim, const py::list &args);
|
|
352
|
-
py::object PYNATIVE_EXPORT Pyboost_BatchNormExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
353
|
-
py::object PYNATIVE_EXPORT Pyboost_FlashAttentionScore_Base(const PrimitivePtr &prim, const py::list &args);
|
|
354
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceFillDiagonal_Base(const PrimitivePtr &prim, const py::list &args);
|
|
355
|
-
py::object PYNATIVE_EXPORT Pyboost_Gcd_Base(const PrimitivePtr &prim, const py::list &args);
|
|
356
|
-
py::object PYNATIVE_EXPORT Pyboost_CumsumExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
357
|
-
py::object PYNATIVE_EXPORT Pyboost_Col2ImExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
358
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceSubExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
359
|
-
py::object PYNATIVE_EXPORT Pyboost_UnstackExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
360
|
-
py::object PYNATIVE_EXPORT Pyboost_Cos_Base(const PrimitivePtr &prim, const py::list &args);
|
|
361
|
-
py::object PYNATIVE_EXPORT Pyboost_IndexAddExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
362
|
-
py::object PYNATIVE_EXPORT Pyboost_Sub_Base(const PrimitivePtr &prim, const py::list &args);
|
|
363
|
-
py::object PYNATIVE_EXPORT Pyboost_Square_Base(const PrimitivePtr &prim, const py::list &args);
|
|
364
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceNormal_Base(const PrimitivePtr &prim, const py::list &args);
|
|
365
|
-
py::object PYNATIVE_EXPORT Pyboost_ArgMaxExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
366
|
-
py::object PYNATIVE_EXPORT Pyboost_Meshgrid_Base(const PrimitivePtr &prim, const py::list &args);
|
|
367
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceSubScalar_Base(const PrimitivePtr &prim, const py::list &args);
|
|
368
|
-
py::object PYNATIVE_EXPORT Pyboost_ReplicationPad3DGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
369
|
-
py::object PYNATIVE_EXPORT Pyboost_AddExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
370
|
-
py::object PYNATIVE_EXPORT Pyboost_InnerCommAllGather_Base(const PrimitivePtr &prim, const py::list &args);
|
|
371
|
-
py::object PYNATIVE_EXPORT Pyboost_TrilExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
372
|
-
py::object PYNATIVE_EXPORT Pyboost_MaxUnpool2DExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
373
|
-
py::object PYNATIVE_EXPORT Pyboost_SoftplusExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
374
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceCopy_Base(const PrimitivePtr &prim, const py::list &args);
|
|
375
|
-
py::object PYNATIVE_EXPORT Pyboost_LogicalOr_Base(const PrimitivePtr &prim, const py::list &args);
|
|
376
|
-
py::object PYNATIVE_EXPORT Pyboost_ReplicationPad2D_Base(const PrimitivePtr &prim, const py::list &args);
|
|
377
|
-
py::object PYNATIVE_EXPORT Pyboost_Std_Base(const PrimitivePtr &prim, const py::list &args);
|
|
378
|
-
py::object PYNATIVE_EXPORT Pyboost_SoftmaxBackward_Base(const PrimitivePtr &prim, const py::list &args);
|
|
379
|
-
py::object PYNATIVE_EXPORT Pyboost_GroupNormGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
380
|
-
py::object PYNATIVE_EXPORT Pyboost_EluGradExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
381
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceDivs_Base(const PrimitivePtr &prim, const py::list &args);
|
|
382
|
-
py::object PYNATIVE_EXPORT Pyboost_Norm_Base(const PrimitivePtr &prim, const py::list &args);
|
|
383
|
-
py::object PYNATIVE_EXPORT Pyboost_Log10_Base(const PrimitivePtr &prim, const py::list &args);
|
|
384
|
-
py::object PYNATIVE_EXPORT Pyboost_TanhGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
385
|
-
py::object PYNATIVE_EXPORT Pyboost_MultiScaleDeformableAttnGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
386
|
-
py::object PYNATIVE_EXPORT Pyboost_Exp2_Base(const PrimitivePtr &prim, const py::list &args);
|
|
387
|
-
py::object PYNATIVE_EXPORT Pyboost_IsClose_Base(const PrimitivePtr &prim, const py::list &args);
|
|
388
|
-
py::object PYNATIVE_EXPORT Pyboost_BCEWithLogitsLoss_Base(const PrimitivePtr &prim, const py::list &args);
|
|
389
|
-
py::object PYNATIVE_EXPORT Pyboost_Conv3DPadding_Base(const PrimitivePtr &prim, const py::list &args);
|
|
390
|
-
py::object PYNATIVE_EXPORT Pyboost_FillScalar_Base(const PrimitivePtr &prim, const py::list &args);
|
|
391
|
-
py::object PYNATIVE_EXPORT Pyboost_MoeTokenUnpermute_Base(const PrimitivePtr &prim, const py::list &args);
|
|
392
|
-
py::object PYNATIVE_EXPORT Pyboost_Dot_Base(const PrimitivePtr &prim, const py::list &args);
|
|
393
|
-
py::object PYNATIVE_EXPORT Pyboost_Select_Base(const PrimitivePtr &prim, const py::list &args);
|
|
394
|
-
py::object PYNATIVE_EXPORT Pyboost_DivMod_Base(const PrimitivePtr &prim, const py::list &args);
|
|
395
|
-
py::object PYNATIVE_EXPORT Pyboost_StdMean_Base(const PrimitivePtr &prim, const py::list &args);
|
|
396
|
-
py::object PYNATIVE_EXPORT Pyboost_ReplicationPad1DGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
397
|
-
py::object PYNATIVE_EXPORT Pyboost_LogSoftmaxGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
398
|
-
py::object PYNATIVE_EXPORT Pyboost_HardtanhGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
399
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceClampScalar_Base(const PrimitivePtr &prim, const py::list &args);
|
|
400
|
-
py::object PYNATIVE_EXPORT Pyboost_AdaptiveAvgPool3DGradExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
401
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceMuls_Base(const PrimitivePtr &prim, const py::list &args);
|
|
402
|
-
py::object PYNATIVE_EXPORT Pyboost_RandIntLike_Base(const PrimitivePtr &prim, const py::list &args);
|
|
403
|
-
py::object PYNATIVE_EXPORT Pyboost_ArgMaxWithValue_Base(const PrimitivePtr &prim, const py::list &args);
|
|
404
|
-
py::object PYNATIVE_EXPORT Pyboost_UpsampleNearest1D_Base(const PrimitivePtr &prim, const py::list &args);
|
|
405
|
-
py::object PYNATIVE_EXPORT Pyboost_PReLU_Base(const PrimitivePtr &prim, const py::list &args);
|
|
406
|
-
py::object PYNATIVE_EXPORT Pyboost_HSwishGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
407
|
-
py::object PYNATIVE_EXPORT Pyboost_MoeTokenPermute_Base(const PrimitivePtr &prim, const py::list &args);
|
|
408
|
-
py::object PYNATIVE_EXPORT Pyboost_ConvTranspose2D_Base(const PrimitivePtr &prim, const py::list &args);
|
|
409
|
-
py::object PYNATIVE_EXPORT Pyboost_DropoutDoMaskExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
410
|
-
py::object PYNATIVE_EXPORT Pyboost_SilentCheckV2_Base(const PrimitivePtr &prim, const py::list &args);
|
|
411
|
-
py::object PYNATIVE_EXPORT Pyboost_L1LossExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
412
|
-
py::object PYNATIVE_EXPORT Pyboost_AsinExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
413
|
-
py::object PYNATIVE_EXPORT Pyboost_RandExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
414
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceAddExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
415
|
-
py::object PYNATIVE_EXPORT Pyboost_Zeros_Base(const PrimitivePtr &prim, const py::list &args);
|
|
416
|
-
py::object PYNATIVE_EXPORT Pyboost_BinaryCrossEntropy_Base(const PrimitivePtr &prim, const py::list &args);
|
|
417
|
-
py::object PYNATIVE_EXPORT Pyboost_TransposeExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
418
|
-
py::object PYNATIVE_EXPORT Pyboost_Ones_Base(const PrimitivePtr &prim, const py::list &args);
|
|
419
|
-
py::object PYNATIVE_EXPORT Pyboost_ReplicationPad3D_Base(const PrimitivePtr &prim, const py::list &args);
|
|
420
|
-
py::object PYNATIVE_EXPORT Pyboost_FmodTensor_Base(const PrimitivePtr &prim, const py::list &args);
|
|
421
|
-
py::object PYNATIVE_EXPORT Pyboost_Erfinv_Base(const PrimitivePtr &prim, const py::list &args);
|
|
422
|
-
py::object PYNATIVE_EXPORT Pyboost_RotaryPositionEmbedding_Base(const PrimitivePtr &prim, const py::list &args);
|
|
423
|
-
py::object PYNATIVE_EXPORT Pyboost_Cast_Base(const PrimitivePtr &prim, const py::list &args);
|
|
424
|
-
py::object PYNATIVE_EXPORT Pyboost_L1LossBackwardExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
425
|
-
py::object PYNATIVE_EXPORT Pyboost_Hardtanh_Base(const PrimitivePtr &prim, const py::list &args);
|
|
426
|
-
py::object PYNATIVE_EXPORT Pyboost_Cross_Base(const PrimitivePtr &prim, const py::list &args);
|
|
427
|
-
py::object PYNATIVE_EXPORT Pyboost_NewOnes_Base(const PrimitivePtr &prim, const py::list &args);
|
|
428
|
-
py::object PYNATIVE_EXPORT Pyboost_Softmax_Base(const PrimitivePtr &prim, const py::list &args);
|
|
429
|
-
py::object PYNATIVE_EXPORT Pyboost_NonZero_Base(const PrimitivePtr &prim, const py::list &args);
|
|
430
|
-
py::object PYNATIVE_EXPORT Pyboost_SmoothL1LossGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
431
|
-
py::object PYNATIVE_EXPORT Pyboost_Eye_Base(const PrimitivePtr &prim, const py::list &args);
|
|
432
|
-
py::object PYNATIVE_EXPORT Pyboost_GatherD_Base(const PrimitivePtr &prim, const py::list &args);
|
|
433
|
-
py::object PYNATIVE_EXPORT Pyboost_AvgPool1D_Base(const PrimitivePtr &prim, const py::list &args);
|
|
434
|
-
py::object PYNATIVE_EXPORT Pyboost_PReLUGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
435
|
-
py::object PYNATIVE_EXPORT Pyboost_SoftMarginLoss_Base(const PrimitivePtr &prim, const py::list &args);
|
|
436
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceScatterValue_Base(const PrimitivePtr &prim, const py::list &args);
|
|
437
|
-
py::object PYNATIVE_EXPORT Pyboost_MultinomialExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
438
|
-
py::object PYNATIVE_EXPORT Pyboost_Conv1DPadding_Base(const PrimitivePtr &prim, const py::list &args);
|
|
439
|
-
py::object PYNATIVE_EXPORT Pyboost_AtanExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
440
|
-
py::object PYNATIVE_EXPORT Pyboost_TopkExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
441
|
-
py::object PYNATIVE_EXPORT Pyboost_ConvolutionGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
442
|
-
py::object PYNATIVE_EXPORT Pyboost_Sinh_Base(const PrimitivePtr &prim, const py::list &args);
|
|
443
|
-
py::object PYNATIVE_EXPORT Pyboost_AsStrided_Base(const PrimitivePtr &prim, const py::list &args);
|
|
444
|
-
py::object PYNATIVE_EXPORT Pyboost_SearchSorted_Base(const PrimitivePtr &prim, const py::list &args);
|
|
445
|
-
py::object PYNATIVE_EXPORT Pyboost_DistCommIsend_Base(const PrimitivePtr &prim, const py::list &args);
|
|
446
|
-
py::object PYNATIVE_EXPORT Pyboost_DistCommAllToAllV_Base(const PrimitivePtr &prim, const py::list &args);
|
|
447
|
-
py::object PYNATIVE_EXPORT Pyboost_Rsqrt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
448
|
-
py::object PYNATIVE_EXPORT Pyboost_IndexSelect_Base(const PrimitivePtr &prim, const py::list &args);
|
|
449
|
-
py::object PYNATIVE_EXPORT Pyboost_ReplicationPad2DGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
450
|
-
py::object PYNATIVE_EXPORT Pyboost_UpsampleNearest2D_Base(const PrimitivePtr &prim, const py::list &args);
|
|
451
|
-
py::object PYNATIVE_EXPORT Pyboost_HSigmoid_Base(const PrimitivePtr &prim, const py::list &args);
|
|
452
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceFloor_Base(const PrimitivePtr &prim, const py::list &args);
|
|
453
|
-
py::object PYNATIVE_EXPORT Pyboost_NormalTensorTensor_Base(const PrimitivePtr &prim, const py::list &args);
|
|
454
|
-
py::object PYNATIVE_EXPORT Pyboost_UniformExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
455
|
-
py::object PYNATIVE_EXPORT Pyboost_Kthvalue_Base(const PrimitivePtr &prim, const py::list &args);
|
|
456
|
-
py::object PYNATIVE_EXPORT Pyboost_ReverseV2_Base(const PrimitivePtr &prim, const py::list &args);
|
|
457
|
-
py::object PYNATIVE_EXPORT Pyboost_InplacePut_Base(const PrimitivePtr &prim, const py::list &args);
|
|
458
|
-
py::object PYNATIVE_EXPORT Pyboost_NLLLoss2d_Base(const PrimitivePtr &prim, const py::list &args);
|
|
459
|
-
py::object PYNATIVE_EXPORT Pyboost_ArgSort_Base(const PrimitivePtr &prim, const py::list &args);
|
|
460
|
-
py::object PYNATIVE_EXPORT Pyboost_Reshape_Base(const PrimitivePtr &prim, const py::list &args);
|
|
461
|
-
py::object PYNATIVE_EXPORT Pyboost_ScatterValue_Base(const PrimitivePtr &prim, const py::list &args);
|
|
462
|
-
py::object PYNATIVE_EXPORT Pyboost_AvgPool3DExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
463
|
-
py::object PYNATIVE_EXPORT Pyboost_Outer_Base(const PrimitivePtr &prim, const py::list &args);
|
|
464
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceDiv_Base(const PrimitivePtr &prim, const py::list &args);
|
|
465
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceThreshold_Base(const PrimitivePtr &prim, const py::list &args);
|
|
466
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceClampTensor_Base(const PrimitivePtr &prim, const py::list &args);
|
|
467
|
-
py::object PYNATIVE_EXPORT Pyboost_SiLU_Base(const PrimitivePtr &prim, const py::list &args);
|
|
468
|
-
py::object PYNATIVE_EXPORT Pyboost_LogSoftmaxExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
469
|
-
py::object PYNATIVE_EXPORT Pyboost_LayerNormGradExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
470
|
-
py::object PYNATIVE_EXPORT Pyboost_UpsampleBilinear2DGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
471
|
-
py::object PYNATIVE_EXPORT Pyboost_GridSampler3DGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
472
|
-
py::object PYNATIVE_EXPORT Pyboost_SeluGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
473
|
-
py::object PYNATIVE_EXPORT Pyboost_Div_Base(const PrimitivePtr &prim, const py::list &args);
|
|
474
|
-
py::object PYNATIVE_EXPORT Pyboost_IsFinite_Base(const PrimitivePtr &prim, const py::list &args);
|
|
475
|
-
py::object PYNATIVE_EXPORT Pyboost_LayerNormExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
476
|
-
py::object PYNATIVE_EXPORT Pyboost_Sinc_Base(const PrimitivePtr &prim, const py::list &args);
|
|
477
|
-
py::object PYNATIVE_EXPORT Pyboost_MaxPoolGradWithIndices_Base(const PrimitivePtr &prim, const py::list &args);
|
|
478
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceErfinv_Base(const PrimitivePtr &prim, const py::list &args);
|
|
479
|
-
py::object PYNATIVE_EXPORT Pyboost_Generator_Base(const PrimitivePtr &prim, const py::list &args);
|
|
480
|
-
py::object PYNATIVE_EXPORT Pyboost_MatMulExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
481
|
-
py::object PYNATIVE_EXPORT Pyboost_GeluGradExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
482
|
-
py::object PYNATIVE_EXPORT Pyboost_InnerCommAllToAllV_Base(const PrimitivePtr &prim, const py::list &args);
|
|
483
|
-
py::object PYNATIVE_EXPORT Pyboost_AcosExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
484
|
-
py::object PYNATIVE_EXPORT Pyboost_LeakyReLUExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
485
|
-
py::object PYNATIVE_EXPORT Pyboost_Minimum_Base(const PrimitivePtr &prim, const py::list &args);
|
|
486
|
-
py::object PYNATIVE_EXPORT Pyboost_SmoothL1Loss_Base(const PrimitivePtr &prim, const py::list &args);
|
|
487
|
-
py::object PYNATIVE_EXPORT Pyboost_RemainderTensorTensor_Base(const PrimitivePtr &prim, const py::list &args);
|
|
488
|
-
py::object PYNATIVE_EXPORT Pyboost_BroadcastTo_Base(const PrimitivePtr &prim, const py::list &args);
|
|
489
|
-
py::object PYNATIVE_EXPORT Pyboost_MSELossExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
490
|
-
py::object PYNATIVE_EXPORT Pyboost_XLogYScalarSelf_Base(const PrimitivePtr &prim, const py::list &args);
|
|
491
|
-
py::object PYNATIVE_EXPORT Pyboost_AdaptiveAvgPool2DGradExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
492
|
-
py::object PYNATIVE_EXPORT Pyboost_Narrow_Base(const PrimitivePtr &prim, const py::list &args);
|
|
493
|
-
py::object PYNATIVE_EXPORT Pyboost_Triu_Base(const PrimitivePtr &prim, const py::list &args);
|
|
494
|
-
py::object PYNATIVE_EXPORT Pyboost_GridSampler2DGrad_Base(const PrimitivePtr &prim, const py::list &args);
|
|
495
|
-
py::object PYNATIVE_EXPORT Pyboost_InnerCommIrecv_Base(const PrimitivePtr &prim, const py::list &args);
|
|
496
|
-
py::object PYNATIVE_EXPORT Pyboost_LogSoftmax_Base(const PrimitivePtr &prim, const py::list &args);
|
|
497
|
-
py::object PYNATIVE_EXPORT Pyboost_Trunc_Base(const PrimitivePtr &prim, const py::list &args);
|
|
498
|
-
py::object PYNATIVE_EXPORT Pyboost_InplaceExp_Base(const PrimitivePtr &prim, const py::list &args);
|
|
499
|
-
py::object PYNATIVE_EXPORT Pyboost_Concat_Base(const PrimitivePtr &prim, const py::list &args);
|
|
500
|
-
py::object PYNATIVE_EXPORT Pyboost_DiagExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
501
|
-
py::object PYNATIVE_EXPORT Pyboost_MatMul_Base(const PrimitivePtr &prim, const py::list &args);
|
|
502
|
-
py::object PYNATIVE_EXPORT Pyboost_AvgPool3DGradExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
503
|
-
py::object PYNATIVE_EXPORT Pyboost_FusedInferAttentionScore_Base(const PrimitivePtr &prim, const py::list &args);
|
|
504
|
-
py::object PYNATIVE_EXPORT Pyboost_QuantV2_Base(const PrimitivePtr &prim, const py::list &args);
|
|
505
|
-
py::object PYNATIVE_EXPORT Pyboost_QuantBatchMatmul_Base(const PrimitivePtr &prim, const py::list &args);
|
|
506
|
-
py::object PYNATIVE_EXPORT Pyboost_GroupedMatmul_Base(const PrimitivePtr &prim, const py::list &args);
|
|
507
|
-
py::object PYNATIVE_EXPORT Pyboost_MoeComputeExpertTokens_Base(const PrimitivePtr &prim, const py::list &args);
|
|
508
|
-
py::object PYNATIVE_EXPORT Pyboost_MoeGatingTopKSoftmax_Base(const PrimitivePtr &prim, const py::list &args);
|
|
509
|
-
py::object PYNATIVE_EXPORT Pyboost_GroupedMatmulV4_Base(const PrimitivePtr &prim, const py::list &args);
|
|
510
|
-
py::object PYNATIVE_EXPORT Pyboost_MoeInitRoutingV2_Base(const PrimitivePtr &prim, const py::list &args);
|
|
511
|
-
py::object PYNATIVE_EXPORT Pyboost_DynamicQuantExt_Base(const PrimitivePtr &prim, const py::list &args);
|
|
512
|
-
py::object PYNATIVE_EXPORT Pyboost_MoeInitRouting_Base(const PrimitivePtr &prim, const py::list &args);
|
|
513
|
-
py::object PYNATIVE_EXPORT Pyboost_KVCacheScatterUpdate_Base(const PrimitivePtr &prim, const py::list &args);
|
|
514
|
-
py::object PYNATIVE_EXPORT Pyboost_AddRmsNormQuantV2_Base(const PrimitivePtr &prim, const py::list &args);
|
|
515
|
-
py::object PYNATIVE_EXPORT Pyboost_WeightQuantBatchMatmul_Base(const PrimitivePtr &prim, const py::list &args);
|
|
516
|
-
py::object PYNATIVE_EXPORT Pyboost_MatmulAllReduceAddRmsNorm_Base(const PrimitivePtr &prim, const py::list &args);
|
|
517
|
-
py::object PYNATIVE_EXPORT Pyboost_MoeFinalizeRouting_Base(const PrimitivePtr &prim, const py::list &args);
|
|
518
|
-
py::object PYNATIVE_EXPORT Pyboost_GroupedMatmulV2_Base(const PrimitivePtr &prim, const py::list &args);
|
|
519
|
-
py::object PYNATIVE_EXPORT Pyboost_PixelShuffle_Base(const PrimitivePtr &prim, const py::list &args);
|
|
520
|
-
|
|
521
|
-
//pyboost op api
|
|
522
|
-
py::object ME_EXPORT Pyboost_GroupNorm_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& num_groups, const std::optional<ValuePtr>& weight, const std::optional<ValuePtr>& bias, const FP32ImmPtr& eps);
|
|
523
|
-
py::object ME_EXPORT Pyboost_GreaterEqualScalar_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ScalarPtr& other);
|
|
524
|
-
py::object ME_EXPORT Pyboost_DistCommAllGatherIntoTensor_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& other, const ValuePtr& input, const Int64ImmPtr& rank_size, const StringImmPtr& group);
|
|
525
|
-
py::object ME_EXPORT Pyboost_BatchNormGatherStatsWithCounts_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& mean, const ValuePtr& invstd, const std::optional<ValuePtr>& running_mean, const std::optional<ValuePtr>& running_var, const FP32ImmPtr& momentum, const FP32ImmPtr& eps, const std::optional<ValuePtr>& counts);
|
|
526
|
-
py::object ME_EXPORT Pyboost_ProdExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const std::optional<Int64ImmPtr>& dim, const BoolImmPtr& keepdim, const std::optional<Int64ImmPtr>& dtype);
|
|
527
|
-
py::object ME_EXPORT Pyboost_NormalTensorFloat_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& mean, const ScalarPtr& std, const ValuePtr& seed, const ValuePtr& offset);
|
|
528
|
-
py::object ME_EXPORT Pyboost_Scatter_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& dim, const ValuePtr& index, const ValuePtr& src, const Int64ImmPtr& reduce);
|
|
529
|
-
py::object ME_EXPORT Pyboost_ReduceAny_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x, const ValueTuplePtr& axis, const BoolImmPtr& keep_dims);
|
|
530
|
-
py::object ME_EXPORT Pyboost_Neg_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
531
|
-
py::object ME_EXPORT Pyboost_Dense_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& weight, const std::optional<ValuePtr>& bias);
|
|
532
|
-
py::object ME_EXPORT Pyboost_UpsampleBicubic2D_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x, const std::optional<ValueTuplePtr>& output_size, const std::optional<ValueTuplePtr>& scales, const BoolImmPtr& align_corners);
|
|
533
|
-
py::object ME_EXPORT Pyboost_RmsNormGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& dy, const ValuePtr& x, const ValuePtr& rstd, const ValuePtr& gamma);
|
|
534
|
-
py::object ME_EXPORT Pyboost_Sqrt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x);
|
|
535
|
-
py::object ME_EXPORT Pyboost_Tan_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
536
|
-
py::object ME_EXPORT Pyboost_RandnLike_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& seed, const ValuePtr& offset, const std::optional<Int64ImmPtr>& dtype);
|
|
537
|
-
py::object ME_EXPORT Pyboost_InplaceScatterSrc_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& dim, const ValuePtr& index, const ValuePtr& src);
|
|
538
|
-
py::object ME_EXPORT Pyboost_BitwiseOrTensor_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other);
|
|
539
|
-
py::object ME_EXPORT Pyboost_ReflectionPad3D_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& padding);
|
|
540
|
-
py::object ME_EXPORT Pyboost_BernoulliExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& seed, const ValuePtr& offset);
|
|
541
|
-
py::object ME_EXPORT Pyboost_Convolution_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& weight, const std::optional<ValuePtr>& bias, const ValueTuplePtr& stride, const ValueTuplePtr& padding, const ValueTuplePtr& dilation, const BoolImmPtr& transposed, const ValueTuplePtr& output_padding, const Int64ImmPtr& groups);
|
|
542
|
-
py::object ME_EXPORT Pyboost_Cummax_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& axis);
|
|
543
|
-
py::object ME_EXPORT Pyboost_ConvolutionStr_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& weight, const std::optional<ValuePtr>& bias, const ValueTuplePtr& stride, const Int64ImmPtr& padding, const ValueTuplePtr& dilation, const BoolImmPtr& transposed, const ValueTuplePtr& output_padding, const Int64ImmPtr& groups);
|
|
544
|
-
py::object ME_EXPORT Pyboost_GeLUGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& dy, const ValuePtr& x, const ValuePtr& y);
|
|
545
|
-
py::object ME_EXPORT Pyboost_TypeAs_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other);
|
|
546
|
-
py::object ME_EXPORT Pyboost_MoeTokenPermuteGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& permuted_tokens_grad, const ValuePtr& sorted_indices, const Int64ImmPtr& num_topk, const BoolImmPtr& padded_mode);
|
|
547
|
-
py::object ME_EXPORT Pyboost_InplaceAddsExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ScalarPtr& other, const ScalarPtr& alpha);
|
|
548
|
-
py::object ME_EXPORT Pyboost_UpsampleBicubic2DGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& dy, const ValueTuplePtr& input_size, const std::optional<ValueTuplePtr>& output_size, const std::optional<ValueTuplePtr>& scales, const BoolImmPtr& align_corners);
|
|
549
|
-
py::object ME_EXPORT Pyboost_ZerosLikeExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const std::optional<Int64ImmPtr>& dtype);
|
|
550
|
-
py::object ME_EXPORT Pyboost_ReduceAll_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const std::optional<ValueTuplePtr>& axis, const BoolImmPtr& keep_dims);
|
|
551
|
-
py::object ME_EXPORT Pyboost_Round_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& decimals);
|
|
552
|
-
py::object ME_EXPORT Pyboost_SilentCheckV3_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& val, const ValuePtr& max, const ValuePtr& avg, const ValuePtr& input_grad, const ValuePtr& step, const FP32ImmPtr& c_thresh_l1, const FP32ImmPtr& c_thresh_l2, const FP32ImmPtr& beta1, const Int64ImmPtr& npu_asd_detect);
|
|
553
|
-
py::object ME_EXPORT Pyboost_Mm_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& mat2);
|
|
554
|
-
py::object ME_EXPORT Pyboost_AdaptiveMaxPool2D_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& output_size);
|
|
555
|
-
py::object ME_EXPORT Pyboost_DistCommScatter_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& other, const ValueTuplePtr& scatter_list, const Int64ImmPtr& rank_size, const Int64ImmPtr& src, const Int64ImmPtr& rank_id, const StringImmPtr& group);
|
|
556
|
-
py::object ME_EXPORT Pyboost_DropoutExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const FP32ImmPtr& p, const ValuePtr& seed, const ValuePtr& offset);
|
|
557
|
-
py::object ME_EXPORT Pyboost_MinDim_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& dim, const BoolImmPtr& keepdim);
|
|
558
|
-
py::object ME_EXPORT Pyboost_NanToNum_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const std::optional<FP32ImmPtr>& nan, const std::optional<FP32ImmPtr>& posinf, const std::optional<FP32ImmPtr>& neginf);
|
|
559
|
-
py::object ME_EXPORT Pyboost_DistCommBatchIsendIrecv_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValueTuplePtr& input, const StringImmPtr& group, const ValueTuplePtr& op_types, const ValueTuplePtr& remotes_ranks);
|
|
560
|
-
py::object ME_EXPORT Pyboost_RotaryPositionEmbeddingGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& dy, const ValuePtr& cos, const ValuePtr& sin, const std::optional<ValuePtr>& dx, const Int64ImmPtr& mode);
|
|
561
|
-
py::object ME_EXPORT Pyboost_HShrink_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const FP32ImmPtr& lambd);
|
|
562
|
-
py::object ME_EXPORT Pyboost_DistCommAllGather_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValueTuplePtr& tensor_list, const ValuePtr& input, const Int64ImmPtr& rank_size, const StringImmPtr& group);
|
|
563
|
-
py::object ME_EXPORT Pyboost_InplaceScatterAdd_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& dim, const ValuePtr& index, const ValuePtr& src);
|
|
564
|
-
py::object ME_EXPORT Pyboost_AddcdivExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& tensor1, const ValuePtr& tensor2, const ScalarPtr& value);
|
|
565
|
-
py::object ME_EXPORT Pyboost_MaxDim_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& dim, const BoolImmPtr& keepdim);
|
|
566
|
-
py::object ME_EXPORT Pyboost_InplaceDivMods_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ScalarPtr& other, const std::optional<Int64ImmPtr>& rounding_mode);
|
|
567
|
-
py::object ME_EXPORT Pyboost_LinSpaceExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ScalarPtr& start, const ScalarPtr& end, const Int64ImmPtr& steps, const std::optional<Int64ImmPtr>& dtype);
|
|
568
|
-
py::object ME_EXPORT Pyboost_Addbmm_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& batch1, const ValuePtr& batch2, const ScalarPtr& beta, const ScalarPtr& alpha);
|
|
569
|
-
py::object ME_EXPORT Pyboost_Embedding_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& weight, const std::optional<Int64ImmPtr>& padding_idx, const std::optional<FP32ImmPtr>& max_norm, const FP32ImmPtr& norm_type, const BoolImmPtr& scale_grad_by_freq);
|
|
570
|
-
py::object ME_EXPORT Pyboost_InplaceMaskedFillScalar_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& mask, const ScalarPtr& value);
|
|
571
|
-
py::object ME_EXPORT Pyboost_InplaceLog_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
572
|
-
py::object ME_EXPORT Pyboost_MaxPoolWithMask_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x, const ValueTuplePtr& kernel_size, const std::optional<ValueTuplePtr>& strides, const ValueTuplePtr& pads, const ValueTuplePtr& dilation, const BoolImmPtr& ceil_mode, const Int64ImmPtr& argmax_type);
|
|
573
|
-
py::object ME_EXPORT Pyboost_DistCommScatterTensor_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& other, const ValuePtr& input, const Int64ImmPtr& rank_size, const Int64ImmPtr& src, const Int64ImmPtr& rank_id, const StringImmPtr& group);
|
|
574
|
-
py::object ME_EXPORT Pyboost_BatchMatMul_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x, const ValuePtr& y, const BoolImmPtr& transpose_a, const BoolImmPtr& transpose_b);
|
|
575
|
-
py::object ME_EXPORT Pyboost_SpeedFusionAttention_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& query, const ValuePtr& key, const ValuePtr& value, const Int64ImmPtr& head_num, const Int64ImmPtr& input_layout, const ValuePtr& seed, const ValuePtr& offset, const std::optional<ValuePtr>& pse, const std::optional<ValuePtr>& padding_mask, const std::optional<ValuePtr>& atten_mask, const FP32ImmPtr& scale, const FP32ImmPtr& keep_prob, const Int64ImmPtr& pre_tokens, const Int64ImmPtr& next_tokens, const Int64ImmPtr& inner_precise, const std::optional<ValueTuplePtr>& prefix, const std::optional<ValueTuplePtr>& actual_seq_qlen, const std::optional<ValueTuplePtr>& actual_seq_kvlen, const Int64ImmPtr& sparse_mode, const BoolImmPtr& gen_mask_parallel, const BoolImmPtr& sync, const Int64ImmPtr& pse_type, const std::optional<ValueTuplePtr>& q_start_idx, const std::optional<ValueTuplePtr>& kv_start_idx);
|
|
576
|
-
py::object ME_EXPORT Pyboost_InnerNonZero_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
577
|
-
py::object ME_EXPORT Pyboost_Mv_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& vec);
|
|
578
|
-
py::object ME_EXPORT Pyboost_UniqueConsecutive_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const BoolImmPtr& return_inverse, const BoolImmPtr& return_counts, const std::optional<Int64ImmPtr>& dim);
|
|
579
|
-
py::object ME_EXPORT Pyboost_ReluGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& y_backprop, const ValuePtr& x);
|
|
580
|
-
py::object ME_EXPORT Pyboost_LogicalAnd_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x, const ValuePtr& y);
|
|
581
|
-
py::object ME_EXPORT Pyboost_InplaceScatterValueReduce_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& dim, const ValuePtr& index, const ScalarPtr& value, const Int64ImmPtr& reduce);
|
|
582
|
-
py::object ME_EXPORT Pyboost_PowScalarTensor_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ScalarPtr& input, const ValuePtr& exponent);
|
|
583
|
-
py::object ME_EXPORT Pyboost_OneHotExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& tensor, const Int64ImmPtr& num_classes, const ValuePtr& on_value, const ValuePtr& off_value, const Int64ImmPtr& axis);
|
|
584
|
-
py::object ME_EXPORT Pyboost_BatchNormElemtGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& dout, const ValuePtr& input, const ValuePtr& mean, const ValuePtr& invstd, const ValuePtr& weight, const ValuePtr& sumd_dy, const ValuePtr& sum_dy_xmu, const ValuePtr& count);
|
|
585
|
-
py::object ME_EXPORT Pyboost_IsNegInf_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
586
|
-
py::object ME_EXPORT Pyboost_FloorDiv_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other);
|
|
587
|
-
py::object ME_EXPORT Pyboost_Contiguous_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
588
|
-
py::object ME_EXPORT Pyboost_SubExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other, const ScalarPtr& alpha);
|
|
589
|
-
py::object ME_EXPORT Pyboost_SplitWithSize_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& split_size, const Int64ImmPtr& dim);
|
|
590
|
-
py::object ME_EXPORT Pyboost_Erfc_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
591
|
-
py::object ME_EXPORT Pyboost_IsInf_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
592
|
-
py::object ME_EXPORT Pyboost_LogicalNot_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
593
|
-
py::object ME_EXPORT Pyboost_AddRmsNorm_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x1, const ValuePtr& x2, const ValuePtr& gamma, const FP32ImmPtr& epsilon);
|
|
594
|
-
py::object ME_EXPORT Pyboost_ArgMinExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const std::optional<Int64ImmPtr>& dim, const BoolImmPtr& keepdim);
|
|
595
|
-
py::object ME_EXPORT Pyboost_UpsampleNearest1DGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& dy, const ValueTuplePtr& input_size, const std::optional<ValueTuplePtr>& output_size, const std::optional<ValueTuplePtr>& scales);
|
|
596
|
-
py::object ME_EXPORT Pyboost_Slice_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& begin, const ValueTuplePtr& size);
|
|
597
|
-
py::object ME_EXPORT Pyboost_Arange_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ScalarPtr& start, const ScalarPtr& end, const ScalarPtr& step, const std::optional<Int64ImmPtr>& dtype);
|
|
598
|
-
py::object ME_EXPORT Pyboost_RepeatInterleaveGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& repeats, const Int64ImmPtr& dim);
|
|
599
|
-
py::object ME_EXPORT Pyboost_Pow_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& exponent);
|
|
600
|
-
py::object ME_EXPORT Pyboost_AllGatherMatmul_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& x2, const StringImmPtr& group, const Int64ImmPtr& world_size, const std::optional<ValuePtr>& bias, const Int64ImmPtr& gather_index, const BoolImmPtr& gather_output, const Int64ImmPtr& comm_turn, const BoolImmPtr& trans_input, const BoolImmPtr& trans_x2);
|
|
601
|
-
py::object ME_EXPORT Pyboost_Less_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other);
|
|
602
|
-
py::object ME_EXPORT Pyboost_InplaceFloorDivides_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ScalarPtr& other);
|
|
603
|
-
py::object ME_EXPORT Pyboost_SoftShrink_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const FP32ImmPtr& lambd);
|
|
604
|
-
py::object ME_EXPORT Pyboost_ConvolutionStrGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& dout, const ValuePtr& input, const ValuePtr& weight, const std::optional<ValuePtr>& bias, const ValueTuplePtr& stride, const Int64ImmPtr& padding, const ValueTuplePtr& dilation, const BoolImmPtr& transposed, const ValueTuplePtr& output_padding, const Int64ImmPtr& groups, const ValueTuplePtr& output_mask);
|
|
605
|
-
py::object ME_EXPORT Pyboost_ViewAs_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other);
|
|
606
|
-
py::object ME_EXPORT Pyboost_FFNExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x, const ValuePtr& weight1, const ValuePtr& weight2, const std::optional<ValueTuplePtr>& expertTokens, const std::optional<ValuePtr>& bias1, const std::optional<ValuePtr>& bias2, const std::optional<ValuePtr>& scale, const std::optional<ValuePtr>& offset, const std::optional<ValuePtr>& deqScale1, const std::optional<ValuePtr>& deqScale2, const std::optional<ValuePtr>& antiquant_scale1, const std::optional<ValuePtr>& antiquant_scale2, const std::optional<ValuePtr>& antiquant_offset1, const std::optional<ValuePtr>& antiquant_offset2, const Int64ImmPtr& activation, const Int64ImmPtr& inner_precise);
|
|
607
|
-
py::object ME_EXPORT Pyboost_LinalgQr_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& A, const Int64ImmPtr& mode);
|
|
608
|
-
py::object ME_EXPORT Pyboost_BinaryCrossEntropyWithLogitsBackward_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& grad_output, const ValuePtr& input, const ValuePtr& target, const std::optional<ValuePtr>& weight, const std::optional<ValuePtr>& posWeight, const Int64ImmPtr& reduction);
|
|
609
|
-
py::object ME_EXPORT Pyboost_EluExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const FP32ImmPtr& alpha);
|
|
610
|
-
py::object ME_EXPORT Pyboost_Maximum_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other);
|
|
611
|
-
py::object ME_EXPORT Pyboost_GridSampler3D_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input_x, const ValuePtr& grid, const Int64ImmPtr& interpolation_mode, const Int64ImmPtr& padding_mode, const BoolImmPtr& align_corners);
|
|
612
|
-
py::object ME_EXPORT Pyboost_MeanExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const std::optional<ValueTuplePtr>& dim, const BoolImmPtr& keepdim, const std::optional<Int64ImmPtr>& dtype);
|
|
613
|
-
py::object ME_EXPORT Pyboost_Atanh_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
614
|
-
py::object ME_EXPORT Pyboost_AvgPool2D_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& kernel_size, const ValueTuplePtr& stride, const ValueTuplePtr& padding, const BoolImmPtr& ceil_mode, const BoolImmPtr& count_include_pad, const std::optional<Int64ImmPtr>& divisor_override);
|
|
615
|
-
py::object ME_EXPORT Pyboost_MultiScaleDeformableAttn_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& value, const ValuePtr& shape, const ValuePtr& offset, const ValuePtr& locations, const ValuePtr& weight);
|
|
616
|
-
py::object ME_EXPORT Pyboost_AddLayerNormGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& dy, const ValuePtr& x1, const ValuePtr& x2, const ValuePtr& rstd, const ValuePtr& mean, const ValuePtr& gamma, const ValuePtr& dsumOptional);
|
|
617
|
-
py::object ME_EXPORT Pyboost_IncreFlashAttention_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& query, const ValueTuplePtr& key, const ValueTuplePtr& value, const std::optional<ValuePtr>& attn_mask, const std::optional<ValuePtr>& actual_seq_lengths, const std::optional<ValuePtr>& pse_shift, const std::optional<ValuePtr>& dequant_scale1, const std::optional<ValuePtr>& quant_scale1, const std::optional<ValuePtr>& dequant_scale2, const std::optional<ValuePtr>& quant_scale2, const std::optional<ValuePtr>& quant_offset2, const std::optional<ValuePtr>& antiquant_scale, const std::optional<ValuePtr>& antiquant_offset, const std::optional<ValuePtr>& block_table, const std::optional<ValuePtr>& kv_padding_size, const Int64ImmPtr& num_heads, const Int64ImmPtr& input_layout, const FP32ImmPtr& scale_value, const Int64ImmPtr& num_key_value_heads, const Int64ImmPtr& block_size, const Int64ImmPtr& inner_precise);
|
|
618
|
-
py::object ME_EXPORT Pyboost_BitwiseAndScalar_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ScalarPtr& other);
|
|
619
|
-
py::object ME_EXPORT Pyboost_InplaceReLU_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
620
|
-
py::object ME_EXPORT Pyboost_Expm1_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
621
|
-
py::object ME_EXPORT Pyboost_FmodScalar_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ScalarPtr& other);
|
|
622
|
-
py::object ME_EXPORT Pyboost_Repeat_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& repeats);
|
|
623
|
-
py::object ME_EXPORT Pyboost_NewZeros_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& size, const std::optional<Int64ImmPtr>& dtype);
|
|
624
|
-
py::object ME_EXPORT Pyboost_SigmoidGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& y, const ValuePtr& dy);
|
|
625
|
-
py::object ME_EXPORT Pyboost_Baddbmm_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& batch1, const ValuePtr& batch2, const ScalarPtr& beta, const ScalarPtr& alpha);
|
|
626
|
-
py::object ME_EXPORT Pyboost_GatherDGradV2_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x, const Int64ImmPtr& dim, const ValuePtr& index, const ValuePtr& dout);
|
|
627
|
-
py::object ME_EXPORT Pyboost_GmmV2Backward_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValueTuplePtr& grad, const ValueTuplePtr& x, const ValueTuplePtr& weight, const std::optional<ValuePtr>& group_list, const Int64ImmPtr& group_list_type);
|
|
628
|
-
py::object ME_EXPORT Pyboost_RmsNorm_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x, const ValuePtr& gamma, const FP32ImmPtr& epsilon);
|
|
629
|
-
py::object ME_EXPORT Pyboost_MaxPoolGradWithMask_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x, const ValuePtr& grad, const ValuePtr& mask, const ValueTuplePtr& kernel_size, const std::optional<ValueTuplePtr>& strides, const ValueTuplePtr& pads, const ValueTuplePtr& dilation, const BoolImmPtr& ceil_mode, const Int64ImmPtr& argmax_type);
|
|
630
|
-
py::object ME_EXPORT Pyboost_SiLUGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& dout, const ValuePtr& x);
|
|
631
|
-
py::object ME_EXPORT Pyboost_Xlogy_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other);
|
|
632
|
-
py::object ME_EXPORT Pyboost_Reciprocal_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
633
|
-
py::object ME_EXPORT Pyboost_AdaptiveMaxPool1D_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& output_size);
|
|
634
|
-
py::object ME_EXPORT Pyboost_StackExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValueTuplePtr& tensors, const Int64ImmPtr& dim);
|
|
635
|
-
py::object ME_EXPORT Pyboost_AdamW_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& var, const ValuePtr& m, const ValuePtr& v, const ValuePtr& max_v, const ValuePtr& gradient, const ValuePtr& step, const FP32ImmPtr& lr, const FP32ImmPtr& beta1, const FP32ImmPtr& beta2, const FP32ImmPtr& decay, const FP32ImmPtr& eps, const BoolImmPtr& amsgrad, const BoolImmPtr& maximize);
|
|
636
|
-
py::object ME_EXPORT Pyboost_BitwiseNot_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
637
|
-
py::object ME_EXPORT Pyboost_RepeatInterleaveTensor_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& repeats, const std::optional<Int64ImmPtr>& dim, const std::optional<Int64ImmPtr>& output_size);
|
|
638
|
-
py::object ME_EXPORT Pyboost_Lerp_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& end, const ValuePtr& weight);
|
|
639
|
-
py::object ME_EXPORT Pyboost_XLogYScalarOther_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ScalarPtr& other);
|
|
640
|
-
py::object ME_EXPORT Pyboost_NLLLoss2dGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& loss_grad, const ValuePtr& input, const ValuePtr& target, const ValuePtr& weight, const Int64ImmPtr& reduction, const Int64ImmPtr& ignore_index, const ValuePtr& total_weight);
|
|
641
|
-
py::object ME_EXPORT Pyboost_IndexFillScalar_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& dim, const ValuePtr& index, const ScalarPtr& value);
|
|
642
|
-
py::object ME_EXPORT Pyboost_CumminExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& dim);
|
|
643
|
-
py::object ME_EXPORT Pyboost_KLDiv_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& target, const Int64ImmPtr& reduction, const BoolImmPtr& log_target);
|
|
644
|
-
py::object ME_EXPORT Pyboost_GmmBackward_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValueTuplePtr& grad, const ValueTuplePtr& x, const ValueTuplePtr& weight, const std::optional<ValueTuplePtr>& group_list);
|
|
645
|
-
py::object ME_EXPORT Pyboost_LogicalXor_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other);
|
|
646
|
-
py::object ME_EXPORT Pyboost_MaskedSelectGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& mask, const ValuePtr& grad);
|
|
647
|
-
py::object ME_EXPORT Pyboost_BatchNormGradExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& dout, const ValuePtr& input, const ValuePtr& weight, const std::optional<ValuePtr>& running_mean, const std::optional<ValuePtr>& running_var, const std::optional<ValuePtr>& saved_mean, const std::optional<ValuePtr>& saved_rstd, const BoolImmPtr& training, const FP32ImmPtr& eps, const ValueTuplePtr& output_mask);
|
|
648
|
-
py::object ME_EXPORT Pyboost_FlattenExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& start_dim, const Int64ImmPtr& end_dim);
|
|
649
|
-
py::object ME_EXPORT Pyboost_UpsampleNearest3D_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x, const std::optional<ValueTuplePtr>& output_size, const std::optional<ValueTuplePtr>& scales);
|
|
650
|
-
py::object ME_EXPORT Pyboost_Exp_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
651
|
-
py::object ME_EXPORT Pyboost_Sin_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
652
|
-
py::object ME_EXPORT Pyboost_Atan2Ext_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other);
|
|
653
|
-
py::object ME_EXPORT Pyboost_RemainderTensorScalar_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ScalarPtr& other);
|
|
654
|
-
py::object ME_EXPORT Pyboost_InnerCommAllReduce_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const StringImmPtr& op_type, const StringImmPtr& group);
|
|
655
|
-
py::object ME_EXPORT Pyboost_FillTensor_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValueTuplePtr& size, const ValuePtr& fill_value, const std::optional<Int64ImmPtr>& dtype);
|
|
656
|
-
py::object ME_EXPORT Pyboost_IndexFillTensor_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& dim, const ValuePtr& index, const ValuePtr& value);
|
|
657
|
-
py::object ME_EXPORT Pyboost_InplaceMaskedFillTensor_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& mask, const ValuePtr& value);
|
|
658
|
-
py::object ME_EXPORT Pyboost_RepeatInterleaveInt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& repeats, const std::optional<Int64ImmPtr>& dim, const std::optional<Int64ImmPtr>& output_size);
|
|
659
|
-
py::object ME_EXPORT Pyboost_Ceil_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
660
|
-
py::object ME_EXPORT Pyboost_Roll_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& shifts, const std::optional<ValueTuplePtr>& dims);
|
|
661
|
-
py::object ME_EXPORT Pyboost_SortExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& dim, const BoolImmPtr& descending, const BoolImmPtr& stable);
|
|
662
|
-
py::object ME_EXPORT Pyboost_InplaceGroupedMatmulAdd_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x, const ValuePtr& weight, const ValuePtr& group_list, const ValuePtr& out);
|
|
663
|
-
py::object ME_EXPORT Pyboost_InplaceTanh_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
664
|
-
py::object ME_EXPORT Pyboost_InplaceStopGradient_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
665
|
-
py::object ME_EXPORT Pyboost_InplaceHardtanh_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ScalarPtr& min_val, const ScalarPtr& max_val);
|
|
666
|
-
py::object ME_EXPORT Pyboost_DivMods_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ScalarPtr& other, const std::optional<Int64ImmPtr>& rounding_mode);
|
|
667
|
-
py::object ME_EXPORT Pyboost_RandLikeExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& tensor, const ValuePtr& seed, const ValuePtr& offset, const std::optional<Int64ImmPtr>& dtype);
|
|
668
|
-
py::object ME_EXPORT Pyboost_LogAddExp2_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other);
|
|
669
|
-
py::object ME_EXPORT Pyboost_ArgMinWithValue_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& axis, const BoolImmPtr& keep_dims);
|
|
670
|
-
py::object ME_EXPORT Pyboost_Im2ColExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& kernel_size, const ValueTuplePtr& dilation, const ValueTuplePtr& padding, const ValueTuplePtr& stride);
|
|
671
|
-
py::object ME_EXPORT Pyboost_Squeeze_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& axis);
|
|
672
|
-
py::object ME_EXPORT Pyboost_Copy_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
673
|
-
py::object ME_EXPORT Pyboost_LinalgVectorNorm_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x, const FP32ImmPtr& ord, const std::optional<ValueTuplePtr>& dim, const BoolImmPtr& keepdim, const std::optional<Int64ImmPtr>& dtype);
|
|
674
|
-
py::object ME_EXPORT Pyboost_Chunk_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& chunks, const Int64ImmPtr& dim);
|
|
675
|
-
py::object ME_EXPORT Pyboost_RandInt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const Int64ImmPtr& low, const Int64ImmPtr& high, const ValueTuplePtr& shape, const ValuePtr& seed, const ValuePtr& offset, const std::optional<Int64ImmPtr>& dtype);
|
|
676
|
-
py::object ME_EXPORT Pyboost_Frac_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
677
|
-
py::object ME_EXPORT Pyboost_BinaryCrossEntropyGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& target, const ValuePtr& grad_output, const std::optional<ValuePtr>& weight, const Int64ImmPtr& reduction);
|
|
678
|
-
py::object ME_EXPORT Pyboost_DropoutGenMaskExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValueTuplePtr& shape, const FP32ImmPtr& p, const ValuePtr& seed, const ValuePtr& offset, const Int64ImmPtr& dtype);
|
|
679
|
-
py::object ME_EXPORT Pyboost_AllFinite_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValueTuplePtr& tensors);
|
|
680
|
-
py::object ME_EXPORT Pyboost_InplaceZero_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
681
|
-
py::object ME_EXPORT Pyboost_LeakyReLUGradExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& dy, const ValuePtr& input, const ScalarPtr& negative_slope, const BoolImmPtr& is_result);
|
|
682
|
-
py::object ME_EXPORT Pyboost_SwigluGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& grad_output, const ValuePtr& input, const Int64ImmPtr& dim);
|
|
683
|
-
py::object ME_EXPORT Pyboost_Polar_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& abs, const ValuePtr& angle);
|
|
684
|
-
py::object ME_EXPORT Pyboost_InplaceRandom_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& from_, const std::optional<Int64ImmPtr>& to, const ValuePtr& seed, const ValuePtr& offset);
|
|
685
|
-
py::object ME_EXPORT Pyboost_Take_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& index);
|
|
686
|
-
py::object ME_EXPORT Pyboost_Cosh_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
687
|
-
py::object ME_EXPORT Pyboost_SoftShrinkGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input_grad, const ValuePtr& input_x, const FP32ImmPtr& lambd);
|
|
688
|
-
py::object ME_EXPORT Pyboost_KLDivGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& grad_output, const ValuePtr& input, const ValuePtr& target, const Int64ImmPtr& reduction, const BoolImmPtr& log_target);
|
|
689
|
-
py::object ME_EXPORT Pyboost_Clone_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
690
|
-
py::object ME_EXPORT Pyboost_SpeedFusionAttentionGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& query, const ValuePtr& key, const ValuePtr& value, const ValuePtr& dy, const Int64ImmPtr& head_num, const Int64ImmPtr& input_layout, const std::optional<ValuePtr>& pse, const std::optional<ValuePtr>& padding_mask, const std::optional<ValuePtr>& atten_mask, const std::optional<ValuePtr>& softmax_max, const std::optional<ValuePtr>& softmax_sum, const std::optional<ValuePtr>& softmax_in, const std::optional<ValuePtr>& attention_in, const FP32ImmPtr& scale_value, const FP32ImmPtr& keep_prob, const Int64ImmPtr& pre_tokens, const Int64ImmPtr& next_tokens, const Int64ImmPtr& inner_precise, const std::optional<ValuePtr>& seed, const std::optional<ValuePtr>& offset, const std::optional<ValuePtr>& numels, const std::optional<ValueTuplePtr>& prefix, const std::optional<ValueTuplePtr>& actual_seq_qlen, const std::optional<ValueTuplePtr>& actual_seq_kvlen, const Int64ImmPtr& sparse_mode, const BoolImmPtr& gen_mask_parallel, const BoolImmPtr& sync, const Int64ImmPtr& pse_type, const std::optional<ValueTuplePtr>& q_start_idx, const std::optional<ValueTuplePtr>& kv_start_idx);
|
|
691
|
-
py::object ME_EXPORT Pyboost_Add_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other);
|
|
692
|
-
py::object ME_EXPORT Pyboost_AddLayerNormV2_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x1, const ValuePtr& x2, const ValuePtr& gamma, const ValuePtr& beta, const FP32ImmPtr& epsilon, const BoolImmPtr& additionalOut);
|
|
693
|
-
py::object ME_EXPORT Pyboost_ReflectionPad1D_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& padding);
|
|
694
|
-
py::object ME_EXPORT Pyboost_UniqueDim_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const BoolImmPtr& sorted, const BoolImmPtr& return_inverse, const Int64ImmPtr& dim);
|
|
695
|
-
py::object ME_EXPORT Pyboost_Abs_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
696
|
-
py::object ME_EXPORT Pyboost_BatchNormElemt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const std::optional<ValuePtr>& weight, const std::optional<ValuePtr>& bias, const std::optional<ValuePtr>& mean, const std::optional<ValuePtr>& invstd, const FP32ImmPtr& eps);
|
|
697
|
-
py::object ME_EXPORT Pyboost_InplaceFillScalar_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ScalarPtr& value);
|
|
698
|
-
py::object ME_EXPORT Pyboost_SubScalar_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ScalarPtr& other, const ScalarPtr& alpha);
|
|
699
|
-
py::object ME_EXPORT Pyboost_FlashAttentionScoreGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& query, const ValuePtr& key, const ValuePtr& value, const ValuePtr& dy, const std::optional<ValuePtr>& pse_shift, const std::optional<ValuePtr>& drop_mask, const std::optional<ValuePtr>& padding_mask, const std::optional<ValuePtr>& atten_mask, const std::optional<ValuePtr>& softmax_max, const std::optional<ValuePtr>& softmax_sum, const std::optional<ValuePtr>& softmax_in, const std::optional<ValuePtr>& attention_in, const std::optional<ValueTuplePtr>& prefix, const std::optional<ValueTuplePtr>& actual_seq_qlen, const std::optional<ValueTuplePtr>& actual_seq_kvlen, const Int64ImmPtr& head_num, const FP32ImmPtr& keep_prob, const FP32ImmPtr& scale_value, const Int64ImmPtr& pre_tokens, const Int64ImmPtr& next_tokens, const Int64ImmPtr& inner_precise, const Int64ImmPtr& input_layout, const Int64ImmPtr& sparse_mode);
|
|
700
|
-
py::object ME_EXPORT Pyboost_NonZeroExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
701
|
-
py::object ME_EXPORT Pyboost_NormalFloatFloat_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ScalarPtr& mean, const ScalarPtr& std, const ValueTuplePtr& size, const ValuePtr& seed, const ValuePtr& offset);
|
|
702
|
-
py::object ME_EXPORT Pyboost_MoeTokenUnpermuteGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& permuted_tokens, const ValuePtr& unpermuted_tokens_grad, const ValuePtr& sorted_indices, const std::optional<ValuePtr>& probs, const BoolImmPtr& padded_mode, const std::optional<ValueTuplePtr>& restore_shape);
|
|
703
|
-
py::object ME_EXPORT Pyboost_MaxPoolWithIndices_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x, const ValueTuplePtr& kernel_size, const std::optional<ValueTuplePtr>& strides, const ValueTuplePtr& pads, const ValueTuplePtr& dilation, const BoolImmPtr& ceil_mode, const Int64ImmPtr& argmax_type);
|
|
704
|
-
py::object ME_EXPORT Pyboost_EqualExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other);
|
|
705
|
-
py::object ME_EXPORT Pyboost_Sigmoid_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
706
|
-
py::object ME_EXPORT Pyboost_InplaceMul_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other);
|
|
707
|
-
py::object ME_EXPORT Pyboost_InplaceElu_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const FP32ImmPtr& alpha);
|
|
708
|
-
py::object ME_EXPORT Pyboost_TriangularSolve_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& b, const ValuePtr& A, const BoolImmPtr& upper, const BoolImmPtr& transpose, const BoolImmPtr& unitriangular);
|
|
709
|
-
py::object ME_EXPORT Pyboost_Unique2_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const BoolImmPtr& sorted, const BoolImmPtr& return_inverse, const BoolImmPtr& return_counts);
|
|
710
|
-
py::object ME_EXPORT Pyboost_GridSampler2D_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input_x, const ValuePtr& grid, const Int64ImmPtr& interpolation_mode, const Int64ImmPtr& padding_mode, const BoolImmPtr& align_corners);
|
|
711
|
-
py::object ME_EXPORT Pyboost_ReplicationPad1D_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& padding);
|
|
712
|
-
py::object ME_EXPORT Pyboost_InnerCommIsend_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& dst, const StringImmPtr& group, const Int64ImmPtr& tag);
|
|
713
|
-
py::object ME_EXPORT Pyboost_RemainderScalarTensor_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ScalarPtr& input, const ValuePtr& other);
|
|
714
|
-
py::object ME_EXPORT Pyboost_MatmulReduceScatter_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& x2, const StringImmPtr& group, const Int64ImmPtr& world_size, const Int64ImmPtr& reduce_op, const std::optional<ValuePtr>& bias, const Int64ImmPtr& comm_turn, const BoolImmPtr& trans_input, const BoolImmPtr& trans_x2);
|
|
715
|
-
py::object ME_EXPORT Pyboost_SelectExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& dim, const Int64ImmPtr& index);
|
|
716
|
-
py::object ME_EXPORT Pyboost_ReflectionPad1DGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& grad_output, const ValuePtr& input, const ValueTuplePtr& padding);
|
|
717
|
-
py::object ME_EXPORT Pyboost_DistCommBroadcast_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& src, const Int64ImmPtr& rank_id, const StringImmPtr& group);
|
|
718
|
-
py::object ME_EXPORT Pyboost_BitwiseXorTensor_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other);
|
|
719
|
-
py::object ME_EXPORT Pyboost_NotEqual_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other);
|
|
720
|
-
py::object ME_EXPORT Pyboost_Threshold_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const FP32ImmPtr& threshold, const FP32ImmPtr& value);
|
|
721
|
-
py::object ME_EXPORT Pyboost_MishExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
722
|
-
py::object ME_EXPORT Pyboost_DistCommReduce_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const StringImmPtr& op_type, const Int64ImmPtr& dst, const StringImmPtr& group);
|
|
723
|
-
py::object ME_EXPORT Pyboost_DistCommIrecv_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& tag, const Int64ImmPtr& src, const StringImmPtr& group);
|
|
724
|
-
py::object ME_EXPORT Pyboost_MaskedSelect_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& mask);
|
|
725
|
-
py::object ME_EXPORT Pyboost_MedianDim_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& dim, const BoolImmPtr& keepdim);
|
|
726
|
-
py::object ME_EXPORT Pyboost_HShrinkGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& gradients, const ValuePtr& features, const FP32ImmPtr& lambd);
|
|
727
|
-
py::object ME_EXPORT Pyboost_LogAddExp_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other);
|
|
728
|
-
py::object ME_EXPORT Pyboost_DistCommGatherIntoTensor_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& other, const ValuePtr& input, const Int64ImmPtr& rank_size, const Int64ImmPtr& dst, const Int64ImmPtr& rank_id, const StringImmPtr& group);
|
|
729
|
-
py::object ME_EXPORT Pyboost_ScatterAddExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& dim, const ValuePtr& index, const ValuePtr& src);
|
|
730
|
-
py::object ME_EXPORT Pyboost_Equal_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other);
|
|
731
|
-
py::object ME_EXPORT Pyboost_DistCommReduceScatterTensor_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& other, const ValuePtr& input, const Int64ImmPtr& rank_size, const StringImmPtr& op_type, const StringImmPtr& group);
|
|
732
|
-
py::object ME_EXPORT Pyboost_InplaceAddmm_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& mat1, const ValuePtr& mat2, const ScalarPtr& beta, const ScalarPtr& alpha);
|
|
733
|
-
py::object ME_EXPORT Pyboost_FloorDivScalar_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ScalarPtr& other);
|
|
734
|
-
py::object ME_EXPORT Pyboost_BatchNormReduceGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& dout, const ValuePtr& input, const ValuePtr& mean, const ValuePtr& invstd, const ValuePtr& weight, const BoolImmPtr& input_g, const BoolImmPtr& weight_g, const BoolImmPtr& bias_g);
|
|
735
|
-
py::object ME_EXPORT Pyboost_Conv1DExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& weight, const std::optional<ValuePtr>& bias, const ValueTuplePtr& stride, const ValueTuplePtr& padding, const ValueTuplePtr& dilation, const Int64ImmPtr& groups);
|
|
736
|
-
py::object ME_EXPORT Pyboost_BatchNormStats_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const FP32ImmPtr& eps);
|
|
737
|
-
py::object ME_EXPORT Pyboost_SumExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const std::optional<ValueTuplePtr>& dim, const BoolImmPtr& keepdim, const std::optional<Int64ImmPtr>& dtype);
|
|
738
|
-
py::object ME_EXPORT Pyboost_DistCommReduceScatter_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& other, const ValueTuplePtr& input_list, const Int64ImmPtr& rank_size, const StringImmPtr& op_type, const StringImmPtr& group);
|
|
739
|
-
py::object ME_EXPORT Pyboost_SelectV2_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& condition, const ValuePtr& input, const ValuePtr& other);
|
|
740
|
-
py::object ME_EXPORT Pyboost_Swiglu_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& dim);
|
|
741
|
-
py::object ME_EXPORT Pyboost_NLLLossGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& logits, const ValuePtr& loss_grad, const ValuePtr& labels, const ValuePtr& weight, const ValuePtr& total_weight, const Int64ImmPtr& reduction, const Int64ImmPtr& ignore_index);
|
|
742
|
-
py::object ME_EXPORT Pyboost_GLU_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x, const Int64ImmPtr& axis);
|
|
743
|
-
py::object ME_EXPORT Pyboost_ReduceMin_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x, const ValueTuplePtr& axis, const BoolImmPtr& keep_dims);
|
|
744
|
-
py::object ME_EXPORT Pyboost_BitwiseXorScalar_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ScalarPtr& other);
|
|
745
|
-
py::object ME_EXPORT Pyboost_MedianExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
746
|
-
py::object ME_EXPORT Pyboost_Erf_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
747
|
-
py::object ME_EXPORT Pyboost_DistCommAllReduce_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const StringImmPtr& op_type, const StringImmPtr& group);
|
|
748
|
-
py::object ME_EXPORT Pyboost_UpsampleBilinear2D_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x, const std::optional<ValueTuplePtr>& output_size, const std::optional<ValueTuplePtr>& scales, const BoolImmPtr& align_corners);
|
|
749
|
-
py::object ME_EXPORT Pyboost_Sign_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
750
|
-
py::object ME_EXPORT Pyboost_SoftMarginLossGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& predict, const ValuePtr& label, const ValuePtr& dout, const Int64ImmPtr& reduction);
|
|
751
|
-
py::object ME_EXPORT Pyboost_InnerCommReduceScatter_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& rank_size, const StringImmPtr& op_type, const StringImmPtr& group);
|
|
752
|
-
py::object ME_EXPORT Pyboost_InplaceUniform_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ScalarPtr& from_, const ScalarPtr& to, const ValuePtr& seed, const ValuePtr& offset);
|
|
753
|
-
py::object ME_EXPORT Pyboost_ClampTensor_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const std::optional<ValuePtr>& min, const std::optional<ValuePtr>& max);
|
|
754
|
-
py::object ME_EXPORT Pyboost_LogSumExp_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& dim, const BoolImmPtr& keepdim);
|
|
755
|
-
py::object ME_EXPORT Pyboost_Split_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input_x, const Int64ImmPtr& axis, const Int64ImmPtr& output_num);
|
|
756
|
-
py::object ME_EXPORT Pyboost_InplaceDivMod_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other, const std::optional<Int64ImmPtr>& rounding_mode);
|
|
757
|
-
py::object ME_EXPORT Pyboost_ReLU_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
758
|
-
py::object ME_EXPORT Pyboost_UpsampleNearest3DGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& dy, const ValueTuplePtr& input_size, const std::optional<ValueTuplePtr>& output_size, const std::optional<ValueTuplePtr>& scales);
|
|
759
|
-
py::object ME_EXPORT Pyboost_OnesLikeExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const std::optional<Int64ImmPtr>& dtype);
|
|
760
|
-
py::object ME_EXPORT Pyboost_BitwiseAndTensor_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other);
|
|
761
|
-
py::object ME_EXPORT Pyboost_AddScalar_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ScalarPtr& other, const ScalarPtr& alpha);
|
|
762
|
-
py::object ME_EXPORT Pyboost_InplaceFillTensor_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& value);
|
|
763
|
-
py::object ME_EXPORT Pyboost_TraceExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
764
|
-
py::object ME_EXPORT Pyboost_SeLUExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
765
|
-
py::object ME_EXPORT Pyboost_LogSigmoid_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
766
|
-
py::object ME_EXPORT Pyboost_ExpandDims_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input_x, const Int64ImmPtr& axis);
|
|
767
|
-
py::object ME_EXPORT Pyboost_PromptFlashAttention_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& query, const ValuePtr& key, const ValuePtr& value, const std::optional<ValuePtr>& attn_mask, const std::optional<ValueTuplePtr>& actual_seq_lengths, const std::optional<ValueTuplePtr>& actual_seq_lengths_kv, const std::optional<ValuePtr>& pse_shift, const std::optional<ValuePtr>& deq_scale1, const std::optional<ValuePtr>& quant_scale1, const std::optional<ValuePtr>& deq_scale2, const std::optional<ValuePtr>& quant_scale2, const std::optional<ValuePtr>& quant_offset2, const Int64ImmPtr& num_heads, const FP32ImmPtr& scale_value, const Int64ImmPtr& pre_tokens, const Int64ImmPtr& next_tokens, const Int64ImmPtr& input_layout, const Int64ImmPtr& num_key_value_heads, const Int64ImmPtr& sparse_mode, const Int64ImmPtr& inner_precise);
|
|
768
|
-
py::object ME_EXPORT Pyboost_BatchMatMulExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& mat2);
|
|
769
|
-
py::object ME_EXPORT Pyboost_LogSigmoidGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& dy, const ValuePtr& input, const ValuePtr& buffer);
|
|
770
|
-
py::object ME_EXPORT Pyboost_SliceExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& dim, const Int64ImmPtr& start, const Int64ImmPtr& end, const Int64ImmPtr& step);
|
|
771
|
-
py::object ME_EXPORT Pyboost_MatrixInverseExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
772
|
-
py::object ME_EXPORT Pyboost_BitwiseOrScalar_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ScalarPtr& other);
|
|
773
|
-
py::object ME_EXPORT Pyboost_ReduceMax_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x, const ValueTuplePtr& axis, const BoolImmPtr& keep_dims);
|
|
774
|
-
py::object ME_EXPORT Pyboost_InnerIndex_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& indices);
|
|
775
|
-
py::object ME_EXPORT Pyboost_ReflectionPad2DGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& grad_output, const ValuePtr& input, const ValueTuplePtr& padding);
|
|
776
|
-
py::object ME_EXPORT Pyboost_Log_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
777
|
-
py::object ME_EXPORT Pyboost_AdaptiveAvgPool3DExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& output_size);
|
|
778
|
-
py::object ME_EXPORT Pyboost_MaskedFill_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input_x, const ValuePtr& mask, const ValuePtr& value);
|
|
779
|
-
py::object ME_EXPORT Pyboost_DropoutGradExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& mask, const FP32ImmPtr& p);
|
|
780
|
-
py::object ME_EXPORT Pyboost_RandpermExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const Int64ImmPtr& n, const ValuePtr& seed, const ValuePtr& offset, const Int64ImmPtr& dtype);
|
|
781
|
-
py::object ME_EXPORT Pyboost_NeScalar_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ScalarPtr& other);
|
|
782
|
-
py::object ME_EXPORT Pyboost_Col2ImGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& kernel_size, const ValueTuplePtr& dilation, const ValueTuplePtr& padding, const ValueTuplePtr& stride);
|
|
783
|
-
py::object ME_EXPORT Pyboost_HistcExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& bins, const ScalarPtr& min, const ScalarPtr& max);
|
|
784
|
-
py::object ME_EXPORT Pyboost_ThresholdGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& grad_output, const ValuePtr& input, const FP32ImmPtr& threshold);
|
|
785
|
-
py::object ME_EXPORT Pyboost_AddcmulExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& tensor1, const ValuePtr& tensor2, const ScalarPtr& value);
|
|
786
|
-
py::object ME_EXPORT Pyboost_UpsampleNearest2DGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& dy, const ValueTuplePtr& input_size, const std::optional<ValueTuplePtr>& output_size, const std::optional<ValueTuplePtr>& scales);
|
|
787
|
-
py::object ME_EXPORT Pyboost_AcoshExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
788
|
-
py::object ME_EXPORT Pyboost_AvgPool2DGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& grad, const ValuePtr& image, const ValueTuplePtr& kernel_size, const ValueTuplePtr& stride, const ValueTuplePtr& padding, const BoolImmPtr& ceil_mode, const BoolImmPtr& count_include_pad, const std::optional<Int64ImmPtr>& divisor_override);
|
|
789
|
-
py::object ME_EXPORT Pyboost_Greater_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other);
|
|
790
|
-
py::object ME_EXPORT Pyboost_InnerInplaceIndexPut_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& indices, const ValuePtr& values, const BoolImmPtr& accumulate);
|
|
791
|
-
py::object ME_EXPORT Pyboost_Mul_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other);
|
|
792
|
-
py::object ME_EXPORT Pyboost_Nansum_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const std::optional<ValueTuplePtr>& dim, const BoolImmPtr& keepdim, const std::optional<Int64ImmPtr>& dtype);
|
|
793
|
-
py::object ME_EXPORT Pyboost_InplaceFloorDivide_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other);
|
|
794
|
-
py::object ME_EXPORT Pyboost_Index_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& indices);
|
|
795
|
-
py::object ME_EXPORT Pyboost_MSELossGradExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& dout, const ValuePtr& x, const ValuePtr& target, const Int64ImmPtr& reduction);
|
|
796
|
-
py::object ME_EXPORT Pyboost_Var_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const std::optional<ValueTuplePtr>& dim, const Int64ImmPtr& correction, const BoolImmPtr& keepdim);
|
|
797
|
-
py::object ME_EXPORT Pyboost_Tile_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& dims);
|
|
798
|
-
py::object ME_EXPORT Pyboost_VarMean_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const std::optional<ValueTuplePtr>& dim, const Int64ImmPtr& correction, const BoolImmPtr& keepdim);
|
|
799
|
-
py::object ME_EXPORT Pyboost_Divs_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ScalarPtr& other);
|
|
800
|
-
py::object ME_EXPORT Pyboost_Conv2DExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& weight, const std::optional<ValuePtr>& bias, const ValueTuplePtr& stride, const ValueTuplePtr& padding, const ValueTuplePtr& dilation, const Int64ImmPtr& groups);
|
|
801
|
-
py::object ME_EXPORT Pyboost_BincountExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const std::optional<ValuePtr>& weights, const Int64ImmPtr& minlength);
|
|
802
|
-
py::object ME_EXPORT Pyboost_Tanh_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
803
|
-
py::object ME_EXPORT Pyboost_UpsampleLinear1D_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x, const std::optional<ValueTuplePtr>& output_size, const std::optional<ValueTuplePtr>& scales, const BoolImmPtr& align_corners);
|
|
804
|
-
py::object ME_EXPORT Pyboost_DistCommGather_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& gather_list, const Int64ImmPtr& rank_size, const Int64ImmPtr& dst, const Int64ImmPtr& rank_id, const StringImmPtr& group);
|
|
805
|
-
py::object ME_EXPORT Pyboost_SoftplusGradExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& dout, const ValuePtr& x, const ScalarPtr& beta, const ScalarPtr& threshold);
|
|
806
|
-
py::object ME_EXPORT Pyboost_InplaceIndexAddExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& dim, const ValuePtr& index, const ValuePtr& source, const ScalarPtr& alpha);
|
|
807
|
-
py::object ME_EXPORT Pyboost_DistCommAllToAllVSingle_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& other, const ValuePtr& input, const StringImmPtr& group, const ValueTuplePtr& send_numel_list, const ValueTuplePtr& recv_numel_list, const Int64ImmPtr& rank_size, const BoolImmPtr& split_sizes_empty);
|
|
808
|
-
py::object ME_EXPORT Pyboost_ReflectionPad3DGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& grad_output, const ValuePtr& input, const ValueTuplePtr& padding);
|
|
809
|
-
py::object ME_EXPORT Pyboost_Addmm_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& mat1, const ValuePtr& mat2, const ScalarPtr& beta, const ScalarPtr& alpha);
|
|
810
|
-
py::object ME_EXPORT Pyboost_Min_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
811
|
-
py::object ME_EXPORT Pyboost_ClampScalar_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const std::optional<ScalarPtr>& min, const std::optional<ScalarPtr>& max);
|
|
812
|
-
py::object ME_EXPORT Pyboost_Log1p_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
813
|
-
py::object ME_EXPORT Pyboost_InplaceScatterSrcReduce_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& dim, const ValuePtr& index, const ValuePtr& src, const Int64ImmPtr& reduce);
|
|
814
|
-
py::object ME_EXPORT Pyboost_ReflectionPad2D_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& padding);
|
|
815
|
-
py::object ME_EXPORT Pyboost_ExpandAs_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other);
|
|
816
|
-
py::object ME_EXPORT Pyboost_AsinhExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
817
|
-
py::object ME_EXPORT Pyboost_Randn_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValueTuplePtr& shape, const ValuePtr& seed, const ValuePtr& offset, const std::optional<Int64ImmPtr>& dtype);
|
|
818
|
-
py::object ME_EXPORT Pyboost_FullLike_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ScalarPtr& fill_value, const std::optional<Int64ImmPtr>& dtype);
|
|
819
|
-
py::object ME_EXPORT Pyboost_ConstantPadND_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& padding, const ScalarPtr& value);
|
|
820
|
-
py::object ME_EXPORT Pyboost_InplaceIndexPut_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& indices, const ValuePtr& values, const BoolImmPtr& accumulate);
|
|
821
|
-
py::object ME_EXPORT Pyboost_TExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
822
|
-
py::object ME_EXPORT Pyboost_PowTensorScalar_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ScalarPtr& exponent);
|
|
823
|
-
py::object ME_EXPORT Pyboost_Max_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
824
|
-
py::object ME_EXPORT Pyboost_SplitTensor_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& split_size, const Int64ImmPtr& dim);
|
|
825
|
-
py::object ME_EXPORT Pyboost_GeLU_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
826
|
-
py::object ME_EXPORT Pyboost_NLLLoss_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& logits, const ValuePtr& labels, const ValuePtr& weight, const Int64ImmPtr& reduction, const Int64ImmPtr& ignore_index);
|
|
827
|
-
py::object ME_EXPORT Pyboost_LessEqual_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other);
|
|
828
|
-
py::object ME_EXPORT Pyboost_Log2_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
829
|
-
py::object ME_EXPORT Pyboost_HSigmoidGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& grads, const ValuePtr& input_x);
|
|
830
|
-
py::object ME_EXPORT Pyboost_Transpose_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& input_perm);
|
|
831
|
-
py::object ME_EXPORT Pyboost_UpsampleLinear1DGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& dy, const ValueTuplePtr& input_size, const std::optional<ValueTuplePtr>& output_size, const std::optional<ValueTuplePtr>& scales, const BoolImmPtr& align_corners);
|
|
832
|
-
py::object ME_EXPORT Pyboost_UpsampleTrilinear3D_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x, const std::optional<ValueTuplePtr>& output_size, const std::optional<ValueTuplePtr>& scales, const BoolImmPtr& align_corners);
|
|
833
|
-
py::object ME_EXPORT Pyboost_GluGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& grads, const ValuePtr& x, const Int64ImmPtr& axis);
|
|
834
|
-
py::object ME_EXPORT Pyboost_GeluExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& approximate);
|
|
835
|
-
py::object ME_EXPORT Pyboost_MishGradExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& dout, const ValuePtr& x);
|
|
836
|
-
py::object ME_EXPORT Pyboost_AdaptiveAvgPool2DExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& output_size);
|
|
837
|
-
py::object ME_EXPORT Pyboost_Floor_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
838
|
-
py::object ME_EXPORT Pyboost_LerpScalar_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& end, const FP32ImmPtr& weight);
|
|
839
|
-
py::object ME_EXPORT Pyboost_GreaterEqual_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other);
|
|
840
|
-
py::object ME_EXPORT Pyboost_Conv3DExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& weight, const std::optional<ValuePtr>& bias, const ValueTuplePtr& stride, const ValueTuplePtr& padding, const ValueTuplePtr& dilation, const Int64ImmPtr& groups);
|
|
841
|
-
py::object ME_EXPORT Pyboost_HSwish_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
842
|
-
py::object ME_EXPORT Pyboost_Muls_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ScalarPtr& other);
|
|
843
|
-
py::object ME_EXPORT Pyboost_DistCommBarrier_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const StringImmPtr& group);
|
|
844
|
-
py::object ME_EXPORT Pyboost_EmbeddingDenseBackward_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& grad, const ValuePtr& indices, const Int64ImmPtr& num_weights, const std::optional<Int64ImmPtr>& padding_idx, const BoolImmPtr& scale_grad_by_freq);
|
|
845
|
-
py::object ME_EXPORT Pyboost_CountNonZero_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const std::optional<ValueTuplePtr>& dim);
|
|
846
|
-
py::object ME_EXPORT Pyboost_UpsampleTrilinear3DGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& dy, const ValueTuplePtr& input_size, const std::optional<ValueTuplePtr>& output_size, const std::optional<ValueTuplePtr>& scales, const BoolImmPtr& align_corners);
|
|
847
|
-
py::object ME_EXPORT Pyboost_NormalFloatTensor_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const FP32ImmPtr& mean, const ValuePtr& std, const ValuePtr& seed, const ValuePtr& offset);
|
|
848
|
-
py::object ME_EXPORT Pyboost_AdaptiveAvgPool1D_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& output_size);
|
|
849
|
-
py::object ME_EXPORT Pyboost_Addmv_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& mat, const ValuePtr& vec, const ScalarPtr& beta, const ScalarPtr& alpha);
|
|
850
|
-
py::object ME_EXPORT Pyboost_Conv2DPadding_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& weight, const std::optional<ValuePtr>& bias, const ValueTuplePtr& stride, const Int64ImmPtr& padding, const ValueTuplePtr& dilation, const Int64ImmPtr& groups);
|
|
851
|
-
py::object ME_EXPORT Pyboost_BatchNormExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& weight, const ValuePtr& bias, const std::optional<ValuePtr>& running_mean, const std::optional<ValuePtr>& runnning_var, const BoolImmPtr& training, const FP32ImmPtr& momentum, const FP32ImmPtr& epsilon);
|
|
852
|
-
py::object ME_EXPORT Pyboost_FlashAttentionScore_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& query, const ValuePtr& key, const ValuePtr& value, const std::optional<ValuePtr>& real_shift, const std::optional<ValuePtr>& drop_mask, const std::optional<ValuePtr>& padding_mask, const std::optional<ValuePtr>& attn_mask, const std::optional<ValueTuplePtr>& prefix, const std::optional<ValueTuplePtr>& actual_seq_qlen, const std::optional<ValueTuplePtr>& actual_seq_kvlen, const Int64ImmPtr& head_num, const FP32ImmPtr& keep_prob, const FP32ImmPtr& scale_value, const Int64ImmPtr& pre_tokens, const Int64ImmPtr& next_tokens, const Int64ImmPtr& inner_precise, const Int64ImmPtr& input_layout, const Int64ImmPtr& sparse_mode);
|
|
853
|
-
py::object ME_EXPORT Pyboost_InplaceFillDiagonal_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ScalarPtr& fill_value, const BoolImmPtr& wrap);
|
|
854
|
-
py::object ME_EXPORT Pyboost_Gcd_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other);
|
|
855
|
-
py::object ME_EXPORT Pyboost_CumsumExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& dim, const std::optional<Int64ImmPtr>& dtype);
|
|
856
|
-
py::object ME_EXPORT Pyboost_Col2ImExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& output_size, const ValueTuplePtr& kernel_size, const ValueTuplePtr& dilation, const ValueTuplePtr& padding, const ValueTuplePtr& stride);
|
|
857
|
-
py::object ME_EXPORT Pyboost_InplaceSubExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other, const ScalarPtr& alpha);
|
|
858
|
-
py::object ME_EXPORT Pyboost_UnstackExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& dim);
|
|
859
|
-
py::object ME_EXPORT Pyboost_Cos_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
860
|
-
py::object ME_EXPORT Pyboost_IndexAddExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& dim, const ValuePtr& index, const ValuePtr& source, const ScalarPtr& alpha);
|
|
861
|
-
py::object ME_EXPORT Pyboost_Sub_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other);
|
|
862
|
-
py::object ME_EXPORT Pyboost_Square_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
863
|
-
py::object ME_EXPORT Pyboost_InplaceNormal_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ScalarPtr& mean, const ScalarPtr& std, const ValuePtr& seed, const ValuePtr& offset);
|
|
864
|
-
py::object ME_EXPORT Pyboost_ArgMaxExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const std::optional<Int64ImmPtr>& dim, const BoolImmPtr& keepdim);
|
|
865
|
-
py::object ME_EXPORT Pyboost_Meshgrid_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValueTuplePtr& inputs, const Int64ImmPtr& indexing);
|
|
866
|
-
py::object ME_EXPORT Pyboost_InplaceSubScalar_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ScalarPtr& other, const ScalarPtr& alpha);
|
|
867
|
-
py::object ME_EXPORT Pyboost_ReplicationPad3DGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& grad_output, const ValuePtr& input, const ValueTuplePtr& padding);
|
|
868
|
-
py::object ME_EXPORT Pyboost_AddExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other, const ScalarPtr& alpha);
|
|
869
|
-
py::object ME_EXPORT Pyboost_InnerCommAllGather_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& rank_size, const StringImmPtr& group);
|
|
870
|
-
py::object ME_EXPORT Pyboost_TrilExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& diagonal);
|
|
871
|
-
py::object ME_EXPORT Pyboost_MaxUnpool2DExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& indices, const ValueTuplePtr& kernel_size, const std::optional<ValueTuplePtr>& stride, const ValueTuplePtr& padding, const std::optional<ValueTuplePtr>& output_size);
|
|
872
|
-
py::object ME_EXPORT Pyboost_SoftplusExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ScalarPtr& beta, const ScalarPtr& threshold);
|
|
873
|
-
py::object ME_EXPORT Pyboost_InplaceCopy_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& src);
|
|
874
|
-
py::object ME_EXPORT Pyboost_LogicalOr_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x, const ValuePtr& y);
|
|
875
|
-
py::object ME_EXPORT Pyboost_ReplicationPad2D_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& padding);
|
|
876
|
-
py::object ME_EXPORT Pyboost_Std_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const std::optional<ValueTuplePtr>& dim, const Int64ImmPtr& correction, const BoolImmPtr& keepdim);
|
|
877
|
-
py::object ME_EXPORT Pyboost_SoftmaxBackward_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& dout, const ValuePtr& out, const Int64ImmPtr& dim);
|
|
878
|
-
py::object ME_EXPORT Pyboost_GroupNormGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& dy, const ValuePtr& x, const ValuePtr& mean, const ValuePtr& rstd, const ValuePtr& gamma_opt, const Int64ImmPtr& num_groups, const BoolImmPtr& dx_is_require, const BoolImmPtr& dgamma_is_require, const BoolImmPtr& dbeta_is_require);
|
|
879
|
-
py::object ME_EXPORT Pyboost_EluGradExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& dout, const ValuePtr& x_or_out, const FP32ImmPtr& alpha, const BoolImmPtr& is_result);
|
|
880
|
-
py::object ME_EXPORT Pyboost_InplaceDivs_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ScalarPtr& other);
|
|
881
|
-
py::object ME_EXPORT Pyboost_Norm_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const FP32ImmPtr& p, const std::optional<ValueTuplePtr>& dim, const BoolImmPtr& keepdim, const std::optional<Int64ImmPtr>& dtype);
|
|
882
|
-
py::object ME_EXPORT Pyboost_Log10_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
883
|
-
py::object ME_EXPORT Pyboost_TanhGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& y, const ValuePtr& dy);
|
|
884
|
-
py::object ME_EXPORT Pyboost_MultiScaleDeformableAttnGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& value, const ValuePtr& shape, const ValuePtr& offset, const ValuePtr& locations_trans, const ValuePtr& weight, const ValuePtr& grad_output);
|
|
885
|
-
py::object ME_EXPORT Pyboost_Exp2_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
886
|
-
py::object ME_EXPORT Pyboost_IsClose_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other, const FP32ImmPtr& rtol, const FP32ImmPtr& atol, const BoolImmPtr& equal_nan);
|
|
887
|
-
py::object ME_EXPORT Pyboost_BCEWithLogitsLoss_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& target, const std::optional<ValuePtr>& weight, const std::optional<ValuePtr>& posWeight, const Int64ImmPtr& reduction);
|
|
888
|
-
py::object ME_EXPORT Pyboost_Conv3DPadding_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& weight, const std::optional<ValuePtr>& bias, const ValueTuplePtr& stride, const Int64ImmPtr& padding, const ValueTuplePtr& dilation, const Int64ImmPtr& groups);
|
|
889
|
-
py::object ME_EXPORT Pyboost_FillScalar_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValueTuplePtr& size, const ScalarPtr& fill_value, const std::optional<Int64ImmPtr>& dtype);
|
|
890
|
-
py::object ME_EXPORT Pyboost_MoeTokenUnpermute_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& permuted_tokens, const ValuePtr& sorted_indices, const std::optional<ValuePtr>& probs, const BoolImmPtr& padded_mode, const std::optional<ValueTuplePtr>& restore_shape);
|
|
891
|
-
py::object ME_EXPORT Pyboost_Dot_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other);
|
|
892
|
-
py::object ME_EXPORT Pyboost_Select_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& condition, const ValuePtr& input, const ValuePtr& other);
|
|
893
|
-
py::object ME_EXPORT Pyboost_DivMod_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other, const std::optional<Int64ImmPtr>& rounding_mode);
|
|
894
|
-
py::object ME_EXPORT Pyboost_StdMean_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const std::optional<ValueTuplePtr>& dim, const Int64ImmPtr& correction, const BoolImmPtr& keepdim);
|
|
895
|
-
py::object ME_EXPORT Pyboost_ReplicationPad1DGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& grad_output, const ValuePtr& input, const ValueTuplePtr& padding);
|
|
896
|
-
py::object ME_EXPORT Pyboost_LogSoftmaxGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& logits, const ValuePtr& grad, const Int64ImmPtr& axis);
|
|
897
|
-
py::object ME_EXPORT Pyboost_HardtanhGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& dout, const ValuePtr& input, const ScalarPtr& min_val, const ScalarPtr& max_val);
|
|
898
|
-
py::object ME_EXPORT Pyboost_InplaceClampScalar_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const std::optional<ScalarPtr>& min, const std::optional<ScalarPtr>& max);
|
|
899
|
-
py::object ME_EXPORT Pyboost_AdaptiveAvgPool3DGradExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input_grad, const ValuePtr& input);
|
|
900
|
-
py::object ME_EXPORT Pyboost_InplaceMuls_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ScalarPtr& other);
|
|
901
|
-
py::object ME_EXPORT Pyboost_RandIntLike_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& low, const Int64ImmPtr& high, const ValuePtr& seed, const ValuePtr& offset, const std::optional<Int64ImmPtr>& dtype);
|
|
902
|
-
py::object ME_EXPORT Pyboost_ArgMaxWithValue_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& axis, const BoolImmPtr& keep_dims);
|
|
903
|
-
py::object ME_EXPORT Pyboost_UpsampleNearest1D_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x, const std::optional<ValueTuplePtr>& output_size, const std::optional<ValueTuplePtr>& scales);
|
|
904
|
-
py::object ME_EXPORT Pyboost_PReLU_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& weight);
|
|
905
|
-
py::object ME_EXPORT Pyboost_HSwishGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& y_grad, const ValuePtr& x);
|
|
906
|
-
py::object ME_EXPORT Pyboost_MoeTokenPermute_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& tokens, const ValuePtr& indices, const std::optional<Int64ImmPtr>& num_out_tokens, const BoolImmPtr& padded_mode);
|
|
907
|
-
py::object ME_EXPORT Pyboost_ConvTranspose2D_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& weight, const std::optional<ValuePtr>& bias, const ValueTuplePtr& stride, const ValueTuplePtr& padding, const ValueTuplePtr& output_padding, const Int64ImmPtr& groups, const ValueTuplePtr& dilation);
|
|
908
|
-
py::object ME_EXPORT Pyboost_DropoutDoMaskExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& mask, const FP32ImmPtr& p);
|
|
909
|
-
py::object ME_EXPORT Pyboost_SilentCheckV2_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& val, const ValuePtr& input_grad, const ValuePtr& sfda, const ValuePtr& step, const Int64ImmPtr& c_min_steps, const FP32ImmPtr& c_thresh_l1, const FP32ImmPtr& c_coeff_l1, const FP32ImmPtr& c_thresh_l2, const FP32ImmPtr& c_coeff_l2, const Int64ImmPtr& npu_asd_detect);
|
|
910
|
-
py::object ME_EXPORT Pyboost_L1LossExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& target, const Int64ImmPtr& reduction);
|
|
911
|
-
py::object ME_EXPORT Pyboost_AsinExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
912
|
-
py::object ME_EXPORT Pyboost_RandExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValueTuplePtr& shape, const ValuePtr& seed, const ValuePtr& offset, const std::optional<Int64ImmPtr>& dtype);
|
|
913
|
-
py::object ME_EXPORT Pyboost_InplaceAddExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other, const ScalarPtr& alpha);
|
|
914
|
-
py::object ME_EXPORT Pyboost_Zeros_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValueTuplePtr& size, const std::optional<Int64ImmPtr>& dtype);
|
|
915
|
-
py::object ME_EXPORT Pyboost_BinaryCrossEntropy_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& target, const std::optional<ValuePtr>& weight, const Int64ImmPtr& reduction);
|
|
916
|
-
py::object ME_EXPORT Pyboost_TransposeExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& dim0, const Int64ImmPtr& dim1);
|
|
917
|
-
py::object ME_EXPORT Pyboost_Ones_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValueTuplePtr& shape, const std::optional<Int64ImmPtr>& dtype);
|
|
918
|
-
py::object ME_EXPORT Pyboost_ReplicationPad3D_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& padding);
|
|
919
|
-
py::object ME_EXPORT Pyboost_FmodTensor_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other);
|
|
920
|
-
py::object ME_EXPORT Pyboost_Erfinv_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
921
|
-
py::object ME_EXPORT Pyboost_RotaryPositionEmbedding_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x, const ValuePtr& cos, const ValuePtr& sin, const Int64ImmPtr& mode);
|
|
922
|
-
py::object ME_EXPORT Pyboost_Cast_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& dtype);
|
|
923
|
-
py::object ME_EXPORT Pyboost_L1LossBackwardExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& grad_output, const ValuePtr& input, const ValuePtr& target, const Int64ImmPtr& reduction);
|
|
924
|
-
py::object ME_EXPORT Pyboost_Hardtanh_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ScalarPtr& min_val, const ScalarPtr& max_val);
|
|
925
|
-
py::object ME_EXPORT Pyboost_Cross_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other, const Int64ImmPtr& dim);
|
|
926
|
-
py::object ME_EXPORT Pyboost_NewOnes_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& size, const std::optional<Int64ImmPtr>& dtype);
|
|
927
|
-
py::object ME_EXPORT Pyboost_Softmax_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& axis);
|
|
928
|
-
py::object ME_EXPORT Pyboost_NonZero_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
929
|
-
py::object ME_EXPORT Pyboost_SmoothL1LossGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& prediction, const ValuePtr& target, const ValuePtr& dout, const FP32ImmPtr& beta, const Int64ImmPtr& reduction);
|
|
930
|
-
py::object ME_EXPORT Pyboost_Eye_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const Int64ImmPtr& n, const Int64ImmPtr& m, const Int64ImmPtr& dtype);
|
|
931
|
-
py::object ME_EXPORT Pyboost_GatherD_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x, const Int64ImmPtr& dim, const ValuePtr& index);
|
|
932
|
-
py::object ME_EXPORT Pyboost_AvgPool1D_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& kernel_size, const std::optional<ValueTuplePtr>& stride, const ValueTuplePtr& padding, const BoolImmPtr& ceil_mode, const BoolImmPtr& count_include_pad);
|
|
933
|
-
py::object ME_EXPORT Pyboost_PReLUGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& dy, const ValuePtr& x, const ValuePtr& weight);
|
|
934
|
-
py::object ME_EXPORT Pyboost_SoftMarginLoss_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& target, const Int64ImmPtr& reduction);
|
|
935
|
-
py::object ME_EXPORT Pyboost_InplaceScatterValue_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& dim, const ValuePtr& index, const ScalarPtr& value);
|
|
936
|
-
py::object ME_EXPORT Pyboost_MultinomialExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& num_samples, const BoolImmPtr& replacement, const ValuePtr& seed, const ValuePtr& offset);
|
|
937
|
-
py::object ME_EXPORT Pyboost_Conv1DPadding_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& weight, const std::optional<ValuePtr>& bias, const ValueTuplePtr& stride, const Int64ImmPtr& padding, const ValueTuplePtr& dilation, const Int64ImmPtr& groups);
|
|
938
|
-
py::object ME_EXPORT Pyboost_AtanExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
939
|
-
py::object ME_EXPORT Pyboost_TopkExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& k, const Int64ImmPtr& dim, const BoolImmPtr& largest, const BoolImmPtr& sorted);
|
|
940
|
-
py::object ME_EXPORT Pyboost_ConvolutionGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& dout, const ValuePtr& input, const ValuePtr& weight, const std::optional<ValuePtr>& bias, const ValueTuplePtr& stride, const ValueTuplePtr& padding, const ValueTuplePtr& dilation, const BoolImmPtr& transposed, const ValueTuplePtr& output_padding, const Int64ImmPtr& groups, const ValueTuplePtr& output_mask);
|
|
941
|
-
py::object ME_EXPORT Pyboost_Sinh_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
942
|
-
py::object ME_EXPORT Pyboost_AsStrided_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& size, const ValueTuplePtr& stride, const Int64ImmPtr& storage_offset);
|
|
943
|
-
py::object ME_EXPORT Pyboost_SearchSorted_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& sorted_sequence, const ValuePtr& values, const std::optional<ValuePtr>& sorter, const Int64ImmPtr& dtype, const BoolImmPtr& right);
|
|
944
|
-
py::object ME_EXPORT Pyboost_DistCommIsend_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& dst, const StringImmPtr& group, const Int64ImmPtr& tag);
|
|
945
|
-
py::object ME_EXPORT Pyboost_DistCommAllToAllV_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValueTuplePtr& other, const ValuePtr& input, const StringImmPtr& group, const ValueTuplePtr& send_numel_list, const ValueTuplePtr& recv_numel_list, const Int64ImmPtr& rank_size);
|
|
946
|
-
py::object ME_EXPORT Pyboost_Rsqrt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
947
|
-
py::object ME_EXPORT Pyboost_IndexSelect_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& dim, const ValuePtr& index);
|
|
948
|
-
py::object ME_EXPORT Pyboost_ReplicationPad2DGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& grad_output, const ValuePtr& input, const ValueTuplePtr& padding);
|
|
949
|
-
py::object ME_EXPORT Pyboost_UpsampleNearest2D_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x, const std::optional<ValueTuplePtr>& output_size, const std::optional<ValueTuplePtr>& scales);
|
|
950
|
-
py::object ME_EXPORT Pyboost_HSigmoid_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
951
|
-
py::object ME_EXPORT Pyboost_InplaceFloor_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
952
|
-
py::object ME_EXPORT Pyboost_NormalTensorTensor_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& mean, const ValuePtr& std, const ValuePtr& seed, const ValuePtr& offset);
|
|
953
|
-
py::object ME_EXPORT Pyboost_UniformExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& tensor, const ScalarPtr& a, const ScalarPtr& b, const ValuePtr& seed, const ValuePtr& offset);
|
|
954
|
-
py::object ME_EXPORT Pyboost_Kthvalue_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& k, const Int64ImmPtr& dim, const BoolImmPtr& keepdim);
|
|
955
|
-
py::object ME_EXPORT Pyboost_ReverseV2_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& axis);
|
|
956
|
-
py::object ME_EXPORT Pyboost_InplacePut_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& index, const ValuePtr& source, const BoolImmPtr& accumulate);
|
|
957
|
-
py::object ME_EXPORT Pyboost_NLLLoss2d_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& target, const ValuePtr& weight, const Int64ImmPtr& reduction, const Int64ImmPtr& ignore_index);
|
|
958
|
-
py::object ME_EXPORT Pyboost_ArgSort_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& dim, const BoolImmPtr& descending, const BoolImmPtr& stable);
|
|
959
|
-
py::object ME_EXPORT Pyboost_Reshape_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& shape);
|
|
960
|
-
py::object ME_EXPORT Pyboost_ScatterValue_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& dim, const ValuePtr& index, const ScalarPtr& src, const Int64ImmPtr& reduce);
|
|
961
|
-
py::object ME_EXPORT Pyboost_AvgPool3DExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& kernel_size, const std::optional<ValueTuplePtr>& stride, const ValueTuplePtr& padding, const BoolImmPtr& ceil_mode, const BoolImmPtr& count_include_pad, const std::optional<Int64ImmPtr>& divisor_override);
|
|
962
|
-
py::object ME_EXPORT Pyboost_Outer_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& vec2);
|
|
963
|
-
py::object ME_EXPORT Pyboost_InplaceDiv_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other);
|
|
964
|
-
py::object ME_EXPORT Pyboost_InplaceThreshold_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const FP32ImmPtr& threshold, const FP32ImmPtr& value);
|
|
965
|
-
py::object ME_EXPORT Pyboost_InplaceClampTensor_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const std::optional<ValuePtr>& min, const std::optional<ValuePtr>& max);
|
|
966
|
-
py::object ME_EXPORT Pyboost_SiLU_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
967
|
-
py::object ME_EXPORT Pyboost_LogSoftmaxExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const std::optional<Int64ImmPtr>& dim, const std::optional<Int64ImmPtr>& dtype);
|
|
968
|
-
py::object ME_EXPORT Pyboost_LayerNormGradExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& dy, const ValuePtr& x, const ValueTuplePtr& normalized_shape, const ValuePtr& mean, const ValuePtr& variance, const ValuePtr& gamma, const ValuePtr& beta, const ValueTuplePtr& output_mask);
|
|
969
|
-
py::object ME_EXPORT Pyboost_UpsampleBilinear2DGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& dy, const ValueTuplePtr& input_size, const std::optional<ValueTuplePtr>& output_size, const std::optional<ValueTuplePtr>& scales, const BoolImmPtr& align_corners);
|
|
970
|
-
py::object ME_EXPORT Pyboost_GridSampler3DGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& grad, const ValuePtr& input_x, const ValuePtr& grid, const Int64ImmPtr& interpolation_mode, const Int64ImmPtr& padding_mode, const BoolImmPtr& align_corners, const ValueTuplePtr& output_mask);
|
|
971
|
-
py::object ME_EXPORT Pyboost_SeluGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& gradient, const ValuePtr& result);
|
|
972
|
-
py::object ME_EXPORT Pyboost_Div_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other);
|
|
973
|
-
py::object ME_EXPORT Pyboost_IsFinite_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
974
|
-
py::object ME_EXPORT Pyboost_LayerNormExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& normalized_shape, const std::optional<ValuePtr>& weight, const std::optional<ValuePtr>& bias, const FP32ImmPtr& eps);
|
|
975
|
-
py::object ME_EXPORT Pyboost_Sinc_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
976
|
-
py::object ME_EXPORT Pyboost_MaxPoolGradWithIndices_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x, const ValuePtr& grad, const ValuePtr& argmax, const ValueTuplePtr& kernel_size, const std::optional<ValueTuplePtr>& strides, const ValueTuplePtr& pads, const ValueTuplePtr& dilation, const BoolImmPtr& ceil_mode, const Int64ImmPtr& argmax_type);
|
|
977
|
-
py::object ME_EXPORT Pyboost_InplaceErfinv_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
978
|
-
py::object ME_EXPORT Pyboost_Generator_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const Int64ImmPtr& cmd, const ValueTuplePtr& inputs);
|
|
979
|
-
py::object ME_EXPORT Pyboost_MatMulExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other);
|
|
980
|
-
py::object ME_EXPORT Pyboost_GeluGradExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& grad, const ValuePtr& input, const Int64ImmPtr& approximate);
|
|
981
|
-
py::object ME_EXPORT Pyboost_InnerCommAllToAllV_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const StringImmPtr& group, const ValueTuplePtr& send_numel_list, const ValueTuplePtr& recv_numel_list, const Int64ImmPtr& rank_size, const BoolImmPtr& split_sizes_empty);
|
|
982
|
-
py::object ME_EXPORT Pyboost_AcosExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
983
|
-
py::object ME_EXPORT Pyboost_LeakyReLUExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ScalarPtr& negative_slope);
|
|
984
|
-
py::object ME_EXPORT Pyboost_Minimum_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other);
|
|
985
|
-
py::object ME_EXPORT Pyboost_SmoothL1Loss_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& prediction, const ValuePtr& target, const FP32ImmPtr& beta, const Int64ImmPtr& reduction);
|
|
986
|
-
py::object ME_EXPORT Pyboost_RemainderTensorTensor_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& other);
|
|
987
|
-
py::object ME_EXPORT Pyboost_BroadcastTo_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValueTuplePtr& shape);
|
|
988
|
-
py::object ME_EXPORT Pyboost_MSELossExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& target, const Int64ImmPtr& reduction);
|
|
989
|
-
py::object ME_EXPORT Pyboost_XLogYScalarSelf_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ScalarPtr& input, const ValuePtr& other);
|
|
990
|
-
py::object ME_EXPORT Pyboost_AdaptiveAvgPool2DGradExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& grad_output, const ValuePtr& x);
|
|
991
|
-
py::object ME_EXPORT Pyboost_Narrow_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& dim, const Int64ImmPtr& start, const Int64ImmPtr& length);
|
|
992
|
-
py::object ME_EXPORT Pyboost_Triu_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& diagonal);
|
|
993
|
-
py::object ME_EXPORT Pyboost_GridSampler2DGrad_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& grad, const ValuePtr& input_x, const ValuePtr& grid, const Int64ImmPtr& interpolation_mode, const Int64ImmPtr& padding_mode, const BoolImmPtr& align_corners, const ValueTuplePtr& output_mask);
|
|
994
|
-
py::object ME_EXPORT Pyboost_InnerCommIrecv_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const Int64ImmPtr& tag, const Int64ImmPtr& src, const ValueTuplePtr& shape, const StringImmPtr& group, const Int64ImmPtr& dtype);
|
|
995
|
-
py::object ME_EXPORT Pyboost_LogSoftmax_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& logits, const Int64ImmPtr& axis);
|
|
996
|
-
py::object ME_EXPORT Pyboost_Trunc_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
997
|
-
py::object ME_EXPORT Pyboost_InplaceExp_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input);
|
|
998
|
-
py::object ME_EXPORT Pyboost_Concat_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValueTuplePtr& tensors, const Int64ImmPtr& axis);
|
|
999
|
-
py::object ME_EXPORT Pyboost_DiagExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& diagonal);
|
|
1000
|
-
py::object ME_EXPORT Pyboost_MatMul_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const ValuePtr& mat2, const BoolImmPtr& transpose_a, const BoolImmPtr& transpose_b);
|
|
1001
|
-
py::object ME_EXPORT Pyboost_AvgPool3DGradExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& grad, const ValuePtr& input, const ValueTuplePtr& kernel_size, const std::optional<ValueTuplePtr>& stride, const ValueTuplePtr& padding, const BoolImmPtr& ceil_mode, const BoolImmPtr& count_include_pad, const std::optional<Int64ImmPtr>& divisor_override);
|
|
1002
|
-
py::object ME_EXPORT Pyboost_FusedInferAttentionScore_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& query, const ValueTuplePtr& key, const ValueTuplePtr& value, const std::optional<ValuePtr>& pse_shift, const std::optional<ValuePtr>& attn_mask, const std::optional<ValuePtr>& actual_seq_lengths, const std::optional<ValuePtr>& actual_seq_lengths_kv, const std::optional<ValuePtr>& dequant_scale1, const std::optional<ValuePtr>& quant_scale1, const std::optional<ValuePtr>& dequant_scale2, const std::optional<ValuePtr>& quant_scale2, const std::optional<ValuePtr>& quant_offset2, const std::optional<ValuePtr>& antiquant_scale, const std::optional<ValuePtr>& antiquant_offset, const std::optional<ValuePtr>& block_table, const std::optional<ValuePtr>& query_padding_size, const std::optional<ValuePtr>& kv_padding_size, const std::optional<ValuePtr>& key_antiquant_scale, const std::optional<ValuePtr>& key_antiquant_offset, const std::optional<ValuePtr>& value_antiquant_scale, const std::optional<ValuePtr>& value_antiquant_offset, const std::optional<ValuePtr>& key_shared_prefix, const std::optional<ValuePtr>& value_shared_prefix, const std::optional<ValuePtr>& actual_shared_prefix_len, const Int64ImmPtr& num_heads, const FP32ImmPtr& scale_value, const Int64ImmPtr& pre_tokens, const Int64ImmPtr& next_tokens, const Int64ImmPtr& input_layout, const Int64ImmPtr& num_key_value_heads, const Int64ImmPtr& sparse_mode, const Int64ImmPtr& inner_precise, const Int64ImmPtr& block_size, const Int64ImmPtr& antiquant_mode, const BoolImmPtr& softmax_lse_flag, const Int64ImmPtr& key_antiquant_mode, const Int64ImmPtr& value_antiquant_mode);
|
|
1003
|
-
py::object ME_EXPORT Pyboost_QuantV2_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x, const ValuePtr& scale, const ValuePtr& offset, const BoolImmPtr& sqrt_mode, const Int64ImmPtr& rounding_mode, const Int64ImmPtr& dst_type);
|
|
1004
|
-
py::object ME_EXPORT Pyboost_QuantBatchMatmul_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x1, const ValuePtr& x2, const ValuePtr& scale, const std::optional<ValuePtr>& offset, const std::optional<ValuePtr>& bias, const std::optional<ValuePtr>& pertokenScaleOptional, const BoolImmPtr& transpose_x1, const BoolImmPtr& transpose_x2, const Int64ImmPtr& dtype);
|
|
1005
|
-
py::object ME_EXPORT Pyboost_GroupedMatmul_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValueTuplePtr& x, const ValueTuplePtr& weight, const std::optional<ValueTuplePtr>& bias, const std::optional<ValueTuplePtr>& scale, const std::optional<ValueTuplePtr>& offset, const std::optional<ValueTuplePtr>& antiquant_scale, const std::optional<ValueTuplePtr>& antiquant_offset, const std::optional<ValuePtr>& group_list, const Int64ImmPtr& split_item, const Int64ImmPtr& group_type, const BoolImmPtr& transpose_a, const BoolImmPtr& transpose_b);
|
|
1006
|
-
py::object ME_EXPORT Pyboost_MoeComputeExpertTokens_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& sorted_experts, const Int64ImmPtr& num_expert);
|
|
1007
|
-
py::object ME_EXPORT Pyboost_MoeGatingTopKSoftmax_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x, const std::optional<ValuePtr>& finished, const Int64ImmPtr& k);
|
|
1008
|
-
py::object ME_EXPORT Pyboost_GroupedMatmulV4_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValueTuplePtr& x, const ValueTuplePtr& weight, const std::optional<ValueTuplePtr>& bias, const std::optional<ValueTuplePtr>& scale, const std::optional<ValueTuplePtr>& offset, const std::optional<ValueTuplePtr>& antiquant_scale, const std::optional<ValueTuplePtr>& antiquant_offset, const std::optional<ValueTuplePtr>& pre_token_scale, const std::optional<ValuePtr>& group_list, const std::optional<ValueTuplePtr>& activation_input, const std::optional<ValueTuplePtr>& activation_quant_scale, const std::optional<ValueTuplePtr>& activation_quant_offset, const Int64ImmPtr& split_item, const Int64ImmPtr& group_type, const Int64ImmPtr& group_list_type, const Int64ImmPtr& act_type);
|
|
1009
|
-
py::object ME_EXPORT Pyboost_MoeInitRoutingV2_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x, const ValuePtr& expert_idx, const Int64ImmPtr& active_num, const Int64ImmPtr& expert_capacity, const Int64ImmPtr& expert_num, const Int64ImmPtr& drop_pad_mode, const Int64ImmPtr& expert_tokens_count_or_cumsum_flag, const BoolImmPtr& expert_tokens_before_capacity_flag);
|
|
1010
|
-
py::object ME_EXPORT Pyboost_DynamicQuantExt_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x, const std::optional<ValuePtr>& smooth_scales);
|
|
1011
|
-
py::object ME_EXPORT Pyboost_MoeInitRouting_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x, const ValuePtr& row_idx, const ValuePtr& expert_idx, const Int64ImmPtr& active_num);
|
|
1012
|
-
py::object ME_EXPORT Pyboost_KVCacheScatterUpdate_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& var, const ValuePtr& indices, const ValuePtr& updates, const Int64ImmPtr& axis, const Int64ImmPtr& reduce);
|
|
1013
|
-
py::object ME_EXPORT Pyboost_AddRmsNormQuantV2_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x1, const ValuePtr& x2, const ValuePtr& gamma, const ValuePtr& scale, const ValuePtr& offset, const FP32ImmPtr& epsilon);
|
|
1014
|
-
py::object ME_EXPORT Pyboost_WeightQuantBatchMatmul_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x, const ValuePtr& weight, const ValuePtr& antiquant_scale, const std::optional<ValuePtr>& antiquant_offset, const std::optional<ValuePtr>& quant_scale, const std::optional<ValuePtr>& quant_offset, const std::optional<ValuePtr>& bias, const BoolImmPtr& transpose_x, const BoolImmPtr& transpose_weight, const Int64ImmPtr& antiquant_group_size);
|
|
1015
|
-
py::object ME_EXPORT Pyboost_MatmulAllReduceAddRmsNorm_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& x1, const ValuePtr& x2, const ValuePtr& bias, const ValuePtr& residual, const ValuePtr& gamma, const FP32ImmPtr& epsilon, const StringImmPtr& group, const Int64ImmPtr& reduce_op, const Int64ImmPtr& comm_turn, const Int64ImmPtr& stream_mode);
|
|
1016
|
-
py::object ME_EXPORT Pyboost_MoeFinalizeRouting_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& expanded_x, const ValuePtr& x1, const std::optional<ValuePtr>& x2, const std::optional<ValuePtr>& bias, const std::optional<ValuePtr>& scales, const std::optional<ValuePtr>& expanded_row_idx, const std::optional<ValuePtr>& expanded_expert_idx);
|
|
1017
|
-
py::object ME_EXPORT Pyboost_GroupedMatmulV2_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValueTuplePtr& x, const ValueTuplePtr& weight, const std::optional<ValueTuplePtr>& bias, const std::optional<ValueTuplePtr>& scale, const std::optional<ValueTuplePtr>& offset, const std::optional<ValueTuplePtr>& antiquant_scale, const std::optional<ValueTuplePtr>& antiquant_offset, const std::optional<ValueTuplePtr>& group_list, const Int64ImmPtr& split_item, const Int64ImmPtr& group_type);
|
|
1018
|
-
py::object ME_EXPORT Pyboost_PixelShuffle_OP(const PrimitivePtr &prim, const std::vector<ops::OP_DTYPE>& source_type, const ValuePtr& input, const Int64ImmPtr& upscale_factor);
|
|
1019
|
-
}// namespace mindspore::pynative
|