==================================================Ascend ============================= test session starts ============================== platform linux -- Python 3.9.19, pytest-6.2.5, py-1.11.0, pluggy-1.5.0 rootdir: /home/jenkins/mindspore/testcases/testcases/tests/st/infer/ops/test_internal_ops, configfile: ../../../../../../../../sault/virtual_test/virtualenv_001/sault/config/pytest.ini plugins: mock-3.14.0, hydra-core-1.3.2, forked-1.6.0, anyio-4.9.0, xdist-1.32.0 collected 16 items test_matmul_split.py [WARNING] ME(152078:281472851701552,MainProcess):2026-01-29-17:37:10.341.643 [mindspore/context.py:1334] For 'context.set_context', the parameter 'device_target' will be deprecated and removed in a future version. Please use the api mindspore.set_device() instead. TotalTime = 1.15054, [21] [bootstrap]: 0.0005812 [type_inference]: 1.05729 [event_method]: 5.228e-05 [auto_monad]: 0.0018452 [graph_reusing]: 8.90999e-06 [inline]: 3.88001e-06 [add_attr]: 0.00820506, [1] [add_attr_with_inline]: 0.00818858, [1] [Cycle 1]: 0.00010675, [2] [tag_attr]: 2.386e-05 [meta_addattr_fg_expand]: 4.87998e-06 [parallel-infer-symbol]: 3.42002e-06 [pre_auto_parallel]: 5.276e-05 [insert-virtual-dataset]: 2.41e-06 [parallel-infer-symbol-second]: 8.10018e-07 [dataset_repeat_opt]: 1.87001e-06 [pipeline_split]: 1.70001e-06 [optimize]: 0.0411684, [53] [py_interpret_to_execute]: 3.274e-05 [rewriter_before_opt_a]: 9.565e-05 [opt_a]: 0.0246379, [2] [Cycle 1]: 0.0229556, [45] [expand_dump_flag]: 2.66999e-06 [switch_simplify]: 3.669e-05 [loop_unroll]: 1.979e-05 [a_1]: 0.00068029 [with_stream_mark]: 2.814e-05 [recompute_prepare]: 2.134e-05 [updatestate_depend_eliminate]: 1.012e-05 [updatestate_assign_eliminate]: 1.146e-05 [updatestate_loads_eliminate]: 1.171e-05 [parameter_eliminate]: 2.17999e-06 [a_2]: 0.00023298 [accelerated_algorithm]: 4.619e-05 [shard]: 2.46e-06 [meta_shard_fg_expand]: 4.23999e-06 [shard_inline]: 1.602e-05 [merge_send_recv]: 1.334e-05 [auto_parallel]: 1.564e-05 [parallel]: 4.317e-05 [flash_sp]: 2.272e-05 [merge_comm]: 1.008e-05 [allreduce_fusion]: 8.36002e-06 [matmul_add_comm_reduction]: 1.771e-05 [allreduce_slice_to_reducescatter]: 7.80012e-07 [virtual_shard_identity]: 2.172e-05 [virtual_dataset]: 0.0120982 [get_grad_eliminate_]: 3.782e-05 [virtual_output]: 1.904e-05 [merge_forward]: 2.272e-05 [cell_reuse_recompute_pass]: 3.91999e-06 [offload_activation]: 3.137e-05 [cell_reuse_handle_not_recompute_node_pass]: 4.792e-05 [merge_recompute_call_nodes]: 3.38e-06 [before_grad]: 2.941e-05 [set_forward_comm_id_for_comm_node_pass]: 1.582e-05 [meta_fg_expand]: 1.091e-05 [flash_sp_send_recv_attached]: 1.293e-05 [receive_attached]: 2.106e-05 [after_resolve]: 2.752e-05 [a_after_grad]: 2.807e-05 [renormalize]: 0.00831139 [add_forward_monad_depend]: 1.999e-05 [auto_monad_grad]: 3.81001e-06 [auto_monad_eliminator]: 6.225e-05 [cse]: 0.00030842 [a_3]: 0.00013803 [Cycle 2]: 0.00166789, [45] [expand_dump_flag]: 3.27997e-06 [switch_simplify]: 1.935e-05 [loop_unroll]: 1.622e-05 [a_1]: 0.00045204 [with_stream_mark]: 3.165e-05 [recompute_prepare]: 2.053e-05 [updatestate_depend_eliminate]: 1.045e-05 [updatestate_assign_eliminate]: 7.86001e-06 [updatestate_loads_eliminate]: 1.036e-05 [parameter_eliminate]: 2.70997e-06 [a_2]: 0.00022132 [accelerated_algorithm]: 2.368e-05 [shard]: 2.11e-06 [meta_shard_fg_expand]: 5.72999e-06 [shard_inline]: 1.557e-05 [merge_send_recv]: 1.553e-05 [auto_parallel]: 1.65e-05 [parallel]: 9.66998e-06 [flash_sp]: 4.06001e-06 [merge_comm]: 8.72e-06 [allreduce_fusion]: 8.16002e-06 [matmul_add_comm_reduction]: 1.742e-05 [allreduce_slice_to_reducescatter]: 7.90023e-07 [virtual_shard_identity]: 2e-05 [virtual_dataset]: 1.475e-05 [get_grad_eliminate_]: 1.508e-05 [virtual_output]: 1.463e-05 [merge_forward]: 9.42999e-06 [cell_reuse_recompute_pass]: 3.3e-06 [offload_activation]: 1.762e-05 [cell_reuse_handle_not_recompute_node_pass]: 2.999e-05 [merge_recompute_call_nodes]: 1.58002e-06 [before_grad]: 2.599e-05 [set_forward_comm_id_for_comm_node_pass]: 9.31e-06 [meta_fg_expand]: 6.36998e-06 [flash_sp_send_recv_attached]: 1.76e-06 [receive_attached]: 2.19001e-06 [after_resolve]: 2.356e-05 [a_after_grad]: 2.606e-05 [renormalize]: 8.00064e-08 [add_forward_monad_depend]: 4.3e-06 [auto_monad_grad]: 2.39001e-06 [auto_monad_eliminator]: 6.151e-05 [cse]: 8.873e-05 [a_3]: 0.00010291 [py_interpret_to_execute_after_opt_a]: 2.856e-05 [slice_cell_reuse_recomputed_activation]: 2.71e-06 [rewriter_after_opt_a]: 0.00041139 [convert_after_rewriter]: 1.881e-05 [order_py_execute_after_rewriter]: 1.069e-05 [mutable_eliminate]: 0.0130995 [opt_b]: 0.00066312, [1] [Cycle 1]: 0.00065005, [7] [b_1]: 0.00043339 [b_2]: 2.12e-05 [updatestate_depend_eliminate]: 2.114e-05 [updatestate_assign_eliminate]: 1.014e-05 [updatestate_loads_eliminate]: 1.263e-05 [renormalize]: 8.2e-07 [cse]: 9.854e-05 [optimize_parallel_all_gather_comm]: 3.771e-05 [overlap_param_gather]: 5.67001e-06 [cconv]: 4.161e-05 [loop_unroll]: 0.00067756 [opt_after_cconv]: 0.00026356, [1] [Cycle 1]: 0.00025465, [7] [c_1]: 0.00010932 [parameter_eliminate]: 5.93998e-06 [updatestate_depend_eliminate]: 1.56e-05 [updatestate_assign_eliminate]: 7.77e-06 [updatestate_loads_eliminate]: 1.057e-05 [cse]: 6.358e-05 [renormalize]: 5.10016e-07 [remove_dup_value]: 7.233e-05 [tuple_transform]: 0.00016673, [1] [Cycle 1]: 0.00015984, [4] [d_1]: 0.00011688 [none_parameter_eliminate]: 2.26e-06 [renormalize]: 2.3999e-07 [switch_simplify]: 1.795e-05 [partial_unused_args_eliminate]: 2.02001e-06 [add_recomputation]: 0.00012062 [cse_after_recomputation]: 5.757e-05, [1] [Cycle 1]: 5.201e-05, [1] [cse]: 4.42e-05 [environ_conv]: 3.476e-05 [swap_dp_allreduce_reducescatter]: 1.254e-05 [bias_add_comm_swap]: 4.82998e-06 [label_micro_interleaved_index]: 6.22001e-06 [label_fine_grained_interleaved_index]: 2.55002e-06 [merge_cast_opt]: 1.39e-06 [slice_recompute_activation]: 2.32999e-06 [micro_interleaved_order_control]: 2.37001e-06 [assign_add_opt]: 1.30001e-06 [ForceFp32Comm]: 7.79983e-07 [remove_cast_before_assign_add]: 1.09e-06 [full_micro_interleaved_order_control]: 2.44001e-06 [reorder_send_recv_between_fp_bp]: 2.91999e-06 [comm_op_add_attrs]: 1.16002e-06 [add_comm_op_reuse_tag]: 9.5999e-07 [interleave_split_concat_branches]: 1.38002e-06 [interleave_parallel_branches]: 9.89996e-07 [overlap_opt_shard_in_pipeline]: 2.623e-05 [overlap_opt_shard_grad_in_pipeline]: 2.02001e-06 [control_data_broadcast_order]: 2.975e-05 [grouped_pairwise_exchange_alltoall]: 1.45999e-06 [offloading_packed_experts]: 8.32e-06 [overlap_recompute_and_grad_model_parallel]: 7.92998e-06 [overlap_grad_matmul_and_grad_allreduce]: 1.15999e-06 [overlap_recompute_allgather_and_fa_grad]: 1.19e-06 [overlap_recompute_comm]: 2.49999e-06 [overlap_grad_ring_attention]: 7.4e-06 [overlap_grad_flash_sp]: 5.325e-05 [begin_end_overlap_inline]: 6.19999e-07 [split_matmul_comm_elemetwise]: 2.10002e-06 [split_layernorm_comm]: 2.10002e-06 [handle_group_info]: 1.02e-06 [symbol_engine_optimizer]: 0.00018103, [1] [Cycle 1]: 0.00017359, [6] [build]: 1.808e-05 [elim_shapecalc]: 2.823e-05 [elim_not_effective]: 3.218e-05 [opt_reshape]: 2.621e-05 [fold_const_symbol]: 2.742e-05 [renormalize]: 3.60014e-07 [detach_backward]: 2.71999e-06 [pipeline_parallel_scheduler]: 1.89999e-06 [auto_monad_reorder]: 5.661e-05 [get_jit_bprop_graph]: 2.06e-06 [rewriter_after_jit_bprop_graph]: 7.08998e-06 [opt_after_jit_grad]: 0.00077677 [validate]: 0.00011934 Sums bootstrap : 0.000581s : 0.05% type_inference : 1.057289s : 96.05% event_method : 0.000052s : 0.00% auto_monad : 0.001845s : 0.17% graph_reusing : 0.000009s : 0.00% inline : 0.000004s : 0.00% add_attr.add_attr_with_inline.tag_attr : 0.000024s : 0.00% add_attr.add_attr_with_inline.meta_addattr_fg_expand : 0.000005s : 0.00% parallel-infer-symbol : 0.000003s : 0.00% pre_auto_parallel : 0.000053s : 0.00% insert-virtual-dataset : 0.000002s : 0.00% parallel-infer-symbol-second : 0.000001s : 0.00% dataset_repeat_opt : 0.000002s : 0.00% pipeline_split : 0.000002s : 0.00% optimize.py_interpret_to_execute : 0.000033s : 0.00% optimize.rewriter_before_opt_a : 0.000096s : 0.01% optimize.opt_a.expand_dump_flag : 0.000006s : 0.00% optimize.opt_a.switch_simplify : 0.000056s : 0.01% optimize.opt_a.loop_unroll : 0.000036s : 0.00% optimize.opt_a.a_1 : 0.001132s : 0.10% optimize.opt_a.with_stream_mark : 0.000060s : 0.01% optimize.opt_a.recompute_prepare : 0.000042s : 0.00% optimize.opt_a.updatestate_depend_eliminate : 0.000021s : 0.00% optimize.opt_a.updatestate_assign_eliminate : 0.000019s : 0.00% optimize.opt_a.updatestate_loads_eliminate : 0.000022s : 0.00% optimize.opt_a.parameter_eliminate : 0.000005s : 0.00% optimize.opt_a.a_2 : 0.000454s : 0.04% optimize.opt_a.accelerated_algorithm : 0.000070s : 0.01% optimize.opt_a.shard : 0.000005s : 0.00% optimize.opt_a.meta_shard_fg_expand : 0.000010s : 0.00% optimize.opt_a.shard_inline : 0.000032s : 0.00% optimize.opt_a.merge_send_recv : 0.000029s : 0.00% optimize.opt_a.auto_parallel : 0.000032s : 0.00% optimize.opt_a.parallel : 0.000053s : 0.00% optimize.opt_a.flash_sp : 0.000027s : 0.00% optimize.opt_a.merge_comm : 0.000019s : 0.00% optimize.opt_a.allreduce_fusion : 0.000017s : 0.00% optimize.opt_a.matmul_add_comm_reduction : 0.000035s : 0.00% optimize.opt_a.allreduce_slice_to_reducescatter : 0.000002s : 0.00% optimize.opt_a.virtual_shard_identity : 0.000042s : 0.00% optimize.opt_a.virtual_dataset : 0.012113s : 1.10% optimize.opt_a.get_grad_eliminate_ : 0.000053s : 0.00% optimize.opt_a.virtual_output : 0.000034s : 0.00% optimize.opt_a.merge_forward : 0.000032s : 0.00% optimize.opt_a.cell_reuse_recompute_pass : 0.000007s : 0.00% optimize.opt_a.offload_activation : 0.000049s : 0.00% optimize.opt_a.cell_reuse_handle_not_recompute_node_pass : 0.000078s : 0.01% optimize.opt_a.merge_recompute_call_nodes : 0.000005s : 0.00% optimize.opt_a.before_grad : 0.000055s : 0.01% optimize.opt_a.set_forward_comm_id_for_comm_node_pass : 0.000025s : 0.00% optimize.opt_a.meta_fg_expand : 0.000017s : 0.00% optimize.opt_a.flash_sp_send_recv_attached : 0.000015s : 0.00% optimize.opt_a.receive_attached : 0.000023s : 0.00% optimize.opt_a.after_resolve : 0.000051s : 0.00% optimize.opt_a.a_after_grad : 0.000054s : 0.00% optimize.opt_a.renormalize : 0.008311s : 0.76% optimize.opt_a.add_forward_monad_depend : 0.000024s : 0.00% optimize.opt_a.auto_monad_grad : 0.000006s : 0.00% optimize.opt_a.auto_monad_eliminator : 0.000124s : 0.01% optimize.opt_a.cse : 0.000397s : 0.04% optimize.opt_a.a_3 : 0.000241s : 0.02% optimize.py_interpret_to_execute_after_opt_a : 0.000029s : 0.00% optimize.slice_cell_reuse_recomputed_activation : 0.000003s : 0.00% optimize.rewriter_after_opt_a : 0.000411s : 0.04% optimize.convert_after_rewriter : 0.000019s : 0.00% optimize.order_py_execute_after_rewriter : 0.000011s : 0.00% optimize.mutable_eliminate : 0.013100s : 1.19% optimize.opt_b.b_1 : 0.000433s : 0.04% optimize.opt_b.b_2 : 0.000021s : 0.00% optimize.opt_b.updatestate_depend_eliminate : 0.000021s : 0.00% optimize.opt_b.updatestate_assign_eliminate : 0.000010s : 0.00% optimize.opt_b.updatestate_loads_eliminate : 0.000013s : 0.00% optimize.opt_b.renormalize : 0.000001s : 0.00% optimize.opt_b.cse : 0.000099s : 0.01% optimize.optimize_parallel_all_gather_comm : 0.000038s : 0.00% optimize.overlap_param_gather : 0.000006s : 0.00% optimize.cconv : 0.000042s : 0.00% optimize.loop_unroll : 0.000678s : 0.06% optimize.opt_after_cconv.c_1 : 0.000109s : 0.01% optimize.opt_after_cconv.parameter_eliminate : 0.000006s : 0.00% optimize.opt_after_cconv.updatestate_depend_eliminate : 0.000016s : 0.00% optimize.opt_after_cconv.updatestate_assign_eliminate : 0.000008s : 0.00% optimize.opt_after_cconv.updatestate_loads_eliminate : 0.000011s : 0.00% optimize.opt_after_cconv.cse : 0.000064s : 0.01% optimize.opt_after_cconv.renormalize : 0.000001s : 0.00% optimize.remove_dup_value : 0.000072s : 0.01% optimize.tuple_transform.d_1 : 0.000117s : 0.01% optimize.tuple_transform.none_parameter_eliminate : 0.000002s : 0.00% optimize.tuple_transform.renormalize : 0.000000s : 0.00% optimize.tuple_transform.switch_simplify : 0.000018s : 0.00% optimize.partial_unused_args_eliminate : 0.000002s : 0.00% optimize.add_recomputation : 0.000121s : 0.01% optimize.cse_after_recomputation.cse : 0.000044s : 0.00% optimize.environ_conv : 0.000035s : 0.00% optimize.swap_dp_allreduce_reducescatter : 0.000013s : 0.00% optimize.bias_add_comm_swap : 0.000005s : 0.00% optimize.label_micro_interleaved_index : 0.000006s : 0.00% optimize.label_fine_grained_interleaved_index : 0.000003s : 0.00% optimize.merge_cast_opt : 0.000001s : 0.00% optimize.slice_recompute_activation : 0.000002s : 0.00% optimize.micro_interleaved_order_control : 0.000002s : 0.00% optimize.assign_add_opt : 0.000001s : 0.00% optimize.ForceFp32Comm : 0.000001s : 0.00% optimize.remove_cast_before_assign_add : 0.000001s : 0.00% optimize.full_micro_interleaved_order_control : 0.000002s : 0.00% optimize.reorder_send_recv_between_fp_bp : 0.000003s : 0.00% optimize.comm_op_add_attrs : 0.000001s : 0.00% optimize.add_comm_op_reuse_tag : 0.000001s : 0.00% optimize.interleave_split_concat_branches : 0.000001s : 0.00% optimize.interleave_parallel_branches : 0.000001s : 0.00% optimize.overlap_opt_shard_in_pipeline : 0.000026s : 0.00% optimize.overlap_opt_shard_grad_in_pipeline : 0.000002s : 0.00% optimize.control_data_broadcast_order : 0.000030s : 0.00% optimize.grouped_pairwise_exchange_alltoall : 0.000001s : 0.00% optimize.offloading_packed_experts : 0.000008s : 0.00% optimize.overlap_recompute_and_grad_model_parallel : 0.000008s : 0.00% optimize.overlap_grad_matmul_and_grad_allreduce : 0.000001s : 0.00% optimize.overlap_recompute_allgather_and_fa_grad : 0.000001s : 0.00% optimize.overlap_recompute_comm : 0.000002s : 0.00% optimize.overlap_grad_ring_attention : 0.000007s : 0.00% optimize.overlap_grad_flash_sp : 0.000053s : 0.00% optimize.begin_end_overlap_inline : 0.000001s : 0.00% optimize.split_matmul_comm_elemetwise : 0.000002s : 0.00% optimize.split_layernorm_comm : 0.000002s : 0.00% optimize.handle_group_info : 0.000001s : 0.00% optimize.symbol_engine_optimizer.build : 0.000018s : 0.00% optimize.symbol_engine_optimizer.elim_shapecalc : 0.000028s : 0.00% optimize.symbol_engine_optimizer.elim_not_effective : 0.000032s : 0.00% optimize.symbol_engine_optimizer.opt_reshape : 0.000026s : 0.00% optimize.symbol_engine_optimizer.fold_const_symbol : 0.000027s : 0.00% optimize.symbol_engine_optimizer.renormalize : 0.000000s : 0.00% detach_backward : 0.000003s : 0.00% pipeline_parallel_scheduler : 0.000002s : 0.00% auto_monad_reorder : 0.000057s : 0.01% get_jit_bprop_graph : 0.000002s : 0.00% rewriter_after_jit_bprop_graph : 0.000007s : 0.00% opt_after_jit_grad : 0.000777s : 0.07% validate : 0.000119s : 0.01% Time group info: ------[substitution.] 0.000309 96 3.67% : 0.000011s : 2: substitution.depend_value_elim 1.26% : 0.000004s : 8: substitution.elim_not_effective 1.31% : 0.000004s : 8: substitution.fold_const_symbol 3.93% : 0.000012s : 13: substitution.graph_param_transform 39.55% : 0.000122s : 1: substitution.inline 3.33% : 0.000010s : 16: substitution.j_node_and_user_rematch 7.89% : 0.000024s : 2: substitution.less_batch_normalization 2.04% : 0.000006s : 6: substitution.load_eliminater 0.71% : 0.000002s : 2: substitution.opt_reshape 4.58% : 0.000014s : 16: substitution.remove_not_recompute_node 3.40% : 0.000010s : 4: substitution.replace_old_param 13.03% : 0.000040s : 4: substitution.reshape_eliminate 8.50% : 0.000026s : 6: substitution.updatestate_pure_node_eliminater 6.80% : 0.000021s : 8: substitution.updatestate_useless_node_eliminater ------[type_inference.] 1.057177 2 99.59% : 1.052832s : 1: type_inference.infer 0.41% : 0.004346s : 1: type_inference.specialize ------[replace.] 0.000021 1 100.00% : 0.000021s : 1: replace.inline ------[match.] 0.000121 1 100.00% : 0.000121s : 1: match.inline ------[predicate.] 0.000455 3002 0.81% : 0.000004s : 27: predicate.accumulaten_eliminater 1.02% : 0.000005s : 13: predicate.ad_related_special_op_eliminate 0.70% : 0.000003s : 26: predicate.addn_check_dump 0.74% : 0.000003s : 27: predicate.addn_zero_filter 0.71% : 0.000003s : 27: predicate.adjust_all_reduce_mul_add 2.04% : 0.000009s : 53: predicate.arithmetic_simplify 0.81% : 0.000004s : 27: predicate.cast_eliminate 0.73% : 0.000003s : 26: predicate.check_bprop_eliminate 0.77% : 0.000003s : 26: predicate.compare_switch_simplify 0.25% : 0.000001s : 13: predicate.const_output_eliminate 0.79% : 0.000004s : 26: predicate.depend_value_elim 0.75% : 0.000003s : 27: predicate.dict_get_item_const_eliminator 0.87% : 0.000004s : 27: predicate.dict_get_item_eliminator 0.77% : 0.000004s : 27: predicate.dict_set_item_eliminator 1.13% : 0.000005s : 26: predicate.dumpgradient_eliminate 0.29% : 0.000001s : 13: predicate.elim_not_effective 0.54% : 0.000002s : 13: predicate.elim_shapecalc_of_broadcastargs 1.08% : 0.000005s : 40: predicate.environ_add_const_eliminate 1.01% : 0.000005s : 40: predicate.environ_get_add_eliminate 1.03% : 0.000005s : 40: predicate.environ_get_depend_swap 1.88% : 0.000009s : 66: predicate.environ_get_eliminate 1.06% : 0.000005s : 40: predicate.environ_get_set_eliminate 0.79% : 0.000004s : 28: predicate.exchange_switch_depend_value 1.37% : 0.000006s : 28: predicate.float_depend_g_call 0.74% : 0.000003s : 26: predicate.float_environ_get_switch 1.10% : 0.000005s : 39: predicate.float_tuple_getitem_switch 0.24% : 0.000001s : 13: predicate.fold_const_symbol 1.69% : 0.000008s : 26: predicate.get_grad_eliminate 0.29% : 0.000001s : 13: predicate.graph_param_transform 0.79% : 0.000004s : 26: predicate.incorporate_call 0.66% : 0.000003s : 26: predicate.incorporate_call_switch 5.24% : 0.000024s : 133: predicate.inline 1.24% : 0.000006s : 26: predicate.inline_without_move 0.40% : 0.000002s : 26: predicate.j_node_and_user_rematch 1.49% : 0.000007s : 27: predicate.less_batch_normalization 1.62% : 0.000007s : 53: predicate.list_to_tuple_eliminator_ 2.20% : 0.000010s : 80: predicate.load_eliminater 1.28% : 0.000006s : 13: predicate.loop_unroll_after_grad 0.97% : 0.000004s : 31: predicate.loop_unroll_before_grad 1.72% : 0.000008s : 53: predicate.make_slice_get_slice_eliminator 0.76% : 0.000003s : 26: predicate.merge_addn 0.75% : 0.000003s : 26: predicate.micro_step_allgather_replace 0.78% : 0.000004s : 26: predicate.mini_step_allgather_replace 0.70% : 0.000003s : 27: predicate.minmaximum_grad 2.54% : 0.000012s : 13: predicate.mutable_eliminate 0.62% : 0.000003s : 13: predicate.opt_reshape 0.54% : 0.000002s : 13: predicate.parallel_virtual_node 1.16% : 0.000005s : 28: predicate.partial_defer_inline 1.21% : 0.000006s : 40: predicate.partial_eliminate 0.69% : 0.000003s : 27: predicate.print_const_string_wrapper 0.76% : 0.000003s : 26: predicate.reduce_all_const_elim 1.10% : 0.000005s : 27: predicate.reduce_eliminate 2.19% : 0.000010s : 80: predicate.redundant_stop_gradient_eliminater 0.62% : 0.000003s : 26: predicate.remove_not_recompute_node 1.36% : 0.000006s : 53: predicate.replace_applicator 0.53% : 0.000002s : 26: predicate.replace_old_param 0.38% : 0.000002s : 13: predicate.reset_defer_inline 0.92% : 0.000004s : 27: predicate.reshape_eliminate 0.82% : 0.000004s : 26: predicate.row_tensor_add_zeros_like 0.45% : 0.000002s : 13: predicate.row_tensor_eliminate 1.08% : 0.000005s : 26: predicate.same_eliminate 0.54% : 0.000002s : 26: predicate.set_cell_output_no_recompute 1.00% : 0.000005s : 26: predicate.shard_identity_eliminate 0.86% : 0.000004s : 26: predicate.special_op_eliminate 0.91% : 0.000004s : 26: predicate.specialize_transform 1.20% : 0.000005s : 26: predicate.split_environ_get_set_with_tuple_value 0.96% : 0.000004s : 26: predicate.stack_unstack_eliminate 0.47% : 0.000002s : 13: predicate.switch_call_monad_eliminater 0.86% : 0.000004s : 28: predicate.switch_defer_inline 1.61% : 0.000007s : 54: predicate.switch_layer_defer_inline 3.37% : 0.000015s : 98: predicate.switch_simplify 0.74% : 0.000003s : 27: predicate.tile_eliminate 0.77% : 0.000004s : 27: predicate.transpose_eliminate 1.60% : 0.000007s : 53: predicate.tuple_list_convert_item_index_to_positive 1.67% : 0.000008s : 53: predicate.tuple_list_get_item_const_eliminator 1.45% : 0.000007s : 53: predicate.tuple_list_get_item_depend_reorder 3.03% : 0.000014s : 79: predicate.tuple_list_get_item_eliminator 1.51% : 0.000007s : 53: predicate.tuple_list_get_set_item_eliminator 2.47% : 0.000011s : 79: predicate.tuple_list_set_item_eliminator 1.57% : 0.000007s : 53: predicate.tuple_to_list_eliminator_ 2.16% : 0.000010s : 80: predicate.updatestate_pure_node_eliminater 3.37% : 0.000015s : 106: predicate.updatestate_useless_node_eliminater 0.46% : 0.000002s : 13: predicate.value_based_eliminate 2.04% : 0.000009s : 26: predicate.virtual_dataset_eliminate 0.87% : 0.000004s : 26: predicate.virtual_output_eliminate 0.38% : 0.000002s : 13: predicate.virtual_view_grad_eliminate 0.56% : 0.000003s : 13: predicate.zero_like_fill_zero ------[func_graph_cloner_run.] 0.038744 58 94.50% : 0.036613s : 55: func_graph_cloner_run.FuncGraphClonerGraph 5.50% : 0.002131s : 3: func_graph_cloner_run.FuncGraphSpecializer ------[meta_graph.] 0.000000 0 ------[manager.] 0.000000 0 ------[pynative] 0.000000 0 ------[others.] 1.223282 192 0.00% : 0.000004s : 1: ForceFp32Comm 0.67% : 0.008212s : 1: add_attr 0.67% : 0.008194s : 1: add_attr_with_inline 0.00% : 0.000004s : 1: add_comm_op_reuse_tag 0.01% : 0.000127s : 1: add_recomputation 0.00% : 0.000004s : 1: assign_add_opt 0.15% : 0.001886s : 1: auto_monad 0.01% : 0.000064s : 1: auto_monad_reorder 0.00% : 0.000004s : 1: begin_end_overlap_inline 0.00% : 0.000008s : 1: bias_add_comm_swap 0.05% : 0.000622s : 1: bootstrap 0.00% : 0.000046s : 1: cconv 0.00% : 0.000004s : 1: comm_op_add_attrs 0.00% : 0.000034s : 1: control_data_broadcast_order 0.00% : 0.000023s : 1: convert_after_rewriter 0.00% : 0.000061s : 1: cse_after_recomputation 0.00% : 0.000005s : 1: dataset_repeat_opt 0.00% : 0.000006s : 1: detach_backward 0.00% : 0.000039s : 1: environ_conv 0.01% : 0.000103s : 1: event_method 0.00% : 0.000005s : 1: full_micro_interleaved_order_control 0.00% : 0.000005s : 1: get_jit_bprop_graph 0.00% : 0.000016s : 1: graph_reusing 0.00% : 0.000005s : 1: grouped_pairwise_exchange_alltoall 0.00% : 0.000004s : 1: handle_group_info 0.00% : 0.000008s : 1: inline 0.00% : 0.000006s : 1: insert-virtual-dataset 0.00% : 0.000004s : 1: interleave_parallel_branches 0.00% : 0.000004s : 1: interleave_split_concat_branches 0.00% : 0.000006s : 1: label_fine_grained_interleaved_index 0.00% : 0.000009s : 1: label_micro_interleaved_index 0.06% : 0.000691s : 1: loop_unroll 0.00% : 0.000004s : 1: merge_cast_opt 0.00% : 0.000005s : 1: micro_interleaved_order_control 1.07% : 0.013126s : 1: mutable_eliminate 0.00% : 0.000011s : 1: offloading_packed_experts 0.00% : 0.000036s : 1: opt.transform.loop_unroll_optimizer 0.00% : 0.000060s : 1: opt.transform.mutable_eliminate 1.18% : 0.014407s : 78: opt.transform.opt_a 0.01% : 0.000108s : 1: opt.transform.opt_after_cconv 0.01% : 0.000066s : 1: opt.transform.opt_after_jit_grad 0.03% : 0.000417s : 28: opt.transform.opt_b 0.01% : 0.000131s : 2: opt.transform.opt_trans_graph 0.01% : 0.000109s : 4: opt.transform.symbol_engine_opt 2.01% : 0.024642s : 1: opt_a 0.02% : 0.000267s : 1: opt_after_cconv 0.06% : 0.000793s : 1: opt_after_jit_grad 0.05% : 0.000669s : 1: opt_b 3.37% : 0.041174s : 1: optimize 0.00% : 0.000043s : 1: optimize_parallel_all_gather_comm 0.00% : 0.000014s : 1: order_py_execute_after_rewriter 0.00% : 0.000060s : 1: overlap_grad_flash_sp 0.00% : 0.000004s : 1: overlap_grad_matmul_and_grad_allreduce 0.00% : 0.000013s : 1: overlap_grad_ring_attention 0.00% : 0.000005s : 1: overlap_opt_shard_grad_in_pipeline 0.00% : 0.000031s : 1: overlap_opt_shard_in_pipeline 0.00% : 0.000010s : 1: overlap_param_gather 0.00% : 0.000004s : 1: overlap_recompute_allgather_and_fa_grad 0.00% : 0.000011s : 1: overlap_recompute_and_grad_model_parallel 0.00% : 0.000006s : 1: overlap_recompute_comm 0.00% : 0.000008s : 1: parallel-infer-symbol 0.00% : 0.000004s : 1: parallel-infer-symbol-second 0.00% : 0.000005s : 1: partial_unused_args_eliminate 0.00% : 0.000005s : 1: pipeline_parallel_scheduler 0.00% : 0.000005s : 1: pipeline_split 0.00% : 0.000057s : 1: pre_auto_parallel 0.00% : 0.000037s : 1: py_interpret_to_execute 0.00% : 0.000033s : 1: py_interpret_to_execute_after_opt_a 0.00% : 0.000004s : 1: remove_cast_before_assign_add 0.01% : 0.000078s : 1: remove_dup_value 0.47% : 0.005780s : 1: renormalize.infer 0.21% : 0.002509s : 1: renormalize.specialize 0.00% : 0.000006s : 1: reorder_send_recv_between_fp_bp 0.00% : 0.000010s : 1: rewriter_after_jit_bprop_graph 0.03% : 0.000425s : 1: rewriter_after_opt_a 0.01% : 0.000102s : 1: rewriter_before_opt_a 0.00% : 0.000006s : 1: slice_cell_reuse_recomputed_activation 0.00% : 0.000005s : 1: slice_recompute_activation 0.00% : 0.000005s : 1: split_layernorm_comm 0.00% : 0.000005s : 1: split_matmul_comm_elemetwise 0.00% : 0.000016s : 1: swap_dp_allreduce_reducescatter 0.02% : 0.000184s : 1: symbol_engine_optimizer 0.01% : 0.000170s : 1: tuple_transform 89.71% : 1.097379s : 1: type_inference . [hook] pytest_runtest_teardown:test_matmul_qkv_bf16_0[False-mstype0-k_n_shape0-1] tests/st/infer/ops/test_internal_ops/test_matmul_split.py::test_matmul_qkv_bf16_0[False-mstype0-k_n_shape0-1],max_mem:120.0M [WARNING] ME(152078:281472851701552,MainProcess):2026-01-29-17:37:59.664.362 [mindspore/context.py:1334] For 'context.set_context', the parameter 'device_target' will be deprecated and removed in a future version. Please use the api mindspore.set_device() instead. TotalTime = 3.3251, [21] [bootstrap]: 0.0352609 [type_inference]: 2.95551 [event_method]: 1.996e-05 [auto_monad]: 0.00030486 [graph_reusing]: 6.62002e-06 [inline]: 3.16001e-06 [add_attr]: 0.162002, [1] [add_attr_with_inline]: 0.161981, [1] [Cycle 1]: 9.083e-05, [2] [tag_attr]: 2.812e-05 [meta_addattr_fg_expand]: 5.09e-06 [parallel-infer-symbol]: 5.70001e-06 [pre_auto_parallel]: 4.879e-05 [insert-virtual-dataset]: 3.03e-06 [parallel-infer-symbol-second]: 7.7e-07 [dataset_repeat_opt]: 2.27999e-06 [pipeline_split]: 1.79e-06 [optimize]: 0.170844, [53] [py_interpret_to_execute]: 4.366e-05 [rewriter_before_opt_a]: 0.00010363 [opt_a]: 0.167073, [2] [Cycle 1]: 0.165142, [45] [expand_dump_flag]: 3.04999e-06 [switch_simplify]: 3.844e-05 [loop_unroll]: 2.037e-05 [a_1]: 0.00084314 [with_stream_mark]: 3.778e-05 [recompute_prepare]: 3.018e-05 [updatestate_depend_eliminate]: 1.462e-05 [updatestate_assign_eliminate]: 1.103e-05 [updatestate_loads_eliminate]: 1.187e-05 [parameter_eliminate]: 2.71e-06 [a_2]: 0.00026825 [accelerated_algorithm]: 5.639e-05 [shard]: 3.4e-06 [meta_shard_fg_expand]: 5.29e-06 [shard_inline]: 1.87e-05 [merge_send_recv]: 1.923e-05 [auto_parallel]: 1.746e-05 [parallel]: 3.133e-05 [flash_sp]: 1.606e-05 [merge_comm]: 1.117e-05 [allreduce_fusion]: 8.92e-06 [matmul_add_comm_reduction]: 1.898e-05 [allreduce_slice_to_reducescatter]: 8.79983e-07 [virtual_shard_identity]: 4.631e-05 [virtual_dataset]: 2.592e-05 [get_grad_eliminate_]: 3.937e-05 [virtual_output]: 2.409e-05 [merge_forward]: 1.092e-05 [cell_reuse_recompute_pass]: 2.70002e-06 [offload_activation]: 1.965e-05 [cell_reuse_handle_not_recompute_node_pass]: 4.001e-05 [merge_recompute_call_nodes]: 1.67001e-06 [before_grad]: 3.216e-05 [set_forward_comm_id_for_comm_node_pass]: 1.088e-05 [meta_fg_expand]: 7.82e-06 [flash_sp_send_recv_attached]: 6.84999e-06 [receive_attached]: 2.34001e-06 [after_resolve]: 2.867e-05 [a_after_grad]: 3.152e-05 [renormalize]: 0.159947 [add_forward_monad_depend]: 1.964e-05 [auto_monad_grad]: 3.40998e-06 [auto_monad_eliminator]: 0.00140661 [cse]: 0.00137734 [a_3]: 0.00015006 [Cycle 2]: 0.00191328, [45] [expand_dump_flag]: 4.46002e-06 [switch_simplify]: 1.912e-05 [loop_unroll]: 1.714e-05 [a_1]: 0.00047241 [with_stream_mark]: 3.594e-05 [recompute_prepare]: 2.051e-05 [updatestate_depend_eliminate]: 1.057e-05 [updatestate_assign_eliminate]: 8.16002e-06 [updatestate_loads_eliminate]: 1.157e-05 [parameter_eliminate]: 2.73e-06 [a_2]: 0.00022192 [accelerated_algorithm]: 2.258e-05 [shard]: 2.61e-06 [meta_shard_fg_expand]: 8.23999e-06 [shard_inline]: 1.586e-05 [merge_send_recv]: 1.619e-05 [auto_parallel]: 1.677e-05 [parallel]: 9.54e-06 [flash_sp]: 8.331e-05 [merge_comm]: 1.025e-05 [allreduce_fusion]: 9.16998e-06 [matmul_add_comm_reduction]: 2.006e-05 [allreduce_slice_to_reducescatter]: 8.70001e-07 [virtual_shard_identity]: 2.062e-05 [virtual_dataset]: 1.545e-05 [get_grad_eliminate_]: 1.51e-05 [virtual_output]: 1.501e-05 [merge_forward]: 9.69e-06 [cell_reuse_recompute_pass]: 3.41001e-06 [offload_activation]: 1.804e-05 [cell_reuse_handle_not_recompute_node_pass]: 3.023e-05 [merge_recompute_call_nodes]: 1.87001e-06 [before_grad]: 2.76e-05 [set_forward_comm_id_for_comm_node_pass]: 9.30001e-06 [meta_fg_expand]: 7.28e-06 [flash_sp_send_recv_attached]: 1.83002e-06 [receive_attached]: 2.96001e-06 [after_resolve]: 2.669e-05 [a_after_grad]: 2.839e-05 [renormalize]: 8.00064e-08 [add_forward_monad_depend]: 4.89003e-06 [auto_monad_grad]: 3.06001e-06 [auto_monad_eliminator]: 6.486e-05 [cse]: 0.00017612 [a_3]: 0.00011821 [py_interpret_to_execute_after_opt_a]: 3.09e-05 [slice_cell_reuse_recomputed_activation]: 2.44999e-06 [rewriter_after_opt_a]: 0.00020913 [convert_after_rewriter]: 1.829e-05 [order_py_execute_after_rewriter]: 1.035e-05 [mutable_eliminate]: 0.00080689 [opt_b]: 0.00055434, [1] [Cycle 1]: 0.00054407, [7] [b_1]: 0.00036917 [b_2]: 1.934e-05 [updatestate_depend_eliminate]: 1.743e-05 [updatestate_assign_eliminate]: 8.22e-06 [updatestate_loads_eliminate]: 1.139e-05 [renormalize]: 9.20001e-07 [cse]: 7.315e-05 [optimize_parallel_all_gather_comm]: 3.467e-05 [overlap_param_gather]: 1.92999e-06 [cconv]: 4.154e-05 [loop_unroll]: 0.00056323 [opt_after_cconv]: 0.00023844, [1] [Cycle 1]: 0.0002291, [7] [c_1]: 0.00010204 [parameter_eliminate]: 5.64998e-06 [updatestate_depend_eliminate]: 1.441e-05 [updatestate_assign_eliminate]: 7.21001e-06 [updatestate_loads_eliminate]: 9.63002e-06 [cse]: 5.087e-05 [renormalize]: 1.09e-06 [remove_dup_value]: 7.387e-05 [tuple_transform]: 0.00016513, [1] [Cycle 1]: 0.0001591, [4] [d_1]: 0.00011443 [none_parameter_eliminate]: 2.71e-06 [renormalize]: 2.00002e-07 [switch_simplify]: 1.778e-05 [partial_unused_args_eliminate]: 2.23002e-06 [add_recomputation]: 0.00011516 [cse_after_recomputation]: 5.591e-05, [1] [Cycle 1]: 4.99e-05, [1] [cse]: 4.355e-05 [environ_conv]: 2.064e-05 [swap_dp_allreduce_reducescatter]: 1.161e-05 [bias_add_comm_swap]: 3.26999e-06 [label_micro_interleaved_index]: 4.83001e-06 [label_fine_grained_interleaved_index]: 2.92002e-06 [merge_cast_opt]: 1.47999e-06 [slice_recompute_activation]: 2.27999e-06 [micro_interleaved_order_control]: 2.47001e-06 [assign_add_opt]: 1.60999e-06 [ForceFp32Comm]: 8.79983e-07 [remove_cast_before_assign_add]: 1.45999e-06 [full_micro_interleaved_order_control]: 2.15002e-06 [reorder_send_recv_between_fp_bp]: 2.63e-06 [comm_op_add_attrs]: 1.21002e-06 [add_comm_op_reuse_tag]: 1.13001e-06 [interleave_split_concat_branches]: 1.17e-06 [interleave_parallel_branches]: 1.20001e-06 [overlap_opt_shard_in_pipeline]: 1.16997e-06 [overlap_opt_shard_grad_in_pipeline]: 1.79e-06 [control_data_broadcast_order]: 2.744e-05 [grouped_pairwise_exchange_alltoall]: 2.15002e-06 [offloading_packed_experts]: 7.35e-06 [overlap_recompute_and_grad_model_parallel]: 7.69002e-06 [overlap_grad_matmul_and_grad_allreduce]: 1.28002e-06 [overlap_recompute_allgather_and_fa_grad]: 1.25999e-06 [overlap_recompute_comm]: 2.31e-06 [overlap_grad_ring_attention]: 8.65999e-06 [overlap_grad_flash_sp]: 4.252e-05 [begin_end_overlap_inline]: 9.50007e-07 [split_matmul_comm_elemetwise]: 2.09999e-06 [split_layernorm_comm]: 1.67001e-06 [handle_group_info]: 1.09e-06 [symbol_engine_optimizer]: 0.00016093, [1] [Cycle 1]: 0.00015427, [6] [build]: 1.909e-05 [elim_shapecalc]: 2.315e-05 [elim_not_effective]: 2.958e-05 [opt_reshape]: 2.28e-05 [fold_const_symbol]: 2.477e-05 [renormalize]: 2.10013e-07 [detach_backward]: 2.48998e-06 [pipeline_parallel_scheduler]: 1.44e-06 [auto_monad_reorder]: 4.763e-05 [get_jit_bprop_graph]: 2.14999e-06 [rewriter_after_jit_bprop_graph]: 7.41001e-06 [opt_after_jit_grad]: 0.0006823 [validate]: 0.00010631 Sums bootstrap : 0.035261s : 1.12% type_inference : 2.955509s : 93.48% event_method : 0.000020s : 0.00% auto_monad : 0.000305s : 0.01% graph_reusing : 0.000007s : 0.00% inline : 0.000003s : 0.00% add_attr.add_attr_with_inline.tag_attr : 0.000028s : 0.00% add_attr.add_attr_with_inline.meta_addattr_fg_expand : 0.000005s : 0.00% parallel-infer-symbol : 0.000006s : 0.00% pre_auto_parallel : 0.000049s : 0.00% insert-virtual-dataset : 0.000003s : 0.00% parallel-infer-symbol-second : 0.000001s : 0.00% dataset_repeat_opt : 0.000002s : 0.00% pipeline_split : 0.000002s : 0.00% optimize.py_interpret_to_execute : 0.000044s : 0.00% optimize.rewriter_before_opt_a : 0.000104s : 0.00% optimize.opt_a.expand_dump_flag : 0.000008s : 0.00% optimize.opt_a.switch_simplify : 0.000058s : 0.00% optimize.opt_a.loop_unroll : 0.000038s : 0.00% optimize.opt_a.a_1 : 0.001316s : 0.04% optimize.opt_a.with_stream_mark : 0.000074s : 0.00% optimize.opt_a.recompute_prepare : 0.000051s : 0.00% optimize.opt_a.updatestate_depend_eliminate : 0.000025s : 0.00% optimize.opt_a.updatestate_assign_eliminate : 0.000019s : 0.00% optimize.opt_a.updatestate_loads_eliminate : 0.000023s : 0.00% optimize.opt_a.parameter_eliminate : 0.000005s : 0.00% optimize.opt_a.a_2 : 0.000490s : 0.02% optimize.opt_a.accelerated_algorithm : 0.000079s : 0.00% optimize.opt_a.shard : 0.000006s : 0.00% optimize.opt_a.meta_shard_fg_expand : 0.000014s : 0.00% optimize.opt_a.shard_inline : 0.000035s : 0.00% optimize.opt_a.merge_send_recv : 0.000035s : 0.00% optimize.opt_a.auto_parallel : 0.000034s : 0.00% optimize.opt_a.parallel : 0.000041s : 0.00% optimize.opt_a.flash_sp : 0.000099s : 0.00% optimize.opt_a.merge_comm : 0.000021s : 0.00% optimize.opt_a.allreduce_fusion : 0.000018s : 0.00% optimize.opt_a.matmul_add_comm_reduction : 0.000039s : 0.00% optimize.opt_a.allreduce_slice_to_reducescatter : 0.000002s : 0.00% optimize.opt_a.virtual_shard_identity : 0.000067s : 0.00% optimize.opt_a.virtual_dataset : 0.000041s : 0.00% optimize.opt_a.get_grad_eliminate_ : 0.000054s : 0.00% optimize.opt_a.virtual_output : 0.000039s : 0.00% optimize.opt_a.merge_forward : 0.000021s : 0.00% optimize.opt_a.cell_reuse_recompute_pass : 0.000006s : 0.00% optimize.opt_a.offload_activation : 0.000038s : 0.00% optimize.opt_a.cell_reuse_handle_not_recompute_node_pass : 0.000070s : 0.00% optimize.opt_a.merge_recompute_call_nodes : 0.000004s : 0.00% optimize.opt_a.before_grad : 0.000060s : 0.00% optimize.opt_a.set_forward_comm_id_for_comm_node_pass : 0.000020s : 0.00% optimize.opt_a.meta_fg_expand : 0.000015s : 0.00% optimize.opt_a.flash_sp_send_recv_attached : 0.000009s : 0.00% optimize.opt_a.receive_attached : 0.000005s : 0.00% optimize.opt_a.after_resolve : 0.000055s : 0.00% optimize.opt_a.a_after_grad : 0.000060s : 0.00% optimize.opt_a.renormalize : 0.159947s : 5.06% optimize.opt_a.add_forward_monad_depend : 0.000025s : 0.00% optimize.opt_a.auto_monad_grad : 0.000006s : 0.00% optimize.opt_a.auto_monad_eliminator : 0.001471s : 0.05% optimize.opt_a.cse : 0.001553s : 0.05% optimize.opt_a.a_3 : 0.000268s : 0.01% optimize.py_interpret_to_execute_after_opt_a : 0.000031s : 0.00% optimize.slice_cell_reuse_recomputed_activation : 0.000002s : 0.00% optimize.rewriter_after_opt_a : 0.000209s : 0.01% optimize.convert_after_rewriter : 0.000018s : 0.00% optimize.order_py_execute_after_rewriter : 0.000010s : 0.00% optimize.mutable_eliminate : 0.000807s : 0.03% optimize.opt_b.b_1 : 0.000369s : 0.01% optimize.opt_b.b_2 : 0.000019s : 0.00% optimize.opt_b.updatestate_depend_eliminate : 0.000017s : 0.00% optimize.opt_b.updatestate_assign_eliminate : 0.000008s : 0.00% optimize.opt_b.updatestate_loads_eliminate : 0.000011s : 0.00% optimize.opt_b.renormalize : 0.000001s : 0.00% optimize.opt_b.cse : 0.000073s : 0.00% optimize.optimize_parallel_all_gather_comm : 0.000035s : 0.00% optimize.overlap_param_gather : 0.000002s : 0.00% optimize.cconv : 0.000042s : 0.00% optimize.loop_unroll : 0.000563s : 0.02% optimize.opt_after_cconv.c_1 : 0.000102s : 0.00% optimize.opt_after_cconv.parameter_eliminate : 0.000006s : 0.00% optimize.opt_after_cconv.updatestate_depend_eliminate : 0.000014s : 0.00% optimize.opt_after_cconv.updatestate_assign_eliminate : 0.000007s : 0.00% optimize.opt_after_cconv.updatestate_loads_eliminate : 0.000010s : 0.00% optimize.opt_after_cconv.cse : 0.000051s : 0.00% optimize.opt_after_cconv.renormalize : 0.000001s : 0.00% optimize.remove_dup_value : 0.000074s : 0.00% optimize.tuple_transform.d_1 : 0.000114s : 0.00% optimize.tuple_transform.none_parameter_eliminate : 0.000003s : 0.00% optimize.tuple_transform.renormalize : 0.000000s : 0.00% optimize.tuple_transform.switch_simplify : 0.000018s : 0.00% optimize.partial_unused_args_eliminate : 0.000002s : 0.00% optimize.add_recomputation : 0.000115s : 0.00% optimize.cse_after_recomputation.cse : 0.000044s : 0.00% optimize.environ_conv : 0.000021s : 0.00% optimize.swap_dp_allreduce_reducescatter : 0.000012s : 0.00% optimize.bias_add_comm_swap : 0.000003s : 0.00% optimize.label_micro_interleaved_index : 0.000005s : 0.00% optimize.label_fine_grained_interleaved_index : 0.000003s : 0.00% optimize.merge_cast_opt : 0.000001s : 0.00% optimize.slice_recompute_activation : 0.000002s : 0.00% optimize.micro_interleaved_order_control : 0.000002s : 0.00% optimize.assign_add_opt : 0.000002s : 0.00% optimize.ForceFp32Comm : 0.000001s : 0.00% optimize.remove_cast_before_assign_add : 0.000001s : 0.00% optimize.full_micro_interleaved_order_control : 0.000002s : 0.00% optimize.reorder_send_recv_between_fp_bp : 0.000003s : 0.00% optimize.comm_op_add_attrs : 0.000001s : 0.00% optimize.add_comm_op_reuse_tag : 0.000001s : 0.00% optimize.interleave_split_concat_branches : 0.000001s : 0.00% optimize.interleave_parallel_branches : 0.000001s : 0.00% optimize.overlap_opt_shard_in_pipeline : 0.000001s : 0.00% optimize.overlap_opt_shard_grad_in_pipeline : 0.000002s : 0.00% optimize.control_data_broadcast_order : 0.000027s : 0.00% optimize.grouped_pairwise_exchange_alltoall : 0.000002s : 0.00% optimize.offloading_packed_experts : 0.000007s : 0.00% optimize.overlap_recompute_and_grad_model_parallel : 0.000008s : 0.00% optimize.overlap_grad_matmul_and_grad_allreduce : 0.000001s : 0.00% optimize.overlap_recompute_allgather_and_fa_grad : 0.000001s : 0.00% optimize.overlap_recompute_comm : 0.000002s : 0.00% optimize.overlap_grad_ring_attention : 0.000009s : 0.00% optimize.overlap_grad_flash_sp : 0.000043s : 0.00% optimize.begin_end_overlap_inline : 0.000001s : 0.00% optimize.split_matmul_comm_elemetwise : 0.000002s : 0.00% optimize.split_layernorm_comm : 0.000002s : 0.00% optimize.handle_group_info : 0.000001s : 0.00% optimize.symbol_engine_optimizer.build : 0.000019s : 0.00% optimize.symbol_engine_optimizer.elim_shapecalc : 0.000023s : 0.00% optimize.symbol_engine_optimizer.elim_not_effective : 0.000030s : 0.00% optimize.symbol_engine_optimizer.opt_reshape : 0.000023s : 0.00% optimize.symbol_engine_optimizer.fold_const_symbol : 0.000025s : 0.00% optimize.symbol_engine_optimizer.renormalize : 0.000000s : 0.00% detach_backward : 0.000002s : 0.00% pipeline_parallel_scheduler : 0.000001s : 0.00% auto_monad_reorder : 0.000048s : 0.00% get_jit_bprop_graph : 0.000002s : 0.00% rewriter_after_jit_bprop_graph : 0.000007s : 0.00% opt_after_jit_grad : 0.000682s : 0.02% validate : 0.000106s : 0.00% Time group info: ------[substitution.] 0.000320 96 3.56% : 0.000011s : 2: substitution.depend_value_elim 1.28% : 0.000004s : 8: substitution.elim_not_effective 0.99% : 0.000003s : 8: substitution.fold_const_symbol 3.73% : 0.000012s : 13: substitution.graph_param_transform 44.03% : 0.000141s : 1: substitution.inline 2.92% : 0.000009s : 16: substitution.j_node_and_user_rematch 9.67% : 0.000031s : 2: substitution.less_batch_normalization 2.16% : 0.000007s : 6: substitution.load_eliminater 0.95% : 0.000003s : 2: substitution.opt_reshape 3.97% : 0.000013s : 16: substitution.remove_not_recompute_node 2.92% : 0.000009s : 4: substitution.replace_old_param 9.71% : 0.000031s : 4: substitution.reshape_eliminate 7.20% : 0.000023s : 6: substitution.updatestate_pure_node_eliminater 6.90% : 0.000022s : 8: substitution.updatestate_useless_node_eliminater ------[type_inference.] 2.955396 2 99.83% : 2.950401s : 1: type_inference.infer 0.17% : 0.004995s : 1: type_inference.specialize ------[replace.] 0.000021 1 100.00% : 0.000021s : 1: replace.inline ------[match.] 0.000140 1 100.00% : 0.000140s : 1: match.inline ------[predicate.] 0.000451 3002 0.79% : 0.000004s : 27: predicate.accumulaten_eliminater 0.98% : 0.000004s : 13: predicate.ad_related_special_op_eliminate 0.71% : 0.000003s : 26: predicate.addn_check_dump 0.78% : 0.000004s : 27: predicate.addn_zero_filter 0.73% : 0.000003s : 27: predicate.adjust_all_reduce_mul_add 2.20% : 0.000010s : 53: predicate.arithmetic_simplify 0.87% : 0.000004s : 27: predicate.cast_eliminate 0.83% : 0.000004s : 26: predicate.check_bprop_eliminate 0.81% : 0.000004s : 26: predicate.compare_switch_simplify 0.25% : 0.000001s : 13: predicate.const_output_eliminate 0.83% : 0.000004s : 26: predicate.depend_value_elim 0.81% : 0.000004s : 27: predicate.dict_get_item_const_eliminator 0.96% : 0.000004s : 27: predicate.dict_get_item_eliminator 0.80% : 0.000004s : 27: predicate.dict_set_item_eliminator 1.03% : 0.000005s : 26: predicate.dumpgradient_eliminate 0.30% : 0.000001s : 13: predicate.elim_not_effective 0.47% : 0.000002s : 13: predicate.elim_shapecalc_of_broadcastargs 1.29% : 0.000006s : 40: predicate.environ_add_const_eliminate 1.05% : 0.000005s : 40: predicate.environ_get_add_eliminate 1.35% : 0.000006s : 40: predicate.environ_get_depend_swap 1.89% : 0.000009s : 66: predicate.environ_get_eliminate 1.07% : 0.000005s : 40: predicate.environ_get_set_eliminate 0.79% : 0.000004s : 28: predicate.exchange_switch_depend_value 1.31% : 0.000006s : 28: predicate.float_depend_g_call 0.74% : 0.000003s : 26: predicate.float_environ_get_switch 1.12% : 0.000005s : 39: predicate.float_tuple_getitem_switch 0.23% : 0.000001s : 13: predicate.fold_const_symbol 0.89% : 0.000004s : 26: predicate.get_grad_eliminate 0.32% : 0.000001s : 13: predicate.graph_param_transform 0.75% : 0.000003s : 26: predicate.incorporate_call 0.69% : 0.000003s : 26: predicate.incorporate_call_switch 5.73% : 0.000026s : 133: predicate.inline 1.09% : 0.000005s : 26: predicate.inline_without_move 0.44% : 0.000002s : 26: predicate.j_node_and_user_rematch 1.33% : 0.000006s : 27: predicate.less_batch_normalization 1.62% : 0.000007s : 53: predicate.list_to_tuple_eliminator_ 2.34% : 0.000011s : 80: predicate.load_eliminater 0.81% : 0.000004s : 13: predicate.loop_unroll_after_grad 0.98% : 0.000004s : 31: predicate.loop_unroll_before_grad 1.71% : 0.000008s : 53: predicate.make_slice_get_slice_eliminator 0.77% : 0.000003s : 26: predicate.merge_addn 0.76% : 0.000003s : 26: predicate.micro_step_allgather_replace 0.76% : 0.000003s : 26: predicate.mini_step_allgather_replace 0.82% : 0.000004s : 27: predicate.minmaximum_grad 1.11% : 0.000005s : 13: predicate.mutable_eliminate 0.57% : 0.000003s : 13: predicate.opt_reshape 0.45% : 0.000002s : 13: predicate.parallel_virtual_node 1.06% : 0.000005s : 28: predicate.partial_defer_inline 1.24% : 0.000006s : 40: predicate.partial_eliminate 0.80% : 0.000004s : 27: predicate.print_const_string_wrapper 0.77% : 0.000003s : 26: predicate.reduce_all_const_elim 1.07% : 0.000005s : 27: predicate.reduce_eliminate 2.19% : 0.000010s : 80: predicate.redundant_stop_gradient_eliminater 0.56% : 0.000003s : 26: predicate.remove_not_recompute_node 1.57% : 0.000007s : 53: predicate.replace_applicator 0.49% : 0.000002s : 26: predicate.replace_old_param 0.45% : 0.000002s : 13: predicate.reset_defer_inline 1.05% : 0.000005s : 27: predicate.reshape_eliminate 0.80% : 0.000004s : 26: predicate.row_tensor_add_zeros_like 0.56% : 0.000003s : 13: predicate.row_tensor_eliminate 1.61% : 0.000007s : 26: predicate.same_eliminate 0.61% : 0.000003s : 26: predicate.set_cell_output_no_recompute 1.10% : 0.000005s : 26: predicate.shard_identity_eliminate 0.88% : 0.000004s : 26: predicate.special_op_eliminate 0.84% : 0.000004s : 26: predicate.specialize_transform 1.20% : 0.000005s : 26: predicate.split_environ_get_set_with_tuple_value 1.08% : 0.000005s : 26: predicate.stack_unstack_eliminate 0.46% : 0.000002s : 13: predicate.switch_call_monad_eliminater 0.90% : 0.000004s : 28: predicate.switch_defer_inline 1.65% : 0.000007s : 54: predicate.switch_layer_defer_inline 3.72% : 0.000017s : 98: predicate.switch_simplify 0.79% : 0.000004s : 27: predicate.tile_eliminate 0.79% : 0.000004s : 27: predicate.transpose_eliminate 1.61% : 0.000007s : 53: predicate.tuple_list_convert_item_index_to_positive 1.77% : 0.000008s : 53: predicate.tuple_list_get_item_const_eliminator 1.77% : 0.000008s : 53: predicate.tuple_list_get_item_depend_reorder 3.10% : 0.000014s : 79: predicate.tuple_list_get_item_eliminator 1.58% : 0.000007s : 53: predicate.tuple_list_get_set_item_eliminator 2.62% : 0.000012s : 79: predicate.tuple_list_set_item_eliminator 1.59% : 0.000007s : 53: predicate.tuple_to_list_eliminator_ 2.22% : 0.000010s : 80: predicate.updatestate_pure_node_eliminater 3.39% : 0.000015s : 106: predicate.updatestate_useless_node_eliminater 0.48% : 0.000002s : 13: predicate.value_based_eliminate 0.87% : 0.000004s : 26: predicate.virtual_dataset_eliminate 0.89% : 0.000004s : 26: predicate.virtual_output_eliminate 0.39% : 0.000002s : 13: predicate.virtual_view_grad_eliminate 0.56% : 0.000003s : 13: predicate.zero_like_fill_zero ------[func_graph_cloner_run.] 0.200209 58 98.20% : 0.196596s : 55: func_graph_cloner_run.FuncGraphClonerGraph 1.80% : 0.003613s : 3: func_graph_cloner_run.FuncGraphSpecializer ------[meta_graph.] 0.000000 0 ------[manager.] 0.000000 0 ------[pynative] 0.000000 0 ------[others.] 3.821077 192 0.00% : 0.000004s : 1: ForceFp32Comm 4.24% : 0.162009s : 1: add_attr 4.24% : 0.161986s : 1: add_attr_with_inline 0.00% : 0.000004s : 1: add_comm_op_reuse_tag 0.00% : 0.000121s : 1: add_recomputation 0.00% : 0.000004s : 1: assign_add_opt 0.01% : 0.000317s : 1: auto_monad 0.00% : 0.000053s : 1: auto_monad_reorder 0.00% : 0.000004s : 1: begin_end_overlap_inline 0.00% : 0.000006s : 1: bias_add_comm_swap 0.92% : 0.035318s : 1: bootstrap 0.00% : 0.000046s : 1: cconv 0.00% : 0.000005s : 1: comm_op_add_attrs 0.00% : 0.000031s : 1: control_data_broadcast_order 0.00% : 0.000023s : 1: convert_after_rewriter 0.00% : 0.000059s : 1: cse_after_recomputation 0.00% : 0.000005s : 1: dataset_repeat_opt 0.00% : 0.000006s : 1: detach_backward 0.00% : 0.000025s : 1: environ_conv 0.00% : 0.000029s : 1: event_method 0.00% : 0.000005s : 1: full_micro_interleaved_order_control 0.00% : 0.000006s : 1: get_jit_bprop_graph 0.00% : 0.000012s : 1: graph_reusing 0.00% : 0.000005s : 1: grouped_pairwise_exchange_alltoall 0.00% : 0.000004s : 1: handle_group_info 0.00% : 0.000006s : 1: inline 0.00% : 0.000008s : 1: insert-virtual-dataset 0.00% : 0.000006s : 1: interleave_parallel_branches 0.00% : 0.000004s : 1: interleave_split_concat_branches 0.00% : 0.000007s : 1: label_fine_grained_interleaved_index 0.00% : 0.000083s : 1: label_micro_interleaved_index 0.02% : 0.000574s : 1: loop_unroll 0.00% : 0.000004s : 1: merge_cast_opt 0.00% : 0.000005s : 1: micro_interleaved_order_control 0.02% : 0.000821s : 1: mutable_eliminate 0.00% : 0.000010s : 1: offloading_packed_experts 0.00% : 0.000029s : 1: opt.transform.loop_unroll_optimizer 0.00% : 0.000037s : 1: opt.transform.mutable_eliminate 0.07% : 0.002638s : 78: opt.transform.opt_a 0.00% : 0.000100s : 1: opt.transform.opt_after_cconv 0.00% : 0.000064s : 1: opt.transform.opt_after_jit_grad 0.01% : 0.000352s : 28: opt.transform.opt_b 0.00% : 0.000129s : 2: opt.transform.opt_trans_graph 0.00% : 0.000096s : 4: opt.transform.symbol_engine_opt 4.37% : 0.167077s : 1: opt_a 0.01% : 0.000243s : 1: opt_after_cconv 0.02% : 0.000696s : 1: opt_after_jit_grad 0.01% : 0.000559s : 1: opt_b 4.47% : 0.170851s : 1: optimize 0.00% : 0.000039s : 1: optimize_parallel_all_gather_comm 0.00% : 0.000014s : 1: order_py_execute_after_rewriter 0.00% : 0.000046s : 1: overlap_grad_flash_sp 0.00% : 0.000004s : 1: overlap_grad_matmul_and_grad_allreduce 0.00% : 0.000013s : 1: overlap_grad_ring_attention 0.00% : 0.000004s : 1: overlap_opt_shard_grad_in_pipeline 0.00% : 0.000004s : 1: overlap_opt_shard_in_pipeline 0.00% : 0.000006s : 1: overlap_param_gather 0.00% : 0.000005s : 1: overlap_recompute_allgather_and_fa_grad 0.00% : 0.000011s : 1: overlap_recompute_and_grad_model_parallel 0.00% : 0.000005s : 1: overlap_recompute_comm 0.00% : 0.000010s : 1: parallel-infer-symbol 0.00% : 0.000004s : 1: parallel-infer-symbol-second 0.00% : 0.000006s : 1: partial_unused_args_eliminate 0.00% : 0.000005s : 1: pipeline_parallel_scheduler 0.00% : 0.000006s : 1: pipeline_split 0.00% : 0.000054s : 1: pre_auto_parallel 0.00% : 0.000049s : 1: py_interpret_to_execute 0.00% : 0.000036s : 1: py_interpret_to_execute_after_opt_a 0.00% : 0.000004s : 1: remove_cast_before_assign_add 0.00% : 0.000079s : 1: remove_dup_value 3.30% : 0.126269s : 1: renormalize.infer 0.88% : 0.033655s : 1: renormalize.specialize 0.00% : 0.000006s : 1: reorder_send_recv_between_fp_bp 0.00% : 0.000011s : 1: rewriter_after_jit_bprop_graph 0.01% : 0.000220s : 1: rewriter_after_opt_a 0.00% : 0.000108s : 1: rewriter_before_opt_a 0.00% : 0.000006s : 1: slice_cell_reuse_recomputed_activation 0.00% : 0.000005s : 1: slice_recompute_activation 0.00% : 0.000005s : 1: split_layernorm_comm 0.00% : 0.000005s : 1: split_matmul_comm_elemetwise 0.00% : 0.000015s : 1: swap_dp_allreduce_reducescatter 0.00% : 0.000164s : 1: symbol_engine_optimizer 0.00% : 0.000168s : 1: tuple_transform 77.35% : 2.955547s : 1: type_inference . [hook] pytest_runtest_teardown:test_matmul_qkv_bf16_0[False-mstype0-k_n_shape0-16] tests/st/infer/ops/test_internal_ops/test_matmul_split.py::test_matmul_qkv_bf16_0[False-mstype0-k_n_shape0-16],max_mem:120.0M [WARNING] ME(152078:281472851701552,MainProcess):2026-01-29-17:38:24.733.267 [mindspore/context.py:1334] For 'context.set_context', the parameter 'device_target' will be deprecated and removed in a future version. Please use the api mindspore.set_device() instead. TotalTime = 2.93287, [21] [bootstrap]: 0.00085986 [type_inference]: 2.69872 [event_method]: 1.802e-05 [auto_monad]: 0.00036572 [graph_reusing]: 6.42001e-06 [inline]: 3.54002e-06 [add_attr]: 0.0565258, [1] [add_attr_with_inline]: 0.0565069, [1] [Cycle 1]: 7.31e-05, [2] [tag_attr]: 2.564e-05 [meta_addattr_fg_expand]: 4.60999e-06 [parallel-infer-symbol]: 5.53002e-06 [pre_auto_parallel]: 4.11e-05 [insert-virtual-dataset]: 2.61e-06 [parallel-infer-symbol-second]: 7.60017e-07 [dataset_repeat_opt]: 1.62001e-06 [pipeline_split]: 1.54e-06 [optimize]: 0.174962, [53] [py_interpret_to_execute]: 3.513e-05 [rewriter_before_opt_a]: 8.975e-05 [opt_a]: 0.132119, [2] [Cycle 1]: 0.102129, [45] [expand_dump_flag]: 3.11001e-06 [switch_simplify]: 3.769e-05 [loop_unroll]: 1.98e-05 [a_1]: 0.00085808 [with_stream_mark]: 3.385e-05 [recompute_prepare]: 3.267e-05 [updatestate_depend_eliminate]: 1.181e-05 [updatestate_assign_eliminate]: 8.71002e-06 [updatestate_loads_eliminate]: 1.145e-05 [parameter_eliminate]: 2.48e-06 [a_2]: 0.00025737 [accelerated_algorithm]: 5.019e-05 [shard]: 2.28002e-06 [meta_shard_fg_expand]: 4.47998e-06 [shard_inline]: 1.897e-05 [merge_send_recv]: 1.605e-05 [auto_parallel]: 1.495e-05 [parallel]: 3.343e-05 [flash_sp]: 1.495e-05 [merge_comm]: 9.97999e-06 [allreduce_fusion]: 9.25001e-06 [matmul_add_comm_reduction]: 1.807e-05 [allreduce_slice_to_reducescatter]: 8.99978e-07 [virtual_shard_identity]: 2.44e-05 [virtual_dataset]: 1.829e-05 [get_grad_eliminate_]: 1.94e-05 [virtual_output]: 1.883e-05 [merge_forward]: 8.85001e-06 [cell_reuse_recompute_pass]: 2.04e-06 [offload_activation]: 1.963e-05 [cell_reuse_handle_not_recompute_node_pass]: 3.939e-05 [merge_recompute_call_nodes]: 1.42e-06 [before_grad]: 5.185e-05 [set_forward_comm_id_for_comm_node_pass]: 1.772e-05 [meta_fg_expand]: 6.71e-06 [flash_sp_send_recv_attached]: 6.73998e-06 [receive_attached]: 2.94001e-06 [after_resolve]: 5.914e-05 [a_after_grad]: 8.953e-05 [renormalize]: 0.0992175 [add_forward_monad_depend]: 2.222e-05 [auto_monad_grad]: 2.79999e-06 [auto_monad_eliminator]: 6.683e-05 [cse]: 0.00044242 [a_3]: 0.00014278 [Cycle 2]: 0.029971, [45] [expand_dump_flag]: 3.63e-06 [switch_simplify]: 2.14e-05 [loop_unroll]: 1.8e-05 [a_1]: 0.00046826 [with_stream_mark]: 3.296e-05 [recompute_prepare]: 2.215e-05 [updatestate_depend_eliminate]: 1.2e-05 [updatestate_assign_eliminate]: 8.52e-06 [updatestate_loads_eliminate]: 1.105e-05 [parameter_eliminate]: 2.78e-06 [a_2]: 0.00024927 [accelerated_algorithm]: 2.407e-05 [shard]: 2.68998e-06 [meta_shard_fg_expand]: 5.42001e-06 [shard_inline]: 1.725e-05 [merge_send_recv]: 1.655e-05 [auto_parallel]: 1.593e-05 [parallel]: 1.071e-05 [flash_sp]: 4.4e-06 [merge_comm]: 8.65001e-06 [allreduce_fusion]: 9.27999e-06 [matmul_add_comm_reduction]: 1.79e-05 [allreduce_slice_to_reducescatter]: 1.03001e-06 [virtual_shard_identity]: 1.995e-05 [virtual_dataset]: 1.555e-05 [get_grad_eliminate_]: 1.478e-05 [virtual_output]: 1.5e-05 [merge_forward]: 9.54999e-06 [cell_reuse_recompute_pass]: 3.19001e-06 [offload_activation]: 1.923e-05 [cell_reuse_handle_not_recompute_node_pass]: 3.607e-05 [merge_recompute_call_nodes]: 1.54998e-06 [before_grad]: 2.555e-05 [set_forward_comm_id_for_comm_node_pass]: 1.183e-05 [meta_fg_expand]: 6.84999e-06 [flash_sp_send_recv_attached]: 1.72999e-06 [receive_attached]: 2.46e-06 [after_resolve]: 2.454e-05 [a_after_grad]: 2.957e-05 [renormalize]: 8.00064e-08 [add_forward_monad_depend]: 4.18001e-06 [auto_monad_grad]: 3.03e-06 [auto_monad_eliminator]: 6.233e-05 [cse]: 0.00010635 [a_3]: 0.0282504 [py_interpret_to_execute_after_opt_a]: 5.333e-05 [slice_cell_reuse_recomputed_activation]: 3.78001e-06 [rewriter_after_opt_a]: 0.00024747 [convert_after_rewriter]: 1.617e-05 [order_py_execute_after_rewriter]: 1.088e-05 [mutable_eliminate]: 0.0196739 [opt_b]: 0.00059793, [1] [Cycle 1]: 0.00058785, [7] [b_1]: 0.00038995 [b_2]: 1.969e-05 [updatestate_depend_eliminate]: 2.062e-05 [updatestate_assign_eliminate]: 8.21002e-06 [updatestate_loads_eliminate]: 1.158e-05 [renormalize]: 9.10019e-07 [cse]: 9.185e-05 [optimize_parallel_all_gather_comm]: 3.595e-05 [overlap_param_gather]: 2.41e-06 [cconv]: 4.107e-05 [loop_unroll]: 0.00069073 [opt_after_cconv]: 0.00024308, [1] [Cycle 1]: 0.00023443, [7] [c_1]: 0.00010465 [parameter_eliminate]: 6.47001e-06 [updatestate_depend_eliminate]: 1.451e-05 [updatestate_assign_eliminate]: 6.93e-06 [updatestate_loads_eliminate]: 9.49e-06 [cse]: 5.416e-05 [renormalize]: 8.90024e-07 [remove_dup_value]: 7.461e-05 [tuple_transform]: 0.00016235, [1] [Cycle 1]: 0.00015659, [4] [d_1]: 0.00011507 [none_parameter_eliminate]: 1.80001e-06 [renormalize]: 2.19996e-07 [switch_simplify]: 1.759e-05 [partial_unused_args_eliminate]: 1.97999e-06 [add_recomputation]: 0.00012779 [cse_after_recomputation]: 5.673e-05, [1] [Cycle 1]: 5.202e-05, [1] [cse]: 4.525e-05 [environ_conv]: 1.992e-05 [swap_dp_allreduce_reducescatter]: 1.33e-05 [bias_add_comm_swap]: 3.9e-06 [label_micro_interleaved_index]: 5.81003e-06 [label_fine_grained_interleaved_index]: 3.21999e-06 [merge_cast_opt]: 1.38002e-06 [slice_recompute_activation]: 2.87002e-06 [micro_interleaved_order_control]: 2.29999e-06 [assign_add_opt]: 1.19e-06 [ForceFp32Comm]: 8.60018e-07 [remove_cast_before_assign_add]: 1.00001e-06 [full_micro_interleaved_order_control]: 2.64001e-06 [reorder_send_recv_between_fp_bp]: 2.88e-06 [comm_op_add_attrs]: 1.22e-06 [add_comm_op_reuse_tag]: 9.70002e-07 [interleave_split_concat_branches]: 1.10001e-06 [interleave_parallel_branches]: 0.0198174 [overlap_opt_shard_in_pipeline]: 3.13998e-06 [overlap_opt_shard_grad_in_pipeline]: 3.5e-06 [control_data_broadcast_order]: 5.802e-05 [grouped_pairwise_exchange_alltoall]: 1.58002e-06 [offloading_packed_experts]: 8.16002e-06 [overlap_recompute_and_grad_model_parallel]: 8.60999e-06 [overlap_grad_matmul_and_grad_allreduce]: 1.28002e-06 [overlap_recompute_allgather_and_fa_grad]: 1.27e-06 [overlap_recompute_comm]: 2.32001e-06 [overlap_grad_ring_attention]: 9.29998e-06 [overlap_grad_flash_sp]: 4.688e-05 [begin_end_overlap_inline]: 6.99976e-07 [split_matmul_comm_elemetwise]: 2.51e-06 [split_layernorm_comm]: 3.11999e-06 [handle_group_info]: 1.12e-06 [symbol_engine_optimizer]: 0.00022235, [1] [Cycle 1]: 0.0002101, [6] [build]: 2.394e-05 [elim_shapecalc]: 4.482e-05 [elim_not_effective]: 4.186e-05 [opt_reshape]: 2.681e-05 [fold_const_symbol]: 2.605e-05 [renormalize]: 4.50003e-07 [detach_backward]: 2.51998e-06 [pipeline_parallel_scheduler]: 1.53002e-06 [auto_monad_reorder]: 6.009e-05 [get_jit_bprop_graph]: 1.86e-06 [rewriter_after_jit_bprop_graph]: 2.628e-05 [opt_after_jit_grad]: 0.00088065 [validate]: 0.00010406 Sums bootstrap : 0.000860s : 0.03% type_inference : 2.698716s : 93.88% event_method : 0.000018s : 0.00% auto_monad : 0.000366s : 0.01% graph_reusing : 0.000006s : 0.00% inline : 0.000004s : 0.00% add_attr.add_attr_with_inline.tag_attr : 0.000026s : 0.00% add_attr.add_attr_with_inline.meta_addattr_fg_expand : 0.000005s : 0.00% parallel-infer-symbol : 0.000006s : 0.00% pre_auto_parallel : 0.000041s : 0.00% insert-virtual-dataset : 0.000003s : 0.00% parallel-infer-symbol-second : 0.000001s : 0.00% dataset_repeat_opt : 0.000002s : 0.00% pipeline_split : 0.000002s : 0.00% optimize.py_interpret_to_execute : 0.000035s : 0.00% optimize.rewriter_before_opt_a : 0.000090s : 0.00% optimize.opt_a.expand_dump_flag : 0.000007s : 0.00% optimize.opt_a.switch_simplify : 0.000059s : 0.00% optimize.opt_a.loop_unroll : 0.000038s : 0.00% optimize.opt_a.a_1 : 0.001326s : 0.05% optimize.opt_a.with_stream_mark : 0.000067s : 0.00% optimize.opt_a.recompute_prepare : 0.000055s : 0.00% optimize.opt_a.updatestate_depend_eliminate : 0.000024s : 0.00% optimize.opt_a.updatestate_assign_eliminate : 0.000017s : 0.00% optimize.opt_a.updatestate_loads_eliminate : 0.000022s : 0.00% optimize.opt_a.parameter_eliminate : 0.000005s : 0.00% optimize.opt_a.a_2 : 0.000507s : 0.02% optimize.opt_a.accelerated_algorithm : 0.000074s : 0.00% optimize.opt_a.shard : 0.000005s : 0.00% optimize.opt_a.meta_shard_fg_expand : 0.000010s : 0.00% optimize.opt_a.shard_inline : 0.000036s : 0.00% optimize.opt_a.merge_send_recv : 0.000033s : 0.00% optimize.opt_a.auto_parallel : 0.000031s : 0.00% optimize.opt_a.parallel : 0.000044s : 0.00% optimize.opt_a.flash_sp : 0.000019s : 0.00% optimize.opt_a.merge_comm : 0.000019s : 0.00% optimize.opt_a.allreduce_fusion : 0.000019s : 0.00% optimize.opt_a.matmul_add_comm_reduction : 0.000036s : 0.00% optimize.opt_a.allreduce_slice_to_reducescatter : 0.000002s : 0.00% optimize.opt_a.virtual_shard_identity : 0.000044s : 0.00% optimize.opt_a.virtual_dataset : 0.000034s : 0.00% optimize.opt_a.get_grad_eliminate_ : 0.000034s : 0.00% optimize.opt_a.virtual_output : 0.000034s : 0.00% optimize.opt_a.merge_forward : 0.000018s : 0.00% optimize.opt_a.cell_reuse_recompute_pass : 0.000005s : 0.00% optimize.opt_a.offload_activation : 0.000039s : 0.00% optimize.opt_a.cell_reuse_handle_not_recompute_node_pass : 0.000075s : 0.00% optimize.opt_a.merge_recompute_call_nodes : 0.000003s : 0.00% optimize.opt_a.before_grad : 0.000077s : 0.00% optimize.opt_a.set_forward_comm_id_for_comm_node_pass : 0.000030s : 0.00% optimize.opt_a.meta_fg_expand : 0.000014s : 0.00% optimize.opt_a.flash_sp_send_recv_attached : 0.000008s : 0.00% optimize.opt_a.receive_attached : 0.000005s : 0.00% optimize.opt_a.after_resolve : 0.000084s : 0.00% optimize.opt_a.a_after_grad : 0.000119s : 0.00% optimize.opt_a.renormalize : 0.099218s : 3.45% optimize.opt_a.add_forward_monad_depend : 0.000026s : 0.00% optimize.opt_a.auto_monad_grad : 0.000006s : 0.00% optimize.opt_a.auto_monad_eliminator : 0.000129s : 0.00% optimize.opt_a.cse : 0.000549s : 0.02% optimize.opt_a.a_3 : 0.028393s : 0.99% optimize.py_interpret_to_execute_after_opt_a : 0.000053s : 0.00% optimize.slice_cell_reuse_recomputed_activation : 0.000004s : 0.00% optimize.rewriter_after_opt_a : 0.000247s : 0.01% optimize.convert_after_rewriter : 0.000016s : 0.00% optimize.order_py_execute_after_rewriter : 0.000011s : 0.00% optimize.mutable_eliminate : 0.019674s : 0.68% optimize.opt_b.b_1 : 0.000390s : 0.01% optimize.opt_b.b_2 : 0.000020s : 0.00% optimize.opt_b.updatestate_depend_eliminate : 0.000021s : 0.00% optimize.opt_b.updatestate_assign_eliminate : 0.000008s : 0.00% optimize.opt_b.updatestate_loads_eliminate : 0.000012s : 0.00% optimize.opt_b.renormalize : 0.000001s : 0.00% optimize.opt_b.cse : 0.000092s : 0.00% optimize.optimize_parallel_all_gather_comm : 0.000036s : 0.00% optimize.overlap_param_gather : 0.000002s : 0.00% optimize.cconv : 0.000041s : 0.00% optimize.loop_unroll : 0.000691s : 0.02% optimize.opt_after_cconv.c_1 : 0.000105s : 0.00% optimize.opt_after_cconv.parameter_eliminate : 0.000006s : 0.00% optimize.opt_after_cconv.updatestate_depend_eliminate : 0.000015s : 0.00% optimize.opt_after_cconv.updatestate_assign_eliminate : 0.000007s : 0.00% optimize.opt_after_cconv.updatestate_loads_eliminate : 0.000009s : 0.00% optimize.opt_after_cconv.cse : 0.000054s : 0.00% optimize.opt_after_cconv.renormalize : 0.000001s : 0.00% optimize.remove_dup_value : 0.000075s : 0.00% optimize.tuple_transform.d_1 : 0.000115s : 0.00% optimize.tuple_transform.none_parameter_eliminate : 0.000002s : 0.00% optimize.tuple_transform.renormalize : 0.000000s : 0.00% optimize.tuple_transform.switch_simplify : 0.000018s : 0.00% optimize.partial_unused_args_eliminate : 0.000002s : 0.00% optimize.add_recomputation : 0.000128s : 0.00% optimize.cse_after_recomputation.cse : 0.000045s : 0.00% optimize.environ_conv : 0.000020s : 0.00% optimize.swap_dp_allreduce_reducescatter : 0.000013s : 0.00% optimize.bias_add_comm_swap : 0.000004s : 0.00% optimize.label_micro_interleaved_index : 0.000006s : 0.00% optimize.label_fine_grained_interleaved_index : 0.000003s : 0.00% optimize.merge_cast_opt : 0.000001s : 0.00% optimize.slice_recompute_activation : 0.000003s : 0.00% optimize.micro_interleaved_order_control : 0.000002s : 0.00% optimize.assign_add_opt : 0.000001s : 0.00% optimize.ForceFp32Comm : 0.000001s : 0.00% optimize.remove_cast_before_assign_add : 0.000001s : 0.00% optimize.full_micro_interleaved_order_control : 0.000003s : 0.00% optimize.reorder_send_recv_between_fp_bp : 0.000003s : 0.00% optimize.comm_op_add_attrs : 0.000001s : 0.00% optimize.add_comm_op_reuse_tag : 0.000001s : 0.00% optimize.interleave_split_concat_branches : 0.000001s : 0.00% optimize.interleave_parallel_branches : 0.019817s : 0.69% optimize.overlap_opt_shard_in_pipeline : 0.000003s : 0.00% optimize.overlap_opt_shard_grad_in_pipeline : 0.000003s : 0.00% optimize.control_data_broadcast_order : 0.000058s : 0.00% optimize.grouped_pairwise_exchange_alltoall : 0.000002s : 0.00% optimize.offloading_packed_experts : 0.000008s : 0.00% optimize.overlap_recompute_and_grad_model_parallel : 0.000009s : 0.00% optimize.overlap_grad_matmul_and_grad_allreduce : 0.000001s : 0.00% optimize.overlap_recompute_allgather_and_fa_grad : 0.000001s : 0.00% optimize.overlap_recompute_comm : 0.000002s : 0.00% optimize.overlap_grad_ring_attention : 0.000009s : 0.00% optimize.overlap_grad_flash_sp : 0.000047s : 0.00% optimize.begin_end_overlap_inline : 0.000001s : 0.00% optimize.split_matmul_comm_elemetwise : 0.000003s : 0.00% optimize.split_layernorm_comm : 0.000003s : 0.00% optimize.handle_group_info : 0.000001s : 0.00% optimize.symbol_engine_optimizer.build : 0.000024s : 0.00% optimize.symbol_engine_optimizer.elim_shapecalc : 0.000045s : 0.00% optimize.symbol_engine_optimizer.elim_not_effective : 0.000042s : 0.00% optimize.symbol_engine_optimizer.opt_reshape : 0.000027s : 0.00% optimize.symbol_engine_optimizer.fold_const_symbol : 0.000026s : 0.00% optimize.symbol_engine_optimizer.renormalize : 0.000000s : 0.00% detach_backward : 0.000003s : 0.00% pipeline_parallel_scheduler : 0.000002s : 0.00% auto_monad_reorder : 0.000060s : 0.00% get_jit_bprop_graph : 0.000002s : 0.00% rewriter_after_jit_bprop_graph : 0.000026s : 0.00% opt_after_jit_grad : 0.000881s : 0.03% validate : 0.000104s : 0.00% Time group info: ------[substitution.] 0.000309 96 3.34% : 0.000010s : 2: substitution.depend_value_elim 1.50% : 0.000005s : 8: substitution.elim_not_effective 1.02% : 0.000003s : 8: substitution.fold_const_symbol 3.96% : 0.000012s : 13: substitution.graph_param_transform 44.19% : 0.000137s : 1: substitution.inline 2.77% : 0.000009s : 16: substitution.j_node_and_user_rematch 8.65% : 0.000027s : 2: substitution.less_batch_normalization 2.39% : 0.000007s : 6: substitution.load_eliminater 1.01% : 0.000003s : 2: substitution.opt_reshape 3.95% : 0.000012s : 16: substitution.remove_not_recompute_node 2.41% : 0.000007s : 4: substitution.replace_old_param 10.84% : 0.000034s : 4: substitution.reshape_eliminate 7.66% : 0.000024s : 6: substitution.updatestate_pure_node_eliminater 6.32% : 0.000020s : 8: substitution.updatestate_useless_node_eliminater ------[type_inference.] 2.698606 2 99.59% : 2.687445s : 1: type_inference.infer 0.41% : 0.011161s : 1: type_inference.specialize ------[replace.] 0.000023 1 100.00% : 0.000023s : 1: replace.inline ------[match.] 0.000135 1 100.00% : 0.000135s : 1: match.inline ------[predicate.] 0.028560 3002 0.01% : 0.000004s : 27: predicate.accumulaten_eliminater 0.02% : 0.000005s : 13: predicate.ad_related_special_op_eliminate 0.01% : 0.000003s : 26: predicate.addn_check_dump 0.01% : 0.000004s : 27: predicate.addn_zero_filter 0.01% : 0.000003s : 27: predicate.adjust_all_reduce_mul_add 0.03% : 0.000010s : 53: predicate.arithmetic_simplify 0.01% : 0.000004s : 27: predicate.cast_eliminate 0.01% : 0.000004s : 26: predicate.check_bprop_eliminate 0.01% : 0.000003s : 26: predicate.compare_switch_simplify 0.00% : 0.000001s : 13: predicate.const_output_eliminate 0.01% : 0.000004s : 26: predicate.depend_value_elim 0.01% : 0.000004s : 27: predicate.dict_get_item_const_eliminator 0.01% : 0.000004s : 27: predicate.dict_get_item_eliminator 0.01% : 0.000004s : 27: predicate.dict_set_item_eliminator 0.02% : 0.000005s : 26: predicate.dumpgradient_eliminate 0.01% : 0.000002s : 13: predicate.elim_not_effective 0.01% : 0.000004s : 13: predicate.elim_shapecalc_of_broadcastargs 0.02% : 0.000005s : 40: predicate.environ_add_const_eliminate 0.02% : 0.000005s : 40: predicate.environ_get_add_eliminate 0.02% : 0.000005s : 40: predicate.environ_get_depend_swap 0.03% : 0.000009s : 66: predicate.environ_get_eliminate 0.02% : 0.000005s : 40: predicate.environ_get_set_eliminate 0.01% : 0.000004s : 28: predicate.exchange_switch_depend_value 0.02% : 0.000006s : 28: predicate.float_depend_g_call 0.01% : 0.000003s : 26: predicate.float_environ_get_switch 0.02% : 0.000005s : 39: predicate.float_tuple_getitem_switch 0.00% : 0.000001s : 13: predicate.fold_const_symbol 0.01% : 0.000004s : 26: predicate.get_grad_eliminate 0.01% : 0.000002s : 13: predicate.graph_param_transform 0.01% : 0.000003s : 26: predicate.incorporate_call 0.01% : 0.000003s : 26: predicate.incorporate_call_switch 0.09% : 0.000025s : 133: predicate.inline 0.02% : 0.000005s : 26: predicate.inline_without_move 0.01% : 0.000002s : 26: predicate.j_node_and_user_rematch 0.02% : 0.000006s : 27: predicate.less_batch_normalization 0.03% : 0.000008s : 53: predicate.list_to_tuple_eliminator_ 0.04% : 0.000011s : 80: predicate.load_eliminater 0.02% : 0.000006s : 13: predicate.loop_unroll_after_grad 0.02% : 0.000005s : 31: predicate.loop_unroll_before_grad 0.03% : 0.000008s : 53: predicate.make_slice_get_slice_eliminator 0.01% : 0.000003s : 26: predicate.merge_addn 0.01% : 0.000004s : 26: predicate.micro_step_allgather_replace 0.01% : 0.000003s : 26: predicate.mini_step_allgather_replace 0.01% : 0.000004s : 27: predicate.minmaximum_grad 0.04% : 0.000010s : 13: predicate.mutable_eliminate 0.01% : 0.000003s : 13: predicate.opt_reshape 0.01% : 0.000002s : 13: predicate.parallel_virtual_node 0.02% : 0.000005s : 28: predicate.partial_defer_inline 0.02% : 0.000006s : 40: predicate.partial_eliminate 0.01% : 0.000004s : 27: predicate.print_const_string_wrapper 0.01% : 0.000004s : 26: predicate.reduce_all_const_elim 0.02% : 0.000005s : 27: predicate.reduce_eliminate 0.04% : 0.000010s : 80: predicate.redundant_stop_gradient_eliminater 0.01% : 0.000003s : 26: predicate.remove_not_recompute_node 0.02% : 0.000007s : 53: predicate.replace_applicator 0.01% : 0.000003s : 26: predicate.replace_old_param 0.01% : 0.000002s : 13: predicate.reset_defer_inline 0.02% : 0.000004s : 27: predicate.reshape_eliminate 0.01% : 0.000004s : 26: predicate.row_tensor_add_zeros_like 0.01% : 0.000002s : 13: predicate.row_tensor_eliminate 0.02% : 0.000006s : 26: predicate.same_eliminate 0.01% : 0.000003s : 26: predicate.set_cell_output_no_recompute 0.01% : 0.000004s : 26: predicate.shard_identity_eliminate 0.01% : 0.000004s : 26: predicate.special_op_eliminate 0.02% : 0.000004s : 26: predicate.specialize_transform 98.40% : 0.028102s : 26: predicate.split_environ_get_set_with_tuple_value 0.02% : 0.000006s : 26: predicate.stack_unstack_eliminate 0.01% : 0.000002s : 13: predicate.switch_call_monad_eliminater 0.01% : 0.000004s : 28: predicate.switch_defer_inline 0.03% : 0.000007s : 54: predicate.switch_layer_defer_inline 0.06% : 0.000017s : 98: predicate.switch_simplify 0.01% : 0.000004s : 27: predicate.tile_eliminate 0.01% : 0.000004s : 27: predicate.transpose_eliminate 0.03% : 0.000008s : 53: predicate.tuple_list_convert_item_index_to_positive 0.03% : 0.000008s : 53: predicate.tuple_list_get_item_const_eliminator 0.03% : 0.000008s : 53: predicate.tuple_list_get_item_depend_reorder 0.05% : 0.000014s : 79: predicate.tuple_list_get_item_eliminator 0.03% : 0.000008s : 53: predicate.tuple_list_get_set_item_eliminator 0.04% : 0.000011s : 79: predicate.tuple_list_set_item_eliminator 0.03% : 0.000008s : 53: predicate.tuple_to_list_eliminator_ 0.04% : 0.000010s : 80: predicate.updatestate_pure_node_eliminater 0.06% : 0.000016s : 106: predicate.updatestate_useless_node_eliminater 0.01% : 0.000002s : 13: predicate.value_based_eliminate 0.01% : 0.000004s : 26: predicate.virtual_dataset_eliminate 0.01% : 0.000004s : 26: predicate.virtual_output_eliminate 0.01% : 0.000002s : 13: predicate.virtual_view_grad_eliminate 0.01% : 0.000002s : 13: predicate.zero_like_fill_zero ------[func_graph_cloner_run.] 0.133953 58 97.96% : 0.131217s : 55: func_graph_cloner_run.FuncGraphClonerGraph 2.04% : 0.002736s : 3: func_graph_cloner_run.FuncGraphSpecializer ------[meta_graph.] 0.000000 0 ------[manager.] 0.000000 0 ------[pynative] 0.000000 0 ------[others.] 3.295029 192 0.00% : 0.000004s : 1: ForceFp32Comm 1.72% : 0.056534s : 1: add_attr 1.72% : 0.056512s : 1: add_attr_with_inline 0.00% : 0.000004s : 1: add_comm_op_reuse_tag 0.00% : 0.000133s : 1: add_recomputation 0.00% : 0.000004s : 1: assign_add_opt 0.01% : 0.000377s : 1: auto_monad 0.00% : 0.000067s : 1: auto_monad_reorder 0.00% : 0.000003s : 1: begin_end_overlap_inline 0.00% : 0.000007s : 1: bias_add_comm_swap 0.03% : 0.000962s : 1: bootstrap 0.00% : 0.000045s : 1: cconv 0.00% : 0.000005s : 1: comm_op_add_attrs 0.00% : 0.000063s : 1: control_data_broadcast_order 0.00% : 0.000021s : 1: convert_after_rewriter 0.00% : 0.000060s : 1: cse_after_recomputation 0.00% : 0.000005s : 1: dataset_repeat_opt 0.00% : 0.000006s : 1: detach_backward 0.00% : 0.000024s : 1: environ_conv 0.00% : 0.000026s : 1: event_method 0.00% : 0.000006s : 1: full_micro_interleaved_order_control 0.00% : 0.000007s : 1: get_jit_bprop_graph 0.00% : 0.000011s : 1: graph_reusing 0.00% : 0.000006s : 1: grouped_pairwise_exchange_alltoall 0.00% : 0.000004s : 1: handle_group_info 0.00% : 0.000007s : 1: inline 0.00% : 0.000007s : 1: insert-virtual-dataset 0.60% : 0.019858s : 1: interleave_parallel_branches 0.00% : 0.000004s : 1: interleave_split_concat_branches 0.00% : 0.000006s : 1: label_fine_grained_interleaved_index 0.00% : 0.000009s : 1: label_micro_interleaved_index 0.02% : 0.000705s : 1: loop_unroll 0.00% : 0.000004s : 1: merge_cast_opt 0.00% : 0.000005s : 1: micro_interleaved_order_control 0.60% : 0.019696s : 1: mutable_eliminate 0.00% : 0.000012s : 1: offloading_packed_experts 0.00% : 0.000034s : 1: opt.transform.loop_unroll_optimizer 0.00% : 0.000059s : 1: opt.transform.mutable_eliminate 0.94% : 0.030836s : 78: opt.transform.opt_a 0.00% : 0.000103s : 1: opt.transform.opt_after_cconv 0.00% : 0.000069s : 1: opt.transform.opt_after_jit_grad 0.01% : 0.000371s : 28: opt.transform.opt_b 0.00% : 0.000130s : 2: opt.transform.opt_trans_graph 0.00% : 0.000130s : 4: opt.transform.symbol_engine_opt 4.01% : 0.132125s : 1: opt_a 0.01% : 0.000247s : 1: opt_after_cconv 0.03% : 0.000895s : 1: opt_after_jit_grad 0.02% : 0.000603s : 1: opt_b 5.31% : 0.174969s : 1: optimize 0.00% : 0.000041s : 1: optimize_parallel_all_gather_comm 0.00% : 0.000014s : 1: order_py_execute_after_rewriter 0.00% : 0.000051s : 1: overlap_grad_flash_sp 0.00% : 0.000004s : 1: overlap_grad_matmul_and_grad_allreduce 0.00% : 0.000014s : 1: overlap_grad_ring_attention 0.00% : 0.000007s : 1: overlap_opt_shard_grad_in_pipeline 0.00% : 0.000011s : 1: overlap_opt_shard_in_pipeline 0.00% : 0.000008s : 1: overlap_param_gather 0.00% : 0.000006s : 1: overlap_recompute_allgather_and_fa_grad 0.00% : 0.000012s : 1: overlap_recompute_and_grad_model_parallel 0.00% : 0.000005s : 1: overlap_recompute_comm 0.00% : 0.000010s : 1: parallel-infer-symbol 0.00% : 0.000004s : 1: parallel-infer-symbol-second 0.00% : 0.000006s : 1: partial_unused_args_eliminate 0.00% : 0.000005s : 1: pipeline_parallel_scheduler 0.00% : 0.000006s : 1: pipeline_split 0.00% : 0.000046s : 1: pre_auto_parallel 0.00% : 0.000039s : 1: py_interpret_to_execute 0.00% : 0.000062s : 1: py_interpret_to_execute_after_opt_a 0.00% : 0.000004s : 1: remove_cast_before_assign_add 0.00% : 0.000080s : 1: remove_dup_value 2.71% : 0.089301s : 1: renormalize.infer 0.30% : 0.009894s : 1: renormalize.specialize 0.00% : 0.000006s : 1: reorder_send_recv_between_fp_bp 0.00% : 0.000031s : 1: rewriter_after_jit_bprop_graph 0.01% : 0.000254s : 1: rewriter_after_opt_a 0.00% : 0.000094s : 1: rewriter_before_opt_a 0.00% : 0.000008s : 1: slice_cell_reuse_recomputed_activation 0.00% : 0.000005s : 1: slice_recompute_activation 0.00% : 0.000010s : 1: split_layernorm_comm 0.00% : 0.000054s : 1: split_matmul_comm_elemetwise 0.00% : 0.000017s : 1: swap_dp_allreduce_reducescatter 0.01% : 0.000225s : 1: symbol_engine_optimizer 0.01% : 0.000166s : 1: tuple_transform 81.90% : 2.698751s : 1: type_inference . [hook] pytest_runtest_teardown:test_matmul_qkv_bf16_0[False-mstype0-k_n_shape1-1] tests/st/infer/ops/test_internal_ops/test_matmul_split.py::test_matmul_qkv_bf16_0[False-mstype0-k_n_shape1-1],max_mem:140.0M [WARNING] ME(152078:281472851701552,MainProcess):2026-01-29-17:38:57.674.610 [mindspore/context.py:1334] For 'context.set_context', the parameter 'device_target' will be deprecated and removed in a future version. Please use the api mindspore.set_device() instead. TotalTime = 3.08868, [21] [bootstrap]: 0.0752168 [type_inference]: 2.73282 [event_method]: 1.96e-05 [auto_monad]: 0.00035872 [graph_reusing]: 6.19999e-06 [inline]: 2.96999e-06 [add_attr]: 0.135979, [1] [add_attr_with_inline]: 0.135961, [1] [Cycle 1]: 7.9e-05, [2] [tag_attr]: 2.377e-05 [meta_addattr_fg_expand]: 4.44002e-06 [parallel-infer-symbol]: 5.67001e-06 [pre_auto_parallel]: 4.382e-05 [insert-virtual-dataset]: 2.66e-06 [parallel-infer-symbol-second]: 6.80011e-07 [dataset_repeat_opt]: 2.38998e-06 [pipeline_split]: 1.64e-06 [optimize]: 0.141425, [53] [py_interpret_to_execute]: 4.063e-05 [rewriter_before_opt_a]: 9.874e-05 [opt_a]: 0.0161988, [2] [Cycle 1]: 0.0144432, [45] [expand_dump_flag]: 3.11001e-06 [switch_simplify]: 3.491e-05 [loop_unroll]: 2.017e-05 [a_1]: 0.00087936 [with_stream_mark]: 3.101e-05 [recompute_prepare]: 2.56e-05 [updatestate_depend_eliminate]: 1.323e-05 [updatestate_assign_eliminate]: 8.32e-06 [updatestate_loads_eliminate]: 1.218e-05 [parameter_eliminate]: 2.70997e-06 [a_2]: 0.00025909 [accelerated_algorithm]: 5.222e-05 [shard]: 3.63e-06 [meta_shard_fg_expand]: 3.6e-06 [shard_inline]: 1.962e-05 [merge_send_recv]: 1.513e-05 [auto_parallel]: 1.644e-05 [parallel]: 3.55e-05 [flash_sp]: 3.604e-05 [merge_comm]: 9.92999e-06 [allreduce_fusion]: 8.45001e-06 [matmul_add_comm_reduction]: 1.815e-05 [allreduce_slice_to_reducescatter]: 9.89996e-07 [virtual_shard_identity]: 7.542e-05 [virtual_dataset]: 1.847e-05 [get_grad_eliminate_]: 1.975e-05 [virtual_output]: 2.004e-05 [merge_forward]: 1.081e-05 [cell_reuse_recompute_pass]: 2.99999e-06 [offload_activation]: 1.871e-05 [cell_reuse_handle_not_recompute_node_pass]: 3.84e-05 [merge_recompute_call_nodes]: 1.83002e-06 [before_grad]: 2.977e-05 [set_forward_comm_id_for_comm_node_pass]: 9.96e-06 [meta_fg_expand]: 6.80998e-06 [flash_sp_send_recv_attached]: 7.10002e-06 [receive_attached]: 2.05002e-06 [after_resolve]: 2.673e-05 [a_after_grad]: 3.193e-05 [renormalize]: 0.0106483 [add_forward_monad_depend]: 1.848e-05 [auto_monad_grad]: 3.09999e-06 [auto_monad_eliminator]: 6.543e-05 [cse]: 0.00133392 [a_3]: 0.00014725 [Cycle 2]: 0.00173775, [45] [expand_dump_flag]: 4.12e-06 [switch_simplify]: 1.976e-05 [loop_unroll]: 1.782e-05 [a_1]: 0.00047214 [with_stream_mark]: 3.692e-05 [recompute_prepare]: 1.936e-05 [updatestate_depend_eliminate]: 1.324e-05 [updatestate_assign_eliminate]: 8.18999e-06 [updatestate_loads_eliminate]: 1.222e-05 [parameter_eliminate]: 2.48e-06 [a_2]: 0.00022413 [accelerated_algorithm]: 2.265e-05 [shard]: 2.31998e-06 [meta_shard_fg_expand]: 6.17999e-06 [shard_inline]: 1.64e-05 [merge_send_recv]: 1.835e-05 [auto_parallel]: 1.596e-05 [parallel]: 1.144e-05 [flash_sp]: 5.25001e-06 [merge_comm]: 9.32999e-06 [allreduce_fusion]: 8.89e-06 [matmul_add_comm_reduction]: 1.674e-05 [allreduce_slice_to_reducescatter]: 1.54e-06 [virtual_shard_identity]: 2.015e-05 [virtual_dataset]: 1.551e-05 [get_grad_eliminate_]: 1.533e-05 [virtual_output]: 1.532e-05 [merge_forward]: 1.001e-05 [cell_reuse_recompute_pass]: 3.04999e-06 [offload_activation]: 1.865e-05 [cell_reuse_handle_not_recompute_node_pass]: 3.003e-05 [merge_recompute_call_nodes]: 2.23002e-06 [before_grad]: 2.551e-05 [set_forward_comm_id_for_comm_node_pass]: 9.49e-06 [meta_fg_expand]: 6.96001e-06 [flash_sp_send_recv_attached]: 1.60999e-06 [receive_attached]: 2.74001e-06 [after_resolve]: 2.398e-05 [a_after_grad]: 2.696e-05 [renormalize]: 6.99947e-08 [add_forward_monad_depend]: 4.13999e-06 [auto_monad_grad]: 3.04999e-06 [auto_monad_eliminator]: 5.751e-05 [cse]: 0.0001166 [a_3]: 0.00010851 [py_interpret_to_execute_after_opt_a]: 2.86e-05 [slice_cell_reuse_recomputed_activation]: 3.08e-06 [rewriter_after_opt_a]: 0.00019668 [convert_after_rewriter]: 1.656e-05 [order_py_execute_after_rewriter]: 1.089e-05 [mutable_eliminate]: 0.12219 [opt_b]: 0.00060259, [1] [Cycle 1]: 0.00058922, [7] [b_1]: 0.00038828 [b_2]: 1.799e-05 [updatestate_depend_eliminate]: 2.165e-05 [updatestate_assign_eliminate]: 8.35001e-06 [updatestate_loads_eliminate]: 1.204e-05 [renormalize]: 1.71e-06 [cse]: 9.437e-05 [optimize_parallel_all_gather_comm]: 3.755e-05 [overlap_param_gather]: 4.42e-06 [cconv]: 4.399e-05 [loop_unroll]: 0.00059509 [opt_after_cconv]: 0.00026009, [1] [Cycle 1]: 0.00025015, [7] [c_1]: 0.00011782 [parameter_eliminate]: 6.28e-06 [updatestate_depend_eliminate]: 1.536e-05 [updatestate_assign_eliminate]: 6.91001e-06 [updatestate_loads_eliminate]: 9.56e-06 [cse]: 5.262e-05 [renormalize]: 5.69999e-07 [remove_dup_value]: 7.484e-05 [tuple_transform]: 0.00016857, [1] [Cycle 1]: 0.00016293, [4] [d_1]: 0.0001172 [none_parameter_eliminate]: 2.39001e-06 [renormalize]: 2.69996e-07 [switch_simplify]: 1.848e-05 [partial_unused_args_eliminate]: 2.35002e-06 [add_recomputation]: 0.00011785 [cse_after_recomputation]: 6.097e-05, [1] [Cycle 1]: 5.493e-05, [1] [cse]: 4.803e-05 [environ_conv]: 2.092e-05 [swap_dp_allreduce_reducescatter]: 1.334e-05 [bias_add_comm_swap]: 3.65e-06 [label_micro_interleaved_index]: 6.11998e-06 [label_fine_grained_interleaved_index]: 3.23e-06 [merge_cast_opt]: 1.96e-06 [slice_recompute_activation]: 2.33002e-06 [micro_interleaved_order_control]: 2.69999e-06 [assign_add_opt]: 1.27e-06 [ForceFp32Comm]: 8.89995e-07 [remove_cast_before_assign_add]: 1.37e-06 [full_micro_interleaved_order_control]: 2.19001e-06 [reorder_send_recv_between_fp_bp]: 2.68998e-06 [comm_op_add_attrs]: 1.52001e-06 [add_comm_op_reuse_tag]: 1.14e-06 [interleave_split_concat_branches]: 1.28002e-06 [interleave_parallel_branches]: 1.17e-06 [overlap_opt_shard_in_pipeline]: 4.33001e-06 [overlap_opt_shard_grad_in_pipeline]: 1.69e-06 [control_data_broadcast_order]: 2.914e-05 [grouped_pairwise_exchange_alltoall]: 1.87001e-06 [offloading_packed_experts]: 7.35998e-06 [overlap_recompute_and_grad_model_parallel]: 8.13999e-06 [overlap_grad_matmul_and_grad_allreduce]: 1.30001e-06 [overlap_recompute_allgather_and_fa_grad]: 1.76e-06 [overlap_recompute_comm]: 2.21e-06 [overlap_grad_ring_attention]: 9.15999e-06 [overlap_grad_flash_sp]: 4.034e-05 [begin_end_overlap_inline]: 6.89994e-07 [split_matmul_comm_elemetwise]: 2.14999e-06 [split_layernorm_comm]: 2.02001e-06 [handle_group_info]: 1.19e-06 [symbol_engine_optimizer]: 0.00017369, [1] [Cycle 1]: 0.00016707, [6] [build]: 1.981e-05 [elim_shapecalc]: 2.644e-05 [elim_not_effective]: 3.286e-05 [opt_reshape]: 2.241e-05 [fold_const_symbol]: 2.518e-05 [renormalize]: 2.19996e-07 [detach_backward]: 2.31e-06 [pipeline_parallel_scheduler]: 1.73002e-06 [auto_monad_reorder]: 5.087e-05 [get_jit_bprop_graph]: 2.18002e-06 [rewriter_after_jit_bprop_graph]: 7.25e-06 [opt_after_jit_grad]: 0.00074844 [validate]: 0.00010571 Sums bootstrap : 0.075217s : 2.55% type_inference : 2.732817s : 92.65% event_method : 0.000020s : 0.00% auto_monad : 0.000359s : 0.01% graph_reusing : 0.000006s : 0.00% inline : 0.000003s : 0.00% add_attr.add_attr_with_inline.tag_attr : 0.000024s : 0.00% add_attr.add_attr_with_inline.meta_addattr_fg_expand : 0.000004s : 0.00% parallel-infer-symbol : 0.000006s : 0.00% pre_auto_parallel : 0.000044s : 0.00% insert-virtual-dataset : 0.000003s : 0.00% parallel-infer-symbol-second : 0.000001s : 0.00% dataset_repeat_opt : 0.000002s : 0.00% pipeline_split : 0.000002s : 0.00% optimize.py_interpret_to_execute : 0.000041s : 0.00% optimize.rewriter_before_opt_a : 0.000099s : 0.00% optimize.opt_a.expand_dump_flag : 0.000007s : 0.00% optimize.opt_a.switch_simplify : 0.000055s : 0.00% optimize.opt_a.loop_unroll : 0.000038s : 0.00% optimize.opt_a.a_1 : 0.001352s : 0.05% optimize.opt_a.with_stream_mark : 0.000068s : 0.00% optimize.opt_a.recompute_prepare : 0.000045s : 0.00% optimize.opt_a.updatestate_depend_eliminate : 0.000026s : 0.00% optimize.opt_a.updatestate_assign_eliminate : 0.000017s : 0.00% optimize.opt_a.updatestate_loads_eliminate : 0.000024s : 0.00% optimize.opt_a.parameter_eliminate : 0.000005s : 0.00% optimize.opt_a.a_2 : 0.000483s : 0.02% optimize.opt_a.accelerated_algorithm : 0.000075s : 0.00% optimize.opt_a.shard : 0.000006s : 0.00% optimize.opt_a.meta_shard_fg_expand : 0.000010s : 0.00% optimize.opt_a.shard_inline : 0.000036s : 0.00% optimize.opt_a.merge_send_recv : 0.000033s : 0.00% optimize.opt_a.auto_parallel : 0.000032s : 0.00% optimize.opt_a.parallel : 0.000047s : 0.00% optimize.opt_a.flash_sp : 0.000041s : 0.00% optimize.opt_a.merge_comm : 0.000019s : 0.00% optimize.opt_a.allreduce_fusion : 0.000017s : 0.00% optimize.opt_a.matmul_add_comm_reduction : 0.000035s : 0.00% optimize.opt_a.allreduce_slice_to_reducescatter : 0.000003s : 0.00% optimize.opt_a.virtual_shard_identity : 0.000096s : 0.00% optimize.opt_a.virtual_dataset : 0.000034s : 0.00% optimize.opt_a.get_grad_eliminate_ : 0.000035s : 0.00% optimize.opt_a.virtual_output : 0.000035s : 0.00% optimize.opt_a.merge_forward : 0.000021s : 0.00% optimize.opt_a.cell_reuse_recompute_pass : 0.000006s : 0.00% optimize.opt_a.offload_activation : 0.000037s : 0.00% optimize.opt_a.cell_reuse_handle_not_recompute_node_pass : 0.000068s : 0.00% optimize.opt_a.merge_recompute_call_nodes : 0.000004s : 0.00% optimize.opt_a.before_grad : 0.000055s : 0.00% optimize.opt_a.set_forward_comm_id_for_comm_node_pass : 0.000019s : 0.00% optimize.opt_a.meta_fg_expand : 0.000014s : 0.00% optimize.opt_a.flash_sp_send_recv_attached : 0.000009s : 0.00% optimize.opt_a.receive_attached : 0.000005s : 0.00% optimize.opt_a.after_resolve : 0.000051s : 0.00% optimize.opt_a.a_after_grad : 0.000059s : 0.00% optimize.opt_a.renormalize : 0.010648s : 0.36% optimize.opt_a.add_forward_monad_depend : 0.000023s : 0.00% optimize.opt_a.auto_monad_grad : 0.000006s : 0.00% optimize.opt_a.auto_monad_eliminator : 0.000123s : 0.00% optimize.opt_a.cse : 0.001451s : 0.05% optimize.opt_a.a_3 : 0.000256s : 0.01% optimize.py_interpret_to_execute_after_opt_a : 0.000029s : 0.00% optimize.slice_cell_reuse_recomputed_activation : 0.000003s : 0.00% optimize.rewriter_after_opt_a : 0.000197s : 0.01% optimize.convert_after_rewriter : 0.000017s : 0.00% optimize.order_py_execute_after_rewriter : 0.000011s : 0.00% optimize.mutable_eliminate : 0.122190s : 4.14% optimize.opt_b.b_1 : 0.000388s : 0.01% optimize.opt_b.b_2 : 0.000018s : 0.00% optimize.opt_b.updatestate_depend_eliminate : 0.000022s : 0.00% optimize.opt_b.updatestate_assign_eliminate : 0.000008s : 0.00% optimize.opt_b.updatestate_loads_eliminate : 0.000012s : 0.00% optimize.opt_b.renormalize : 0.000002s : 0.00% optimize.opt_b.cse : 0.000094s : 0.00% optimize.optimize_parallel_all_gather_comm : 0.000038s : 0.00% optimize.overlap_param_gather : 0.000004s : 0.00% optimize.cconv : 0.000044s : 0.00% optimize.loop_unroll : 0.000595s : 0.02% optimize.opt_after_cconv.c_1 : 0.000118s : 0.00% optimize.opt_after_cconv.parameter_eliminate : 0.000006s : 0.00% optimize.opt_after_cconv.updatestate_depend_eliminate : 0.000015s : 0.00% optimize.opt_after_cconv.updatestate_assign_eliminate : 0.000007s : 0.00% optimize.opt_after_cconv.updatestate_loads_eliminate : 0.000010s : 0.00% optimize.opt_after_cconv.cse : 0.000053s : 0.00% optimize.opt_after_cconv.renormalize : 0.000001s : 0.00% optimize.remove_dup_value : 0.000075s : 0.00% optimize.tuple_transform.d_1 : 0.000117s : 0.00% optimize.tuple_transform.none_parameter_eliminate : 0.000002s : 0.00% optimize.tuple_transform.renormalize : 0.000000s : 0.00% optimize.tuple_transform.switch_simplify : 0.000018s : 0.00% optimize.partial_unused_args_eliminate : 0.000002s : 0.00% optimize.add_recomputation : 0.000118s : 0.00% optimize.cse_after_recomputation.cse : 0.000048s : 0.00% optimize.environ_conv : 0.000021s : 0.00% optimize.swap_dp_allreduce_reducescatter : 0.000013s : 0.00% optimize.bias_add_comm_swap : 0.000004s : 0.00% optimize.label_micro_interleaved_index : 0.000006s : 0.00% optimize.label_fine_grained_interleaved_index : 0.000003s : 0.00% optimize.merge_cast_opt : 0.000002s : 0.00% optimize.slice_recompute_activation : 0.000002s : 0.00% optimize.micro_interleaved_order_control : 0.000003s : 0.00% optimize.assign_add_opt : 0.000001s : 0.00% optimize.ForceFp32Comm : 0.000001s : 0.00% optimize.remove_cast_before_assign_add : 0.000001s : 0.00% optimize.full_micro_interleaved_order_control : 0.000002s : 0.00% optimize.reorder_send_recv_between_fp_bp : 0.000003s : 0.00% optimize.comm_op_add_attrs : 0.000002s : 0.00% optimize.add_comm_op_reuse_tag : 0.000001s : 0.00% optimize.interleave_split_concat_branches : 0.000001s : 0.00% optimize.interleave_parallel_branches : 0.000001s : 0.00% optimize.overlap_opt_shard_in_pipeline : 0.000004s : 0.00% optimize.overlap_opt_shard_grad_in_pipeline : 0.000002s : 0.00% optimize.control_data_broadcast_order : 0.000029s : 0.00% optimize.grouped_pairwise_exchange_alltoall : 0.000002s : 0.00% optimize.offloading_packed_experts : 0.000007s : 0.00% optimize.overlap_recompute_and_grad_model_parallel : 0.000008s : 0.00% optimize.overlap_grad_matmul_and_grad_allreduce : 0.000001s : 0.00% optimize.overlap_recompute_allgather_and_fa_grad : 0.000002s : 0.00% optimize.overlap_recompute_comm : 0.000002s : 0.00% optimize.overlap_grad_ring_attention : 0.000009s : 0.00% optimize.overlap_grad_flash_sp : 0.000040s : 0.00% optimize.begin_end_overlap_inline : 0.000001s : 0.00% optimize.split_matmul_comm_elemetwise : 0.000002s : 0.00% optimize.split_layernorm_comm : 0.000002s : 0.00% optimize.handle_group_info : 0.000001s : 0.00% optimize.symbol_engine_optimizer.build : 0.000020s : 0.00% optimize.symbol_engine_optimizer.elim_shapecalc : 0.000026s : 0.00% optimize.symbol_engine_optimizer.elim_not_effective : 0.000033s : 0.00% optimize.symbol_engine_optimizer.opt_reshape : 0.000022s : 0.00% optimize.symbol_engine_optimizer.fold_const_symbol : 0.000025s : 0.00% optimize.symbol_engine_optimizer.renormalize : 0.000000s : 0.00% detach_backward : 0.000002s : 0.00% pipeline_parallel_scheduler : 0.000002s : 0.00% auto_monad_reorder : 0.000051s : 0.00% get_jit_bprop_graph : 0.000002s : 0.00% rewriter_after_jit_bprop_graph : 0.000007s : 0.00% opt_after_jit_grad : 0.000748s : 0.03% validate : 0.000106s : 0.00% Time group info: ------[substitution.] 0.000313 96 3.68% : 0.000012s : 2: substitution.depend_value_elim 1.20% : 0.000004s : 8: substitution.elim_not_effective 1.04% : 0.000003s : 8: substitution.fold_const_symbol 3.86% : 0.000012s : 13: substitution.graph_param_transform 43.89% : 0.000137s : 1: substitution.inline 2.72% : 0.000009s : 16: substitution.j_node_and_user_rematch 8.87% : 0.000028s : 2: substitution.less_batch_normalization 1.96% : 0.000006s : 6: substitution.load_eliminater 0.97% : 0.000003s : 2: substitution.opt_reshape 3.71% : 0.000012s : 16: substitution.remove_not_recompute_node 2.50% : 0.000008s : 4: substitution.replace_old_param 10.37% : 0.000032s : 4: substitution.reshape_eliminate 8.66% : 0.000027s : 6: substitution.updatestate_pure_node_eliminater 6.58% : 0.000021s : 8: substitution.updatestate_useless_node_eliminater ------[type_inference.] 2.732706 2 99.81% : 2.727534s : 1: type_inference.infer 0.19% : 0.005172s : 1: type_inference.specialize ------[replace.] 0.000019 1 100.00% : 0.000019s : 1: replace.inline ------[match.] 0.000136 1 100.00% : 0.000136s : 1: match.inline ------[predicate.] 0.000468 3002 0.83% : 0.000004s : 27: predicate.accumulaten_eliminater 1.13% : 0.000005s : 13: predicate.ad_related_special_op_eliminate 0.70% : 0.000003s : 26: predicate.addn_check_dump 0.76% : 0.000004s : 27: predicate.addn_zero_filter 0.69% : 0.000003s : 27: predicate.adjust_all_reduce_mul_add 2.28% : 0.000011s : 53: predicate.arithmetic_simplify 0.95% : 0.000004s : 27: predicate.cast_eliminate 0.75% : 0.000004s : 26: predicate.check_bprop_eliminate 0.78% : 0.000004s : 26: predicate.compare_switch_simplify 0.27% : 0.000001s : 13: predicate.const_output_eliminate 0.99% : 0.000005s : 26: predicate.depend_value_elim 0.77% : 0.000004s : 27: predicate.dict_get_item_const_eliminator 0.85% : 0.000004s : 27: predicate.dict_get_item_eliminator 0.73% : 0.000003s : 27: predicate.dict_set_item_eliminator 1.21% : 0.000006s : 26: predicate.dumpgradient_eliminate 0.26% : 0.000001s : 13: predicate.elim_not_effective 0.52% : 0.000002s : 13: predicate.elim_shapecalc_of_broadcastargs 1.09% : 0.000005s : 40: predicate.environ_add_const_eliminate 1.07% : 0.000005s : 40: predicate.environ_get_add_eliminate 0.99% : 0.000005s : 40: predicate.environ_get_depend_swap 1.93% : 0.000009s : 66: predicate.environ_get_eliminate 1.06% : 0.000005s : 40: predicate.environ_get_set_eliminate 0.73% : 0.000003s : 28: predicate.exchange_switch_depend_value 1.25% : 0.000006s : 28: predicate.float_depend_g_call 0.72% : 0.000003s : 26: predicate.float_environ_get_switch 1.13% : 0.000005s : 39: predicate.float_tuple_getitem_switch 0.23% : 0.000001s : 13: predicate.fold_const_symbol 0.86% : 0.000004s : 26: predicate.get_grad_eliminate 0.31% : 0.000001s : 13: predicate.graph_param_transform 0.73% : 0.000003s : 26: predicate.incorporate_call 0.66% : 0.000003s : 26: predicate.incorporate_call_switch 5.22% : 0.000024s : 133: predicate.inline 0.95% : 0.000004s : 26: predicate.inline_without_move 0.43% : 0.000002s : 26: predicate.j_node_and_user_rematch 1.19% : 0.000006s : 27: predicate.less_batch_normalization 1.57% : 0.000007s : 53: predicate.list_to_tuple_eliminator_ 2.26% : 0.000011s : 80: predicate.load_eliminater 0.99% : 0.000005s : 13: predicate.loop_unroll_after_grad 1.06% : 0.000005s : 31: predicate.loop_unroll_before_grad 1.74% : 0.000008s : 53: predicate.make_slice_get_slice_eliminator 0.74% : 0.000003s : 26: predicate.merge_addn 0.78% : 0.000004s : 26: predicate.micro_step_allgather_replace 0.76% : 0.000004s : 26: predicate.mini_step_allgather_replace 0.79% : 0.000004s : 27: predicate.minmaximum_grad 2.59% : 0.000012s : 13: predicate.mutable_eliminate 0.50% : 0.000002s : 13: predicate.opt_reshape 0.49% : 0.000002s : 13: predicate.parallel_virtual_node 0.97% : 0.000005s : 28: predicate.partial_defer_inline 1.21% : 0.000006s : 40: predicate.partial_eliminate 0.85% : 0.000004s : 27: predicate.print_const_string_wrapper 0.77% : 0.000004s : 26: predicate.reduce_all_const_elim 1.04% : 0.000005s : 27: predicate.reduce_eliminate 2.21% : 0.000010s : 80: predicate.redundant_stop_gradient_eliminater 0.49% : 0.000002s : 26: predicate.remove_not_recompute_node 1.35% : 0.000006s : 53: predicate.replace_applicator 0.61% : 0.000003s : 26: predicate.replace_old_param 0.38% : 0.000002s : 13: predicate.reset_defer_inline 1.06% : 0.000005s : 27: predicate.reshape_eliminate 0.80% : 0.000004s : 26: predicate.row_tensor_add_zeros_like 0.47% : 0.000002s : 13: predicate.row_tensor_eliminate 1.34% : 0.000006s : 26: predicate.same_eliminate 0.57% : 0.000003s : 26: predicate.set_cell_output_no_recompute 1.02% : 0.000005s : 26: predicate.shard_identity_eliminate 0.92% : 0.000004s : 26: predicate.special_op_eliminate 0.89% : 0.000004s : 26: predicate.specialize_transform 1.43% : 0.000007s : 26: predicate.split_environ_get_set_with_tuple_value 1.01% : 0.000005s : 26: predicate.stack_unstack_eliminate 0.49% : 0.000002s : 13: predicate.switch_call_monad_eliminater 0.85% : 0.000004s : 28: predicate.switch_defer_inline 1.63% : 0.000008s : 54: predicate.switch_layer_defer_inline 3.59% : 0.000017s : 98: predicate.switch_simplify 0.79% : 0.000004s : 27: predicate.tile_eliminate 0.74% : 0.000003s : 27: predicate.transpose_eliminate 1.58% : 0.000007s : 53: predicate.tuple_list_convert_item_index_to_positive 1.80% : 0.000008s : 53: predicate.tuple_list_get_item_const_eliminator 1.69% : 0.000008s : 53: predicate.tuple_list_get_item_depend_reorder 3.07% : 0.000014s : 79: predicate.tuple_list_get_item_eliminator 1.73% : 0.000008s : 53: predicate.tuple_list_get_set_item_eliminator 2.99% : 0.000014s : 79: predicate.tuple_list_set_item_eliminator 1.61% : 0.000008s : 53: predicate.tuple_to_list_eliminator_ 2.28% : 0.000011s : 80: predicate.updatestate_pure_node_eliminater 3.36% : 0.000016s : 106: predicate.updatestate_useless_node_eliminater 0.44% : 0.000002s : 13: predicate.value_based_eliminate 0.85% : 0.000004s : 26: predicate.virtual_dataset_eliminate 0.90% : 0.000004s : 26: predicate.virtual_output_eliminate 0.45% : 0.000002s : 13: predicate.virtual_view_grad_eliminate 0.49% : 0.000002s : 13: predicate.zero_like_fill_zero ------[func_graph_cloner_run.] 0.177826 58 98.12% : 0.174487s : 55: func_graph_cloner_run.FuncGraphClonerGraph 1.88% : 0.003339s : 3: func_graph_cloner_run.FuncGraphSpecializer ------[meta_graph.] 0.000000 0 ------[manager.] 0.000000 0 ------[pynative] 0.000000 0 ------[others.] 3.379991 192 0.00% : 0.000004s : 1: ForceFp32Comm 4.02% : 0.135987s : 1: add_attr 4.02% : 0.135966s : 1: add_attr_with_inline 0.00% : 0.000004s : 1: add_comm_op_reuse_tag 0.00% : 0.000124s : 1: add_recomputation 0.00% : 0.000004s : 1: assign_add_opt 0.01% : 0.000372s : 1: auto_monad 0.00% : 0.000057s : 1: auto_monad_reorder 0.00% : 0.000004s : 1: begin_end_overlap_inline 0.00% : 0.000007s : 1: bias_add_comm_swap 2.28% : 0.076905s : 1: bootstrap 0.00% : 0.000048s : 1: cconv 0.00% : 0.000005s : 1: comm_op_add_attrs 0.00% : 0.000033s : 1: control_data_broadcast_order 0.00% : 0.000022s : 1: convert_after_rewriter 0.00% : 0.000064s : 1: cse_after_recomputation 0.00% : 0.000006s : 1: dataset_repeat_opt 0.00% : 0.000005s : 1: detach_backward 0.00% : 0.000026s : 1: environ_conv 0.00% : 0.000028s : 1: event_method 0.00% : 0.000005s : 1: full_micro_interleaved_order_control 0.00% : 0.000007s : 1: get_jit_bprop_graph 0.00% : 0.000011s : 1: graph_reusing 0.00% : 0.000005s : 1: grouped_pairwise_exchange_alltoall 0.00% : 0.000004s : 1: handle_group_info 0.00% : 0.000006s : 1: inline 0.00% : 0.000008s : 1: insert-virtual-dataset 0.00% : 0.000005s : 1: interleave_parallel_branches 0.00% : 0.000004s : 1: interleave_split_concat_branches 0.00% : 0.000006s : 1: label_fine_grained_interleaved_index 0.00% : 0.000009s : 1: label_micro_interleaved_index 0.02% : 0.000608s : 1: loop_unroll 0.00% : 0.000005s : 1: merge_cast_opt 0.00% : 0.000006s : 1: micro_interleaved_order_control 3.62% : 0.122214s : 1: mutable_eliminate 0.00% : 0.000011s : 1: offloading_packed_experts 0.00% : 0.000031s : 1: opt.transform.loop_unroll_optimizer 0.00% : 0.000063s : 1: opt.transform.mutable_eliminate 0.08% : 0.002642s : 78: opt.transform.opt_a 0.00% : 0.000116s : 1: opt.transform.opt_after_cconv 0.00% : 0.000070s : 1: opt.transform.opt_after_jit_grad 0.01% : 0.000368s : 28: opt.transform.opt_b 0.00% : 0.000132s : 2: opt.transform.opt_trans_graph 0.00% : 0.000102s : 4: opt.transform.symbol_engine_opt 0.48% : 0.016204s : 1: opt_a 0.01% : 0.000264s : 1: opt_after_cconv 0.02% : 0.000766s : 1: opt_after_jit_grad 0.02% : 0.000608s : 1: opt_b 4.18% : 0.141432s : 1: optimize 0.00% : 0.000042s : 1: optimize_parallel_all_gather_comm 0.00% : 0.000015s : 1: order_py_execute_after_rewriter 0.00% : 0.000044s : 1: overlap_grad_flash_sp 0.00% : 0.000004s : 1: overlap_grad_matmul_and_grad_allreduce 0.00% : 0.000014s : 1: overlap_grad_ring_attention 0.00% : 0.000005s : 1: overlap_opt_shard_grad_in_pipeline 0.00% : 0.000008s : 1: overlap_opt_shard_in_pipeline 0.00% : 0.000010s : 1: overlap_param_gather 0.00% : 0.000006s : 1: overlap_recompute_allgather_and_fa_grad 0.00% : 0.000011s : 1: overlap_recompute_and_grad_model_parallel 0.00% : 0.000005s : 1: overlap_recompute_comm 0.00% : 0.000010s : 1: parallel-infer-symbol 0.00% : 0.000004s : 1: parallel-infer-symbol-second 0.00% : 0.000006s : 1: partial_unused_args_eliminate 0.00% : 0.000005s : 1: pipeline_parallel_scheduler 0.00% : 0.000006s : 1: pipeline_split 0.00% : 0.000049s : 1: pre_auto_parallel 0.00% : 0.000045s : 1: py_interpret_to_execute 0.00% : 0.000035s : 1: py_interpret_to_execute_after_opt_a 0.00% : 0.000004s : 1: remove_cast_before_assign_add 0.00% : 0.000080s : 1: remove_dup_value 0.20% : 0.006904s : 1: renormalize.infer 0.11% : 0.003726s : 1: renormalize.specialize 0.00% : 0.000006s : 1: reorder_send_recv_between_fp_bp 0.00% : 0.000011s : 1: rewriter_after_jit_bprop_graph 0.01% : 0.000207s : 1: rewriter_after_opt_a 0.00% : 0.000104s : 1: rewriter_before_opt_a 0.00% : 0.000008s : 1: slice_cell_reuse_recomputed_activation 0.00% : 0.000005s : 1: slice_recompute_activation 0.00% : 0.000007s : 1: split_layernorm_comm 0.00% : 0.000005s : 1: split_matmul_comm_elemetwise 0.00% : 0.000017s : 1: swap_dp_allreduce_reducescatter 0.01% : 0.000177s : 1: symbol_engine_optimizer 0.01% : 0.000172s : 1: tuple_transform 80.85% : 2.732854s : 1: type_inference . [hook] pytest_runtest_teardown:test_matmul_qkv_bf16_0[False-mstype0-k_n_shape1-16] tests/st/infer/ops/test_internal_ops/test_matmul_split.py::test_matmul_qkv_bf16_0[False-mstype0-k_n_shape1-16],max_mem:140.0M [WARNING] ME(152078:281472851701552,MainProcess):2026-01-29-17:39:40.174.950 [mindspore/context.py:1334] For 'context.set_context', the parameter 'device_target' will be deprecated and removed in a future version. Please use the api mindspore.set_device() instead. TotalTime = 2.6713, [21] [bootstrap]: 0.00082485 [type_inference]: 2.27877 [event_method]: 2.062e-05 [auto_monad]: 0.00037878 [graph_reusing]: 6.59001e-06 [inline]: 3.66001e-06 [add_attr]: 0.00717382, [1] [add_attr_with_inline]: 0.00715606, [1] [Cycle 1]: 8.006e-05, [2] [tag_attr]: 2.428e-05 [meta_addattr_fg_expand]: 4.62e-06 [parallel-infer-symbol]: 4.57e-06 [pre_auto_parallel]: 0.23244 [insert-virtual-dataset]: 4.43999e-06 [parallel-infer-symbol-second]: 3.29001e-06 [dataset_repeat_opt]: 3.31999e-06 [pipeline_split]: 2.22001e-06 [optimize]: 0.150297, [53] [py_interpret_to_execute]: 5.556e-05 [rewriter_before_opt_a]: 9.799e-05 [opt_a]: 0.0154901, [2] [Cycle 1]: 0.0137398, [45] [expand_dump_flag]: 3.98001e-06 [switch_simplify]: 3.747e-05 [loop_unroll]: 2.133e-05 [a_1]: 0.00090264 [with_stream_mark]: 3.084e-05 [recompute_prepare]: 2.64e-05 [updatestate_depend_eliminate]: 1.14e-05 [updatestate_assign_eliminate]: 9.86e-06 [updatestate_loads_eliminate]: 1.089e-05 [parameter_eliminate]: 2.37001e-06 [a_2]: 0.00025609 [accelerated_algorithm]: 5.186e-05 [shard]: 2.53003e-06 [meta_shard_fg_expand]: 4.57e-06 [shard_inline]: 1.849e-05 [merge_send_recv]: 1.681e-05 [auto_parallel]: 1.487e-05 [parallel]: 3.096e-05 [flash_sp]: 1.522e-05 [merge_comm]: 8.71002e-06 [allreduce_fusion]: 8.58001e-06 [matmul_add_comm_reduction]: 1.791e-05 [allreduce_slice_to_reducescatter]: 1.27e-06 [virtual_shard_identity]: 2.127e-05 [virtual_dataset]: 1.836e-05 [get_grad_eliminate_]: 1.903e-05 [virtual_output]: 1.801e-05 [merge_forward]: 9.06002e-06 [cell_reuse_recompute_pass]: 2.29999e-06 [offload_activation]: 2.056e-05 [cell_reuse_handle_not_recompute_node_pass]: 4.131e-05 [merge_recompute_call_nodes]: 1.69e-06 [before_grad]: 3.403e-05 [set_forward_comm_id_for_comm_node_pass]: 1.114e-05 [meta_fg_expand]: 6.91999e-06 [flash_sp_send_recv_attached]: 6.94001e-06 [receive_attached]: 2.36998e-06 [after_resolve]: 2.578e-05 [a_after_grad]: 3.523e-05 [renormalize]: 0.0111504 [add_forward_monad_depend]: 2e-05 [auto_monad_grad]: 3.55998e-06 [auto_monad_eliminator]: 6.952e-05 [cse]: 0.00020487 [a_3]: 0.00014365 [Cycle 2]: 0.00172565, [45] [expand_dump_flag]: 3.04001e-06 [switch_simplify]: 2.021e-05 [loop_unroll]: 1.699e-05 [a_1]: 0.00046973 [with_stream_mark]: 3.104e-05 [recompute_prepare]: 2.008e-05 [updatestate_depend_eliminate]: 1.053e-05 [updatestate_assign_eliminate]: 9.15001e-06 [updatestate_loads_eliminate]: 1.075e-05 [parameter_eliminate]: 2.58998e-06 [a_2]: 0.00022272 [accelerated_algorithm]: 2.469e-05 [shard]: 2.26998e-06 [meta_shard_fg_expand]: 6.20002e-06 [shard_inline]: 1.565e-05 [merge_send_recv]: 1.564e-05 [auto_parallel]: 1.758e-05 [parallel]: 1.066e-05 [flash_sp]: 4.95999e-06 [merge_comm]: 8.23999e-06 [allreduce_fusion]: 9.81e-06 [matmul_add_comm_reduction]: 1.846e-05 [allreduce_slice_to_reducescatter]: 8.00006e-07 [virtual_shard_identity]: 2.019e-05 [virtual_dataset]: 1.563e-05 [get_grad_eliminate_]: 1.561e-05 [virtual_output]: 1.555e-05 [merge_forward]: 9.47001e-06 [cell_reuse_recompute_pass]: 2.54999e-06 [offload_activation]: 1.967e-05 [cell_reuse_handle_not_recompute_node_pass]: 3.303e-05 [merge_recompute_call_nodes]: 1.87999e-06 [before_grad]: 2.561e-05 [set_forward_comm_id_for_comm_node_pass]: 1.005e-05 [meta_fg_expand]: 6.94999e-06 [flash_sp_send_recv_attached]: 2.36e-06 [receive_attached]: 3.09999e-06 [after_resolve]: 2.394e-05 [a_after_grad]: 2.735e-05 [renormalize]: 4.00003e-07 [add_forward_monad_depend]: 3.91001e-06 [auto_monad_grad]: 2.69999e-06 [auto_monad_eliminator]: 5.83e-05 [cse]: 7.524e-05 [a_3]: 0.00010543 [py_interpret_to_execute_after_opt_a]: 3.024e-05 [slice_cell_reuse_recomputed_activation]: 2.91999e-06 [rewriter_after_opt_a]: 0.00148517 [convert_after_rewriter]: 2.842e-05 [order_py_execute_after_rewriter]: 1.295e-05 [mutable_eliminate]: 0.00086577 [opt_b]: 0.00058668, [1] [Cycle 1]: 0.00057533, [7] [b_1]: 0.00038212 [b_2]: 1.967e-05 [updatestate_depend_eliminate]: 1.69e-05 [updatestate_assign_eliminate]: 8.47e-06 [updatestate_loads_eliminate]: 1.205e-05 [renormalize]: 1.12e-06 [cse]: 8.601e-05 [optimize_parallel_all_gather_comm]: 3.96e-05 [overlap_param_gather]: 2.40002e-06 [cconv]: 4.705e-05 [loop_unroll]: 0.00069047 [opt_after_cconv]: 0.12973, [1] [Cycle 1]: 0.129716, [7] [c_1]: 0.129475 [parameter_eliminate]: 7.7e-06 [updatestate_depend_eliminate]: 2.685e-05 [updatestate_assign_eliminate]: 8.70999e-06 [updatestate_loads_eliminate]: 1.291e-05 [cse]: 8.792e-05 [renormalize]: 1.22e-06 [remove_dup_value]: 8.342e-05 [tuple_transform]: 0.00018446, [1] [Cycle 1]: 0.00017312, [4] [d_1]: 0.00012948 [none_parameter_eliminate]: 1.94999e-06 [renormalize]: 1.50001e-07 [switch_simplify]: 1.794e-05 [partial_unused_args_eliminate]: 2.20002e-06 [add_recomputation]: 0.00012822 [cse_after_recomputation]: 5.567e-05, [1] [Cycle 1]: 5.058e-05, [1] [cse]: 4.399e-05 [environ_conv]: 2.069e-05 [swap_dp_allreduce_reducescatter]: 1.199e-05 [bias_add_comm_swap]: 3.79002e-06 [label_micro_interleaved_index]: 7.41001e-06 [label_fine_grained_interleaved_index]: 3.31999e-06 [merge_cast_opt]: 1.55999e-06 [slice_recompute_activation]: 2.21e-06 [micro_interleaved_order_control]: 2.45002e-06 [assign_add_opt]: 1.63002e-06 [ForceFp32Comm]: 9.49978e-07 [remove_cast_before_assign_add]: 1.23002e-06 [full_micro_interleaved_order_control]: 2.73998e-06 [reorder_send_recv_between_fp_bp]: 2.56e-06 [comm_op_add_attrs]: 1.42e-06 [add_comm_op_reuse_tag]: 1.35999e-06 [interleave_split_concat_branches]: 1.22999e-06 [interleave_parallel_branches]: 1.13001e-06 [overlap_opt_shard_in_pipeline]: 1.39998e-06 [overlap_opt_shard_grad_in_pipeline]: 2.16e-06 [control_data_broadcast_order]: 2.949e-05 [grouped_pairwise_exchange_alltoall]: 1.89e-06 [offloading_packed_experts]: 7.75998e-06 [overlap_recompute_and_grad_model_parallel]: 7.89002e-06 [overlap_grad_matmul_and_grad_allreduce]: 1.65001e-06 [overlap_recompute_allgather_and_fa_grad]: 1.40999e-06 [overlap_recompute_comm]: 2.63e-06 [overlap_grad_ring_attention]: 8.15999e-06 [overlap_grad_flash_sp]: 4.293e-05 [begin_end_overlap_inline]: 5.50004e-07 [split_matmul_comm_elemetwise]: 2.48e-06 [split_layernorm_comm]: 1.55001e-06 [handle_group_info]: 1.29e-06 [symbol_engine_optimizer]: 0.00017058, [1] [Cycle 1]: 0.00016389, [6] [build]: 1.919e-05 [elim_shapecalc]: 2.475e-05 [elim_not_effective]: 3.258e-05 [opt_reshape]: 2.57e-05 [fold_const_symbol]: 2.472e-05 [renormalize]: 4.00003e-07 [detach_backward]: 2.49001e-06 [pipeline_parallel_scheduler]: 1.92001e-06 [auto_monad_reorder]: 4.911e-05 [get_jit_bprop_graph]: 3.06001e-06 [rewriter_after_jit_bprop_graph]: 7.5e-06 [opt_after_jit_grad]: 0.00087285 [validate]: 0.00010461 Sums bootstrap : 0.000825s : 0.03% type_inference : 2.278770s : 85.59% event_method : 0.000021s : 0.00% auto_monad : 0.000379s : 0.01% graph_reusing : 0.000007s : 0.00% inline : 0.000004s : 0.00% add_attr.add_attr_with_inline.tag_attr : 0.000024s : 0.00% add_attr.add_attr_with_inline.meta_addattr_fg_expand : 0.000005s : 0.00% parallel-infer-symbol : 0.000005s : 0.00% pre_auto_parallel : 0.232440s : 8.73% insert-virtual-dataset : 0.000004s : 0.00% parallel-infer-symbol-second : 0.000003s : 0.00% dataset_repeat_opt : 0.000003s : 0.00% pipeline_split : 0.000002s : 0.00% optimize.py_interpret_to_execute : 0.000056s : 0.00% optimize.rewriter_before_opt_a : 0.000098s : 0.00% optimize.opt_a.expand_dump_flag : 0.000007s : 0.00% optimize.opt_a.switch_simplify : 0.000058s : 0.00% optimize.opt_a.loop_unroll : 0.000038s : 0.00% optimize.opt_a.a_1 : 0.001372s : 0.05% optimize.opt_a.with_stream_mark : 0.000062s : 0.00% optimize.opt_a.recompute_prepare : 0.000046s : 0.00% optimize.opt_a.updatestate_depend_eliminate : 0.000022s : 0.00% optimize.opt_a.updatestate_assign_eliminate : 0.000019s : 0.00% optimize.opt_a.updatestate_loads_eliminate : 0.000022s : 0.00% optimize.opt_a.parameter_eliminate : 0.000005s : 0.00% optimize.opt_a.a_2 : 0.000479s : 0.02% optimize.opt_a.accelerated_algorithm : 0.000077s : 0.00% optimize.opt_a.shard : 0.000005s : 0.00% optimize.opt_a.meta_shard_fg_expand : 0.000011s : 0.00% optimize.opt_a.shard_inline : 0.000034s : 0.00% optimize.opt_a.merge_send_recv : 0.000032s : 0.00% optimize.opt_a.auto_parallel : 0.000032s : 0.00% optimize.opt_a.parallel : 0.000042s : 0.00% optimize.opt_a.flash_sp : 0.000020s : 0.00% optimize.opt_a.merge_comm : 0.000017s : 0.00% optimize.opt_a.allreduce_fusion : 0.000018s : 0.00% optimize.opt_a.matmul_add_comm_reduction : 0.000036s : 0.00% optimize.opt_a.allreduce_slice_to_reducescatter : 0.000002s : 0.00% optimize.opt_a.virtual_shard_identity : 0.000041s : 0.00% optimize.opt_a.virtual_dataset : 0.000034s : 0.00% optimize.opt_a.get_grad_eliminate_ : 0.000035s : 0.00% optimize.opt_a.virtual_output : 0.000034s : 0.00% optimize.opt_a.merge_forward : 0.000019s : 0.00% optimize.opt_a.cell_reuse_recompute_pass : 0.000005s : 0.00% optimize.opt_a.offload_activation : 0.000040s : 0.00% optimize.opt_a.cell_reuse_handle_not_recompute_node_pass : 0.000074s : 0.00% optimize.opt_a.merge_recompute_call_nodes : 0.000004s : 0.00% optimize.opt_a.before_grad : 0.000060s : 0.00% optimize.opt_a.set_forward_comm_id_for_comm_node_pass : 0.000021s : 0.00% optimize.opt_a.meta_fg_expand : 0.000014s : 0.00% optimize.opt_a.flash_sp_send_recv_attached : 0.000009s : 0.00% optimize.opt_a.receive_attached : 0.000005s : 0.00% optimize.opt_a.after_resolve : 0.000050s : 0.00% optimize.opt_a.a_after_grad : 0.000063s : 0.00% optimize.opt_a.renormalize : 0.011151s : 0.42% optimize.opt_a.add_forward_monad_depend : 0.000024s : 0.00% optimize.opt_a.auto_monad_grad : 0.000006s : 0.00% optimize.opt_a.auto_monad_eliminator : 0.000128s : 0.00% optimize.opt_a.cse : 0.000280s : 0.01% optimize.opt_a.a_3 : 0.000249s : 0.01% optimize.py_interpret_to_execute_after_opt_a : 0.000030s : 0.00% optimize.slice_cell_reuse_recomputed_activation : 0.000003s : 0.00% optimize.rewriter_after_opt_a : 0.001485s : 0.06% optimize.convert_after_rewriter : 0.000028s : 0.00% optimize.order_py_execute_after_rewriter : 0.000013s : 0.00% optimize.mutable_eliminate : 0.000866s : 0.03% optimize.opt_b.b_1 : 0.000382s : 0.01% optimize.opt_b.b_2 : 0.000020s : 0.00% optimize.opt_b.updatestate_depend_eliminate : 0.000017s : 0.00% optimize.opt_b.updatestate_assign_eliminate : 0.000008s : 0.00% optimize.opt_b.updatestate_loads_eliminate : 0.000012s : 0.00% optimize.opt_b.renormalize : 0.000001s : 0.00% optimize.opt_b.cse : 0.000086s : 0.00% optimize.optimize_parallel_all_gather_comm : 0.000040s : 0.00% optimize.overlap_param_gather : 0.000002s : 0.00% optimize.cconv : 0.000047s : 0.00% optimize.loop_unroll : 0.000690s : 0.03% optimize.opt_after_cconv.c_1 : 0.129475s : 4.86% optimize.opt_after_cconv.parameter_eliminate : 0.000008s : 0.00% optimize.opt_after_cconv.updatestate_depend_eliminate : 0.000027s : 0.00% optimize.opt_after_cconv.updatestate_assign_eliminate : 0.000009s : 0.00% optimize.opt_after_cconv.updatestate_loads_eliminate : 0.000013s : 0.00% optimize.opt_after_cconv.cse : 0.000088s : 0.00% optimize.opt_after_cconv.renormalize : 0.000001s : 0.00% optimize.remove_dup_value : 0.000083s : 0.00% optimize.tuple_transform.d_1 : 0.000129s : 0.00% optimize.tuple_transform.none_parameter_eliminate : 0.000002s : 0.00% optimize.tuple_transform.renormalize : 0.000000s : 0.00% optimize.tuple_transform.switch_simplify : 0.000018s : 0.00% optimize.partial_unused_args_eliminate : 0.000002s : 0.00% optimize.add_recomputation : 0.000128s : 0.00% optimize.cse_after_recomputation.cse : 0.000044s : 0.00% optimize.environ_conv : 0.000021s : 0.00% optimize.swap_dp_allreduce_reducescatter : 0.000012s : 0.00% optimize.bias_add_comm_swap : 0.000004s : 0.00% optimize.label_micro_interleaved_index : 0.000007s : 0.00% optimize.label_fine_grained_interleaved_index : 0.000003s : 0.00% optimize.merge_cast_opt : 0.000002s : 0.00% optimize.slice_recompute_activation : 0.000002s : 0.00% optimize.micro_interleaved_order_control : 0.000002s : 0.00% optimize.assign_add_opt : 0.000002s : 0.00% optimize.ForceFp32Comm : 0.000001s : 0.00% optimize.remove_cast_before_assign_add : 0.000001s : 0.00% optimize.full_micro_interleaved_order_control : 0.000003s : 0.00% optimize.reorder_send_recv_between_fp_bp : 0.000003s : 0.00% optimize.comm_op_add_attrs : 0.000001s : 0.00% optimize.add_comm_op_reuse_tag : 0.000001s : 0.00% optimize.interleave_split_concat_branches : 0.000001s : 0.00% optimize.interleave_parallel_branches : 0.000001s : 0.00% optimize.overlap_opt_shard_in_pipeline : 0.000001s : 0.00% optimize.overlap_opt_shard_grad_in_pipeline : 0.000002s : 0.00% optimize.control_data_broadcast_order : 0.000029s : 0.00% optimize.grouped_pairwise_exchange_alltoall : 0.000002s : 0.00% optimize.offloading_packed_experts : 0.000008s : 0.00% optimize.overlap_recompute_and_grad_model_parallel : 0.000008s : 0.00% optimize.overlap_grad_matmul_and_grad_allreduce : 0.000002s : 0.00% optimize.overlap_recompute_allgather_and_fa_grad : 0.000001s : 0.00% optimize.overlap_recompute_comm : 0.000003s : 0.00% optimize.overlap_grad_ring_attention : 0.000008s : 0.00% optimize.overlap_grad_flash_sp : 0.000043s : 0.00% optimize.begin_end_overlap_inline : 0.000001s : 0.00% optimize.split_matmul_comm_elemetwise : 0.000002s : 0.00% optimize.split_layernorm_comm : 0.000002s : 0.00% optimize.handle_group_info : 0.000001s : 0.00% optimize.symbol_engine_optimizer.build : 0.000019s : 0.00% optimize.symbol_engine_optimizer.elim_shapecalc : 0.000025s : 0.00% optimize.symbol_engine_optimizer.elim_not_effective : 0.000033s : 0.00% optimize.symbol_engine_optimizer.opt_reshape : 0.000026s : 0.00% optimize.symbol_engine_optimizer.fold_const_symbol : 0.000025s : 0.00% optimize.symbol_engine_optimizer.renormalize : 0.000000s : 0.00% detach_backward : 0.000002s : 0.00% pipeline_parallel_scheduler : 0.000002s : 0.00% auto_monad_reorder : 0.000049s : 0.00% get_jit_bprop_graph : 0.000003s : 0.00% rewriter_after_jit_bprop_graph : 0.000007s : 0.00% opt_after_jit_grad : 0.000873s : 0.03% validate : 0.000105s : 0.00% Time group info: ------[substitution.] 0.000338 96 3.74% : 0.000013s : 2: substitution.depend_value_elim 1.28% : 0.000004s : 8: substitution.elim_not_effective 0.94% : 0.000003s : 8: substitution.fold_const_symbol 3.79% : 0.000013s : 13: substitution.graph_param_transform 40.16% : 0.000136s : 1: substitution.inline 2.89% : 0.000010s : 16: substitution.j_node_and_user_rematch 8.74% : 0.000029s : 2: substitution.less_batch_normalization 2.33% : 0.000008s : 6: substitution.load_eliminater 0.93% : 0.000003s : 2: substitution.opt_reshape 3.93% : 0.000013s : 16: substitution.remove_not_recompute_node 2.18% : 0.000007s : 4: substitution.replace_old_param 10.56% : 0.000036s : 4: substitution.reshape_eliminate 12.48% : 0.000042s : 6: substitution.updatestate_pure_node_eliminater 6.06% : 0.000020s : 8: substitution.updatestate_useless_node_eliminater ------[type_inference.] 2.278656 2 99.76% : 2.273224s : 1: type_inference.infer 0.24% : 0.005431s : 1: type_inference.specialize ------[replace.] 0.000021 1 100.00% : 0.000021s : 1: replace.inline ------[match.] 0.000135 1 100.00% : 0.000135s : 1: match.inline ------[predicate.] 0.000473 3002 0.85% : 0.000004s : 27: predicate.accumulaten_eliminater 1.05% : 0.000005s : 13: predicate.ad_related_special_op_eliminate 0.78% : 0.000004s : 26: predicate.addn_check_dump 0.81% : 0.000004s : 27: predicate.addn_zero_filter 0.71% : 0.000003s : 27: predicate.adjust_all_reduce_mul_add 2.17% : 0.000010s : 53: predicate.arithmetic_simplify 0.94% : 0.000004s : 27: predicate.cast_eliminate 0.87% : 0.000004s : 26: predicate.check_bprop_eliminate 0.72% : 0.000003s : 26: predicate.compare_switch_simplify 0.23% : 0.000001s : 13: predicate.const_output_eliminate 0.80% : 0.000004s : 26: predicate.depend_value_elim 0.79% : 0.000004s : 27: predicate.dict_get_item_const_eliminator 0.86% : 0.000004s : 27: predicate.dict_get_item_eliminator 0.75% : 0.000004s : 27: predicate.dict_set_item_eliminator 1.08% : 0.000005s : 26: predicate.dumpgradient_eliminate 0.26% : 0.000001s : 13: predicate.elim_not_effective 0.54% : 0.000003s : 13: predicate.elim_shapecalc_of_broadcastargs 1.18% : 0.000006s : 40: predicate.environ_add_const_eliminate 1.09% : 0.000005s : 40: predicate.environ_get_add_eliminate 1.05% : 0.000005s : 40: predicate.environ_get_depend_swap 1.83% : 0.000009s : 66: predicate.environ_get_eliminate 1.02% : 0.000005s : 40: predicate.environ_get_set_eliminate 0.76% : 0.000004s : 28: predicate.exchange_switch_depend_value 1.28% : 0.000006s : 28: predicate.float_depend_g_call 0.73% : 0.000003s : 26: predicate.float_environ_get_switch 1.10% : 0.000005s : 39: predicate.float_tuple_getitem_switch 0.22% : 0.000001s : 13: predicate.fold_const_symbol 0.83% : 0.000004s : 26: predicate.get_grad_eliminate 0.31% : 0.000001s : 13: predicate.graph_param_transform 0.77% : 0.000004s : 26: predicate.incorporate_call 0.66% : 0.000003s : 26: predicate.incorporate_call_switch 5.85% : 0.000028s : 133: predicate.inline 1.12% : 0.000005s : 26: predicate.inline_without_move 0.41% : 0.000002s : 26: predicate.j_node_and_user_rematch 1.42% : 0.000007s : 27: predicate.less_batch_normalization 1.69% : 0.000008s : 53: predicate.list_to_tuple_eliminator_ 2.85% : 0.000013s : 80: predicate.load_eliminater 1.33% : 0.000006s : 13: predicate.loop_unroll_after_grad 1.03% : 0.000005s : 31: predicate.loop_unroll_before_grad 2.05% : 0.000010s : 53: predicate.make_slice_get_slice_eliminator 0.84% : 0.000004s : 26: predicate.merge_addn 0.71% : 0.000003s : 26: predicate.micro_step_allgather_replace 0.76% : 0.000004s : 26: predicate.mini_step_allgather_replace 0.72% : 0.000003s : 27: predicate.minmaximum_grad 1.11% : 0.000005s : 13: predicate.mutable_eliminate 0.68% : 0.000003s : 13: predicate.opt_reshape 0.42% : 0.000002s : 13: predicate.parallel_virtual_node 1.02% : 0.000005s : 28: predicate.partial_defer_inline 1.25% : 0.000006s : 40: predicate.partial_eliminate 0.82% : 0.000004s : 27: predicate.print_const_string_wrapper 0.75% : 0.000004s : 26: predicate.reduce_all_const_elim 1.12% : 0.000005s : 27: predicate.reduce_eliminate 2.32% : 0.000011s : 80: predicate.redundant_stop_gradient_eliminater 0.53% : 0.000003s : 26: predicate.remove_not_recompute_node 1.29% : 0.000006s : 53: predicate.replace_applicator 0.58% : 0.000003s : 26: predicate.replace_old_param 0.37% : 0.000002s : 13: predicate.reset_defer_inline 0.90% : 0.000004s : 27: predicate.reshape_eliminate 0.78% : 0.000004s : 26: predicate.row_tensor_add_zeros_like 0.59% : 0.000003s : 13: predicate.row_tensor_eliminate 1.00% : 0.000005s : 26: predicate.same_eliminate 0.51% : 0.000002s : 26: predicate.set_cell_output_no_recompute 1.08% : 0.000005s : 26: predicate.shard_identity_eliminate 0.87% : 0.000004s : 26: predicate.special_op_eliminate 0.83% : 0.000004s : 26: predicate.specialize_transform 1.30% : 0.000006s : 26: predicate.split_environ_get_set_with_tuple_value 0.96% : 0.000005s : 26: predicate.stack_unstack_eliminate 0.85% : 0.000004s : 13: predicate.switch_call_monad_eliminater 0.88% : 0.000004s : 28: predicate.switch_defer_inline 1.57% : 0.000007s : 54: predicate.switch_layer_defer_inline 3.47% : 0.000016s : 98: predicate.switch_simplify 0.96% : 0.000005s : 27: predicate.tile_eliminate 0.79% : 0.000004s : 27: predicate.transpose_eliminate 1.76% : 0.000008s : 53: predicate.tuple_list_convert_item_index_to_positive 1.67% : 0.000008s : 53: predicate.tuple_list_get_item_const_eliminator 1.58% : 0.000007s : 53: predicate.tuple_list_get_item_depend_reorder 3.24% : 0.000015s : 79: predicate.tuple_list_get_item_eliminator 1.60% : 0.000008s : 53: predicate.tuple_list_get_set_item_eliminator 2.53% : 0.000012s : 79: predicate.tuple_list_set_item_eliminator 1.68% : 0.000008s : 53: predicate.tuple_to_list_eliminator_ 2.25% : 0.000011s : 80: predicate.updatestate_pure_node_eliminater 3.27% : 0.000015s : 106: predicate.updatestate_useless_node_eliminater 0.42% : 0.000002s : 13: predicate.value_based_eliminate 0.82% : 0.000004s : 26: predicate.virtual_dataset_eliminate 0.81% : 0.000004s : 26: predicate.virtual_output_eliminate 0.42% : 0.000002s : 13: predicate.virtual_view_grad_eliminate 0.50% : 0.000002s : 13: predicate.zero_like_fill_zero ------[func_graph_cloner_run.] 0.017312 58 79.11% : 0.013695s : 55: func_graph_cloner_run.FuncGraphClonerGraph 20.89% : 0.003617s : 3: func_graph_cloner_run.FuncGraphSpecializer ------[meta_graph.] 0.000000 0 ------[manager.] 0.000000 0 ------[pynative] 0.000000 0 ------[others.] 2.972483 192 0.00% : 0.000004s : 1: ForceFp32Comm 0.24% : 0.007181s : 1: add_attr 0.24% : 0.007161s : 1: add_attr_with_inline 0.00% : 0.000004s : 1: add_comm_op_reuse_tag 0.00% : 0.000133s : 1: add_recomputation 0.00% : 0.000004s : 1: assign_add_opt 0.01% : 0.000393s : 1: auto_monad 0.00% : 0.000055s : 1: auto_monad_reorder 0.00% : 0.000003s : 1: begin_end_overlap_inline 0.00% : 0.000007s : 1: bias_add_comm_swap 0.03% : 0.000906s : 1: bootstrap 0.00% : 0.000053s : 1: cconv 0.00% : 0.000005s : 1: comm_op_add_attrs 0.00% : 0.000033s : 1: control_data_broadcast_order 0.00% : 0.000035s : 1: convert_after_rewriter 0.00% : 0.000059s : 1: cse_after_recomputation 0.00% : 0.000007s : 1: dataset_repeat_opt 0.00% : 0.000006s : 1: detach_backward 0.00% : 0.000025s : 1: environ_conv 0.00% : 0.000029s : 1: event_method 0.00% : 0.000006s : 1: full_micro_interleaved_order_control 0.00% : 0.000008s : 1: get_jit_bprop_graph 0.00% : 0.000011s : 1: graph_reusing 0.00% : 0.000005s : 1: grouped_pairwise_exchange_alltoall 0.00% : 0.000004s : 1: handle_group_info 0.00% : 0.000007s : 1: inline 0.00% : 0.000016s : 1: insert-virtual-dataset 0.00% : 0.000005s : 1: interleave_parallel_branches 0.00% : 0.000004s : 1: interleave_split_concat_branches 0.00% : 0.000006s : 1: label_fine_grained_interleaved_index 0.00% : 0.000010s : 1: label_micro_interleaved_index 0.02% : 0.000705s : 1: loop_unroll 0.00% : 0.000004s : 1: merge_cast_opt 0.00% : 0.000005s : 1: micro_interleaved_order_control 0.03% : 0.000881s : 1: mutable_eliminate 0.00% : 0.000011s : 1: offloading_packed_experts 0.00% : 0.000039s : 1: opt.transform.loop_unroll_optimizer 0.00% : 0.000040s : 1: opt.transform.mutable_eliminate 0.09% : 0.002614s : 78: opt.transform.opt_a 4.36% : 0.129469s : 1: opt.transform.opt_after_cconv 0.00% : 0.000068s : 1: opt.transform.opt_after_jit_grad 0.01% : 0.000363s : 28: opt.transform.opt_b 0.00% : 0.000143s : 2: opt.transform.opt_trans_graph 0.00% : 0.000104s : 4: opt.transform.symbol_engine_opt 0.52% : 0.015495s : 1: opt_a 4.36% : 0.129736s : 1: opt_after_cconv 0.03% : 0.000886s : 1: opt_after_jit_grad 0.02% : 0.000591s : 1: opt_b 5.06% : 0.150305s : 1: optimize 0.00% : 0.000045s : 1: optimize_parallel_all_gather_comm 0.00% : 0.000017s : 1: order_py_execute_after_rewriter 0.00% : 0.000047s : 1: overlap_grad_flash_sp 0.00% : 0.000005s : 1: overlap_grad_matmul_and_grad_allreduce 0.00% : 0.000012s : 1: overlap_grad_ring_attention 0.00% : 0.000005s : 1: overlap_opt_shard_grad_in_pipeline 0.00% : 0.000005s : 1: overlap_opt_shard_in_pipeline 0.00% : 0.000007s : 1: overlap_param_gather 0.00% : 0.000006s : 1: overlap_recompute_allgather_and_fa_grad 0.00% : 0.000011s : 1: overlap_recompute_and_grad_model_parallel 0.00% : 0.000005s : 1: overlap_recompute_comm 0.00% : 0.000009s : 1: parallel-infer-symbol 0.00% : 0.000007s : 1: parallel-infer-symbol-second 0.00% : 0.000006s : 1: partial_unused_args_eliminate 0.00% : 0.000005s : 1: pipeline_parallel_scheduler 0.00% : 0.000006s : 1: pipeline_split 7.82% : 0.232487s : 1: pre_auto_parallel 0.00% : 0.000061s : 1: py_interpret_to_execute 0.00% : 0.000036s : 1: py_interpret_to_execute_after_opt_a 0.00% : 0.000004s : 1: remove_cast_before_assign_add 0.00% : 0.000089s : 1: remove_dup_value 0.24% : 0.007153s : 1: renormalize.infer 0.13% : 0.003977s : 1: renormalize.specialize 0.00% : 0.000006s : 1: reorder_send_recv_between_fp_bp 0.00% : 0.000011s : 1: rewriter_after_jit_bprop_graph 0.05% : 0.001506s : 1: rewriter_after_opt_a 0.00% : 0.000103s : 1: rewriter_before_opt_a 0.00% : 0.000006s : 1: slice_cell_reuse_recomputed_activation 0.00% : 0.000005s : 1: slice_recompute_activation 0.00% : 0.000006s : 1: split_layernorm_comm 0.00% : 0.000005s : 1: split_matmul_comm_elemetwise 0.00% : 0.000016s : 1: swap_dp_allreduce_reducescatter 0.01% : 0.000173s : 1: symbol_engine_optimizer 0.01% : 0.000187s : 1: tuple_transform 76.66% : 2.278811s : 1: type_inference . [hook] pytest_runtest_teardown:test_matmul_qkv_bf16_0[False-mstype0-k_n_shape2-1] tests/st/infer/ops/test_internal_ops/test_matmul_split.py::test_matmul_qkv_bf16_0[False-mstype0-k_n_shape2-1],max_mem:196.0M [WARNING] ME(152078:281472851701552,MainProcess):2026-01-29-17:40:29.439.957 [mindspore/context.py:1334] For 'context.set_context', the parameter 'device_target' will be deprecated and removed in a future version. Please use the api mindspore.set_device() instead. TotalTime = 2.89203, [21] [bootstrap]: 0.00074112 [type_inference]: 2.76067 [event_method]: 2.124e-05 [auto_monad]: 0.00043242 [graph_reusing]: 8.58001e-06 [inline]: 3.12002e-06 [add_attr]: 0.0069262, [1] [add_attr_with_inline]: 0.00690778, [1] [Cycle 1]: 7.328e-05, [2] [tag_attr]: 2.315e-05 [meta_addattr_fg_expand]: 4.29002e-06 [parallel-infer-symbol]: 5.02999e-06 [pre_auto_parallel]: 4.2e-05 [insert-virtual-dataset]: 2.86999e-06 [parallel-infer-symbol-second]: 7.30011e-07 [dataset_repeat_opt]: 1.92001e-06 [pipeline_split]: 1.76e-06 [optimize]: 0.12205, [53] [py_interpret_to_execute]: 2.952e-05 [rewriter_before_opt_a]: 8.572e-05 [opt_a]: 0.118196, [2] [Cycle 1]: 0.115487, [45] [expand_dump_flag]: 3.16001e-06 [switch_simplify]: 3.43e-05 [loop_unroll]: 1.931e-05 [a_1]: 0.00103768 [with_stream_mark]: 3.131e-05 [recompute_prepare]: 2.774e-05 [updatestate_depend_eliminate]: 1.226e-05 [updatestate_assign_eliminate]: 8.22998e-06 [updatestate_loads_eliminate]: 1.026e-05 [parameter_eliminate]: 2.01e-06 [a_2]: 0.0002688 [accelerated_algorithm]: 4.706e-05 [shard]: 2.64999e-06 [meta_shard_fg_expand]: 4.15999e-06 [shard_inline]: 1.679e-05 [merge_send_recv]: 1.495e-05 [auto_parallel]: 1.259e-05 [parallel]: 3.016e-05 [flash_sp]: 1.226e-05 [merge_comm]: 9.20999e-06 [allreduce_fusion]: 8.17e-06 [matmul_add_comm_reduction]: 1.8e-05 [allreduce_slice_to_reducescatter]: 8.09989e-07 [virtual_shard_identity]: 2.07e-05 [virtual_dataset]: 1.763e-05 [get_grad_eliminate_]: 1.959e-05 [virtual_output]: 1.791e-05 [merge_forward]: 9.22999e-06 [cell_reuse_recompute_pass]: 1.99e-06 [offload_activation]: 2.003e-05 [cell_reuse_handle_not_recompute_node_pass]: 3.482e-05 [merge_recompute_call_nodes]: 1.82001e-06 [before_grad]: 3.096e-05 [set_forward_comm_id_for_comm_node_pass]: 1.022e-05 [meta_fg_expand]: 6.09001e-06 [flash_sp_send_recv_attached]: 6.69999e-06 [receive_attached]: 2.19001e-06 [after_resolve]: 3.012e-05 [a_after_grad]: 3.031e-05 [renormalize]: 0.112888 [add_forward_monad_depend]: 1.712e-05 [auto_monad_grad]: 2.57001e-06 [auto_monad_eliminator]: 6.352e-05 [cse]: 0.00015493 [a_3]: 0.00013938 [Cycle 2]: 0.00269388, [45] [expand_dump_flag]: 2.53998e-06 [switch_simplify]: 1.988e-05 [loop_unroll]: 1.631e-05 [a_1]: 0.00046202 [with_stream_mark]: 2.531e-05 [recompute_prepare]: 1.793e-05 [updatestate_depend_eliminate]: 9.66e-06 [updatestate_assign_eliminate]: 8e-06 [updatestate_loads_eliminate]: 1.056e-05 [parameter_eliminate]: 2.43e-06 [a_2]: 0.00023925 [accelerated_algorithm]: 2.287e-05 [shard]: 3.38e-06 [meta_shard_fg_expand]: 4.15e-06 [shard_inline]: 1.572e-05 [merge_send_recv]: 1.596e-05 [auto_parallel]: 1.628e-05 [parallel]: 1.111e-05 [flash_sp]: 4.55001e-06 [merge_comm]: 9.34e-06 [allreduce_fusion]: 9.72999e-06 [matmul_add_comm_reduction]: 1.766e-05 [allreduce_slice_to_reducescatter]: 9.50007e-07 [virtual_shard_identity]: 1.79e-05 [virtual_dataset]: 1.6e-05 [get_grad_eliminate_]: 1.551e-05 [virtual_output]: 1.531e-05 [merge_forward]: 9.17001e-06 [cell_reuse_recompute_pass]: 3.13998e-06 [offload_activation]: 1.71e-05 [cell_reuse_handle_not_recompute_node_pass]: 2.92e-05 [merge_recompute_call_nodes]: 1.57999e-06 [before_grad]: 2.64e-05 [set_forward_comm_id_for_comm_node_pass]: 9.71e-06 [meta_fg_expand]: 6.62002e-06 [flash_sp_send_recv_attached]: 1.79998e-06 [receive_attached]: 2.66e-06 [after_resolve]: 2.123e-05 [a_after_grad]: 2.557e-05 [renormalize]: 6.09987e-07 [add_forward_monad_depend]: 3.13e-06 [auto_monad_grad]: 1.84e-06 [auto_monad_eliminator]: 3.651e-05 [cse]: 0.00110924 [a_3]: 0.00012019 [py_interpret_to_execute_after_opt_a]: 2.925e-05 [slice_cell_reuse_recomputed_activation]: 3.03998e-06 [rewriter_after_opt_a]: 0.00020522 [convert_after_rewriter]: 1.442e-05 [order_py_execute_after_rewriter]: 1.043e-05 [mutable_eliminate]: 0.00078286 [opt_b]: 0.00054323, [1] [Cycle 1]: 0.00053416, [7] [b_1]: 0.0003742 [b_2]: 1.827e-05 [updatestate_depend_eliminate]: 1.29e-05 [updatestate_assign_eliminate]: 8.41002e-06 [updatestate_loads_eliminate]: 9.82999e-06 [renormalize]: 6.50005e-07 [cse]: 6.927e-05 [optimize_parallel_all_gather_comm]: 3.067e-05 [overlap_param_gather]: 2.03002e-06 [cconv]: 3.788e-05 [loop_unroll]: 0.00055139 [opt_after_cconv]: 0.00026662, [1] [Cycle 1]: 0.00025824, [7] [c_1]: 0.00010299 [parameter_eliminate]: 4.35999e-06 [updatestate_depend_eliminate]: 1.335e-05 [updatestate_assign_eliminate]: 7.02002e-06 [updatestate_loads_eliminate]: 8.18001e-06 [cse]: 7.919e-05 [renormalize]: 3.69997e-07 [remove_dup_value]: 7.576e-05 [tuple_transform]: 0.00037211, [1] [Cycle 1]: 0.00036505, [4] [d_1]: 0.00011926 [none_parameter_eliminate]: 2.37999e-06 [renormalize]: 2.40019e-07 [switch_simplify]: 0.00021356 [partial_unused_args_eliminate]: 3.56999e-06 [add_recomputation]: 0.00012346 [cse_after_recomputation]: 6.825e-05, [1] [Cycle 1]: 6.203e-05, [1] [cse]: 5.237e-05 [environ_conv]: 2.013e-05 [swap_dp_allreduce_reducescatter]: 1.248e-05 [bias_add_comm_swap]: 3.45e-06 [label_micro_interleaved_index]: 4.84e-06 [label_fine_grained_interleaved_index]: 2.96001e-06 [merge_cast_opt]: 1.47999e-06 [slice_recompute_activation]: 2.32999e-06 [micro_interleaved_order_control]: 2.68998e-06 [assign_add_opt]: 1.56002e-06 [ForceFp32Comm]: 1.09e-06 [remove_cast_before_assign_add]: 1.42999e-06 [full_micro_interleaved_order_control]: 2.21998e-06 [reorder_send_recv_between_fp_bp]: 3.45e-06 [comm_op_add_attrs]: 1.22999e-06 [add_comm_op_reuse_tag]: 1.27999e-06 [interleave_split_concat_branches]: 1.34e-06 [interleave_parallel_branches]: 1.09e-06 [overlap_opt_shard_in_pipeline]: 1.24998e-06 [overlap_opt_shard_grad_in_pipeline]: 1.71e-06 [control_data_broadcast_order]: 2.969e-05 [grouped_pairwise_exchange_alltoall]: 1.97999e-06 [offloading_packed_experts]: 6.58e-06 [overlap_recompute_and_grad_model_parallel]: 7.41001e-06 [overlap_grad_matmul_and_grad_allreduce]: 1.37e-06 [overlap_recompute_allgather_and_fa_grad]: 1.44998e-06 [overlap_recompute_comm]: 2.37999e-06 [overlap_grad_ring_attention]: 7.8e-06 [overlap_grad_flash_sp]: 3.911e-05 [begin_end_overlap_inline]: 4.99975e-07 [split_matmul_comm_elemetwise]: 2.56e-06 [split_layernorm_comm]: 2.43e-06 [handle_group_info]: 1.00001e-06 [symbol_engine_optimizer]: 0.00016794, [1] [Cycle 1]: 0.00016075, [6] [build]: 1.779e-05 [elim_shapecalc]: 2.604e-05 [elim_not_effective]: 3.218e-05 [opt_reshape]: 2.31e-05 [fold_const_symbol]: 2.433e-05 [renormalize]: 3.4002e-07 [detach_backward]: 2.16e-06 [pipeline_parallel_scheduler]: 1.57001e-06 [auto_monad_reorder]: 4.821e-05 [get_jit_bprop_graph]: 2.41e-06 [rewriter_after_jit_bprop_graph]: 6.61e-06 [opt_after_jit_grad]: 0.00069822 [validate]: 9.686e-05 Sums bootstrap : 0.000741s : 0.03% type_inference : 2.760666s : 95.73% event_method : 0.000021s : 0.00% auto_monad : 0.000432s : 0.01% graph_reusing : 0.000009s : 0.00% inline : 0.000003s : 0.00% add_attr.add_attr_with_inline.tag_attr : 0.000023s : 0.00% add_attr.add_attr_with_inline.meta_addattr_fg_expand : 0.000004s : 0.00% parallel-infer-symbol : 0.000005s : 0.00% pre_auto_parallel : 0.000042s : 0.00% insert-virtual-dataset : 0.000003s : 0.00% parallel-infer-symbol-second : 0.000001s : 0.00% dataset_repeat_opt : 0.000002s : 0.00% pipeline_split : 0.000002s : 0.00% optimize.py_interpret_to_execute : 0.000030s : 0.00% optimize.rewriter_before_opt_a : 0.000086s : 0.00% optimize.opt_a.expand_dump_flag : 0.000006s : 0.00% optimize.opt_a.switch_simplify : 0.000054s : 0.00% optimize.opt_a.loop_unroll : 0.000036s : 0.00% optimize.opt_a.a_1 : 0.001500s : 0.05% optimize.opt_a.with_stream_mark : 0.000057s : 0.00% optimize.opt_a.recompute_prepare : 0.000046s : 0.00% optimize.opt_a.updatestate_depend_eliminate : 0.000022s : 0.00% optimize.opt_a.updatestate_assign_eliminate : 0.000016s : 0.00% optimize.opt_a.updatestate_loads_eliminate : 0.000021s : 0.00% optimize.opt_a.parameter_eliminate : 0.000004s : 0.00% optimize.opt_a.a_2 : 0.000508s : 0.02% optimize.opt_a.accelerated_algorithm : 0.000070s : 0.00% optimize.opt_a.shard : 0.000006s : 0.00% optimize.opt_a.meta_shard_fg_expand : 0.000008s : 0.00% optimize.opt_a.shard_inline : 0.000033s : 0.00% optimize.opt_a.merge_send_recv : 0.000031s : 0.00% optimize.opt_a.auto_parallel : 0.000029s : 0.00% optimize.opt_a.parallel : 0.000041s : 0.00% optimize.opt_a.flash_sp : 0.000017s : 0.00% optimize.opt_a.merge_comm : 0.000019s : 0.00% optimize.opt_a.allreduce_fusion : 0.000018s : 0.00% optimize.opt_a.matmul_add_comm_reduction : 0.000036s : 0.00% optimize.opt_a.allreduce_slice_to_reducescatter : 0.000002s : 0.00% optimize.opt_a.virtual_shard_identity : 0.000039s : 0.00% optimize.opt_a.virtual_dataset : 0.000034s : 0.00% optimize.opt_a.get_grad_eliminate_ : 0.000035s : 0.00% optimize.opt_a.virtual_output : 0.000033s : 0.00% optimize.opt_a.merge_forward : 0.000018s : 0.00% optimize.opt_a.cell_reuse_recompute_pass : 0.000005s : 0.00% optimize.opt_a.offload_activation : 0.000037s : 0.00% optimize.opt_a.cell_reuse_handle_not_recompute_node_pass : 0.000064s : 0.00% optimize.opt_a.merge_recompute_call_nodes : 0.000003s : 0.00% optimize.opt_a.before_grad : 0.000057s : 0.00% optimize.opt_a.set_forward_comm_id_for_comm_node_pass : 0.000020s : 0.00% optimize.opt_a.meta_fg_expand : 0.000013s : 0.00% optimize.opt_a.flash_sp_send_recv_attached : 0.000008s : 0.00% optimize.opt_a.receive_attached : 0.000005s : 0.00% optimize.opt_a.after_resolve : 0.000051s : 0.00% optimize.opt_a.a_after_grad : 0.000056s : 0.00% optimize.opt_a.renormalize : 0.112889s : 3.91% optimize.opt_a.add_forward_monad_depend : 0.000020s : 0.00% optimize.opt_a.auto_monad_grad : 0.000004s : 0.00% optimize.opt_a.auto_monad_eliminator : 0.000100s : 0.00% optimize.opt_a.cse : 0.001264s : 0.04% optimize.opt_a.a_3 : 0.000260s : 0.01% optimize.py_interpret_to_execute_after_opt_a : 0.000029s : 0.00% optimize.slice_cell_reuse_recomputed_activation : 0.000003s : 0.00% optimize.rewriter_after_opt_a : 0.000205s : 0.01% optimize.convert_after_rewriter : 0.000014s : 0.00% optimize.order_py_execute_after_rewriter : 0.000010s : 0.00% optimize.mutable_eliminate : 0.000783s : 0.03% optimize.opt_b.b_1 : 0.000374s : 0.01% optimize.opt_b.b_2 : 0.000018s : 0.00% optimize.opt_b.updatestate_depend_eliminate : 0.000013s : 0.00% optimize.opt_b.updatestate_assign_eliminate : 0.000008s : 0.00% optimize.opt_b.updatestate_loads_eliminate : 0.000010s : 0.00% optimize.opt_b.renormalize : 0.000001s : 0.00% optimize.opt_b.cse : 0.000069s : 0.00% optimize.optimize_parallel_all_gather_comm : 0.000031s : 0.00% optimize.overlap_param_gather : 0.000002s : 0.00% optimize.cconv : 0.000038s : 0.00% optimize.loop_unroll : 0.000551s : 0.02% optimize.opt_after_cconv.c_1 : 0.000103s : 0.00% optimize.opt_after_cconv.parameter_eliminate : 0.000004s : 0.00% optimize.opt_after_cconv.updatestate_depend_eliminate : 0.000013s : 0.00% optimize.opt_after_cconv.updatestate_assign_eliminate : 0.000007s : 0.00% optimize.opt_after_cconv.updatestate_loads_eliminate : 0.000008s : 0.00% optimize.opt_after_cconv.cse : 0.000079s : 0.00% optimize.opt_after_cconv.renormalize : 0.000000s : 0.00% optimize.remove_dup_value : 0.000076s : 0.00% optimize.tuple_transform.d_1 : 0.000119s : 0.00% optimize.tuple_transform.none_parameter_eliminate : 0.000002s : 0.00% optimize.tuple_transform.renormalize : 0.000000s : 0.00% optimize.tuple_transform.switch_simplify : 0.000214s : 0.01% optimize.partial_unused_args_eliminate : 0.000004s : 0.00% optimize.add_recomputation : 0.000123s : 0.00% optimize.cse_after_recomputation.cse : 0.000052s : 0.00% optimize.environ_conv : 0.000020s : 0.00% optimize.swap_dp_allreduce_reducescatter : 0.000012s : 0.00% optimize.bias_add_comm_swap : 0.000003s : 0.00% optimize.label_micro_interleaved_index : 0.000005s : 0.00% optimize.label_fine_grained_interleaved_index : 0.000003s : 0.00% optimize.merge_cast_opt : 0.000001s : 0.00% optimize.slice_recompute_activation : 0.000002s : 0.00% optimize.micro_interleaved_order_control : 0.000003s : 0.00% optimize.assign_add_opt : 0.000002s : 0.00% optimize.ForceFp32Comm : 0.000001s : 0.00% optimize.remove_cast_before_assign_add : 0.000001s : 0.00% optimize.full_micro_interleaved_order_control : 0.000002s : 0.00% optimize.reorder_send_recv_between_fp_bp : 0.000003s : 0.00% optimize.comm_op_add_attrs : 0.000001s : 0.00% optimize.add_comm_op_reuse_tag : 0.000001s : 0.00% optimize.interleave_split_concat_branches : 0.000001s : 0.00% optimize.interleave_parallel_branches : 0.000001s : 0.00% optimize.overlap_opt_shard_in_pipeline : 0.000001s : 0.00% optimize.overlap_opt_shard_grad_in_pipeline : 0.000002s : 0.00% optimize.control_data_broadcast_order : 0.000030s : 0.00% optimize.grouped_pairwise_exchange_alltoall : 0.000002s : 0.00% optimize.offloading_packed_experts : 0.000007s : 0.00% optimize.overlap_recompute_and_grad_model_parallel : 0.000007s : 0.00% optimize.overlap_grad_matmul_and_grad_allreduce : 0.000001s : 0.00% optimize.overlap_recompute_allgather_and_fa_grad : 0.000001s : 0.00% optimize.overlap_recompute_comm : 0.000002s : 0.00% optimize.overlap_grad_ring_attention : 0.000008s : 0.00% optimize.overlap_grad_flash_sp : 0.000039s : 0.00% optimize.begin_end_overlap_inline : 0.000000s : 0.00% optimize.split_matmul_comm_elemetwise : 0.000003s : 0.00% optimize.split_layernorm_comm : 0.000002s : 0.00% optimize.handle_group_info : 0.000001s : 0.00% optimize.symbol_engine_optimizer.build : 0.000018s : 0.00% optimize.symbol_engine_optimizer.elim_shapecalc : 0.000026s : 0.00% optimize.symbol_engine_optimizer.elim_not_effective : 0.000032s : 0.00% optimize.symbol_engine_optimizer.opt_reshape : 0.000023s : 0.00% optimize.symbol_engine_optimizer.fold_const_symbol : 0.000024s : 0.00% optimize.symbol_engine_optimizer.renormalize : 0.000000s : 0.00% detach_backward : 0.000002s : 0.00% pipeline_parallel_scheduler : 0.000002s : 0.00% auto_monad_reorder : 0.000048s : 0.00% get_jit_bprop_graph : 0.000002s : 0.00% rewriter_after_jit_bprop_graph : 0.000007s : 0.00% opt_after_jit_grad : 0.000698s : 0.02% validate : 0.000097s : 0.00% Time group info: ------[substitution.] 0.000449 96 2.69% : 0.000012s : 2: substitution.depend_value_elim 0.82% : 0.000004s : 8: substitution.elim_not_effective 0.68% : 0.000003s : 8: substitution.fold_const_symbol 2.69% : 0.000012s : 13: substitution.graph_param_transform 63.45% : 0.000285s : 1: substitution.inline 1.96% : 0.000009s : 16: substitution.j_node_and_user_rematch 5.57% : 0.000025s : 2: substitution.less_batch_normalization 1.34% : 0.000006s : 6: substitution.load_eliminater 0.69% : 0.000003s : 2: substitution.opt_reshape 2.74% : 0.000012s : 16: substitution.remove_not_recompute_node 1.59% : 0.000007s : 4: substitution.replace_old_param 7.26% : 0.000033s : 4: substitution.reshape_eliminate 4.23% : 0.000019s : 6: substitution.updatestate_pure_node_eliminater 4.31% : 0.000019s : 8: substitution.updatestate_useless_node_eliminater ------[type_inference.] 2.760554 2 99.79% : 2.754850s : 1: type_inference.infer 0.21% : 0.005704s : 1: type_inference.specialize ------[replace.] 0.000018 1 100.00% : 0.000018s : 1: replace.inline ------[match.] 0.000284 1 100.00% : 0.000284s : 1: match.inline ------[predicate.] 0.000449 3002 0.77% : 0.000003s : 27: predicate.accumulaten_eliminater 0.92% : 0.000004s : 13: predicate.ad_related_special_op_eliminate 0.82% : 0.000004s : 26: predicate.addn_check_dump 0.77% : 0.000003s : 27: predicate.addn_zero_filter 0.71% : 0.000003s : 27: predicate.adjust_all_reduce_mul_add 2.17% : 0.000010s : 53: predicate.arithmetic_simplify 0.84% : 0.000004s : 27: predicate.cast_eliminate 0.90% : 0.000004s : 26: predicate.check_bprop_eliminate 0.79% : 0.000004s : 26: predicate.compare_switch_simplify 0.26% : 0.000001s : 13: predicate.const_output_eliminate 0.88% : 0.000004s : 26: predicate.depend_value_elim 0.85% : 0.000004s : 27: predicate.dict_get_item_const_eliminator 0.90% : 0.000004s : 27: predicate.dict_get_item_eliminator 0.77% : 0.000003s : 27: predicate.dict_set_item_eliminator 1.19% : 0.000005s : 26: predicate.dumpgradient_eliminate 0.36% : 0.000002s : 13: predicate.elim_not_effective 0.57% : 0.000003s : 13: predicate.elim_shapecalc_of_broadcastargs 1.17% : 0.000005s : 40: predicate.environ_add_const_eliminate 1.20% : 0.000005s : 40: predicate.environ_get_add_eliminate 1.07% : 0.000005s : 40: predicate.environ_get_depend_swap 1.98% : 0.000009s : 66: predicate.environ_get_eliminate 1.09% : 0.000005s : 40: predicate.environ_get_set_eliminate 0.76% : 0.000003s : 28: predicate.exchange_switch_depend_value 1.40% : 0.000006s : 28: predicate.float_depend_g_call 0.76% : 0.000003s : 26: predicate.float_environ_get_switch 1.22% : 0.000005s : 39: predicate.float_tuple_getitem_switch 0.23% : 0.000001s : 13: predicate.fold_const_symbol 0.89% : 0.000004s : 26: predicate.get_grad_eliminate 0.32% : 0.000001s : 13: predicate.graph_param_transform 0.80% : 0.000004s : 26: predicate.incorporate_call 0.69% : 0.000003s : 26: predicate.incorporate_call_switch 5.38% : 0.000024s : 133: predicate.inline 1.19% : 0.000005s : 26: predicate.inline_without_move 0.44% : 0.000002s : 26: predicate.j_node_and_user_rematch 1.34% : 0.000006s : 27: predicate.less_batch_normalization 1.71% : 0.000008s : 53: predicate.list_to_tuple_eliminator_ 2.27% : 0.000010s : 80: predicate.load_eliminater 0.98% : 0.000004s : 13: predicate.loop_unroll_after_grad 0.97% : 0.000004s : 31: predicate.loop_unroll_before_grad 1.73% : 0.000008s : 53: predicate.make_slice_get_slice_eliminator 0.78% : 0.000003s : 26: predicate.merge_addn 0.78% : 0.000004s : 26: predicate.micro_step_allgather_replace 0.84% : 0.000004s : 26: predicate.mini_step_allgather_replace 0.73% : 0.000003s : 27: predicate.minmaximum_grad 1.08% : 0.000005s : 13: predicate.mutable_eliminate 0.60% : 0.000003s : 13: predicate.opt_reshape 0.45% : 0.000002s : 13: predicate.parallel_virtual_node 1.10% : 0.000005s : 28: predicate.partial_defer_inline 1.30% : 0.000006s : 40: predicate.partial_eliminate 0.82% : 0.000004s : 27: predicate.print_const_string_wrapper 0.82% : 0.000004s : 26: predicate.reduce_all_const_elim 1.18% : 0.000005s : 27: predicate.reduce_eliminate 2.27% : 0.000010s : 80: predicate.redundant_stop_gradient_eliminater 0.51% : 0.000002s : 26: predicate.remove_not_recompute_node 1.50% : 0.000007s : 53: predicate.replace_applicator 0.49% : 0.000002s : 26: predicate.replace_old_param 0.35% : 0.000002s : 13: predicate.reset_defer_inline 0.98% : 0.000004s : 27: predicate.reshape_eliminate 0.86% : 0.000004s : 26: predicate.row_tensor_add_zeros_like 0.50% : 0.000002s : 13: predicate.row_tensor_eliminate 1.25% : 0.000006s : 26: predicate.same_eliminate 0.59% : 0.000003s : 26: predicate.set_cell_output_no_recompute 1.02% : 0.000005s : 26: predicate.shard_identity_eliminate 0.91% : 0.000004s : 26: predicate.special_op_eliminate 0.88% : 0.000004s : 26: predicate.specialize_transform 1.24% : 0.000006s : 26: predicate.split_environ_get_set_with_tuple_value 0.94% : 0.000004s : 26: predicate.stack_unstack_eliminate 0.48% : 0.000002s : 13: predicate.switch_call_monad_eliminater 0.96% : 0.000004s : 28: predicate.switch_defer_inline 1.60% : 0.000007s : 54: predicate.switch_layer_defer_inline 3.44% : 0.000015s : 98: predicate.switch_simplify 0.83% : 0.000004s : 27: predicate.tile_eliminate 0.80% : 0.000004s : 27: predicate.transpose_eliminate 1.66% : 0.000007s : 53: predicate.tuple_list_convert_item_index_to_positive 1.82% : 0.000008s : 53: predicate.tuple_list_get_item_const_eliminator 1.68% : 0.000008s : 53: predicate.tuple_list_get_item_depend_reorder 3.10% : 0.000014s : 79: predicate.tuple_list_get_item_eliminator 1.71% : 0.000008s : 53: predicate.tuple_list_get_set_item_eliminator 2.61% : 0.000012s : 79: predicate.tuple_list_set_item_eliminator 1.68% : 0.000008s : 53: predicate.tuple_to_list_eliminator_ 2.25% : 0.000010s : 80: predicate.updatestate_pure_node_eliminater 3.30% : 0.000015s : 106: predicate.updatestate_useless_node_eliminater 0.50% : 0.000002s : 13: predicate.value_based_eliminate 0.86% : 0.000004s : 26: predicate.virtual_dataset_eliminate 0.96% : 0.000004s : 26: predicate.virtual_output_eliminate 0.41% : 0.000002s : 13: predicate.virtual_view_grad_eliminate 0.52% : 0.000002s : 13: predicate.zero_like_fill_zero ------[func_graph_cloner_run.] 0.481762 58 78.05% : 0.376017s : 55: func_graph_cloner_run.FuncGraphClonerGraph 21.95% : 0.105745s : 3: func_graph_cloner_run.FuncGraphSpecializer ------[meta_graph.] 0.000000 0 ------[manager.] 0.000000 0 ------[pynative] 0.000000 0 ------[others.] 3.137402 192 0.00% : 0.000004s : 1: ForceFp32Comm 0.22% : 0.006933s : 1: add_attr 0.22% : 0.006913s : 1: add_attr_with_inline 0.00% : 0.000004s : 1: add_comm_op_reuse_tag 0.00% : 0.000130s : 1: add_recomputation 0.00% : 0.000004s : 1: assign_add_opt 0.01% : 0.000453s : 1: auto_monad 0.00% : 0.000055s : 1: auto_monad_reorder 0.00% : 0.000003s : 1: begin_end_overlap_inline 0.00% : 0.000006s : 1: bias_add_comm_swap 0.03% : 0.000824s : 1: bootstrap 0.00% : 0.000042s : 1: cconv 0.00% : 0.000005s : 1: comm_op_add_attrs 0.00% : 0.000033s : 1: control_data_broadcast_order 0.00% : 0.000019s : 1: convert_after_rewriter 0.00% : 0.000072s : 1: cse_after_recomputation 0.00% : 0.000005s : 1: dataset_repeat_opt 0.00% : 0.000006s : 1: detach_backward 0.00% : 0.000025s : 1: environ_conv 0.00% : 0.000030s : 1: event_method 0.00% : 0.000005s : 1: full_micro_interleaved_order_control 0.00% : 0.000007s : 1: get_jit_bprop_graph 0.00% : 0.000015s : 1: graph_reusing 0.00% : 0.000005s : 1: grouped_pairwise_exchange_alltoall 0.00% : 0.000004s : 1: handle_group_info 0.00% : 0.000006s : 1: inline 0.00% : 0.000007s : 1: insert-virtual-dataset 0.00% : 0.000005s : 1: interleave_parallel_branches 0.00% : 0.000005s : 1: interleave_split_concat_branches 0.00% : 0.000006s : 1: label_fine_grained_interleaved_index 0.00% : 0.000008s : 1: label_micro_interleaved_index 0.02% : 0.000562s : 1: loop_unroll 0.00% : 0.000004s : 1: merge_cast_opt 0.00% : 0.000006s : 1: micro_interleaved_order_control 0.03% : 0.000793s : 1: mutable_eliminate 0.00% : 0.000010s : 1: offloading_packed_experts 0.00% : 0.000030s : 1: opt.transform.loop_unroll_optimizer 0.00% : 0.000032s : 1: opt.transform.mutable_eliminate 0.09% : 0.002745s : 78: opt.transform.opt_a 0.00% : 0.000101s : 1: opt.transform.opt_after_cconv 0.00% : 0.000060s : 1: opt.transform.opt_after_jit_grad 0.01% : 0.000356s : 28: opt.transform.opt_b 0.01% : 0.000330s : 2: opt.transform.opt_trans_graph 0.00% : 0.000101s : 4: opt.transform.symbol_engine_opt 3.77% : 0.118200s : 1: opt_a 0.01% : 0.000271s : 1: opt_after_cconv 0.02% : 0.000712s : 1: opt_after_jit_grad 0.02% : 0.000547s : 1: opt_b 3.89% : 0.122057s : 1: optimize 0.00% : 0.000035s : 1: optimize_parallel_all_gather_comm 0.00% : 0.000014s : 1: order_py_execute_after_rewriter 0.00% : 0.000043s : 1: overlap_grad_flash_sp 0.00% : 0.000004s : 1: overlap_grad_matmul_and_grad_allreduce 0.00% : 0.000012s : 1: overlap_grad_ring_attention 0.00% : 0.000005s : 1: overlap_opt_shard_grad_in_pipeline 0.00% : 0.000004s : 1: overlap_opt_shard_in_pipeline 0.00% : 0.000006s : 1: overlap_param_gather 0.00% : 0.000006s : 1: overlap_recompute_allgather_and_fa_grad 0.00% : 0.000010s : 1: overlap_recompute_and_grad_model_parallel 0.00% : 0.000005s : 1: overlap_recompute_comm 0.00% : 0.000009s : 1: parallel-infer-symbol 0.00% : 0.000004s : 1: parallel-infer-symbol-second 0.00% : 0.000008s : 1: partial_unused_args_eliminate 0.00% : 0.000005s : 1: pipeline_parallel_scheduler 0.00% : 0.000006s : 1: pipeline_split 0.00% : 0.000047s : 1: pre_auto_parallel 0.00% : 0.000034s : 1: py_interpret_to_execute 0.00% : 0.000033s : 1: py_interpret_to_execute_after_opt_a 0.00% : 0.000005s : 1: remove_cast_before_assign_add 0.00% : 0.000080s : 1: remove_dup_value 0.21% : 0.006519s : 1: renormalize.infer 3.39% : 0.106348s : 1: renormalize.specialize 0.00% : 0.000006s : 1: reorder_send_recv_between_fp_bp 0.00% : 0.000010s : 1: rewriter_after_jit_bprop_graph 0.01% : 0.000212s : 1: rewriter_after_opt_a 0.00% : 0.000090s : 1: rewriter_before_opt_a 0.00% : 0.000006s : 1: slice_cell_reuse_recomputed_activation 0.00% : 0.000005s : 1: slice_recompute_activation 0.00% : 0.000006s : 1: split_layernorm_comm 0.00% : 0.000005s : 1: split_matmul_comm_elemetwise 0.00% : 0.000016s : 1: swap_dp_allreduce_reducescatter 0.01% : 0.000171s : 1: symbol_engine_optimizer 0.01% : 0.000376s : 1: tuple_transform 87.99% : 2.760697s : 1: type_inference . [hook] pytest_runtest_teardown:test_matmul_qkv_bf16_0[False-mstype0-k_n_shape2-16] tests/st/infer/ops/test_internal_ops/test_matmul_split.py::test_matmul_qkv_bf16_0[False-mstype0-k_n_shape2-16],max_mem:196.0M [WARNING] ME(152078:281472851701552,MainProcess):2026-01-29-17:41:26.281.126 [mindspore/context.py:1334] For 'context.set_context', the parameter 'device_target' will be deprecated and removed in a future version. Please use the api mindspore.set_device() instead. TotalTime = 2.63063, [21] [bootstrap]: 0.0006863 [type_inference]: 2.26302 [event_method]: 1.996e-05 [auto_monad]: 0.00040224 [graph_reusing]: 7.09001e-06 [inline]: 3.25e-06 [add_attr]: 0.164257, [1] [add_attr_with_inline]: 0.16424, [1] [Cycle 1]: 6.818e-05, [2] [tag_attr]: 2.176e-05 [meta_addattr_fg_expand]: 4.58999e-06 [parallel-infer-symbol]: 4.90001e-06 [pre_auto_parallel]: 3.932e-05 [insert-virtual-dataset]: 2.59001e-06 [parallel-infer-symbol-second]: 7.29982e-07 [dataset_repeat_opt]: 1.81998e-06 [pipeline_split]: 1.50001e-06 [optimize]: 0.0205421, [53] [py_interpret_to_execute]: 3.455e-05 [rewriter_before_opt_a]: 8.291e-05 [opt_a]: 0.0164373, [2] [Cycle 1]: 0.0135274, [45] [expand_dump_flag]: 2.89999e-06 [switch_simplify]: 3.403e-05 [loop_unroll]: 1.933e-05 [a_1]: 0.00089511 [with_stream_mark]: 3.392e-05 [recompute_prepare]: 2.813e-05 [updatestate_depend_eliminate]: 1.21e-05 [updatestate_assign_eliminate]: 8.90001e-06 [updatestate_loads_eliminate]: 1.125e-05 [parameter_eliminate]: 2.59999e-06 [a_2]: 0.00025777 [accelerated_algorithm]: 5.002e-05 [shard]: 2.56e-06 [meta_shard_fg_expand]: 3.91999e-06 [shard_inline]: 1.777e-05 [merge_send_recv]: 1.549e-05 [auto_parallel]: 1.464e-05 [parallel]: 3.175e-05 [flash_sp]: 1.437e-05 [merge_comm]: 9.24e-06 [allreduce_fusion]: 8.33001e-06 [matmul_add_comm_reduction]: 1.806e-05 [allreduce_slice_to_reducescatter]: 8.99978e-07 [virtual_shard_identity]: 2.197e-05 [virtual_dataset]: 1.732e-05 [get_grad_eliminate_]: 1.834e-05 [virtual_output]: 1.767e-05 [merge_forward]: 8.57e-06 [cell_reuse_recompute_pass]: 3.04001e-06 [offload_activation]: 1.907e-05 [cell_reuse_handle_not_recompute_node_pass]: 3.666e-05 [merge_recompute_call_nodes]: 1.68002e-06 [before_grad]: 2.968e-05 [set_forward_comm_id_for_comm_node_pass]: 1.012e-05 [meta_fg_expand]: 6.16e-06 [flash_sp_send_recv_attached]: 6.19999e-06 [receive_attached]: 2.25002e-06 [after_resolve]: 2.661e-05 [a_after_grad]: 3.631e-05 [renormalize]: 0.0110209 [add_forward_monad_depend]: 1.92e-05 [auto_monad_grad]: 3.54002e-06 [auto_monad_eliminator]: 6.356e-05 [cse]: 0.00016726 [a_3]: 0.00013743 [Cycle 2]: 0.00289239, [45] [expand_dump_flag]: 2.29999e-06 [switch_simplify]: 1.967e-05 [loop_unroll]: 1.605e-05 [a_1]: 0.00046332 [with_stream_mark]: 2.856e-05 [recompute_prepare]: 1.787e-05 [updatestate_depend_eliminate]: 9.64e-06 [updatestate_assign_eliminate]: 7.92e-06 [updatestate_loads_eliminate]: 1.087e-05 [parameter_eliminate]: 2.48e-06 [a_2]: 0.00022183 [accelerated_algorithm]: 2.361e-05 [shard]: 3.07002e-06 [meta_shard_fg_expand]: 4.37998e-06 [shard_inline]: 1.553e-05 [merge_send_recv]: 1.402e-05 [auto_parallel]: 1.588e-05 [parallel]: 1.016e-05 [flash_sp]: 4.33999e-06 [merge_comm]: 8.45999e-06 [allreduce_fusion]: 9.31e-06 [matmul_add_comm_reduction]: 1.705e-05 [allreduce_slice_to_reducescatter]: 9.60019e-07 [virtual_shard_identity]: 1.851e-05 [virtual_dataset]: 1.569e-05 [get_grad_eliminate_]: 1.537e-05 [virtual_output]: 1.53e-05 [merge_forward]: 9.42001e-06 [cell_reuse_recompute_pass]: 3.09001e-06 [offload_activation]: 1.912e-05 [cell_reuse_handle_not_recompute_node_pass]: 3.121e-05 [merge_recompute_call_nodes]: 2.12999e-06 [before_grad]: 2.568e-05 [set_forward_comm_id_for_comm_node_pass]: 1.058e-05 [meta_fg_expand]: 7.24001e-06 [flash_sp_send_recv_attached]: 1.69998e-06 [receive_attached]: 3.03998e-06 [after_resolve]: 2.4e-05 [a_after_grad]: 2.699e-05 [renormalize]: 8.00064e-08 [add_forward_monad_depend]: 3.51001e-06 [auto_monad_grad]: 2.79001e-06 [auto_monad_eliminator]: 5.084e-05 [cse]: 0.00124507 [a_3]: 0.00013844 [py_interpret_to_execute_after_opt_a]: 3.203e-05 [slice_cell_reuse_recomputed_activation]: 3.36999e-06 [rewriter_after_opt_a]: 0.00036347 [convert_after_rewriter]: 2.002e-05 [order_py_execute_after_rewriter]: 1.089e-05 [mutable_eliminate]: 0.00088165 [opt_b]: 0.00060104, [1] [Cycle 1]: 0.00059038, [7] [b_1]: 0.00038865 [b_2]: 1.954e-05 [updatestate_depend_eliminate]: 1.832e-05 [updatestate_assign_eliminate]: 8.29998e-06 [updatestate_loads_eliminate]: 1.223e-05 [renormalize]: 6.80011e-07 [cse]: 9.338e-05 [optimize_parallel_all_gather_comm]: 3.596e-05 [overlap_param_gather]: 2.14999e-06 [cconv]: 4.152e-05 [loop_unroll]: 0.00069689 [opt_after_cconv]: 0.00026224, [1] [Cycle 1]: 0.0002524, [7] [c_1]: 0.00010938 [parameter_eliminate]: 6.14999e-06 [updatestate_depend_eliminate]: 1.705e-05 [updatestate_assign_eliminate]: 7.71001e-06 [updatestate_loads_eliminate]: 1.009e-05 [cse]: 6.093e-05 [renormalize]: 7.50006e-07 [remove_dup_value]: 7.699e-05 [tuple_transform]: 0.00016366, [1] [Cycle 1]: 0.00015807, [4] [d_1]: 0.00011483 [none_parameter_eliminate]: 2.39999e-06 [renormalize]: 2.19996e-07 [switch_simplify]: 1.855e-05 [partial_unused_args_eliminate]: 2.86e-06 [add_recomputation]: 0.00010942 [cse_after_recomputation]: 5.927e-05, [1] [Cycle 1]: 5.405e-05, [1] [cse]: 4.663e-05 [environ_conv]: 2.01e-05 [swap_dp_allreduce_reducescatter]: 1.225e-05 [bias_add_comm_swap]: 3.6e-06 [label_micro_interleaved_index]: 6.38e-06 [label_fine_grained_interleaved_index]: 2.68e-06 [merge_cast_opt]: 2.06e-06 [slice_recompute_activation]: 2.61e-06 [micro_interleaved_order_control]: 2.63998e-06 [assign_add_opt]: 1.22e-06 [ForceFp32Comm]: 9.09989e-07 [remove_cast_before_assign_add]: 1.40999e-06 [full_micro_interleaved_order_control]: 2.16e-06 [reorder_send_recv_between_fp_bp]: 2.99001e-06 [comm_op_add_attrs]: 1.23002e-06 [add_comm_op_reuse_tag]: 1.18001e-06 [interleave_split_concat_branches]: 1.15999e-06 [interleave_parallel_branches]: 1.09e-06 [overlap_opt_shard_in_pipeline]: 1.30999e-06 [overlap_opt_shard_grad_in_pipeline]: 2.02999e-06 [control_data_broadcast_order]: 2.677e-05 [grouped_pairwise_exchange_alltoall]: 1.55001e-06 [offloading_packed_experts]: 6.72002e-06 [overlap_recompute_and_grad_model_parallel]: 7.79002e-06 [overlap_grad_matmul_and_grad_allreduce]: 1.31002e-06 [overlap_recompute_allgather_and_fa_grad]: 1.35999e-06 [overlap_recompute_comm]: 2.22001e-06 [overlap_grad_ring_attention]: 7.95e-06 [overlap_grad_flash_sp]: 3.841e-05 [begin_end_overlap_inline]: 5.69999e-07 [split_matmul_comm_elemetwise]: 2.43998e-06 [split_layernorm_comm]: 1.95001e-06 [handle_group_info]: 8.79983e-07 [symbol_engine_optimizer]: 0.00015716, [1] [Cycle 1]: 0.00015123, [6] [build]: 1.758e-05 [elim_shapecalc]: 2.153e-05 [elim_not_effective]: 2.971e-05 [opt_reshape]: 2.34e-05 [fold_const_symbol]: 2.426e-05 [renormalize]: 2.60014e-07 [detach_backward]: 2.02999e-06 [pipeline_parallel_scheduler]: 1.27999e-06 [auto_monad_reorder]: 4.795e-05 [get_jit_bprop_graph]: 2.01e-06 [rewriter_after_jit_bprop_graph]: 7.78999e-06 [opt_after_jit_grad]: 0.181198 [validate]: 0.00011259 Sums bootstrap : 0.000686s : 0.03% type_inference : 2.263025s : 91.81% event_method : 0.000020s : 0.00% auto_monad : 0.000402s : 0.02% graph_reusing : 0.000007s : 0.00% inline : 0.000003s : 0.00% add_attr.add_attr_with_inline.tag_attr : 0.000022s : 0.00% add_attr.add_attr_with_inline.meta_addattr_fg_expand : 0.000005s : 0.00% parallel-infer-symbol : 0.000005s : 0.00% pre_auto_parallel : 0.000039s : 0.00% insert-virtual-dataset : 0.000003s : 0.00% parallel-infer-symbol-second : 0.000001s : 0.00% dataset_repeat_opt : 0.000002s : 0.00% pipeline_split : 0.000002s : 0.00% optimize.py_interpret_to_execute : 0.000035s : 0.00% optimize.rewriter_before_opt_a : 0.000083s : 0.00% optimize.opt_a.expand_dump_flag : 0.000005s : 0.00% optimize.opt_a.switch_simplify : 0.000054s : 0.00% optimize.opt_a.loop_unroll : 0.000035s : 0.00% optimize.opt_a.a_1 : 0.001358s : 0.06% optimize.opt_a.with_stream_mark : 0.000062s : 0.00% optimize.opt_a.recompute_prepare : 0.000046s : 0.00% optimize.opt_a.updatestate_depend_eliminate : 0.000022s : 0.00% optimize.opt_a.updatestate_assign_eliminate : 0.000017s : 0.00% optimize.opt_a.updatestate_loads_eliminate : 0.000022s : 0.00% optimize.opt_a.parameter_eliminate : 0.000005s : 0.00% optimize.opt_a.a_2 : 0.000480s : 0.02% optimize.opt_a.accelerated_algorithm : 0.000074s : 0.00% optimize.opt_a.shard : 0.000006s : 0.00% optimize.opt_a.meta_shard_fg_expand : 0.000008s : 0.00% optimize.opt_a.shard_inline : 0.000033s : 0.00% optimize.opt_a.merge_send_recv : 0.000030s : 0.00% optimize.opt_a.auto_parallel : 0.000031s : 0.00% optimize.opt_a.parallel : 0.000042s : 0.00% optimize.opt_a.flash_sp : 0.000019s : 0.00% optimize.opt_a.merge_comm : 0.000018s : 0.00% optimize.opt_a.allreduce_fusion : 0.000018s : 0.00% optimize.opt_a.matmul_add_comm_reduction : 0.000035s : 0.00% optimize.opt_a.allreduce_slice_to_reducescatter : 0.000002s : 0.00% optimize.opt_a.virtual_shard_identity : 0.000040s : 0.00% optimize.opt_a.virtual_dataset : 0.000033s : 0.00% optimize.opt_a.get_grad_eliminate_ : 0.000034s : 0.00% optimize.opt_a.virtual_output : 0.000033s : 0.00% optimize.opt_a.merge_forward : 0.000018s : 0.00% optimize.opt_a.cell_reuse_recompute_pass : 0.000006s : 0.00% optimize.opt_a.offload_activation : 0.000038s : 0.00% optimize.opt_a.cell_reuse_handle_not_recompute_node_pass : 0.000068s : 0.00% optimize.opt_a.merge_recompute_call_nodes : 0.000004s : 0.00% optimize.opt_a.before_grad : 0.000055s : 0.00% optimize.opt_a.set_forward_comm_id_for_comm_node_pass : 0.000021s : 0.00% optimize.opt_a.meta_fg_expand : 0.000013s : 0.00% optimize.opt_a.flash_sp_send_recv_attached : 0.000008s : 0.00% optimize.opt_a.receive_attached : 0.000005s : 0.00% optimize.opt_a.after_resolve : 0.000051s : 0.00% optimize.opt_a.a_after_grad : 0.000063s : 0.00% optimize.opt_a.renormalize : 0.011021s : 0.45% optimize.opt_a.add_forward_monad_depend : 0.000023s : 0.00% optimize.opt_a.auto_monad_grad : 0.000006s : 0.00% optimize.opt_a.auto_monad_eliminator : 0.000114s : 0.00% optimize.opt_a.cse : 0.001412s : 0.06% optimize.opt_a.a_3 : 0.000276s : 0.01% optimize.py_interpret_to_execute_after_opt_a : 0.000032s : 0.00% optimize.slice_cell_reuse_recomputed_activation : 0.000003s : 0.00% optimize.rewriter_after_opt_a : 0.000363s : 0.01% optimize.convert_after_rewriter : 0.000020s : 0.00% optimize.order_py_execute_after_rewriter : 0.000011s : 0.00% optimize.mutable_eliminate : 0.000882s : 0.04% optimize.opt_b.b_1 : 0.000389s : 0.02% optimize.opt_b.b_2 : 0.000020s : 0.00% optimize.opt_b.updatestate_depend_eliminate : 0.000018s : 0.00% optimize.opt_b.updatestate_assign_eliminate : 0.000008s : 0.00% optimize.opt_b.updatestate_loads_eliminate : 0.000012s : 0.00% optimize.opt_b.renormalize : 0.000001s : 0.00% optimize.opt_b.cse : 0.000093s : 0.00% optimize.optimize_parallel_all_gather_comm : 0.000036s : 0.00% optimize.overlap_param_gather : 0.000002s : 0.00% optimize.cconv : 0.000042s : 0.00% optimize.loop_unroll : 0.000697s : 0.03% optimize.opt_after_cconv.c_1 : 0.000109s : 0.00% optimize.opt_after_cconv.parameter_eliminate : 0.000006s : 0.00% optimize.opt_after_cconv.updatestate_depend_eliminate : 0.000017s : 0.00% optimize.opt_after_cconv.updatestate_assign_eliminate : 0.000008s : 0.00% optimize.opt_after_cconv.updatestate_loads_eliminate : 0.000010s : 0.00% optimize.opt_after_cconv.cse : 0.000061s : 0.00% optimize.opt_after_cconv.renormalize : 0.000001s : 0.00% optimize.remove_dup_value : 0.000077s : 0.00% optimize.tuple_transform.d_1 : 0.000115s : 0.00% optimize.tuple_transform.none_parameter_eliminate : 0.000002s : 0.00% optimize.tuple_transform.renormalize : 0.000000s : 0.00% optimize.tuple_transform.switch_simplify : 0.000019s : 0.00% optimize.partial_unused_args_eliminate : 0.000003s : 0.00% optimize.add_recomputation : 0.000109s : 0.00% optimize.cse_after_recomputation.cse : 0.000047s : 0.00% optimize.environ_conv : 0.000020s : 0.00% optimize.swap_dp_allreduce_reducescatter : 0.000012s : 0.00% optimize.bias_add_comm_swap : 0.000004s : 0.00% optimize.label_micro_interleaved_index : 0.000006s : 0.00% optimize.label_fine_grained_interleaved_index : 0.000003s : 0.00% optimize.merge_cast_opt : 0.000002s : 0.00% optimize.slice_recompute_activation : 0.000003s : 0.00% optimize.micro_interleaved_order_control : 0.000003s : 0.00% optimize.assign_add_opt : 0.000001s : 0.00% optimize.ForceFp32Comm : 0.000001s : 0.00% optimize.remove_cast_before_assign_add : 0.000001s : 0.00% optimize.full_micro_interleaved_order_control : 0.000002s : 0.00% optimize.reorder_send_recv_between_fp_bp : 0.000003s : 0.00% optimize.comm_op_add_attrs : 0.000001s : 0.00% optimize.add_comm_op_reuse_tag : 0.000001s : 0.00% optimize.interleave_split_concat_branches : 0.000001s : 0.00% optimize.interleave_parallel_branches : 0.000001s : 0.00% optimize.overlap_opt_shard_in_pipeline : 0.000001s : 0.00% optimize.overlap_opt_shard_grad_in_pipeline : 0.000002s : 0.00% optimize.control_data_broadcast_order : 0.000027s : 0.00% optimize.grouped_pairwise_exchange_alltoall : 0.000002s : 0.00% optimize.offloading_packed_experts : 0.000007s : 0.00% optimize.overlap_recompute_and_grad_model_parallel : 0.000008s : 0.00% optimize.overlap_grad_matmul_and_grad_allreduce : 0.000001s : 0.00% optimize.overlap_recompute_allgather_and_fa_grad : 0.000001s : 0.00% optimize.overlap_recompute_comm : 0.000002s : 0.00% optimize.overlap_grad_ring_attention : 0.000008s : 0.00% optimize.overlap_grad_flash_sp : 0.000038s : 0.00% optimize.begin_end_overlap_inline : 0.000001s : 0.00% optimize.split_matmul_comm_elemetwise : 0.000002s : 0.00% optimize.split_layernorm_comm : 0.000002s : 0.00% optimize.handle_group_info : 0.000001s : 0.00% optimize.symbol_engine_optimizer.build : 0.000018s : 0.00% optimize.symbol_engine_optimizer.elim_shapecalc : 0.000022s : 0.00% optimize.symbol_engine_optimizer.elim_not_effective : 0.000030s : 0.00% optimize.symbol_engine_optimizer.opt_reshape : 0.000023s : 0.00% optimize.symbol_engine_optimizer.fold_const_symbol : 0.000024s : 0.00% optimize.symbol_engine_optimizer.renormalize : 0.000000s : 0.00% detach_backward : 0.000002s : 0.00% pipeline_parallel_scheduler : 0.000001s : 0.00% auto_monad_reorder : 0.000048s : 0.00% get_jit_bprop_graph : 0.000002s : 0.00% rewriter_after_jit_bprop_graph : 0.000008s : 0.00% opt_after_jit_grad : 0.181198s : 7.35% validate : 0.000113s : 0.00% Time group info: ------[substitution.] 0.000294 96 3.38% : 0.000010s : 2: substitution.depend_value_elim 1.29% : 0.000004s : 8: substitution.elim_not_effective 1.08% : 0.000003s : 8: substitution.fold_const_symbol 4.03% : 0.000012s : 13: substitution.graph_param_transform 42.01% : 0.000124s : 1: substitution.inline 2.89% : 0.000009s : 16: substitution.j_node_and_user_rematch 9.11% : 0.000027s : 2: substitution.less_batch_normalization 2.22% : 0.000007s : 6: substitution.load_eliminater 1.08% : 0.000003s : 2: substitution.opt_reshape 4.28% : 0.000013s : 16: substitution.remove_not_recompute_node 2.77% : 0.000008s : 4: substitution.replace_old_param 11.13% : 0.000033s : 4: substitution.reshape_eliminate 7.91% : 0.000023s : 6: substitution.updatestate_pure_node_eliminater 6.82% : 0.000020s : 8: substitution.updatestate_useless_node_eliminater ------[type_inference.] 2.262921 2 99.77% : 2.257620s : 1: type_inference.infer 0.23% : 0.005301s : 1: type_inference.specialize ------[replace.] 0.000019 1 100.00% : 0.000019s : 1: replace.inline ------[match.] 0.000123 1 100.00% : 0.000123s : 1: match.inline ------[predicate.] 0.000462 3002 0.82% : 0.000004s : 27: predicate.accumulaten_eliminater 1.89% : 0.000009s : 13: predicate.ad_related_special_op_eliminate 0.72% : 0.000003s : 26: predicate.addn_check_dump 0.77% : 0.000004s : 27: predicate.addn_zero_filter 0.69% : 0.000003s : 27: predicate.adjust_all_reduce_mul_add 2.16% : 0.000010s : 53: predicate.arithmetic_simplify 0.89% : 0.000004s : 27: predicate.cast_eliminate 0.86% : 0.000004s : 26: predicate.check_bprop_eliminate 0.75% : 0.000003s : 26: predicate.compare_switch_simplify 0.24% : 0.000001s : 13: predicate.const_output_eliminate 0.80% : 0.000004s : 26: predicate.depend_value_elim 0.83% : 0.000004s : 27: predicate.dict_get_item_const_eliminator 0.86% : 0.000004s : 27: predicate.dict_get_item_eliminator 0.78% : 0.000004s : 27: predicate.dict_set_item_eliminator 1.40% : 0.000006s : 26: predicate.dumpgradient_eliminate 0.32% : 0.000001s : 13: predicate.elim_not_effective 0.49% : 0.000002s : 13: predicate.elim_shapecalc_of_broadcastargs 1.14% : 0.000005s : 40: predicate.environ_add_const_eliminate 1.13% : 0.000005s : 40: predicate.environ_get_add_eliminate 1.12% : 0.000005s : 40: predicate.environ_get_depend_swap 1.89% : 0.000009s : 66: predicate.environ_get_eliminate 1.08% : 0.000005s : 40: predicate.environ_get_set_eliminate 0.80% : 0.000004s : 28: predicate.exchange_switch_depend_value 1.29% : 0.000006s : 28: predicate.float_depend_g_call 0.74% : 0.000003s : 26: predicate.float_environ_get_switch 1.16% : 0.000005s : 39: predicate.float_tuple_getitem_switch 0.22% : 0.000001s : 13: predicate.fold_const_symbol 0.87% : 0.000004s : 26: predicate.get_grad_eliminate 0.27% : 0.000001s : 13: predicate.graph_param_transform 0.74% : 0.000003s : 26: predicate.incorporate_call 0.66% : 0.000003s : 26: predicate.incorporate_call_switch 5.15% : 0.000024s : 133: predicate.inline 1.16% : 0.000005s : 26: predicate.inline_without_move 0.41% : 0.000002s : 26: predicate.j_node_and_user_rematch 1.31% : 0.000006s : 27: predicate.less_batch_normalization 1.61% : 0.000007s : 53: predicate.list_to_tuple_eliminator_ 2.30% : 0.000011s : 80: predicate.load_eliminater 1.22% : 0.000006s : 13: predicate.loop_unroll_after_grad 0.94% : 0.000004s : 31: predicate.loop_unroll_before_grad 1.76% : 0.000008s : 53: predicate.make_slice_get_slice_eliminator 0.77% : 0.000004s : 26: predicate.merge_addn 0.76% : 0.000003s : 26: predicate.micro_step_allgather_replace 0.77% : 0.000004s : 26: predicate.mini_step_allgather_replace 0.74% : 0.000003s : 27: predicate.minmaximum_grad 1.33% : 0.000006s : 13: predicate.mutable_eliminate 0.58% : 0.000003s : 13: predicate.opt_reshape 0.51% : 0.000002s : 13: predicate.parallel_virtual_node 1.00% : 0.000005s : 28: predicate.partial_defer_inline 1.22% : 0.000006s : 40: predicate.partial_eliminate 0.76% : 0.000004s : 27: predicate.print_const_string_wrapper 0.75% : 0.000003s : 26: predicate.reduce_all_const_elim 1.07% : 0.000005s : 27: predicate.reduce_eliminate 2.28% : 0.000011s : 80: predicate.redundant_stop_gradient_eliminater 0.51% : 0.000002s : 26: predicate.remove_not_recompute_node 1.58% : 0.000007s : 53: predicate.replace_applicator 0.64% : 0.000003s : 26: predicate.replace_old_param 0.42% : 0.000002s : 13: predicate.reset_defer_inline 0.99% : 0.000005s : 27: predicate.reshape_eliminate 0.83% : 0.000004s : 26: predicate.row_tensor_add_zeros_like 0.53% : 0.000002s : 13: predicate.row_tensor_eliminate 1.35% : 0.000006s : 26: predicate.same_eliminate 0.56% : 0.000003s : 26: predicate.set_cell_output_no_recompute 0.98% : 0.000005s : 26: predicate.shard_identity_eliminate 1.01% : 0.000005s : 26: predicate.special_op_eliminate 0.94% : 0.000004s : 26: predicate.specialize_transform 1.41% : 0.000007s : 26: predicate.split_environ_get_set_with_tuple_value 1.03% : 0.000005s : 26: predicate.stack_unstack_eliminate 0.46% : 0.000002s : 13: predicate.switch_call_monad_eliminater 0.90% : 0.000004s : 28: predicate.switch_defer_inline 1.63% : 0.000008s : 54: predicate.switch_layer_defer_inline 3.30% : 0.000015s : 98: predicate.switch_simplify 0.82% : 0.000004s : 27: predicate.tile_eliminate 0.75% : 0.000003s : 27: predicate.transpose_eliminate 1.69% : 0.000008s : 53: predicate.tuple_list_convert_item_index_to_positive 1.69% : 0.000008s : 53: predicate.tuple_list_get_item_const_eliminator 1.59% : 0.000007s : 53: predicate.tuple_list_get_item_depend_reorder 3.09% : 0.000014s : 79: predicate.tuple_list_get_item_eliminator 1.63% : 0.000008s : 53: predicate.tuple_list_get_set_item_eliminator 2.52% : 0.000012s : 79: predicate.tuple_list_set_item_eliminator 1.73% : 0.000008s : 53: predicate.tuple_to_list_eliminator_ 2.27% : 0.000010s : 80: predicate.updatestate_pure_node_eliminater 3.38% : 0.000016s : 106: predicate.updatestate_useless_node_eliminater 0.43% : 0.000002s : 13: predicate.value_based_eliminate 0.82% : 0.000004s : 26: predicate.virtual_dataset_eliminate 0.84% : 0.000004s : 26: predicate.virtual_output_eliminate 0.42% : 0.000002s : 13: predicate.virtual_view_grad_eliminate 0.47% : 0.000002s : 13: predicate.zero_like_fill_zero ------[func_graph_cloner_run.] 0.327997 58 98.91% : 0.324417s : 55: func_graph_cloner_run.FuncGraphClonerGraph 1.09% : 0.003580s : 3: func_graph_cloner_run.FuncGraphSpecializer ------[meta_graph.] 0.000000 0 ------[manager.] 0.000000 0 ------[pynative] 0.000000 0 ------[others.] 2.829653 192 0.00% : 0.000004s : 1: ForceFp32Comm 5.81% : 0.164265s : 1: add_attr 5.80% : 0.164246s : 1: add_attr_with_inline 0.00% : 0.000004s : 1: add_comm_op_reuse_tag 0.00% : 0.000115s : 1: add_recomputation 0.00% : 0.000004s : 1: assign_add_opt 0.01% : 0.000415s : 1: auto_monad 0.00% : 0.000054s : 1: auto_monad_reorder 0.00% : 0.000003s : 1: begin_end_overlap_inline 0.00% : 0.000006s : 1: bias_add_comm_swap 0.03% : 0.000759s : 1: bootstrap 0.00% : 0.000046s : 1: cconv 0.00% : 0.000005s : 1: comm_op_add_attrs 0.00% : 0.000030s : 1: control_data_broadcast_order 0.00% : 0.000026s : 1: convert_after_rewriter 0.00% : 0.000062s : 1: cse_after_recomputation 0.00% : 0.000005s : 1: dataset_repeat_opt 0.00% : 0.000005s : 1: detach_backward 0.00% : 0.000024s : 1: environ_conv 0.00% : 0.000029s : 1: event_method 0.00% : 0.000005s : 1: full_micro_interleaved_order_control 0.00% : 0.000007s : 1: get_jit_bprop_graph 0.00% : 0.000012s : 1: graph_reusing 0.00% : 0.000005s : 1: grouped_pairwise_exchange_alltoall 0.00% : 0.000003s : 1: handle_group_info 0.00% : 0.000006s : 1: inline 0.00% : 0.000007s : 1: insert-virtual-dataset 0.00% : 0.000005s : 1: interleave_parallel_branches 0.00% : 0.000004s : 1: interleave_split_concat_branches 0.00% : 0.000006s : 1: label_fine_grained_interleaved_index 0.00% : 0.000009s : 1: label_micro_interleaved_index 0.03% : 0.000711s : 1: loop_unroll 0.00% : 0.000005s : 1: merge_cast_opt 0.00% : 0.000005s : 1: micro_interleaved_order_control 0.03% : 0.000898s : 1: mutable_eliminate 0.00% : 0.000010s : 1: offloading_packed_experts 0.00% : 0.000037s : 1: opt.transform.loop_unroll_optimizer 0.00% : 0.000044s : 1: opt.transform.mutable_eliminate 0.09% : 0.002596s : 78: opt.transform.opt_a 0.00% : 0.000107s : 1: opt.transform.opt_after_cconv 0.00% : 0.000090s : 1: opt.transform.opt_after_jit_grad 0.01% : 0.000370s : 28: opt.transform.opt_b 0.00% : 0.000130s : 2: opt.transform.opt_trans_graph 0.00% : 0.000095s : 4: opt.transform.symbol_engine_opt 0.58% : 0.016442s : 1: opt_a 0.01% : 0.000266s : 1: opt_after_cconv 6.40% : 0.181223s : 1: opt_after_jit_grad 0.02% : 0.000605s : 1: opt_b 0.73% : 0.020549s : 1: optimize 0.00% : 0.000041s : 1: optimize_parallel_all_gather_comm 0.00% : 0.000015s : 1: order_py_execute_after_rewriter 0.00% : 0.000043s : 1: overlap_grad_flash_sp 0.00% : 0.000004s : 1: overlap_grad_matmul_and_grad_allreduce 0.00% : 0.000013s : 1: overlap_grad_ring_attention 0.00% : 0.000005s : 1: overlap_opt_shard_grad_in_pipeline 0.00% : 0.000004s : 1: overlap_opt_shard_in_pipeline 0.00% : 0.000007s : 1: overlap_param_gather 0.00% : 0.000005s : 1: overlap_recompute_allgather_and_fa_grad 0.00% : 0.000011s : 1: overlap_recompute_and_grad_model_parallel 0.00% : 0.000005s : 1: overlap_recompute_comm 0.00% : 0.000009s : 1: parallel-infer-symbol 0.00% : 0.000004s : 1: parallel-infer-symbol-second 0.00% : 0.000006s : 1: partial_unused_args_eliminate 0.00% : 0.000005s : 1: pipeline_parallel_scheduler 0.00% : 0.000006s : 1: pipeline_split 0.00% : 0.000044s : 1: pre_auto_parallel 0.00% : 0.000039s : 1: py_interpret_to_execute 0.00% : 0.000037s : 1: py_interpret_to_execute_after_opt_a 0.00% : 0.000004s : 1: remove_cast_before_assign_add 0.00% : 0.000081s : 1: remove_dup_value 0.25% : 0.006987s : 1: renormalize.infer 0.14% : 0.004016s : 1: renormalize.specialize 0.00% : 0.000006s : 1: reorder_send_recv_between_fp_bp 0.00% : 0.000011s : 1: rewriter_after_jit_bprop_graph 0.01% : 0.000374s : 1: rewriter_after_opt_a 0.00% : 0.000087s : 1: rewriter_before_opt_a 0.00% : 0.000007s : 1: slice_cell_reuse_recomputed_activation 0.00% : 0.000005s : 1: slice_recompute_activation 0.00% : 0.000006s : 1: split_layernorm_comm 0.00% : 0.000005s : 1: split_matmul_comm_elemetwise 0.00% : 0.000016s : 1: swap_dp_allreduce_reducescatter 0.01% : 0.000160s : 1: symbol_engine_optimizer 0.01% : 0.000167s : 1: tuple_transform 79.98% : 2.263058s : 1: type_inference . [hook] pytest_runtest_teardown:test_matmul_qkv_bf16_0[False-mstype0-k_n_shape3-1] tests/st/infer/ops/test_internal_ops/test_matmul_split.py::test_matmul_qkv_bf16_0[False-mstype0-k_n_shape3-1],max_mem:208.0M [WARNING] ME(152078:281472851701552,MainProcess):2026-01-29-17:42:15.100.780 [mindspore/context.py:1334] For 'context.set_context', the parameter 'device_target' will be deprecated and removed in a future version. Please use the api mindspore.set_device() instead. TotalTime = 0.781274, [21] [bootstrap]: 0.00065846 [type_inference]: 0.711463 [event_method]: 1.913e-05 [auto_monad]: 0.00032349 [graph_reusing]: 6.83e-06 [inline]: 3.2e-06 [add_attr]: 0.0095316, [1] [add_attr_with_inline]: 0.00951492, [1] [Cycle 1]: 7.469e-05, [2] [tag_attr]: 2.277e-05 [meta_addattr_fg_expand]: 4.38001e-06 [parallel-infer-symbol]: 5.43002e-06 [pre_auto_parallel]: 4.181e-05 [insert-virtual-dataset]: 2.41e-06 [parallel-infer-symbol-second]: 6.69999e-07 [dataset_repeat_opt]: 1.86998e-06 [pipeline_split]: 1.51998e-06 [optimize]: 0.0484061, [53] [py_interpret_to_execute]: 3.449e-05 [rewriter_before_opt_a]: 8.341e-05 [opt_a]: 0.0390465, [2] [Cycle 1]: 0.0325083, [45] [expand_dump_flag]: 2.64001e-06 [switch_simplify]: 3.673e-05 [loop_unroll]: 2.023e-05 [a_1]: 0.00087568 [with_stream_mark]: 3.085e-05 [recompute_prepare]: 2.608e-05 [updatestate_depend_eliminate]: 1.139e-05 [updatestate_assign_eliminate]: 9.00001e-06 [updatestate_loads_eliminate]: 1.056e-05 [parameter_eliminate]: 2.36998e-06 [a_2]: 0.00027275 [accelerated_algorithm]: 5.234e-05 [shard]: 2.41998e-06 [meta_shard_fg_expand]: 4e-06 [shard_inline]: 0.00224593 [merge_send_recv]: 3.758e-05 [auto_parallel]: 2.386e-05 [parallel]: 4.104e-05 [flash_sp]: 1.77e-05 [merge_comm]: 9.44e-06 [allreduce_fusion]: 8.99e-06 [matmul_add_comm_reduction]: 2.121e-05 [allreduce_slice_to_reducescatter]: 1.00999e-06 [virtual_shard_identity]: 4.185e-05 [virtual_dataset]: 2.178e-05 [get_grad_eliminate_]: 2.175e-05 [virtual_output]: 2.019e-05 [merge_forward]: 1.101e-05 [cell_reuse_recompute_pass]: 3.06999e-06 [offload_activation]: 1.993e-05 [cell_reuse_handle_not_recompute_node_pass]: 5.233e-05 [merge_recompute_call_nodes]: 1.62999e-06 [before_grad]: 3.271e-05 [set_forward_comm_id_for_comm_node_pass]: 1.053e-05 [meta_fg_expand]: 1.092e-05 [flash_sp_send_recv_attached]: 7.63999e-06 [receive_attached]: 2.48998e-06 [after_resolve]: 2.874e-05 [a_after_grad]: 3.397e-05 [renormalize]: 0.019984 [add_forward_monad_depend]: 2.086e-05 [auto_monad_grad]: 2.31e-06 [auto_monad_eliminator]: 6.124e-05 [cse]: 0.00590432 [a_3]: 0.00185439 [Cycle 2]: 0.00651944, [45] [expand_dump_flag]: 5.60001e-06 [switch_simplify]: 2.77e-05 [loop_unroll]: 1.911e-05 [a_1]: 0.00402896 [with_stream_mark]: 0.00057092 [recompute_prepare]: 4.482e-05 [updatestate_depend_eliminate]: 2.136e-05 [updatestate_assign_eliminate]: 1.06e-05 [updatestate_loads_eliminate]: 1.339e-05 [parameter_eliminate]: 5.14e-06 [a_2]: 0.00025545 [accelerated_algorithm]: 2.564e-05 [shard]: 3.35e-06 [meta_shard_fg_expand]: 8.35001e-06 [shard_inline]: 1.62e-05 [merge_send_recv]: 1.721e-05 [auto_parallel]: 1.601e-05 [parallel]: 1.021e-05 [flash_sp]: 4.82998e-06 [merge_comm]: 9.07001e-06 [allreduce_fusion]: 9.74e-06 [matmul_add_comm_reduction]: 1.89e-05 [allreduce_slice_to_reducescatter]: 9.10019e-07 [virtual_shard_identity]: 1.883e-05 [virtual_dataset]: 1.62e-05 [get_grad_eliminate_]: 1.588e-05 [virtual_output]: 1.507e-05 [merge_forward]: 8.72e-06 [cell_reuse_recompute_pass]: 3.13e-06 [offload_activation]: 1.797e-05 [cell_reuse_handle_not_recompute_node_pass]: 3.018e-05 [merge_recompute_call_nodes]: 1.47001e-06 [before_grad]: 2.517e-05 [set_forward_comm_id_for_comm_node_pass]: 9.13002e-06 [meta_fg_expand]: 6.61999e-06 [flash_sp_send_recv_attached]: 2.22999e-06 [receive_attached]: 2.43e-06 [after_resolve]: 2.499e-05 [a_after_grad]: 2.996e-05 [renormalize]: 9.00181e-08 [add_forward_monad_depend]: 4.50001e-06 [auto_monad_grad]: 2.83998e-06 [auto_monad_eliminator]: 6.64e-05 [cse]: 0.00016681 [a_3]: 0.00010854 [py_interpret_to_execute_after_opt_a]: 2.932e-05 [slice_cell_reuse_recomputed_activation]: 2.76e-06 [rewriter_after_opt_a]: 0.00021253 [convert_after_rewriter]: 1.68e-05 [order_py_execute_after_rewriter]: 1.144e-05 [mutable_eliminate]: 0.0009128 [opt_b]: 0.00056573, [1] [Cycle 1]: 0.00055524, [7] [b_1]: 0.00037465 [b_2]: 1.919e-05 [updatestate_depend_eliminate]: 1.506e-05 [updatestate_assign_eliminate]: 8.47e-06 [updatestate_loads_eliminate]: 1.024e-05 [renormalize]: 8.89995e-07 [cse]: 8.017e-05 [optimize_parallel_all_gather_comm]: 3.321e-05 [overlap_param_gather]: 1.84e-06 [cconv]: 3.974e-05 [loop_unroll]: 0.00054773 [opt_after_cconv]: 0.00022803, [1] [Cycle 1]: 0.00021989, [7] [c_1]: 9.882e-05 [parameter_eliminate]: 5.72001e-06 [updatestate_depend_eliminate]: 1.466e-05 [updatestate_assign_eliminate]: 7.13998e-06 [updatestate_loads_eliminate]: 9.55001e-06 [cse]: 4.76e-05 [renormalize]: 4.19997e-07 [remove_dup_value]: 7.473e-05 [tuple_transform]: 0.0001627, [1] [Cycle 1]: 0.00015674, [4] [d_1]: 0.0001127 [none_parameter_eliminate]: 2.89001e-06 [renormalize]: 2.60014e-07 [switch_simplify]: 1.694e-05 [partial_unused_args_eliminate]: 1.90001e-06 [add_recomputation]: 0.00011361 [cse_after_recomputation]: 6.153e-05, [1] [Cycle 1]: 5.55e-05, [1] [cse]: 4.869e-05 [environ_conv]: 2.039e-05 [swap_dp_allreduce_reducescatter]: 1.285e-05 [bias_add_comm_swap]: 3.5e-06 [label_micro_interleaved_index]: 4.49002e-06 [label_fine_grained_interleaved_index]: 2.81e-06 [merge_cast_opt]: 1.35999e-06 [slice_recompute_activation]: 2.11e-06 [micro_interleaved_order_control]: 2.19999e-06 [assign_add_opt]: 1.32e-06 [ForceFp32Comm]: 8.49977e-07 [remove_cast_before_assign_add]: 1.17e-06 [full_micro_interleaved_order_control]: 2.08002e-06 [reorder_send_recv_between_fp_bp]: 3.06001e-06 [comm_op_add_attrs]: 1.55001e-06 [add_comm_op_reuse_tag]: 1.03001e-06 [interleave_split_concat_branches]: 1.12e-06 [interleave_parallel_branches]: 1.00001e-06 [overlap_opt_shard_in_pipeline]: 1.17e-06 [overlap_opt_shard_grad_in_pipeline]: 1.71e-06 [control_data_broadcast_order]: 2.744e-05 [grouped_pairwise_exchange_alltoall]: 2.22999e-06 [offloading_packed_experts]: 7.95e-06 [overlap_recompute_and_grad_model_parallel]: 7.75e-06 [overlap_grad_matmul_and_grad_allreduce]: 1.29e-06 [overlap_recompute_allgather_and_fa_grad]: 1.37e-06 [overlap_recompute_comm]: 2.22001e-06 [overlap_grad_ring_attention]: 7.9e-06 [overlap_grad_flash_sp]: 0.00024398 [begin_end_overlap_inline]: 2.07999e-06 [split_matmul_comm_elemetwise]: 4.11001e-06 [split_layernorm_comm]: 1.62999e-06 [handle_group_info]: 2.02001e-06 [symbol_engine_optimizer]: 0.00426462, [1] [Cycle 1]: 0.00424731, [6] [build]: 0.00016729 [elim_shapecalc]: 0.00019568 [elim_not_effective]: 0.00020946 [opt_reshape]: 0.00019062 [fold_const_symbol]: 0.00303363 [renormalize]: 1.80001e-06 [detach_backward]: 5.25001e-06 [pipeline_parallel_scheduler]: 2.41e-06 [auto_monad_reorder]: 0.00039149 [get_jit_bprop_graph]: 2.66999e-06 [rewriter_after_jit_bprop_graph]: 0.00015891 [opt_after_jit_grad]: 0.00878764 [validate]: 0.00046247 Sums bootstrap : 0.000658s : 0.09% type_inference : 0.711463s : 92.74% event_method : 0.000019s : 0.00% auto_monad : 0.000323s : 0.04% graph_reusing : 0.000007s : 0.00% inline : 0.000003s : 0.00% add_attr.add_attr_with_inline.tag_attr : 0.000023s : 0.00% add_attr.add_attr_with_inline.meta_addattr_fg_expand : 0.000004s : 0.00% parallel-infer-symbol : 0.000005s : 0.00% pre_auto_parallel : 0.000042s : 0.01% insert-virtual-dataset : 0.000002s : 0.00% parallel-infer-symbol-second : 0.000001s : 0.00% dataset_repeat_opt : 0.000002s : 0.00% pipeline_split : 0.000002s : 0.00% optimize.py_interpret_to_execute : 0.000034s : 0.00% optimize.rewriter_before_opt_a : 0.000083s : 0.01% optimize.opt_a.expand_dump_flag : 0.000008s : 0.00% optimize.opt_a.switch_simplify : 0.000064s : 0.01% optimize.opt_a.loop_unroll : 0.000039s : 0.01% optimize.opt_a.a_1 : 0.004905s : 0.64% optimize.opt_a.with_stream_mark : 0.000602s : 0.08% optimize.opt_a.recompute_prepare : 0.000071s : 0.01% optimize.opt_a.updatestate_depend_eliminate : 0.000033s : 0.00% optimize.opt_a.updatestate_assign_eliminate : 0.000020s : 0.00% optimize.opt_a.updatestate_loads_eliminate : 0.000024s : 0.00% optimize.opt_a.parameter_eliminate : 0.000008s : 0.00% optimize.opt_a.a_2 : 0.000528s : 0.07% optimize.opt_a.accelerated_algorithm : 0.000078s : 0.01% optimize.opt_a.shard : 0.000006s : 0.00% optimize.opt_a.meta_shard_fg_expand : 0.000012s : 0.00% optimize.opt_a.shard_inline : 0.002262s : 0.29% optimize.opt_a.merge_send_recv : 0.000055s : 0.01% optimize.opt_a.auto_parallel : 0.000040s : 0.01% optimize.opt_a.parallel : 0.000051s : 0.01% optimize.opt_a.flash_sp : 0.000023s : 0.00% optimize.opt_a.merge_comm : 0.000019s : 0.00% optimize.opt_a.allreduce_fusion : 0.000019s : 0.00% optimize.opt_a.matmul_add_comm_reduction : 0.000040s : 0.01% optimize.opt_a.allreduce_slice_to_reducescatter : 0.000002s : 0.00% optimize.opt_a.virtual_shard_identity : 0.000061s : 0.01% optimize.opt_a.virtual_dataset : 0.000038s : 0.00% optimize.opt_a.get_grad_eliminate_ : 0.000038s : 0.00% optimize.opt_a.virtual_output : 0.000035s : 0.00% optimize.opt_a.merge_forward : 0.000020s : 0.00% optimize.opt_a.cell_reuse_recompute_pass : 0.000006s : 0.00% optimize.opt_a.offload_activation : 0.000038s : 0.00% optimize.opt_a.cell_reuse_handle_not_recompute_node_pass : 0.000083s : 0.01% optimize.opt_a.merge_recompute_call_nodes : 0.000003s : 0.00% optimize.opt_a.before_grad : 0.000058s : 0.01% optimize.opt_a.set_forward_comm_id_for_comm_node_pass : 0.000020s : 0.00% optimize.opt_a.meta_fg_expand : 0.000018s : 0.00% optimize.opt_a.flash_sp_send_recv_attached : 0.000010s : 0.00% optimize.opt_a.receive_attached : 0.000005s : 0.00% optimize.opt_a.after_resolve : 0.000054s : 0.01% optimize.opt_a.a_after_grad : 0.000064s : 0.01% optimize.opt_a.renormalize : 0.019984s : 2.60% optimize.opt_a.add_forward_monad_depend : 0.000025s : 0.00% optimize.opt_a.auto_monad_grad : 0.000005s : 0.00% optimize.opt_a.auto_monad_eliminator : 0.000128s : 0.02% optimize.opt_a.cse : 0.006071s : 0.79% optimize.opt_a.a_3 : 0.001963s : 0.26% optimize.py_interpret_to_execute_after_opt_a : 0.000029s : 0.00% optimize.slice_cell_reuse_recomputed_activation : 0.000003s : 0.00% optimize.rewriter_after_opt_a : 0.000213s : 0.03% optimize.convert_after_rewriter : 0.000017s : 0.00% optimize.order_py_execute_after_rewriter : 0.000011s : 0.00% optimize.mutable_eliminate : 0.000913s : 0.12% optimize.opt_b.b_1 : 0.000375s : 0.05% optimize.opt_b.b_2 : 0.000019s : 0.00% optimize.opt_b.updatestate_depend_eliminate : 0.000015s : 0.00% optimize.opt_b.updatestate_assign_eliminate : 0.000008s : 0.00% optimize.opt_b.updatestate_loads_eliminate : 0.000010s : 0.00% optimize.opt_b.renormalize : 0.000001s : 0.00% optimize.opt_b.cse : 0.000080s : 0.01% optimize.optimize_parallel_all_gather_comm : 0.000033s : 0.00% optimize.overlap_param_gather : 0.000002s : 0.00% optimize.cconv : 0.000040s : 0.01% optimize.loop_unroll : 0.000548s : 0.07% optimize.opt_after_cconv.c_1 : 0.000099s : 0.01% optimize.opt_after_cconv.parameter_eliminate : 0.000006s : 0.00% optimize.opt_after_cconv.updatestate_depend_eliminate : 0.000015s : 0.00% optimize.opt_after_cconv.updatestate_assign_eliminate : 0.000007s : 0.00% optimize.opt_after_cconv.updatestate_loads_eliminate : 0.000010s : 0.00% optimize.opt_after_cconv.cse : 0.000048s : 0.01% optimize.opt_after_cconv.renormalize : 0.000000s : 0.00% optimize.remove_dup_value : 0.000075s : 0.01% optimize.tuple_transform.d_1 : 0.000113s : 0.01% optimize.tuple_transform.none_parameter_eliminate : 0.000003s : 0.00% optimize.tuple_transform.renormalize : 0.000000s : 0.00% optimize.tuple_transform.switch_simplify : 0.000017s : 0.00% optimize.partial_unused_args_eliminate : 0.000002s : 0.00% optimize.add_recomputation : 0.000114s : 0.01% optimize.cse_after_recomputation.cse : 0.000049s : 0.01% optimize.environ_conv : 0.000020s : 0.00% optimize.swap_dp_allreduce_reducescatter : 0.000013s : 0.00% optimize.bias_add_comm_swap : 0.000003s : 0.00% optimize.label_micro_interleaved_index : 0.000004s : 0.00% optimize.label_fine_grained_interleaved_index : 0.000003s : 0.00% optimize.merge_cast_opt : 0.000001s : 0.00% optimize.slice_recompute_activation : 0.000002s : 0.00% optimize.micro_interleaved_order_control : 0.000002s : 0.00% optimize.assign_add_opt : 0.000001s : 0.00% optimize.ForceFp32Comm : 0.000001s : 0.00% optimize.remove_cast_before_assign_add : 0.000001s : 0.00% optimize.full_micro_interleaved_order_control : 0.000002s : 0.00% optimize.reorder_send_recv_between_fp_bp : 0.000003s : 0.00% optimize.comm_op_add_attrs : 0.000002s : 0.00% optimize.add_comm_op_reuse_tag : 0.000001s : 0.00% optimize.interleave_split_concat_branches : 0.000001s : 0.00% optimize.interleave_parallel_branches : 0.000001s : 0.00% optimize.overlap_opt_shard_in_pipeline : 0.000001s : 0.00% optimize.overlap_opt_shard_grad_in_pipeline : 0.000002s : 0.00% optimize.control_data_broadcast_order : 0.000027s : 0.00% optimize.grouped_pairwise_exchange_alltoall : 0.000002s : 0.00% optimize.offloading_packed_experts : 0.000008s : 0.00% optimize.overlap_recompute_and_grad_model_parallel : 0.000008s : 0.00% optimize.overlap_grad_matmul_and_grad_allreduce : 0.000001s : 0.00% optimize.overlap_recompute_allgather_and_fa_grad : 0.000001s : 0.00% optimize.overlap_recompute_comm : 0.000002s : 0.00% optimize.overlap_grad_ring_attention : 0.000008s : 0.00% optimize.overlap_grad_flash_sp : 0.000244s : 0.03% optimize.begin_end_overlap_inline : 0.000002s : 0.00% optimize.split_matmul_comm_elemetwise : 0.000004s : 0.00% optimize.split_layernorm_comm : 0.000002s : 0.00% optimize.handle_group_info : 0.000002s : 0.00% optimize.symbol_engine_optimizer.build : 0.000167s : 0.02% optimize.symbol_engine_optimizer.elim_shapecalc : 0.000196s : 0.03% optimize.symbol_engine_optimizer.elim_not_effective : 0.000209s : 0.03% optimize.symbol_engine_optimizer.opt_reshape : 0.000191s : 0.02% optimize.symbol_engine_optimizer.fold_const_symbol : 0.003034s : 0.40% optimize.symbol_engine_optimizer.renormalize : 0.000002s : 0.00% detach_backward : 0.000005s : 0.00% pipeline_parallel_scheduler : 0.000002s : 0.00% auto_monad_reorder : 0.000391s : 0.05% get_jit_bprop_graph : 0.000003s : 0.00% rewriter_after_jit_bprop_graph : 0.000159s : 0.02% opt_after_jit_grad : 0.008788s : 1.15% validate : 0.000462s : 0.06% Time group info: ------[substitution.] 0.000338 96 3.35% : 0.000011s : 2: substitution.depend_value_elim 1.37% : 0.000005s : 8: substitution.elim_not_effective 1.91% : 0.000006s : 8: substitution.fold_const_symbol 3.44% : 0.000012s : 13: substitution.graph_param_transform 37.46% : 0.000127s : 1: substitution.inline 2.83% : 0.000010s : 16: substitution.j_node_and_user_rematch 7.88% : 0.000027s : 2: substitution.less_batch_normalization 2.47% : 0.000008s : 6: substitution.load_eliminater 1.18% : 0.000004s : 2: substitution.opt_reshape 3.61% : 0.000012s : 16: substitution.remove_not_recompute_node 2.85% : 0.000010s : 4: substitution.replace_old_param 15.80% : 0.000053s : 4: substitution.reshape_eliminate 8.69% : 0.000029s : 6: substitution.updatestate_pure_node_eliminater 7.16% : 0.000024s : 8: substitution.updatestate_useless_node_eliminater ------[type_inference.] 0.711364 2 99.34% : 0.706675s : 1: type_inference.infer 0.66% : 0.004688s : 1: type_inference.specialize ------[replace.] 0.000019 1 100.00% : 0.000019s : 1: replace.inline ------[match.] 0.000125 1 100.00% : 0.000125s : 1: match.inline ------[predicate.] 0.002187 3002 0.23% : 0.000005s : 27: predicate.accumulaten_eliminater 0.43% : 0.000009s : 13: predicate.ad_related_special_op_eliminate 0.16% : 0.000004s : 26: predicate.addn_check_dump 0.26% : 0.000006s : 27: predicate.addn_zero_filter 0.18% : 0.000004s : 27: predicate.adjust_all_reduce_mul_add 0.74% : 0.000016s : 53: predicate.arithmetic_simplify 0.22% : 0.000005s : 27: predicate.cast_eliminate 0.31% : 0.000007s : 26: predicate.check_bprop_eliminate 0.16% : 0.000004s : 26: predicate.compare_switch_simplify 0.05% : 0.000001s : 13: predicate.const_output_eliminate 0.19% : 0.000004s : 26: predicate.depend_value_elim 0.17% : 0.000004s : 27: predicate.dict_get_item_const_eliminator 0.29% : 0.000006s : 27: predicate.dict_get_item_eliminator 9.71% : 0.000212s : 27: predicate.dict_set_item_eliminator 0.34% : 0.000008s : 26: predicate.dumpgradient_eliminate 0.08% : 0.000002s : 13: predicate.elim_not_effective 0.28% : 0.000006s : 13: predicate.elim_shapecalc_of_broadcastargs 0.28% : 0.000006s : 40: predicate.environ_add_const_eliminate 0.26% : 0.000006s : 40: predicate.environ_get_add_eliminate 0.36% : 0.000008s : 40: predicate.environ_get_depend_swap 0.51% : 0.000011s : 66: predicate.environ_get_eliminate 6.61% : 0.000145s : 40: predicate.environ_get_set_eliminate 0.17% : 0.000004s : 28: predicate.exchange_switch_depend_value 7.00% : 0.000153s : 28: predicate.float_depend_g_call 0.17% : 0.000004s : 26: predicate.float_environ_get_switch 0.26% : 0.000006s : 39: predicate.float_tuple_getitem_switch 0.18% : 0.000004s : 13: predicate.fold_const_symbol 0.20% : 0.000004s : 26: predicate.get_grad_eliminate 0.06% : 0.000001s : 13: predicate.graph_param_transform 0.18% : 0.000004s : 26: predicate.incorporate_call 0.15% : 0.000003s : 26: predicate.incorporate_call_switch 1.47% : 0.000032s : 133: predicate.inline 0.25% : 0.000005s : 26: predicate.inline_without_move 0.09% : 0.000002s : 26: predicate.j_node_and_user_rematch 0.26% : 0.000006s : 27: predicate.less_batch_normalization 0.42% : 0.000009s : 53: predicate.list_to_tuple_eliminator_ 0.57% : 0.000012s : 80: predicate.load_eliminater 0.17% : 0.000004s : 13: predicate.loop_unroll_after_grad 0.22% : 0.000005s : 31: predicate.loop_unroll_before_grad 0.39% : 0.000009s : 53: predicate.make_slice_get_slice_eliminator 0.19% : 0.000004s : 26: predicate.merge_addn 46.11% : 0.001009s : 26: predicate.micro_step_allgather_replace 0.20% : 0.000004s : 26: predicate.mini_step_allgather_replace 0.29% : 0.000006s : 27: predicate.minmaximum_grad 0.22% : 0.000005s : 13: predicate.mutable_eliminate 0.22% : 0.000005s : 13: predicate.opt_reshape 0.10% : 0.000002s : 13: predicate.parallel_virtual_node 0.51% : 0.000011s : 28: predicate.partial_defer_inline 0.29% : 0.000006s : 40: predicate.partial_eliminate 0.18% : 0.000004s : 27: predicate.print_const_string_wrapper 0.18% : 0.000004s : 26: predicate.reduce_all_const_elim 0.32% : 0.000007s : 27: predicate.reduce_eliminate 0.53% : 0.000012s : 80: predicate.redundant_stop_gradient_eliminater 0.14% : 0.000003s : 26: predicate.remove_not_recompute_node 0.37% : 0.000008s : 53: predicate.replace_applicator 0.11% : 0.000002s : 26: predicate.replace_old_param 0.06% : 0.000001s : 13: predicate.reset_defer_inline 0.29% : 0.000006s : 27: predicate.reshape_eliminate 0.24% : 0.000005s : 26: predicate.row_tensor_add_zeros_like 0.10% : 0.000002s : 13: predicate.row_tensor_eliminate 0.37% : 0.000008s : 26: predicate.same_eliminate 0.17% : 0.000004s : 26: predicate.set_cell_output_no_recompute 0.25% : 0.000005s : 26: predicate.shard_identity_eliminate 6.59% : 0.000144s : 26: predicate.special_op_eliminate 0.20% : 0.000004s : 26: predicate.specialize_transform 0.32% : 0.000007s : 26: predicate.split_environ_get_set_with_tuple_value 0.21% : 0.000005s : 26: predicate.stack_unstack_eliminate 0.09% : 0.000002s : 13: predicate.switch_call_monad_eliminater 0.24% : 0.000005s : 28: predicate.switch_defer_inline 0.40% : 0.000009s : 54: predicate.switch_layer_defer_inline 0.76% : 0.000017s : 98: predicate.switch_simplify 0.20% : 0.000004s : 27: predicate.tile_eliminate 0.19% : 0.000004s : 27: predicate.transpose_eliminate 0.42% : 0.000009s : 53: predicate.tuple_list_convert_item_index_to_positive 0.43% : 0.000009s : 53: predicate.tuple_list_get_item_const_eliminator 0.36% : 0.000008s : 53: predicate.tuple_list_get_item_depend_reorder 0.73% : 0.000016s : 79: predicate.tuple_list_get_item_eliminator 0.41% : 0.000009s : 53: predicate.tuple_list_get_set_item_eliminator 0.66% : 0.000014s : 79: predicate.tuple_list_set_item_eliminator 0.40% : 0.000009s : 53: predicate.tuple_to_list_eliminator_ 0.48% : 0.000011s : 80: predicate.updatestate_pure_node_eliminater 0.75% : 0.000016s : 106: predicate.updatestate_useless_node_eliminater 0.09% : 0.000002s : 13: predicate.value_based_eliminate 0.20% : 0.000004s : 26: predicate.virtual_dataset_eliminate 0.21% : 0.000005s : 26: predicate.virtual_output_eliminate 0.10% : 0.000002s : 13: predicate.virtual_view_grad_eliminate 0.10% : 0.000002s : 13: predicate.zero_like_fill_zero ------[func_graph_cloner_run.] 0.030955 58 87.70% : 0.027146s : 55: func_graph_cloner_run.FuncGraphClonerGraph 12.30% : 0.003808s : 3: func_graph_cloner_run.FuncGraphSpecializer ------[meta_graph.] 0.000000 0 ------[manager.] 0.000000 0 ------[pynative] 0.000000 0 ------[others.] 0.872591 192 0.00% : 0.000004s : 1: ForceFp32Comm 1.09% : 0.009539s : 1: add_attr 1.09% : 0.009520s : 1: add_attr_with_inline 0.00% : 0.000004s : 1: add_comm_op_reuse_tag 0.01% : 0.000119s : 1: add_recomputation 0.00% : 0.000004s : 1: assign_add_opt 0.04% : 0.000337s : 1: auto_monad 0.05% : 0.000407s : 1: auto_monad_reorder 0.00% : 0.000015s : 1: begin_end_overlap_inline 0.00% : 0.000007s : 1: bias_add_comm_swap 0.08% : 0.000727s : 1: bootstrap 0.01% : 0.000044s : 1: cconv 0.00% : 0.000006s : 1: comm_op_add_attrs 0.00% : 0.000031s : 1: control_data_broadcast_order 0.00% : 0.000022s : 1: convert_after_rewriter 0.01% : 0.000065s : 1: cse_after_recomputation 0.00% : 0.000005s : 1: dataset_repeat_opt 0.00% : 0.000015s : 1: detach_backward 0.00% : 0.000026s : 1: environ_conv 0.00% : 0.000027s : 1: event_method 0.00% : 0.000005s : 1: full_micro_interleaved_order_control 0.00% : 0.000012s : 1: get_jit_bprop_graph 0.00% : 0.000012s : 1: graph_reusing 0.00% : 0.000005s : 1: grouped_pairwise_exchange_alltoall 0.00% : 0.000005s : 1: handle_group_info 0.00% : 0.000007s : 1: inline 0.00% : 0.000007s : 1: insert-virtual-dataset 0.00% : 0.000005s : 1: interleave_parallel_branches 0.00% : 0.000004s : 1: interleave_split_concat_branches 0.00% : 0.000006s : 1: label_fine_grained_interleaved_index 0.00% : 0.000007s : 1: label_micro_interleaved_index 0.06% : 0.000558s : 1: loop_unroll 0.00% : 0.000004s : 1: merge_cast_opt 0.00% : 0.000005s : 1: micro_interleaved_order_control 0.11% : 0.000929s : 1: mutable_eliminate 0.00% : 0.000011s : 1: offloading_packed_experts 0.00% : 0.000030s : 1: opt.transform.loop_unroll_optimizer 0.00% : 0.000038s : 1: opt.transform.mutable_eliminate 1.11% : 0.009692s : 78: opt.transform.opt_a 0.01% : 0.000097s : 1: opt.transform.opt_after_cconv 0.04% : 0.000384s : 1: opt.transform.opt_after_jit_grad 0.04% : 0.000355s : 28: opt.transform.opt_b 0.01% : 0.000127s : 2: opt.transform.opt_trans_graph 0.41% : 0.003610s : 4: opt.transform.symbol_engine_opt 4.48% : 0.039051s : 1: opt_a 0.03% : 0.000231s : 1: opt_after_cconv 1.05% : 0.009205s : 1: opt_after_jit_grad 0.07% : 0.000570s : 1: opt_b 5.57% : 0.048568s : 1: optimize 0.00% : 0.000038s : 1: optimize_parallel_all_gather_comm 0.00% : 0.000016s : 1: order_py_execute_after_rewriter 0.03% : 0.000261s : 1: overlap_grad_flash_sp 0.00% : 0.000004s : 1: overlap_grad_matmul_and_grad_allreduce 0.13% : 0.001121s : 1: overlap_grad_ring_attention 0.00% : 0.000004s : 1: overlap_opt_shard_grad_in_pipeline 0.00% : 0.000004s : 1: overlap_opt_shard_in_pipeline 0.00% : 0.000006s : 1: overlap_param_gather 0.00% : 0.000005s : 1: overlap_recompute_allgather_and_fa_grad 0.00% : 0.000011s : 1: overlap_recompute_and_grad_model_parallel 0.00% : 0.000005s : 1: overlap_recompute_comm 0.00% : 0.000010s : 1: parallel-infer-symbol 0.00% : 0.000004s : 1: parallel-infer-symbol-second 0.00% : 0.000005s : 1: partial_unused_args_eliminate 0.00% : 0.000007s : 1: pipeline_parallel_scheduler 0.00% : 0.000006s : 1: pipeline_split 0.01% : 0.000047s : 1: pre_auto_parallel 0.00% : 0.000039s : 1: py_interpret_to_execute 0.00% : 0.000034s : 1: py_interpret_to_execute_after_opt_a 0.00% : 0.000004s : 1: remove_cast_before_assign_add 0.01% : 0.000080s : 1: remove_dup_value 1.50% : 0.013112s : 1: renormalize.infer 0.78% : 0.006849s : 1: renormalize.specialize 0.00% : 0.000006s : 1: reorder_send_recv_between_fp_bp 0.02% : 0.000172s : 1: rewriter_after_jit_bprop_graph 0.03% : 0.000223s : 1: rewriter_after_opt_a 0.01% : 0.000088s : 1: rewriter_before_opt_a 0.00% : 0.000006s : 1: slice_cell_reuse_recomputed_activation 0.00% : 0.000005s : 1: slice_recompute_activation 0.00% : 0.000007s : 1: split_layernorm_comm 0.00% : 0.000008s : 1: split_matmul_comm_elemetwise 0.00% : 0.000016s : 1: swap_dp_allreduce_reducescatter 0.49% : 0.004272s : 1: symbol_engine_optimizer 0.02% : 0.000166s : 1: tuple_transform 81.54% : 0.711490s : 1: type_inference . [hook] pytest_runtest_teardown:test_matmul_qkv_bf16_0[False-mstype0-k_n_shape3-16] tests/st/infer/ops/test_internal_ops/test_matmul_split.py::test_matmul_qkv_bf16_0[False-mstype0-k_n_shape3-16],max_mem:208.0M [WARNING] ME(152078:281472851701552,MainProcess):2026-01-29-17:42:54.131.167 [mindspore/context.py:1334] For 'context.set_context', the parameter 'device_target' will be deprecated and removed in a future version. Please use the api mindspore.set_device() instead. TotalTime = 1.79473, [21] [bootstrap]: 0.00067761 [type_inference]: 1.48387 [event_method]: 0.00021452 [auto_monad]: 0.00076096 [graph_reusing]: 4.798e-05 [inline]: 3.35e-06 [add_attr]: 0.00540498, [1] [add_attr_with_inline]: 0.00538811, [1] [Cycle 1]: 0.00029949, [2] [tag_attr]: 0.0001959 [meta_addattr_fg_expand]: 5.03e-05 [parallel-infer-symbol]: 3.9e-06 [pre_auto_parallel]: 0.00024739 [insert-virtual-dataset]: 2.53998e-06 [parallel-infer-symbol-second]: 1.39e-06 [dataset_repeat_opt]: 2.07999e-06 [pipeline_split]: 1.49e-06 [optimize]: 0.302186, [53] [py_interpret_to_execute]: 0.0002318 [rewriter_before_opt_a]: 0.00077528 [opt_a]: 0.15595, [2] [Cycle 1]: 0.153815, [45] [expand_dump_flag]: 1.732e-05 [switch_simplify]: 0.0007922 [loop_unroll]: 0.0003143 [a_1]: 0.0092021 [with_stream_mark]: 6.253e-05 [recompute_prepare]: 4.051e-05 [updatestate_depend_eliminate]: 2.232e-05 [updatestate_assign_eliminate]: 1.614e-05 [updatestate_loads_eliminate]: 1.928e-05 [parameter_eliminate]: 2.81e-06 [a_2]: 0.00079618 [accelerated_algorithm]: 6.871e-05 [shard]: 2.90002e-06 [meta_shard_fg_expand]: 2.975e-05 [shard_inline]: 2.394e-05 [merge_send_recv]: 2.187e-05 [auto_parallel]: 2.147e-05 [parallel]: 4.215e-05 [flash_sp]: 1.581e-05 [merge_comm]: 1.445e-05 [allreduce_fusion]: 1.277e-05 [matmul_add_comm_reduction]: 2.122e-05 [allreduce_slice_to_reducescatter]: 1.27e-06 [virtual_shard_identity]: 2.908e-05 [virtual_dataset]: 2.267e-05 [get_grad_eliminate_]: 2.112e-05 [virtual_output]: 2.233e-05 [merge_forward]: 1.367e-05 [cell_reuse_recompute_pass]: 3.51001e-06 [offload_activation]: 2.321e-05 [cell_reuse_handle_not_recompute_node_pass]: 0.124583 [merge_recompute_call_nodes]: 3.9e-06 [before_grad]: 6.844e-05 [set_forward_comm_id_for_comm_node_pass]: 3.663e-05 [meta_fg_expand]: 4.209e-05 [flash_sp_send_recv_attached]: 1.29e-05 [receive_attached]: 2.42001e-06 [after_resolve]: 3.5e-05 [a_after_grad]: 3.82e-05 [renormalize]: 0.0153356 [add_forward_monad_depend]: 2.174e-05 [auto_monad_grad]: 3.55e-06 [auto_monad_eliminator]: 8.873e-05 [cse]: 0.00117399 [a_3]: 0.00017359 [Cycle 2]: 0.00211647, [45] [expand_dump_flag]: 3.06999e-06 [switch_simplify]: 2.639e-05 [loop_unroll]: 2.234e-05 [a_1]: 0.00066871 [with_stream_mark]: 3.505e-05 [recompute_prepare]: 2.17e-05 [updatestate_depend_eliminate]: 1.647e-05 [updatestate_assign_eliminate]: 1.173e-05 [updatestate_loads_eliminate]: 1.461e-05 [parameter_eliminate]: 2.46e-06 [a_2]: 0.00031 [accelerated_algorithm]: 2.965e-05 [shard]: 2.23002e-06 [meta_shard_fg_expand]: 9.59999e-06 [shard_inline]: 2.098e-05 [merge_send_recv]: 2.052e-05 [auto_parallel]: 2.072e-05 [parallel]: 1.137e-05 [flash_sp]: 4.84998e-06 [merge_comm]: 1.34e-05 [allreduce_fusion]: 1.25e-05 [matmul_add_comm_reduction]: 2.158e-05 [allreduce_slice_to_reducescatter]: 9.29984e-07 [virtual_shard_identity]: 2.321e-05 [virtual_dataset]: 2.043e-05 [get_grad_eliminate_]: 2.073e-05 [virtual_output]: 1.954e-05 [merge_forward]: 1.232e-05 [cell_reuse_recompute_pass]: 3.10998e-06 [offload_activation]: 2.489e-05 [cell_reuse_handle_not_recompute_node_pass]: 3.985e-05 [merge_recompute_call_nodes]: 1.44e-06 [before_grad]: 3.506e-05 [set_forward_comm_id_for_comm_node_pass]: 1.381e-05 [meta_fg_expand]: 1.015e-05 [flash_sp_send_recv_attached]: 1.69e-06 [receive_attached]: 2.48e-06 [after_resolve]: 2.676e-05 [a_after_grad]: 3.354e-05 [renormalize]: 8.00064e-08 [add_forward_monad_depend]: 2.74001e-06 [auto_monad_grad]: 2.65002e-06 [auto_monad_eliminator]: 6.227e-05 [cse]: 8.452e-05 [a_3]: 0.00013846 [py_interpret_to_execute_after_opt_a]: 3.489e-05 [slice_cell_reuse_recomputed_activation]: 2.63998e-06 [rewriter_after_opt_a]: 0.140594 [convert_after_rewriter]: 4.211e-05 [order_py_execute_after_rewriter]: 1.505e-05 [mutable_eliminate]: 0.00088182 [opt_b]: 0.00085727, [1] [Cycle 1]: 0.00084626, [7] [b_1]: 0.00059788 [b_2]: 2.586e-05 [updatestate_depend_eliminate]: 2.21e-05 [updatestate_assign_eliminate]: 1.273e-05 [updatestate_loads_eliminate]: 1.491e-05 [renormalize]: 9.09989e-07 [cse]: 0.000121 [optimize_parallel_all_gather_comm]: 4.583e-05 [overlap_param_gather]: 4.89e-06 [cconv]: 4.655e-05 [loop_unroll]: 0.00068674 [opt_after_cconv]: 0.00031928, [1] [Cycle 1]: 0.00030928, [7] [c_1]: 0.00013125 [parameter_eliminate]: 5.70001e-06 [updatestate_depend_eliminate]: 1.948e-05 [updatestate_assign_eliminate]: 1.113e-05 [updatestate_loads_eliminate]: 1.329e-05 [cse]: 8.835e-05 [renormalize]: 9.49978e-07 [remove_dup_value]: 9.781e-05 [tuple_transform]: 0.00025759, [1] [Cycle 1]: 0.00025102, [4] [d_1]: 0.00020429 [none_parameter_eliminate]: 2.48e-06 [renormalize]: 3.10014e-07 [switch_simplify]: 2.173e-05 [partial_unused_args_eliminate]: 2.00002e-06 [add_recomputation]: 0.00014679 [cse_after_recomputation]: 6.947e-05, [1] [Cycle 1]: 6.325e-05, [1] [cse]: 5.665e-05 [environ_conv]: 2.503e-05 [swap_dp_allreduce_reducescatter]: 1.867e-05 [bias_add_comm_swap]: 3.08998e-06 [label_micro_interleaved_index]: 6.44001e-06 [label_fine_grained_interleaved_index]: 3.17002e-06 [merge_cast_opt]: 1.83002e-06 [slice_recompute_activation]: 2.36e-06 [micro_interleaved_order_control]: 2.84001e-06 [assign_add_opt]: 1.40999e-06 [ForceFp32Comm]: 1.15001e-06 [remove_cast_before_assign_add]: 1.29e-06 [full_micro_interleaved_order_control]: 2.04999e-06 [reorder_send_recv_between_fp_bp]: 2.88e-06 [comm_op_add_attrs]: 1.34998e-06 [add_comm_op_reuse_tag]: 9.80013e-07 [interleave_split_concat_branches]: 1.13001e-06 [interleave_parallel_branches]: 1.02e-06 [overlap_opt_shard_in_pipeline]: 5.49998e-06 [overlap_opt_shard_grad_in_pipeline]: 1.60001e-06 [control_data_broadcast_order]: 3.805e-05 [grouped_pairwise_exchange_alltoall]: 1.89e-06 [offloading_packed_experts]: 1.095e-05 [overlap_recompute_and_grad_model_parallel]: 1.056e-05 [overlap_grad_matmul_and_grad_allreduce]: 1.17999e-06 [overlap_recompute_allgather_and_fa_grad]: 1.32e-06 [overlap_recompute_comm]: 2.68998e-06 [overlap_grad_ring_attention]: 9.87001e-06 [overlap_grad_flash_sp]: 5.12e-05 [begin_end_overlap_inline]: 5.8001e-07 [split_matmul_comm_elemetwise]: 2.58e-06 [split_layernorm_comm]: 1.71998e-06 [handle_group_info]: 1.02e-06 [symbol_engine_optimizer]: 0.00057246, [1] [Cycle 1]: 0.00056678, [6] [build]: 0.00032161 [elim_shapecalc]: 3.416e-05 [elim_not_effective]: 6.677e-05 [opt_reshape]: 3.331e-05 [fold_const_symbol]: 6.126e-05 [renormalize]: 2.69996e-07 [detach_backward]: 2.65002e-06 [pipeline_parallel_scheduler]: 1.74e-06 [auto_monad_reorder]: 6.283e-05 [get_jit_bprop_graph]: 2.28998e-06 [rewriter_after_jit_bprop_graph]: 7.40003e-06 [opt_after_jit_grad]: 0.00082458 [validate]: 0.00014511 Sums bootstrap : 0.000678s : 0.04% type_inference : 1.483871s : 82.99% event_method : 0.000215s : 0.01% auto_monad : 0.000761s : 0.04% graph_reusing : 0.000048s : 0.00% inline : 0.000003s : 0.00% add_attr.add_attr_with_inline.tag_attr : 0.000196s : 0.01% add_attr.add_attr_with_inline.meta_addattr_fg_expand : 0.000050s : 0.00% parallel-infer-symbol : 0.000004s : 0.00% pre_auto_parallel : 0.000247s : 0.01% insert-virtual-dataset : 0.000003s : 0.00% parallel-infer-symbol-second : 0.000001s : 0.00% dataset_repeat_opt : 0.000002s : 0.00% pipeline_split : 0.000001s : 0.00% optimize.py_interpret_to_execute : 0.000232s : 0.01% optimize.rewriter_before_opt_a : 0.000775s : 0.04% optimize.opt_a.expand_dump_flag : 0.000020s : 0.00% optimize.opt_a.switch_simplify : 0.000819s : 0.05% optimize.opt_a.loop_unroll : 0.000337s : 0.02% optimize.opt_a.a_1 : 0.009871s : 0.55% optimize.opt_a.with_stream_mark : 0.000098s : 0.01% optimize.opt_a.recompute_prepare : 0.000062s : 0.00% optimize.opt_a.updatestate_depend_eliminate : 0.000039s : 0.00% optimize.opt_a.updatestate_assign_eliminate : 0.000028s : 0.00% optimize.opt_a.updatestate_loads_eliminate : 0.000034s : 0.00% optimize.opt_a.parameter_eliminate : 0.000005s : 0.00% optimize.opt_a.a_2 : 0.001106s : 0.06% optimize.opt_a.accelerated_algorithm : 0.000098s : 0.01% optimize.opt_a.shard : 0.000005s : 0.00% optimize.opt_a.meta_shard_fg_expand : 0.000039s : 0.00% optimize.opt_a.shard_inline : 0.000045s : 0.00% optimize.opt_a.merge_send_recv : 0.000042s : 0.00% optimize.opt_a.auto_parallel : 0.000042s : 0.00% optimize.opt_a.parallel : 0.000054s : 0.00% optimize.opt_a.flash_sp : 0.000021s : 0.00% optimize.opt_a.merge_comm : 0.000028s : 0.00% optimize.opt_a.allreduce_fusion : 0.000025s : 0.00% optimize.opt_a.matmul_add_comm_reduction : 0.000043s : 0.00% optimize.opt_a.allreduce_slice_to_reducescatter : 0.000002s : 0.00% optimize.opt_a.virtual_shard_identity : 0.000052s : 0.00% optimize.opt_a.virtual_dataset : 0.000043s : 0.00% optimize.opt_a.get_grad_eliminate_ : 0.000042s : 0.00% optimize.opt_a.virtual_output : 0.000042s : 0.00% optimize.opt_a.merge_forward : 0.000026s : 0.00% optimize.opt_a.cell_reuse_recompute_pass : 0.000007s : 0.00% optimize.opt_a.offload_activation : 0.000048s : 0.00% optimize.opt_a.cell_reuse_handle_not_recompute_node_pass : 0.124622s : 6.97% optimize.opt_a.merge_recompute_call_nodes : 0.000005s : 0.00% optimize.opt_a.before_grad : 0.000104s : 0.01% optimize.opt_a.set_forward_comm_id_for_comm_node_pass : 0.000050s : 0.00% optimize.opt_a.meta_fg_expand : 0.000052s : 0.00% optimize.opt_a.flash_sp_send_recv_attached : 0.000015s : 0.00% optimize.opt_a.receive_attached : 0.000005s : 0.00% optimize.opt_a.after_resolve : 0.000062s : 0.00% optimize.opt_a.a_after_grad : 0.000072s : 0.00% optimize.opt_a.renormalize : 0.015336s : 0.86% optimize.opt_a.add_forward_monad_depend : 0.000024s : 0.00% optimize.opt_a.auto_monad_grad : 0.000006s : 0.00% optimize.opt_a.auto_monad_eliminator : 0.000151s : 0.01% optimize.opt_a.cse : 0.001259s : 0.07% optimize.opt_a.a_3 : 0.000312s : 0.02% optimize.py_interpret_to_execute_after_opt_a : 0.000035s : 0.00% optimize.slice_cell_reuse_recomputed_activation : 0.000003s : 0.00% optimize.rewriter_after_opt_a : 0.140594s : 7.86% optimize.convert_after_rewriter : 0.000042s : 0.00% optimize.order_py_execute_after_rewriter : 0.000015s : 0.00% optimize.mutable_eliminate : 0.000882s : 0.05% optimize.opt_b.b_1 : 0.000598s : 0.03% optimize.opt_b.b_2 : 0.000026s : 0.00% optimize.opt_b.updatestate_depend_eliminate : 0.000022s : 0.00% optimize.opt_b.updatestate_assign_eliminate : 0.000013s : 0.00% optimize.opt_b.updatestate_loads_eliminate : 0.000015s : 0.00% optimize.opt_b.renormalize : 0.000001s : 0.00% optimize.opt_b.cse : 0.000121s : 0.01% optimize.optimize_parallel_all_gather_comm : 0.000046s : 0.00% optimize.overlap_param_gather : 0.000005s : 0.00% optimize.cconv : 0.000047s : 0.00% optimize.loop_unroll : 0.000687s : 0.04% optimize.opt_after_cconv.c_1 : 0.000131s : 0.01% optimize.opt_after_cconv.parameter_eliminate : 0.000006s : 0.00% optimize.opt_after_cconv.updatestate_depend_eliminate : 0.000019s : 0.00% optimize.opt_after_cconv.updatestate_assign_eliminate : 0.000011s : 0.00% optimize.opt_after_cconv.updatestate_loads_eliminate : 0.000013s : 0.00% optimize.opt_after_cconv.cse : 0.000088s : 0.00% optimize.opt_after_cconv.renormalize : 0.000001s : 0.00% optimize.remove_dup_value : 0.000098s : 0.01% optimize.tuple_transform.d_1 : 0.000204s : 0.01% optimize.tuple_transform.none_parameter_eliminate : 0.000002s : 0.00% optimize.tuple_transform.renormalize : 0.000000s : 0.00% optimize.tuple_transform.switch_simplify : 0.000022s : 0.00% optimize.partial_unused_args_eliminate : 0.000002s : 0.00% optimize.add_recomputation : 0.000147s : 0.01% optimize.cse_after_recomputation.cse : 0.000057s : 0.00% optimize.environ_conv : 0.000025s : 0.00% optimize.swap_dp_allreduce_reducescatter : 0.000019s : 0.00% optimize.bias_add_comm_swap : 0.000003s : 0.00% optimize.label_micro_interleaved_index : 0.000006s : 0.00% optimize.label_fine_grained_interleaved_index : 0.000003s : 0.00% optimize.merge_cast_opt : 0.000002s : 0.00% optimize.slice_recompute_activation : 0.000002s : 0.00% optimize.micro_interleaved_order_control : 0.000003s : 0.00% optimize.assign_add_opt : 0.000001s : 0.00% optimize.ForceFp32Comm : 0.000001s : 0.00% optimize.remove_cast_before_assign_add : 0.000001s : 0.00% optimize.full_micro_interleaved_order_control : 0.000002s : 0.00% optimize.reorder_send_recv_between_fp_bp : 0.000003s : 0.00% optimize.comm_op_add_attrs : 0.000001s : 0.00% optimize.add_comm_op_reuse_tag : 0.000001s : 0.00% optimize.interleave_split_concat_branches : 0.000001s : 0.00% optimize.interleave_parallel_branches : 0.000001s : 0.00% optimize.overlap_opt_shard_in_pipeline : 0.000005s : 0.00% optimize.overlap_opt_shard_grad_in_pipeline : 0.000002s : 0.00% optimize.control_data_broadcast_order : 0.000038s : 0.00% optimize.grouped_pairwise_exchange_alltoall : 0.000002s : 0.00% optimize.offloading_packed_experts : 0.000011s : 0.00% optimize.overlap_recompute_and_grad_model_parallel : 0.000011s : 0.00% optimize.overlap_grad_matmul_and_grad_allreduce : 0.000001s : 0.00% optimize.overlap_recompute_allgather_and_fa_grad : 0.000001s : 0.00% optimize.overlap_recompute_comm : 0.000003s : 0.00% optimize.overlap_grad_ring_attention : 0.000010s : 0.00% optimize.overlap_grad_flash_sp : 0.000051s : 0.00% optimize.begin_end_overlap_inline : 0.000001s : 0.00% optimize.split_matmul_comm_elemetwise : 0.000003s : 0.00% optimize.split_layernorm_comm : 0.000002s : 0.00% optimize.handle_group_info : 0.000001s : 0.00% optimize.symbol_engine_optimizer.build : 0.000322s : 0.02% optimize.symbol_engine_optimizer.elim_shapecalc : 0.000034s : 0.00% optimize.symbol_engine_optimizer.elim_not_effective : 0.000067s : 0.00% optimize.symbol_engine_optimizer.opt_reshape : 0.000033s : 0.00% optimize.symbol_engine_optimizer.fold_const_symbol : 0.000061s : 0.00% optimize.symbol_engine_optimizer.renormalize : 0.000000s : 0.00% detach_backward : 0.000003s : 0.00% pipeline_parallel_scheduler : 0.000002s : 0.00% auto_monad_reorder : 0.000063s : 0.00% get_jit_bprop_graph : 0.000002s : 0.00% rewriter_after_jit_bprop_graph : 0.000007s : 0.00% opt_after_jit_grad : 0.000825s : 0.05% validate : 0.000145s : 0.01% Time group info: ------[substitution.] 0.127923 569 0.01% : 0.000017s : 8: substitution.depend_value_elim 0.02% : 0.000027s : 14: substitution.elim_not_effective 0.01% : 0.000018s : 12: substitution.float_tuple_getitem_switch 0.02% : 0.000028s : 14: substitution.fold_const_symbol 0.01% : 0.000015s : 17: substitution.graph_param_transform 1.79% : 0.002284s : 71: substitution.inline 0.01% : 0.000019s : 29: substitution.j_node_and_user_rematch 0.03% : 0.000036s : 6: substitution.less_batch_normalization 0.05% : 0.000062s : 2: substitution.list_to_tuple_eliminator_ 0.01% : 0.000007s : 6: substitution.load_eliminater 0.03% : 0.000038s : 21: substitution.minmaximum_grad 0.01% : 0.000008s : 2: substitution.opt_reshape 97.32% : 0.124489s : 29: substitution.remove_not_recompute_node 0.01% : 0.000011s : 4: substitution.replace_old_param 0.06% : 0.000072s : 18: substitution.reshape_eliminate 0.04% : 0.000048s : 24: substitution.switch_simplify 0.13% : 0.000161s : 51: substitution.tuple_list_convert_item_index_to_positive 0.07% : 0.000091s : 51: substitution.tuple_list_get_item_const_eliminator 0.09% : 0.000115s : 51: substitution.tuple_list_get_item_depend_reorder 0.17% : 0.000222s : 73: substitution.tuple_list_get_item_eliminator 0.08% : 0.000108s : 51: substitution.tuple_list_get_set_item_eliminator 0.02% : 0.000027s : 6: substitution.updatestate_pure_node_eliminater 0.02% : 0.000019s : 9: substitution.updatestate_useless_node_eliminater ------[type_inference.] 1.483314 2 94.55% : 1.402421s : 1: type_inference.infer 5.45% : 0.080893s : 1: type_inference.specialize ------[replace.] 0.001149 115 4.38% : 0.000050s : 5: replace.depend_value_elim 57.42% : 0.000660s : 71: replace.inline 3.61% : 0.000042s : 2: replace.list_to_tuple_eliminator_ 25.44% : 0.000292s : 24: replace.switch_simplify 9.15% : 0.000105s : 13: replace.tuple_list_get_item_eliminator ------[match.] 0.002371 115 0.09% : 0.000002s : 5: match.depend_value_elim 94.39% : 0.002238s : 71: match.inline 2.47% : 0.000059s : 2: match.list_to_tuple_eliminator_ 1.42% : 0.000034s : 24: match.switch_simplify 1.63% : 0.000039s : 13: match.tuple_list_get_item_eliminator ------[predicate.] 0.001621 9701 1.19% : 0.000019s : 123: predicate.accumulaten_eliminater 0.39% : 0.000006s : 17: predicate.ad_related_special_op_eliminate 0.49% : 0.000008s : 58: predicate.addn_check_dump 1.13% : 0.000018s : 123: predicate.addn_zero_filter 1.10% : 0.000018s : 123: predicate.adjust_all_reduce_mul_add 3.43% : 0.000056s : 181: predicate.arithmetic_simplify 1.20% : 0.000019s : 123: predicate.cast_eliminate 0.31% : 0.000005s : 34: predicate.check_bprop_eliminate 0.49% : 0.000008s : 58: predicate.compare_switch_simplify 0.08% : 0.000001s : 17: predicate.const_output_eliminate 0.60% : 0.000010s : 61: predicate.depend_value_elim 1.17% : 0.000019s : 123: predicate.dict_get_item_const_eliminator 1.27% : 0.000021s : 123: predicate.dict_get_item_eliminator 1.08% : 0.000017s : 123: predicate.dict_set_item_eliminator 0.38% : 0.000006s : 34: predicate.dumpgradient_eliminate 0.09% : 0.000001s : 17: predicate.elim_not_effective 0.20% : 0.000003s : 17: predicate.elim_shapecalc_of_broadcastargs 1.22% : 0.000020s : 140: predicate.environ_add_const_eliminate 1.38% : 0.000022s : 140: predicate.environ_get_add_eliminate 2.24% : 0.000036s : 140: predicate.environ_get_depend_swap 1.76% : 0.000029s : 198: predicate.environ_get_eliminate 1.24% : 0.000020s : 140: predicate.environ_get_set_eliminate 1.90% : 0.000031s : 209: predicate.exchange_switch_depend_value 2.68% : 0.000043s : 209: predicate.float_depend_g_call 0.50% : 0.000008s : 58: predicate.float_environ_get_switch 0.68% : 0.000011s : 75: predicate.float_tuple_getitem_switch 0.08% : 0.000001s : 17: predicate.fold_const_symbol 0.33% : 0.000005s : 35: predicate.get_grad_eliminate 0.10% : 0.000002s : 17: predicate.graph_param_transform 0.50% : 0.000008s : 58: predicate.incorporate_call 0.45% : 0.000007s : 58: predicate.incorporate_call_switch 5.70% : 0.000092s : 459: predicate.inline 0.43% : 0.000007s : 35: predicate.inline_without_move 0.15% : 0.000002s : 35: predicate.j_node_and_user_rematch 0.50% : 0.000008s : 36: predicate.less_batch_normalization 1.64% : 0.000027s : 172: predicate.list_to_tuple_eliminator_ 2.68% : 0.000043s : 295: predicate.load_eliminater 0.46% : 0.000007s : 17: predicate.loop_unroll_after_grad 3.00% : 0.000049s : 314: predicate.loop_unroll_before_grad 1.55% : 0.000025s : 157: predicate.make_slice_get_slice_eliminator 0.52% : 0.000008s : 58: predicate.merge_addn 0.30% : 0.000005s : 34: predicate.micro_step_allgather_replace 0.30% : 0.000005s : 34: predicate.mini_step_allgather_replace 1.22% : 0.000020s : 123: predicate.minmaximum_grad 0.52% : 0.000008s : 17: predicate.mutable_eliminate 0.19% : 0.000003s : 17: predicate.opt_reshape 0.18% : 0.000003s : 17: predicate.parallel_virtual_node 3.18% : 0.000052s : 209: predicate.partial_defer_inline 1.49% : 0.000024s : 155: predicate.partial_eliminate 1.27% : 0.000021s : 123: predicate.print_const_string_wrapper 0.47% : 0.000008s : 53: predicate.reduce_all_const_elim 1.54% : 0.000025s : 123: predicate.reduce_eliminate 2.67% : 0.000043s : 295: predicate.redundant_stop_gradient_eliminater 0.25% : 0.000004s : 35: predicate.remove_not_recompute_node 1.09% : 0.000018s : 172: predicate.replace_applicator 0.18% : 0.000003s : 35: predicate.replace_old_param 0.11% : 0.000002s : 17: predicate.reset_defer_inline 1.31% : 0.000021s : 123: predicate.reshape_eliminate 0.32% : 0.000005s : 34: predicate.row_tensor_add_zeros_like 0.20% : 0.000003s : 17: predicate.row_tensor_eliminate 0.49% : 0.000008s : 34: predicate.same_eliminate 0.22% : 0.000004s : 40: predicate.set_cell_output_no_recompute 0.38% : 0.000006s : 35: predicate.shard_identity_eliminate 0.34% : 0.000006s : 34: predicate.special_op_eliminate 0.60% : 0.000010s : 58: predicate.specialize_transform 0.41% : 0.000007s : 34: predicate.split_environ_get_set_with_tuple_value 0.41% : 0.000007s : 35: predicate.stack_unstack_eliminate 0.17% : 0.000003s : 17: predicate.switch_call_monad_eliminater 2.15% : 0.000035s : 209: predicate.switch_defer_inline 2.34% : 0.000038s : 243: predicate.switch_layer_defer_inline 6.58% : 0.000107s : 646: predicate.switch_simplify 1.20% : 0.000019s : 123: predicate.tile_eliminate 1.06% : 0.000017s : 123: predicate.transpose_eliminate 1.70% : 0.000028s : 157: predicate.tuple_list_convert_item_index_to_positive 1.76% : 0.000029s : 157: predicate.tuple_list_get_item_const_eliminator 1.57% : 0.000025s : 157: predicate.tuple_list_get_item_depend_reorder 3.10% : 0.000050s : 228: predicate.tuple_list_get_item_eliminator 1.67% : 0.000027s : 157: predicate.tuple_list_get_set_item_eliminator 2.22% : 0.000036s : 215: predicate.tuple_list_set_item_eliminator 1.60% : 0.000026s : 170: predicate.tuple_to_list_eliminator_ 2.55% : 0.000041s : 295: predicate.updatestate_pure_node_eliminater 3.13% : 0.000051s : 353: predicate.updatestate_useless_node_eliminater 0.21% : 0.000003s : 17: predicate.value_based_eliminate 0.35% : 0.000006s : 35: predicate.virtual_dataset_eliminate 0.39% : 0.000006s : 35: predicate.virtual_output_eliminate 0.15% : 0.000003s : 17: predicate.virtual_view_grad_eliminate 0.20% : 0.000003s : 17: predicate.zero_like_fill_zero ------[func_graph_cloner_run.] 0.015230 135 65.01% : 0.009900s : 55: func_graph_cloner_run.FuncGraphClonerGraph 34.99% : 0.005330s : 80: func_graph_cloner_run.FuncGraphSpecializer ------[meta_graph.] 0.000000 0 ------[manager.] 0.000000 0 ------[pynative] 0.000000 0 ------[others.] 2.256179 209 0.00% : 0.000004s : 1: ForceFp32Comm 0.24% : 0.005412s : 1: add_attr 0.24% : 0.005394s : 1: add_attr_with_inline 0.00% : 0.000005s : 1: add_comm_op_reuse_tag 0.01% : 0.000152s : 1: add_recomputation 0.00% : 0.000005s : 1: assign_add_opt 0.03% : 0.000777s : 1: auto_monad 0.00% : 0.000068s : 1: auto_monad_reorder 0.00% : 0.000004s : 1: begin_end_overlap_inline 0.00% : 0.000006s : 1: bias_add_comm_swap 0.03% : 0.000739s : 1: bootstrap 0.00% : 0.000050s : 1: cconv 0.00% : 0.000004s : 1: comm_op_add_attrs 0.00% : 0.000042s : 1: control_data_broadcast_order 0.00% : 0.000050s : 1: convert_after_rewriter 0.00% : 0.000073s : 1: cse_after_recomputation 0.00% : 0.000005s : 1: dataset_repeat_opt 0.00% : 0.000006s : 1: detach_backward 0.00% : 0.000029s : 1: environ_conv 0.01% : 0.000227s : 1: event_method 0.00% : 0.000006s : 1: full_micro_interleaved_order_control 0.00% : 0.000007s : 1: get_jit_bprop_graph 0.00% : 0.000054s : 1: graph_reusing 0.00% : 0.000005s : 1: grouped_pairwise_exchange_alltoall 0.00% : 0.000005s : 1: handle_group_info 0.00% : 0.000007s : 1: inline 0.00% : 0.000006s : 1: insert-virtual-dataset 0.00% : 0.000004s : 1: interleave_parallel_branches 0.00% : 0.000004s : 1: interleave_split_concat_branches 0.00% : 0.000006s : 1: label_fine_grained_interleaved_index 0.00% : 0.000009s : 1: label_micro_interleaved_index 0.03% : 0.000701s : 1: loop_unroll 0.00% : 0.000006s : 1: merge_cast_opt 0.00% : 0.000006s : 1: micro_interleaved_order_control 0.04% : 0.000894s : 1: mutable_eliminate 0.00% : 0.000014s : 1: offloading_packed_experts 0.00% : 0.000041s : 1: opt.transform.loop_unroll_optimizer 0.00% : 0.000053s : 1: opt.transform.mutable_eliminate 6.10% : 0.137535s : 95: opt.transform.opt_a 0.01% : 0.000130s : 1: opt.transform.opt_after_cconv 0.00% : 0.000079s : 1: opt.transform.opt_after_jit_grad 0.03% : 0.000578s : 28: opt.transform.opt_b 0.01% : 0.000223s : 2: opt.transform.opt_trans_graph 0.01% : 0.000189s : 4: opt.transform.symbol_engine_opt 6.91% : 0.155954s : 1: opt_a 0.01% : 0.000324s : 1: opt_after_cconv 0.04% : 0.000840s : 1: opt_after_jit_grad 0.04% : 0.000863s : 1: opt_b 13.39% : 0.302194s : 1: optimize 0.00% : 0.000051s : 1: optimize_parallel_all_gather_comm 0.00% : 0.000020s : 1: order_py_execute_after_rewriter 0.00% : 0.000055s : 1: overlap_grad_flash_sp 0.00% : 0.000004s : 1: overlap_grad_matmul_and_grad_allreduce 0.00% : 0.000013s : 1: overlap_grad_ring_attention 0.00% : 0.000006s : 1: overlap_opt_shard_grad_in_pipeline 0.00% : 0.000009s : 1: overlap_opt_shard_in_pipeline 0.00% : 0.000010s : 1: overlap_param_gather 0.00% : 0.000004s : 1: overlap_recompute_allgather_and_fa_grad 0.00% : 0.000014s : 1: overlap_recompute_and_grad_model_parallel 0.00% : 0.000007s : 1: overlap_recompute_comm 0.00% : 0.000009s : 1: parallel-infer-symbol 0.00% : 0.000004s : 1: parallel-infer-symbol-second 0.00% : 0.000005s : 1: partial_unused_args_eliminate 0.00% : 0.000005s : 1: pipeline_parallel_scheduler 0.00% : 0.000004s : 1: pipeline_split 0.01% : 0.000256s : 1: pre_auto_parallel 0.01% : 0.000240s : 1: py_interpret_to_execute 0.00% : 0.000040s : 1: py_interpret_to_execute_after_opt_a 0.00% : 0.000004s : 1: remove_cast_before_assign_add 0.00% : 0.000103s : 1: remove_dup_value 0.45% : 0.010066s : 1: renormalize.infer 0.23% : 0.005248s : 1: renormalize.specialize 0.00% : 0.000007s : 1: reorder_send_recv_between_fp_bp 0.00% : 0.000011s : 1: rewriter_after_jit_bprop_graph 6.23% : 0.140621s : 1: rewriter_after_opt_a 0.03% : 0.000788s : 1: rewriter_before_opt_a 0.00% : 0.000006s : 1: slice_cell_reuse_recomputed_activation 0.00% : 0.000005s : 1: slice_recompute_activation 0.00% : 0.000004s : 1: split_layernorm_comm 0.00% : 0.000006s : 1: split_matmul_comm_elemetwise 0.00% : 0.000022s : 1: swap_dp_allreduce_reducescatter 0.03% : 0.000576s : 1: symbol_engine_optimizer 0.01% : 0.000262s : 1: tuple_transform 65.77% : 1.483904s : 1: type_inference . [hook] pytest_runtest_teardown:test_matmul_qkv_bf16_0[True-mstype0-k_n_shape0-1] tests/st/infer/ops/test_internal_ops/test_matmul_split.py::test_matmul_qkv_bf16_0[True-mstype0-k_n_shape0-1],max_mem:208.0M [WARNING] ME(152078:281472851701552,MainProcess):2026-01-29-17:43:02.882.667 [mindspore/context.py:1334] For 'context.set_context', the parameter 'device_target' will be deprecated and removed in a future version. Please use the api mindspore.set_device() instead. TotalTime = 1.53606, [21] [bootstrap]: 0.081001 [type_inference]: 1.28626 [event_method]: 0.0002041 [auto_monad]: 0.00070553 [graph_reusing]: 4.631e-05 [inline]: 2.86999e-06 [add_attr]: 0.045601, [1] [add_attr_with_inline]: 0.0455815, [1] [Cycle 1]: 0.00030871, [2] [tag_attr]: 0.00020845 [meta_addattr_fg_expand]: 5.029e-05 [parallel-infer-symbol]: 4.25e-06 [pre_auto_parallel]: 0.00025686 [insert-virtual-dataset]: 2.71999e-06 [parallel-infer-symbol-second]: 1.07998e-06 [dataset_repeat_opt]: 2.68e-06 [pipeline_split]: 1.79e-06 [optimize]: 0.120794, [53] [py_interpret_to_execute]: 0.00023336 [rewriter_before_opt_a]: 0.0007812 [opt_a]: 0.114482, [2] [Cycle 1]: 0.112251, [45] [expand_dump_flag]: 1.764e-05 [switch_simplify]: 0.00080164 [loop_unroll]: 0.0003339 [a_1]: 0.0094897 [with_stream_mark]: 6.081e-05 [recompute_prepare]: 4.01e-05 [updatestate_depend_eliminate]: 2.113e-05 [updatestate_assign_eliminate]: 1.813e-05 [updatestate_loads_eliminate]: 1.891e-05 [parameter_eliminate]: 2.90998e-06 [a_2]: 0.00080218 [accelerated_algorithm]: 6.926e-05 [shard]: 2.79001e-06 [meta_shard_fg_expand]: 2.498e-05 [shard_inline]: 2.325e-05 [merge_send_recv]: 2.057e-05 [auto_parallel]: 2.177e-05 [parallel]: 3.136e-05 [flash_sp]: 1.605e-05 [merge_comm]: 1.383e-05 [allreduce_fusion]: 1.239e-05 [matmul_add_comm_reduction]: 2.154e-05 [allreduce_slice_to_reducescatter]: 9.70002e-07 [virtual_shard_identity]: 2.845e-05 [virtual_dataset]: 2.384e-05 [get_grad_eliminate_]: 2.201e-05 [virtual_output]: 2.208e-05 [merge_forward]: 1.45e-05 [cell_reuse_recompute_pass]: 2.56e-06 [offload_activation]: 2.501e-05 [cell_reuse_handle_not_recompute_node_pass]: 4.883e-05 [merge_recompute_call_nodes]: 1.69998e-06 [before_grad]: 3.96e-05 [set_forward_comm_id_for_comm_node_pass]: 1.433e-05 [meta_fg_expand]: 1.744e-05 [flash_sp_send_recv_attached]: 5.61998e-06 [receive_attached]: 2.34001e-06 [after_resolve]: 3.147e-05 [a_after_grad]: 3.796e-05 [renormalize]: 0.0357722 [add_forward_monad_depend]: 1.969e-05 [auto_monad_grad]: 3.36001e-06 [auto_monad_eliminator]: 8.429e-05 [cse]: 0.0635364 [a_3]: 0.00019415 [Cycle 2]: 0.0022145, [45] [expand_dump_flag]: 3.87998e-06 [switch_simplify]: 2.834e-05 [loop_unroll]: 2.45e-05 [a_1]: 0.00065815 [with_stream_mark]: 4.027e-05 [recompute_prepare]: 2.279e-05 [updatestate_depend_eliminate]: 1.533e-05 [updatestate_assign_eliminate]: 1.162e-05 [updatestate_loads_eliminate]: 1.406e-05 [parameter_eliminate]: 2.41e-06 [a_2]: 0.00036027 [accelerated_algorithm]: 3.221e-05 [shard]: 2.36e-06 [meta_shard_fg_expand]: 9.10999e-06 [shard_inline]: 2.139e-05 [merge_send_recv]: 2.035e-05 [auto_parallel]: 2.025e-05 [parallel]: 1.127e-05 [flash_sp]: 4.68001e-06 [merge_comm]: 1.315e-05 [allreduce_fusion]: 1.28e-05 [matmul_add_comm_reduction]: 2.267e-05 [allreduce_slice_to_reducescatter]: 1.32e-06 [virtual_shard_identity]: 2.363e-05 [virtual_dataset]: 2.034e-05 [get_grad_eliminate_]: 2.182e-05 [virtual_output]: 1.951e-05 [merge_forward]: 1.324e-05 [cell_reuse_recompute_pass]: 3.44001e-06 [offload_activation]: 2.591e-05 [cell_reuse_handle_not_recompute_node_pass]: 4.193e-05 [merge_recompute_call_nodes]: 1.57001e-06 [before_grad]: 3.738e-05 [set_forward_comm_id_for_comm_node_pass]: 1.52e-05 [meta_fg_expand]: 9.39e-06 [flash_sp_send_recv_attached]: 2.74001e-06 [receive_attached]: 2.37001e-06 [after_resolve]: 2.839e-05 [a_after_grad]: 3.346e-05 [renormalize]: 7.99773e-08 [add_forward_monad_depend]: 4.20999e-06 [auto_monad_grad]: 2.86999e-06 [auto_monad_eliminator]: 6.916e-05 [cse]: 8.669e-05 [a_3]: 0.0001442 [py_interpret_to_execute_after_opt_a]: 3.697e-05 [slice_cell_reuse_recomputed_activation]: 2.59999e-06 [rewriter_after_opt_a]: 0.00089656 [convert_after_rewriter]: 2.888e-05 [order_py_execute_after_rewriter]: 1.577e-05 [mutable_eliminate]: 0.00086975 [opt_b]: 0.00077812, [1] [Cycle 1]: 0.0007692, [7] [b_1]: 0.00055543 [b_2]: 2.444e-05 [updatestate_depend_eliminate]: 2.09e-05 [updatestate_assign_eliminate]: 1.26e-05 [updatestate_loads_eliminate]: 1.443e-05 [renormalize]: 6.89994e-07 [cse]: 9.434e-05 [optimize_parallel_all_gather_comm]: 4.416e-05 [overlap_param_gather]: 2.07999e-06 [cconv]: 5.454e-05 [loop_unroll]: 0.00060676 [opt_after_cconv]: 0.00030774, [1] [Cycle 1]: 0.00029868, [7] [c_1]: 0.00013048 [parameter_eliminate]: 5.84999e-06 [updatestate_depend_eliminate]: 2.058e-05 [updatestate_assign_eliminate]: 1.113e-05 [updatestate_loads_eliminate]: 1.376e-05 [cse]: 7.773e-05 [renormalize]: 6.69999e-07 [remove_dup_value]: 0.00010209 [tuple_transform]: 0.00026178, [1] [Cycle 1]: 0.00025559, [4] [d_1]: 0.00020559 [none_parameter_eliminate]: 2.53e-06 [renormalize]: 1.80007e-07 [switch_simplify]: 2.251e-05 [partial_unused_args_eliminate]: 2.19001e-06 [add_recomputation]: 0.00015336 [cse_after_recomputation]: 7.367e-05, [1] [Cycle 1]: 6.631e-05, [1] [cse]: 5.784e-05 [environ_conv]: 2.4e-05 [swap_dp_allreduce_reducescatter]: 1.978e-05 [bias_add_comm_swap]: 3.06001e-06 [label_micro_interleaved_index]: 6.26e-06 [label_fine_grained_interleaved_index]: 2.58e-06 [merge_cast_opt]: 1.55001e-06 [slice_recompute_activation]: 2.15002e-06 [micro_interleaved_order_control]: 3.3e-06 [assign_add_opt]: 1.54e-06 [ForceFp32Comm]: 8.2e-07 [remove_cast_before_assign_add]: 1.08001e-06 [full_micro_interleaved_order_control]: 2.04e-06 [reorder_send_recv_between_fp_bp]: 2.96999e-06 [comm_op_add_attrs]: 1.22e-06 [add_comm_op_reuse_tag]: 9.79984e-07 [interleave_split_concat_branches]: 1.17999e-06 [interleave_parallel_branches]: 1.37999e-06 [overlap_opt_shard_in_pipeline]: 1.20001e-06 [overlap_opt_shard_grad_in_pipeline]: 1.73002e-06 [control_data_broadcast_order]: 4.214e-05 [grouped_pairwise_exchange_alltoall]: 1.60999e-06 [offloading_packed_experts]: 1.121e-05 [overlap_recompute_and_grad_model_parallel]: 1.069e-05 [overlap_grad_matmul_and_grad_allreduce]: 1.26002e-06 [overlap_recompute_allgather_and_fa_grad]: 1.31002e-06 [overlap_recompute_comm]: 2.41e-06 [overlap_grad_ring_attention]: 9.09e-06 [overlap_grad_flash_sp]: 5.134e-05 [begin_end_overlap_inline]: 5.39992e-07 [split_matmul_comm_elemetwise]: 2.54999e-06 [split_layernorm_comm]: 1.76e-06 [handle_group_info]: 9.30013e-07 [symbol_engine_optimizer]: 0.00052044, [1] [Cycle 1]: 0.00051454, [6] [build]: 0.00029209 [elim_shapecalc]: 3.629e-05 [elim_not_effective]: 5.016e-05 [opt_reshape]: 3.431e-05 [fold_const_symbol]: 5.589e-05 [renormalize]: 3.4002e-07 [detach_backward]: 2.85002e-06 [pipeline_parallel_scheduler]: 1.79e-06 [auto_monad_reorder]: 6.495e-05 [get_jit_bprop_graph]: 2.56e-06 [rewriter_after_jit_bprop_graph]: 5.96e-06 [opt_after_jit_grad]: 0.0007339 [validate]: 0.00011548 Sums bootstrap : 0.081001s : 5.44% type_inference : 1.286263s : 86.37% event_method : 0.000204s : 0.01% auto_monad : 0.000706s : 0.05% graph_reusing : 0.000046s : 0.00% inline : 0.000003s : 0.00% add_attr.add_attr_with_inline.tag_attr : 0.000208s : 0.01% add_attr.add_attr_with_inline.meta_addattr_fg_expand : 0.000050s : 0.00% parallel-infer-symbol : 0.000004s : 0.00% pre_auto_parallel : 0.000257s : 0.02% insert-virtual-dataset : 0.000003s : 0.00% parallel-infer-symbol-second : 0.000001s : 0.00% dataset_repeat_opt : 0.000003s : 0.00% pipeline_split : 0.000002s : 0.00% optimize.py_interpret_to_execute : 0.000233s : 0.02% optimize.rewriter_before_opt_a : 0.000781s : 0.05% optimize.opt_a.expand_dump_flag : 0.000022s : 0.00% optimize.opt_a.switch_simplify : 0.000830s : 0.06% optimize.opt_a.loop_unroll : 0.000358s : 0.02% optimize.opt_a.a_1 : 0.010148s : 0.68% optimize.opt_a.with_stream_mark : 0.000101s : 0.01% optimize.opt_a.recompute_prepare : 0.000063s : 0.00% optimize.opt_a.updatestate_depend_eliminate : 0.000036s : 0.00% optimize.opt_a.updatestate_assign_eliminate : 0.000030s : 0.00% optimize.opt_a.updatestate_loads_eliminate : 0.000033s : 0.00% optimize.opt_a.parameter_eliminate : 0.000005s : 0.00% optimize.opt_a.a_2 : 0.001162s : 0.08% optimize.opt_a.accelerated_algorithm : 0.000101s : 0.01% optimize.opt_a.shard : 0.000005s : 0.00% optimize.opt_a.meta_shard_fg_expand : 0.000034s : 0.00% optimize.opt_a.shard_inline : 0.000045s : 0.00% optimize.opt_a.merge_send_recv : 0.000041s : 0.00% optimize.opt_a.auto_parallel : 0.000042s : 0.00% optimize.opt_a.parallel : 0.000043s : 0.00% optimize.opt_a.flash_sp : 0.000021s : 0.00% optimize.opt_a.merge_comm : 0.000027s : 0.00% optimize.opt_a.allreduce_fusion : 0.000025s : 0.00% optimize.opt_a.matmul_add_comm_reduction : 0.000044s : 0.00% optimize.opt_a.allreduce_slice_to_reducescatter : 0.000002s : 0.00% optimize.opt_a.virtual_shard_identity : 0.000052s : 0.00% optimize.opt_a.virtual_dataset : 0.000044s : 0.00% optimize.opt_a.get_grad_eliminate_ : 0.000044s : 0.00% optimize.opt_a.virtual_output : 0.000042s : 0.00% optimize.opt_a.merge_forward : 0.000028s : 0.00% optimize.opt_a.cell_reuse_recompute_pass : 0.000006s : 0.00% optimize.opt_a.offload_activation : 0.000051s : 0.00% optimize.opt_a.cell_reuse_handle_not_recompute_node_pass : 0.000091s : 0.01% optimize.opt_a.merge_recompute_call_nodes : 0.000003s : 0.00% optimize.opt_a.before_grad : 0.000077s : 0.01% optimize.opt_a.set_forward_comm_id_for_comm_node_pass : 0.000030s : 0.00% optimize.opt_a.meta_fg_expand : 0.000027s : 0.00% optimize.opt_a.flash_sp_send_recv_attached : 0.000008s : 0.00% optimize.opt_a.receive_attached : 0.000005s : 0.00% optimize.opt_a.after_resolve : 0.000060s : 0.00% optimize.opt_a.a_after_grad : 0.000071s : 0.00% optimize.opt_a.renormalize : 0.035772s : 2.40% optimize.opt_a.add_forward_monad_depend : 0.000024s : 0.00% optimize.opt_a.auto_monad_grad : 0.000006s : 0.00% optimize.opt_a.auto_monad_eliminator : 0.000153s : 0.01% optimize.opt_a.cse : 0.063623s : 4.27% optimize.opt_a.a_3 : 0.000338s : 0.02% optimize.py_interpret_to_execute_after_opt_a : 0.000037s : 0.00% optimize.slice_cell_reuse_recomputed_activation : 0.000003s : 0.00% optimize.rewriter_after_opt_a : 0.000897s : 0.06% optimize.convert_after_rewriter : 0.000029s : 0.00% optimize.order_py_execute_after_rewriter : 0.000016s : 0.00% optimize.mutable_eliminate : 0.000870s : 0.06% optimize.opt_b.b_1 : 0.000555s : 0.04% optimize.opt_b.b_2 : 0.000024s : 0.00% optimize.opt_b.updatestate_depend_eliminate : 0.000021s : 0.00% optimize.opt_b.updatestate_assign_eliminate : 0.000013s : 0.00% optimize.opt_b.updatestate_loads_eliminate : 0.000014s : 0.00% optimize.opt_b.renormalize : 0.000001s : 0.00% optimize.opt_b.cse : 0.000094s : 0.01% optimize.optimize_parallel_all_gather_comm : 0.000044s : 0.00% optimize.overlap_param_gather : 0.000002s : 0.00% optimize.cconv : 0.000055s : 0.00% optimize.loop_unroll : 0.000607s : 0.04% optimize.opt_after_cconv.c_1 : 0.000130s : 0.01% optimize.opt_after_cconv.parameter_eliminate : 0.000006s : 0.00% optimize.opt_after_cconv.updatestate_depend_eliminate : 0.000021s : 0.00% optimize.opt_after_cconv.updatestate_assign_eliminate : 0.000011s : 0.00% optimize.opt_after_cconv.updatestate_loads_eliminate : 0.000014s : 0.00% optimize.opt_after_cconv.cse : 0.000078s : 0.01% optimize.opt_after_cconv.renormalize : 0.000001s : 0.00% optimize.remove_dup_value : 0.000102s : 0.01% optimize.tuple_transform.d_1 : 0.000206s : 0.01% optimize.tuple_transform.none_parameter_eliminate : 0.000003s : 0.00% optimize.tuple_transform.renormalize : 0.000000s : 0.00% optimize.tuple_transform.switch_simplify : 0.000023s : 0.00% optimize.partial_unused_args_eliminate : 0.000002s : 0.00% optimize.add_recomputation : 0.000153s : 0.01% optimize.cse_after_recomputation.cse : 0.000058s : 0.00% optimize.environ_conv : 0.000024s : 0.00% optimize.swap_dp_allreduce_reducescatter : 0.000020s : 0.00% optimize.bias_add_comm_swap : 0.000003s : 0.00% optimize.label_micro_interleaved_index : 0.000006s : 0.00% optimize.label_fine_grained_interleaved_index : 0.000003s : 0.00% optimize.merge_cast_opt : 0.000002s : 0.00% optimize.slice_recompute_activation : 0.000002s : 0.00% optimize.micro_interleaved_order_control : 0.000003s : 0.00% optimize.assign_add_opt : 0.000002s : 0.00% optimize.ForceFp32Comm : 0.000001s : 0.00% optimize.remove_cast_before_assign_add : 0.000001s : 0.00% optimize.full_micro_interleaved_order_control : 0.000002s : 0.00% optimize.reorder_send_recv_between_fp_bp : 0.000003s : 0.00% optimize.comm_op_add_attrs : 0.000001s : 0.00% optimize.add_comm_op_reuse_tag : 0.000001s : 0.00% optimize.interleave_split_concat_branches : 0.000001s : 0.00% optimize.interleave_parallel_branches : 0.000001s : 0.00% optimize.overlap_opt_shard_in_pipeline : 0.000001s : 0.00% optimize.overlap_opt_shard_grad_in_pipeline : 0.000002s : 0.00% optimize.control_data_broadcast_order : 0.000042s : 0.00% optimize.grouped_pairwise_exchange_alltoall : 0.000002s : 0.00% optimize.offloading_packed_experts : 0.000011s : 0.00% optimize.overlap_recompute_and_grad_model_parallel : 0.000011s : 0.00% optimize.overlap_grad_matmul_and_grad_allreduce : 0.000001s : 0.00% optimize.overlap_recompute_allgather_and_fa_grad : 0.000001s : 0.00% optimize.overlap_recompute_comm : 0.000002s : 0.00% optimize.overlap_grad_ring_attention : 0.000009s : 0.00% optimize.overlap_grad_flash_sp : 0.000051s : 0.00% optimize.begin_end_overlap_inline : 0.000001s : 0.00% optimize.split_matmul_comm_elemetwise : 0.000003s : 0.00% optimize.split_layernorm_comm : 0.000002s : 0.00% optimize.handle_group_info : 0.000001s : 0.00% optimize.symbol_engine_optimizer.build : 0.000292s : 0.02% optimize.symbol_engine_optimizer.elim_shapecalc : 0.000036s : 0.00% optimize.symbol_engine_optimizer.elim_not_effective : 0.000050s : 0.00% optimize.symbol_engine_optimizer.opt_reshape : 0.000034s : 0.00% optimize.symbol_engine_optimizer.fold_const_symbol : 0.000056s : 0.00% optimize.symbol_engine_optimizer.renormalize : 0.000000s : 0.00% detach_backward : 0.000003s : 0.00% pipeline_parallel_scheduler : 0.000002s : 0.00% auto_monad_reorder : 0.000065s : 0.00% get_jit_bprop_graph : 0.000003s : 0.00% rewriter_after_jit_bprop_graph : 0.000006s : 0.00% opt_after_jit_grad : 0.000734s : 0.05% validate : 0.000115s : 0.01% Time group info: ------[substitution.] 0.003631 569 0.49% : 0.000018s : 8: substitution.depend_value_elim 0.37% : 0.000013s : 14: substitution.elim_not_effective 0.49% : 0.000018s : 12: substitution.float_tuple_getitem_switch 0.68% : 0.000025s : 14: substitution.fold_const_symbol 0.44% : 0.000016s : 17: substitution.graph_param_transform 69.06% : 0.002507s : 71: substitution.inline 0.38% : 0.000014s : 29: substitution.j_node_and_user_rematch 1.03% : 0.000037s : 6: substitution.less_batch_normalization 1.62% : 0.000059s : 2: substitution.list_to_tuple_eliminator_ 0.18% : 0.000007s : 6: substitution.load_eliminater 1.03% : 0.000038s : 21: substitution.minmaximum_grad 0.24% : 0.000009s : 2: substitution.opt_reshape 0.54% : 0.000020s : 29: substitution.remove_not_recompute_node 0.24% : 0.000009s : 4: substitution.replace_old_param 1.81% : 0.000066s : 18: substitution.reshape_eliminate 1.37% : 0.000050s : 24: substitution.switch_simplify 4.36% : 0.000158s : 51: substitution.tuple_list_convert_item_index_to_positive 2.30% : 0.000083s : 51: substitution.tuple_list_get_item_const_eliminator 3.01% : 0.000109s : 51: substitution.tuple_list_get_item_depend_reorder 6.03% : 0.000219s : 73: substitution.tuple_list_get_item_eliminator 2.99% : 0.000109s : 51: substitution.tuple_list_get_set_item_eliminator 0.69% : 0.000025s : 6: substitution.updatestate_pure_node_eliminater 0.64% : 0.000023s : 9: substitution.updatestate_useless_node_eliminater ------[type_inference.] 1.285856 2 98.58% : 1.267570s : 1: type_inference.infer 1.42% : 0.018286s : 1: type_inference.specialize ------[replace.] 0.001136 115 4.02% : 0.000046s : 5: replace.depend_value_elim 57.93% : 0.000658s : 71: replace.inline 3.65% : 0.000042s : 2: replace.list_to_tuple_eliminator_ 25.52% : 0.000290s : 24: replace.switch_simplify 8.88% : 0.000101s : 13: replace.tuple_list_get_item_eliminator ------[match.] 0.002594 115 0.11% : 0.000003s : 5: match.depend_value_elim 95.02% : 0.002465s : 71: match.inline 2.18% : 0.000057s : 2: match.list_to_tuple_eliminator_ 1.34% : 0.000035s : 24: match.switch_simplify 1.34% : 0.000035s : 13: match.tuple_list_get_item_eliminator ------[predicate.] 0.001609 9701 1.22% : 0.000020s : 123: predicate.accumulaten_eliminater 0.38% : 0.000006s : 17: predicate.ad_related_special_op_eliminate 0.50% : 0.000008s : 58: predicate.addn_check_dump 1.17% : 0.000019s : 123: predicate.addn_zero_filter 1.18% : 0.000019s : 123: predicate.adjust_all_reduce_mul_add 2.40% : 0.000039s : 181: predicate.arithmetic_simplify 1.21% : 0.000020s : 123: predicate.cast_eliminate 0.33% : 0.000005s : 34: predicate.check_bprop_eliminate 0.50% : 0.000008s : 58: predicate.compare_switch_simplify 0.08% : 0.000001s : 17: predicate.const_output_eliminate 0.59% : 0.000009s : 61: predicate.depend_value_elim 1.22% : 0.000020s : 123: predicate.dict_get_item_const_eliminator 1.31% : 0.000021s : 123: predicate.dict_get_item_eliminator 1.19% : 0.000019s : 123: predicate.dict_set_item_eliminator 0.39% : 0.000006s : 34: predicate.dumpgradient_eliminate 0.09% : 0.000001s : 17: predicate.elim_not_effective 0.21% : 0.000003s : 17: predicate.elim_shapecalc_of_broadcastargs 1.24% : 0.000020s : 140: predicate.environ_add_const_eliminate 1.26% : 0.000020s : 140: predicate.environ_get_add_eliminate 1.26% : 0.000020s : 140: predicate.environ_get_depend_swap 1.82% : 0.000029s : 198: predicate.environ_get_eliminate 1.20% : 0.000019s : 140: predicate.environ_get_set_eliminate 1.89% : 0.000030s : 209: predicate.exchange_switch_depend_value 2.80% : 0.000045s : 209: predicate.float_depend_g_call 0.53% : 0.000008s : 58: predicate.float_environ_get_switch 0.73% : 0.000012s : 75: predicate.float_tuple_getitem_switch 0.08% : 0.000001s : 17: predicate.fold_const_symbol 0.35% : 0.000006s : 35: predicate.get_grad_eliminate 0.11% : 0.000002s : 17: predicate.graph_param_transform 0.53% : 0.000009s : 58: predicate.incorporate_call 0.46% : 0.000007s : 58: predicate.incorporate_call_switch 6.17% : 0.000099s : 459: predicate.inline 0.47% : 0.000008s : 35: predicate.inline_without_move 0.15% : 0.000002s : 35: predicate.j_node_and_user_rematch 0.51% : 0.000008s : 36: predicate.less_batch_normalization 1.67% : 0.000027s : 172: predicate.list_to_tuple_eliminator_ 2.60% : 0.000042s : 295: predicate.load_eliminater 0.36% : 0.000006s : 17: predicate.loop_unroll_after_grad 3.22% : 0.000052s : 314: predicate.loop_unroll_before_grad 1.72% : 0.000028s : 157: predicate.make_slice_get_slice_eliminator 0.54% : 0.000009s : 58: predicate.merge_addn 0.33% : 0.000005s : 34: predicate.micro_step_allgather_replace 0.34% : 0.000005s : 34: predicate.mini_step_allgather_replace 1.19% : 0.000019s : 123: predicate.minmaximum_grad 0.39% : 0.000006s : 17: predicate.mutable_eliminate 0.19% : 0.000003s : 17: predicate.opt_reshape 0.19% : 0.000003s : 17: predicate.parallel_virtual_node 3.21% : 0.000052s : 209: predicate.partial_defer_inline 1.52% : 0.000024s : 155: predicate.partial_eliminate 1.18% : 0.000019s : 123: predicate.print_const_string_wrapper 0.49% : 0.000008s : 53: predicate.reduce_all_const_elim 1.62% : 0.000026s : 123: predicate.reduce_eliminate 2.62% : 0.000042s : 295: predicate.redundant_stop_gradient_eliminater 0.17% : 0.000003s : 35: predicate.remove_not_recompute_node 1.08% : 0.000017s : 172: predicate.replace_applicator 0.20% : 0.000003s : 35: predicate.replace_old_param 0.13% : 0.000002s : 17: predicate.reset_defer_inline 1.32% : 0.000021s : 123: predicate.reshape_eliminate 0.34% : 0.000005s : 34: predicate.row_tensor_add_zeros_like 0.20% : 0.000003s : 17: predicate.row_tensor_eliminate 0.54% : 0.000009s : 34: predicate.same_eliminate 0.26% : 0.000004s : 40: predicate.set_cell_output_no_recompute 0.36% : 0.000006s : 35: predicate.shard_identity_eliminate 0.34% : 0.000006s : 34: predicate.special_op_eliminate 0.60% : 0.000010s : 58: predicate.specialize_transform 0.43% : 0.000007s : 34: predicate.split_environ_get_set_with_tuple_value 0.43% : 0.000007s : 35: predicate.stack_unstack_eliminate 0.19% : 0.000003s : 17: predicate.switch_call_monad_eliminater 2.16% : 0.000035s : 209: predicate.switch_defer_inline 2.35% : 0.000038s : 243: predicate.switch_layer_defer_inline 6.62% : 0.000107s : 646: predicate.switch_simplify 1.18% : 0.000019s : 123: predicate.tile_eliminate 1.19% : 0.000019s : 123: predicate.transpose_eliminate 1.90% : 0.000031s : 157: predicate.tuple_list_convert_item_index_to_positive 1.76% : 0.000028s : 157: predicate.tuple_list_get_item_const_eliminator 1.59% : 0.000026s : 157: predicate.tuple_list_get_item_depend_reorder 3.19% : 0.000051s : 228: predicate.tuple_list_get_item_eliminator 1.63% : 0.000026s : 157: predicate.tuple_list_get_set_item_eliminator 2.29% : 0.000037s : 215: predicate.tuple_list_set_item_eliminator 1.63% : 0.000026s : 170: predicate.tuple_to_list_eliminator_ 2.51% : 0.000040s : 295: predicate.updatestate_pure_node_eliminater 3.25% : 0.000052s : 353: predicate.updatestate_useless_node_eliminater 0.17% : 0.000003s : 17: predicate.value_based_eliminate 0.36% : 0.000006s : 35: predicate.virtual_dataset_eliminate 0.38% : 0.000006s : 35: predicate.virtual_output_eliminate 0.16% : 0.000003s : 17: predicate.virtual_view_grad_eliminate 0.19% : 0.000003s : 17: predicate.zero_like_fill_zero ------[func_graph_cloner_run.] 0.028824 135 75.47% : 0.021754s : 55: func_graph_cloner_run.FuncGraphClonerGraph 24.53% : 0.007070s : 80: func_graph_cloner_run.FuncGraphSpecializer ------[meta_graph.] 0.000000 0 ------[manager.] 0.000000 0 ------[pynative] 0.000000 0 ------[others.] 1.752552 209 0.00% : 0.000003s : 1: ForceFp32Comm 2.60% : 0.045608s : 1: add_attr 2.60% : 0.045587s : 1: add_attr_with_inline 0.00% : 0.000005s : 1: add_comm_op_reuse_tag 0.01% : 0.000161s : 1: add_recomputation 0.00% : 0.000005s : 1: assign_add_opt 0.04% : 0.000723s : 1: auto_monad 0.00% : 0.000072s : 1: auto_monad_reorder 0.00% : 0.000005s : 1: begin_end_overlap_inline 0.00% : 0.000006s : 1: bias_add_comm_swap 4.62% : 0.081046s : 1: bootstrap 0.00% : 0.000060s : 1: cconv 0.00% : 0.000004s : 1: comm_op_add_attrs 0.00% : 0.000045s : 1: control_data_broadcast_order 0.00% : 0.000036s : 1: convert_after_rewriter 0.00% : 0.000078s : 1: cse_after_recomputation 0.00% : 0.000006s : 1: dataset_repeat_opt 0.00% : 0.000006s : 1: detach_backward 0.00% : 0.000027s : 1: environ_conv 0.01% : 0.000218s : 1: event_method 0.00% : 0.000006s : 1: full_micro_interleaved_order_control 0.00% : 0.000007s : 1: get_jit_bprop_graph 0.00% : 0.000053s : 1: graph_reusing 0.00% : 0.000004s : 1: grouped_pairwise_exchange_alltoall 0.00% : 0.000005s : 1: handle_group_info 0.00% : 0.000006s : 1: inline 0.00% : 0.000006s : 1: insert-virtual-dataset 0.00% : 0.000004s : 1: interleave_parallel_branches 0.00% : 0.000004s : 1: interleave_split_concat_branches 0.00% : 0.000006s : 1: label_fine_grained_interleaved_index 0.00% : 0.000009s : 1: label_micro_interleaved_index 0.04% : 0.000617s : 1: loop_unroll 0.00% : 0.000005s : 1: merge_cast_opt 0.00% : 0.000007s : 1: micro_interleaved_order_control 0.05% : 0.000883s : 1: mutable_eliminate 0.00% : 0.000015s : 1: offloading_packed_experts 0.00% : 0.000039s : 1: opt.transform.loop_unroll_optimizer 0.00% : 0.000044s : 1: opt.transform.mutable_eliminate 0.76% : 0.013378s : 95: opt.transform.opt_a 0.01% : 0.000129s : 1: opt.transform.opt_after_cconv 0.00% : 0.000078s : 1: opt.transform.opt_after_jit_grad 0.03% : 0.000543s : 28: opt.transform.opt_b 0.01% : 0.000226s : 2: opt.transform.opt_trans_graph 0.01% : 0.000170s : 4: opt.transform.symbol_engine_opt 6.53% : 0.114487s : 1: opt_a 0.02% : 0.000312s : 1: opt_after_cconv 0.04% : 0.000748s : 1: opt_after_jit_grad 0.04% : 0.000782s : 1: opt_b 6.89% : 0.120800s : 1: optimize 0.00% : 0.000050s : 1: optimize_parallel_all_gather_comm 0.00% : 0.000020s : 1: order_py_execute_after_rewriter 0.00% : 0.000056s : 1: overlap_grad_flash_sp 0.00% : 0.000004s : 1: overlap_grad_matmul_and_grad_allreduce 0.00% : 0.000012s : 1: overlap_grad_ring_attention 0.00% : 0.000005s : 1: overlap_opt_shard_grad_in_pipeline 0.00% : 0.000004s : 1: overlap_opt_shard_in_pipeline 0.00% : 0.000007s : 1: overlap_param_gather 0.00% : 0.000004s : 1: overlap_recompute_allgather_and_fa_grad 0.00% : 0.000014s : 1: overlap_recompute_and_grad_model_parallel 0.00% : 0.000006s : 1: overlap_recompute_comm 0.00% : 0.000011s : 1: parallel-infer-symbol 0.00% : 0.000004s : 1: parallel-infer-symbol-second 0.00% : 0.000006s : 1: partial_unused_args_eliminate 0.00% : 0.000005s : 1: pipeline_parallel_scheduler 0.00% : 0.000005s : 1: pipeline_split 0.02% : 0.000267s : 1: pre_auto_parallel 0.01% : 0.000242s : 1: py_interpret_to_execute 0.00% : 0.000042s : 1: py_interpret_to_execute_after_opt_a 0.00% : 0.000004s : 1: remove_cast_before_assign_add 0.01% : 0.000108s : 1: remove_dup_value 1.64% : 0.028812s : 1: renormalize.infer 0.40% : 0.006935s : 1: renormalize.specialize 0.00% : 0.000007s : 1: reorder_send_recv_between_fp_bp 0.00% : 0.000009s : 1: rewriter_after_jit_bprop_graph 0.05% : 0.000913s : 1: rewriter_after_opt_a 0.05% : 0.000793s : 1: rewriter_before_opt_a 0.00% : 0.000006s : 1: slice_cell_reuse_recomputed_activation 0.00% : 0.000005s : 1: slice_recompute_activation 0.00% : 0.000005s : 1: split_layernorm_comm 0.00% : 0.000005s : 1: split_matmul_comm_elemetwise 0.00% : 0.000023s : 1: swap_dp_allreduce_reducescatter 0.03% : 0.000524s : 1: symbol_engine_optimizer 0.02% : 0.000266s : 1: tuple_transform 73.40% : 1.286296s : 1: type_inference . [hook] pytest_runtest_teardown:test_matmul_qkv_bf16_0[True-mstype0-k_n_shape0-16] tests/st/infer/ops/test_internal_ops/test_matmul_split.py::test_matmul_qkv_bf16_0[True-mstype0-k_n_shape0-16],max_mem:208.0M [WARNING] ME(152078:281472851701552,MainProcess):2026-01-29-17:43:18.615.308 [mindspore/context.py:1334] For 'context.set_context', the parameter 'device_target' will be deprecated and removed in a future version. Please use the api mindspore.set_device() instead. TotalTime = 0.8831, [21] [bootstrap]: 0.00079502 [type_inference]: 0.740024 [event_method]: 0.00020367 [auto_monad]: 0.00237314 [graph_reusing]: 5.396e-05 [inline]: 2.98e-06 [add_attr]: 0.00521606, [1] [add_attr_with_inline]: 0.00519986, [1] [Cycle 1]: 0.00030879, [2] [tag_attr]: 0.00020759 [meta_addattr_fg_expand]: 4.977e-05 [parallel-infer-symbol]: 4.02e-06 [pre_auto_parallel]: 0.00024497 [insert-virtual-dataset]: 2.43998e-06 [parallel-infer-symbol-second]: 1.34e-06 [dataset_repeat_opt]: 2.11e-06 [pipeline_split]: 1.74e-06 [optimize]: 0.13307, [53] [py_interpret_to_execute]: 0.00021821 [rewriter_before_opt_a]: 0.00076174 [opt_a]: 0.127065, [2] [Cycle 1]: 0.071271, [45] [expand_dump_flag]: 1.586e-05 [switch_simplify]: 0.00082413 [loop_unroll]: 0.00034528 [a_1]: 0.0518144 [with_stream_mark]: 5.675e-05 [recompute_prepare]: 4.067e-05 [updatestate_depend_eliminate]: 1.976e-05 [updatestate_assign_eliminate]: 1.56e-05 [updatestate_loads_eliminate]: 1.834e-05 [parameter_eliminate]: 3.45e-06 [a_2]: 0.00083922 [accelerated_algorithm]: 6.653e-05 [shard]: 3.06999e-06 [meta_shard_fg_expand]: 2.112e-05 [shard_inline]: 2.297e-05 [merge_send_recv]: 2.057e-05 [auto_parallel]: 2.018e-05 [parallel]: 3.237e-05 [flash_sp]: 1.539e-05 [merge_comm]: 1.382e-05 [allreduce_fusion]: 1.25e-05 [matmul_add_comm_reduction]: 2.1e-05 [allreduce_slice_to_reducescatter]: 1.05999e-06 [virtual_shard_identity]: 2.692e-05 [virtual_dataset]: 2.334e-05 [get_grad_eliminate_]: 2.162e-05 [virtual_output]: 2.177e-05 [merge_forward]: 1.318e-05 [cell_reuse_recompute_pass]: 2.61999e-06 [offload_activation]: 2.246e-05 [cell_reuse_handle_not_recompute_node_pass]: 4.327e-05 [merge_recompute_call_nodes]: 1.47999e-06 [before_grad]: 3.989e-05 [set_forward_comm_id_for_comm_node_pass]: 1.415e-05 [meta_fg_expand]: 1.526e-05 [flash_sp_send_recv_attached]: 6.21998e-06 [receive_attached]: 2.34999e-06 [after_resolve]: 2.978e-05 [a_after_grad]: 3.557e-05 [renormalize]: 0.0118854 [add_forward_monad_depend]: 2.117e-05 [auto_monad_grad]: 3.08e-06 [auto_monad_eliminator]: 8.087e-05 [cse]: 0.00400218 [a_3]: 0.00017929 [Cycle 2]: 0.0557755, [45] [expand_dump_flag]: 2.98e-06 [switch_simplify]: 2.639e-05 [loop_unroll]: 2.156e-05 [a_1]: 0.00064615 [with_stream_mark]: 3.948e-05 [recompute_prepare]: 2.157e-05 [updatestate_depend_eliminate]: 1.633e-05 [updatestate_assign_eliminate]: 1.204e-05 [updatestate_loads_eliminate]: 1.419e-05 [parameter_eliminate]: 2.64999e-06 [a_2]: 0.00030727 [accelerated_algorithm]: 3.01e-05 [shard]: 2.16e-06 [meta_shard_fg_expand]: 8.97999e-06 [shard_inline]: 2.024e-05 [merge_send_recv]: 1.968e-05 [auto_parallel]: 1.966e-05 [parallel]: 1.054e-05 [flash_sp]: 3.95e-06 [merge_comm]: 1.281e-05 [allreduce_fusion]: 1.205e-05 [matmul_add_comm_reduction]: 2.195e-05 [allreduce_slice_to_reducescatter]: 1.21002e-06 [virtual_shard_identity]: 2.222e-05 [virtual_dataset]: 1.998e-05 [get_grad_eliminate_]: 2.063e-05 [virtual_output]: 1.912e-05 [merge_forward]: 1.225e-05 [cell_reuse_recompute_pass]: 2.86e-06 [offload_activation]: 2.342e-05 [cell_reuse_handle_not_recompute_node_pass]: 3.899e-05 [merge_recompute_call_nodes]: 1.36002e-06 [before_grad]: 3.541e-05 [set_forward_comm_id_for_comm_node_pass]: 1.31e-05 [meta_fg_expand]: 9.34e-06 [flash_sp_send_recv_attached]: 1.70001e-06 [receive_attached]: 2.51e-06 [after_resolve]: 2.66e-05 [a_after_grad]: 3.166e-05 [renormalize]: 6.99947e-08 [add_forward_monad_depend]: 3.52997e-06 [auto_monad_grad]: 3.33e-06 [auto_monad_eliminator]: 0.0536426 [cse]: 0.00012531 [a_3]: 0.00017719 [py_interpret_to_execute_after_opt_a]: 4.284e-05 [slice_cell_reuse_recomputed_activation]: 3.60998e-06 [rewriter_after_opt_a]: 0.00088909 [convert_after_rewriter]: 2.391e-05 [order_py_execute_after_rewriter]: 1.437e-05 [mutable_eliminate]: 0.00078265 [opt_b]: 0.00077112, [1] [Cycle 1]: 0.00076164, [7] [b_1]: 0.00055796 [b_2]: 2.469e-05 [updatestate_depend_eliminate]: 1.904e-05 [updatestate_assign_eliminate]: 1.188e-05 [updatestate_loads_eliminate]: 1.477e-05 [renormalize]: 1.19e-06 [cse]: 8.887e-05 [optimize_parallel_all_gather_comm]: 4.195e-05 [overlap_param_gather]: 2.14e-06 [cconv]: 4.184e-05 [loop_unroll]: 0.00054009 [opt_after_cconv]: 0.00029535, [1] [Cycle 1]: 0.0002862, [7] [c_1]: 0.00012778 [parameter_eliminate]: 5.37001e-06 [updatestate_depend_eliminate]: 1.882e-05 [updatestate_assign_eliminate]: 1.054e-05 [updatestate_loads_eliminate]: 1.283e-05 [cse]: 7.207e-05 [renormalize]: 5.29981e-07 [remove_dup_value]: 9.791e-05 [tuple_transform]: 0.0002468, [1] [Cycle 1]: 0.00024102, [4] [d_1]: 0.00019547 [none_parameter_eliminate]: 2.76e-06 [renormalize]: 2.00002e-07 [switch_simplify]: 2.151e-05 [partial_unused_args_eliminate]: 1.86e-06 [add_recomputation]: 0.00015101 [cse_after_recomputation]: 6.935e-05, [1] [Cycle 1]: 6.306e-05, [1] [cse]: 5.623e-05 [environ_conv]: 2.185e-05 [swap_dp_allreduce_reducescatter]: 1.776e-05 [bias_add_comm_swap]: 3.4e-06 [label_micro_interleaved_index]: 5.36998e-06 [label_fine_grained_interleaved_index]: 2.92002e-06 [merge_cast_opt]: 1.33002e-06 [slice_recompute_activation]: 2.24999e-06 [micro_interleaved_order_control]: 2.70002e-06 [assign_add_opt]: 1.57999e-06 [ForceFp32Comm]: 1.05001e-06 [remove_cast_before_assign_add]: 1.07998e-06 [full_micro_interleaved_order_control]: 1.97999e-06 [reorder_send_recv_between_fp_bp]: 2.73998e-06 [comm_op_add_attrs]: 1.15001e-06 [add_comm_op_reuse_tag]: 9.70002e-07 [interleave_split_concat_branches]: 1.19e-06 [interleave_parallel_branches]: 1.09e-06 [overlap_opt_shard_in_pipeline]: 1.22999e-06 [overlap_opt_shard_grad_in_pipeline]: 1.97999e-06 [control_data_broadcast_order]: 3.84e-05 [grouped_pairwise_exchange_alltoall]: 1.50001e-06 [offloading_packed_experts]: 1.128e-05 [overlap_recompute_and_grad_model_parallel]: 1.063e-05 [overlap_grad_matmul_and_grad_allreduce]: 1.52001e-06 [overlap_recompute_allgather_and_fa_grad]: 1.35001e-06 [overlap_recompute_comm]: 2.13998e-06 [overlap_grad_ring_attention]: 9.12999e-06 [overlap_grad_flash_sp]: 5.057e-05 [begin_end_overlap_inline]: 5.50004e-07 [split_matmul_comm_elemetwise]: 2.07999e-06 [split_layernorm_comm]: 1.65001e-06 [handle_group_info]: 9.09989e-07 [symbol_engine_optimizer]: 0.00050585, [1] [Cycle 1]: 0.00050048, [6] [build]: 0.00028289 [elim_shapecalc]: 3.293e-05 [elim_not_effective]: 4.813e-05 [opt_reshape]: 3.348e-05 [fold_const_symbol]: 5.864e-05 [renormalize]: 4.00003e-07 [detach_backward]: 2.66e-06 [pipeline_parallel_scheduler]: 1.60999e-06 [auto_monad_reorder]: 6.141e-05 [get_jit_bprop_graph]: 1.76e-06 [rewriter_after_jit_bprop_graph]: 5.44998e-06 [opt_after_jit_grad]: 0.00066915 [validate]: 9.714e-05 Sums bootstrap : 0.000795s : 0.09% type_inference : 0.740024s : 84.42% event_method : 0.000204s : 0.02% auto_monad : 0.002373s : 0.27% graph_reusing : 0.000054s : 0.01% inline : 0.000003s : 0.00% add_attr.add_attr_with_inline.tag_attr : 0.000208s : 0.02% add_attr.add_attr_with_inline.meta_addattr_fg_expand : 0.000050s : 0.01% parallel-infer-symbol : 0.000004s : 0.00% pre_auto_parallel : 0.000245s : 0.03% insert-virtual-dataset : 0.000002s : 0.00% parallel-infer-symbol-second : 0.000001s : 0.00% dataset_repeat_opt : 0.000002s : 0.00% pipeline_split : 0.000002s : 0.00% optimize.py_interpret_to_execute : 0.000218s : 0.02% optimize.rewriter_before_opt_a : 0.000762s : 0.09% optimize.opt_a.expand_dump_flag : 0.000019s : 0.00% optimize.opt_a.switch_simplify : 0.000851s : 0.10% optimize.opt_a.loop_unroll : 0.000367s : 0.04% optimize.opt_a.a_1 : 0.052461s : 5.98% optimize.opt_a.with_stream_mark : 0.000096s : 0.01% optimize.opt_a.recompute_prepare : 0.000062s : 0.01% optimize.opt_a.updatestate_depend_eliminate : 0.000036s : 0.00% optimize.opt_a.updatestate_assign_eliminate : 0.000028s : 0.00% optimize.opt_a.updatestate_loads_eliminate : 0.000033s : 0.00% optimize.opt_a.parameter_eliminate : 0.000006s : 0.00% optimize.opt_a.a_2 : 0.001146s : 0.13% optimize.opt_a.accelerated_algorithm : 0.000097s : 0.01% optimize.opt_a.shard : 0.000005s : 0.00% optimize.opt_a.meta_shard_fg_expand : 0.000030s : 0.00% optimize.opt_a.shard_inline : 0.000043s : 0.00% optimize.opt_a.merge_send_recv : 0.000040s : 0.00% optimize.opt_a.auto_parallel : 0.000040s : 0.00% optimize.opt_a.parallel : 0.000043s : 0.00% optimize.opt_a.flash_sp : 0.000019s : 0.00% optimize.opt_a.merge_comm : 0.000027s : 0.00% optimize.opt_a.allreduce_fusion : 0.000025s : 0.00% optimize.opt_a.matmul_add_comm_reduction : 0.000043s : 0.00% optimize.opt_a.allreduce_slice_to_reducescatter : 0.000002s : 0.00% optimize.opt_a.virtual_shard_identity : 0.000049s : 0.01% optimize.opt_a.virtual_dataset : 0.000043s : 0.00% optimize.opt_a.get_grad_eliminate_ : 0.000042s : 0.00% optimize.opt_a.virtual_output : 0.000041s : 0.00% optimize.opt_a.merge_forward : 0.000025s : 0.00% optimize.opt_a.cell_reuse_recompute_pass : 0.000005s : 0.00% optimize.opt_a.offload_activation : 0.000046s : 0.01% optimize.opt_a.cell_reuse_handle_not_recompute_node_pass : 0.000082s : 0.01% optimize.opt_a.merge_recompute_call_nodes : 0.000003s : 0.00% optimize.opt_a.before_grad : 0.000075s : 0.01% optimize.opt_a.set_forward_comm_id_for_comm_node_pass : 0.000027s : 0.00% optimize.opt_a.meta_fg_expand : 0.000025s : 0.00% optimize.opt_a.flash_sp_send_recv_attached : 0.000008s : 0.00% optimize.opt_a.receive_attached : 0.000005s : 0.00% optimize.opt_a.after_resolve : 0.000056s : 0.01% optimize.opt_a.a_after_grad : 0.000067s : 0.01% optimize.opt_a.renormalize : 0.011885s : 1.36% optimize.opt_a.add_forward_monad_depend : 0.000025s : 0.00% optimize.opt_a.auto_monad_grad : 0.000006s : 0.00% optimize.opt_a.auto_monad_eliminator : 0.053723s : 6.13% optimize.opt_a.cse : 0.004127s : 0.47% optimize.opt_a.a_3 : 0.000356s : 0.04% optimize.py_interpret_to_execute_after_opt_a : 0.000043s : 0.00% optimize.slice_cell_reuse_recomputed_activation : 0.000004s : 0.00% optimize.rewriter_after_opt_a : 0.000889s : 0.10% optimize.convert_after_rewriter : 0.000024s : 0.00% optimize.order_py_execute_after_rewriter : 0.000014s : 0.00% optimize.mutable_eliminate : 0.000783s : 0.09% optimize.opt_b.b_1 : 0.000558s : 0.06% optimize.opt_b.b_2 : 0.000025s : 0.00% optimize.opt_b.updatestate_depend_eliminate : 0.000019s : 0.00% optimize.opt_b.updatestate_assign_eliminate : 0.000012s : 0.00% optimize.opt_b.updatestate_loads_eliminate : 0.000015s : 0.00% optimize.opt_b.renormalize : 0.000001s : 0.00% optimize.opt_b.cse : 0.000089s : 0.01% optimize.optimize_parallel_all_gather_comm : 0.000042s : 0.00% optimize.overlap_param_gather : 0.000002s : 0.00% optimize.cconv : 0.000042s : 0.00% optimize.loop_unroll : 0.000540s : 0.06% optimize.opt_after_cconv.c_1 : 0.000128s : 0.01% optimize.opt_after_cconv.parameter_eliminate : 0.000005s : 0.00% optimize.opt_after_cconv.updatestate_depend_eliminate : 0.000019s : 0.00% optimize.opt_after_cconv.updatestate_assign_eliminate : 0.000011s : 0.00% optimize.opt_after_cconv.updatestate_loads_eliminate : 0.000013s : 0.00% optimize.opt_after_cconv.cse : 0.000072s : 0.01% optimize.opt_after_cconv.renormalize : 0.000001s : 0.00% optimize.remove_dup_value : 0.000098s : 0.01% optimize.tuple_transform.d_1 : 0.000195s : 0.02% optimize.tuple_transform.none_parameter_eliminate : 0.000003s : 0.00% optimize.tuple_transform.renormalize : 0.000000s : 0.00% optimize.tuple_transform.switch_simplify : 0.000022s : 0.00% optimize.partial_unused_args_eliminate : 0.000002s : 0.00% optimize.add_recomputation : 0.000151s : 0.02% optimize.cse_after_recomputation.cse : 0.000056s : 0.01% optimize.environ_conv : 0.000022s : 0.00% optimize.swap_dp_allreduce_reducescatter : 0.000018s : 0.00% optimize.bias_add_comm_swap : 0.000003s : 0.00% optimize.label_micro_interleaved_index : 0.000005s : 0.00% optimize.label_fine_grained_interleaved_index : 0.000003s : 0.00% optimize.merge_cast_opt : 0.000001s : 0.00% optimize.slice_recompute_activation : 0.000002s : 0.00% optimize.micro_interleaved_order_control : 0.000003s : 0.00% optimize.assign_add_opt : 0.000002s : 0.00% optimize.ForceFp32Comm : 0.000001s : 0.00% optimize.remove_cast_before_assign_add : 0.000001s : 0.00% optimize.full_micro_interleaved_order_control : 0.000002s : 0.00% optimize.reorder_send_recv_between_fp_bp : 0.000003s : 0.00% optimize.comm_op_add_attrs : 0.000001s : 0.00% optimize.add_comm_op_reuse_tag : 0.000001s : 0.00% optimize.interleave_split_concat_branches : 0.000001s : 0.00% optimize.interleave_parallel_branches : 0.000001s : 0.00% optimize.overlap_opt_shard_in_pipeline : 0.000001s : 0.00% optimize.overlap_opt_shard_grad_in_pipeline : 0.000002s : 0.00% optimize.control_data_broadcast_order : 0.000038s : 0.00% optimize.grouped_pairwise_exchange_alltoall : 0.000002s : 0.00% optimize.offloading_packed_experts : 0.000011s : 0.00% optimize.overlap_recompute_and_grad_model_parallel : 0.000011s : 0.00% optimize.overlap_grad_matmul_and_grad_allreduce : 0.000002s : 0.00% optimize.overlap_recompute_allgather_and_fa_grad : 0.000001s : 0.00% optimize.overlap_recompute_comm : 0.000002s : 0.00% optimize.overlap_grad_ring_attention : 0.000009s : 0.00% optimize.overlap_grad_flash_sp : 0.000051s : 0.01% optimize.begin_end_overlap_inline : 0.000001s : 0.00% optimize.split_matmul_comm_elemetwise : 0.000002s : 0.00% optimize.split_layernorm_comm : 0.000002s : 0.00% optimize.handle_group_info : 0.000001s : 0.00% optimize.symbol_engine_optimizer.build : 0.000283s : 0.03% optimize.symbol_engine_optimizer.elim_shapecalc : 0.000033s : 0.00% optimize.symbol_engine_optimizer.elim_not_effective : 0.000048s : 0.01% optimize.symbol_engine_optimizer.opt_reshape : 0.000033s : 0.00% optimize.symbol_engine_optimizer.fold_const_symbol : 0.000059s : 0.01% optimize.symbol_engine_optimizer.renormalize : 0.000000s : 0.00% detach_backward : 0.000003s : 0.00% pipeline_parallel_scheduler : 0.000002s : 0.00% auto_monad_reorder : 0.000061s : 0.01% get_jit_bprop_graph : 0.000002s : 0.00% rewriter_after_jit_bprop_graph : 0.000005s : 0.00% opt_after_jit_grad : 0.000669s : 0.08% validate : 0.000097s : 0.01% Time group info: ------[substitution.] 0.003545 569 0.46% : 0.000016s : 8: substitution.depend_value_elim 0.34% : 0.000012s : 14: substitution.elim_not_effective 0.55% : 0.000019s : 12: substitution.float_tuple_getitem_switch 0.78% : 0.000028s : 14: substitution.fold_const_symbol 0.41% : 0.000015s : 17: substitution.graph_param_transform 68.03% : 0.002412s : 71: substitution.inline 0.38% : 0.000013s : 29: substitution.j_node_and_user_rematch 1.06% : 0.000037s : 6: substitution.less_batch_normalization 1.70% : 0.000060s : 2: substitution.list_to_tuple_eliminator_ 0.18% : 0.000006s : 6: substitution.load_eliminater 0.89% : 0.000032s : 21: substitution.minmaximum_grad 0.22% : 0.000008s : 2: substitution.opt_reshape 0.55% : 0.000019s : 29: substitution.remove_not_recompute_node 0.21% : 0.000008s : 4: substitution.replace_old_param 1.78% : 0.000063s : 18: substitution.reshape_eliminate 1.48% : 0.000052s : 24: substitution.switch_simplify 4.57% : 0.000162s : 51: substitution.tuple_list_convert_item_index_to_positive 2.51% : 0.000089s : 51: substitution.tuple_list_get_item_const_eliminator 3.46% : 0.000123s : 51: substitution.tuple_list_get_item_depend_reorder 6.05% : 0.000214s : 73: substitution.tuple_list_get_item_eliminator 3.17% : 0.000112s : 51: substitution.tuple_list_get_set_item_eliminator 0.63% : 0.000022s : 6: substitution.updatestate_pure_node_eliminater 0.60% : 0.000021s : 9: substitution.updatestate_useless_node_eliminater ------[type_inference.] 0.739669 2 96.22% : 0.711696s : 1: type_inference.infer 3.78% : 0.027973s : 1: type_inference.specialize ------[replace.] 0.001123 115 4.24% : 0.000048s : 5: replace.depend_value_elim 55.70% : 0.000626s : 71: replace.inline 4.12% : 0.000046s : 2: replace.list_to_tuple_eliminator_ 26.95% : 0.000303s : 24: replace.switch_simplify 8.99% : 0.000101s : 13: replace.tuple_list_get_item_eliminator ------[match.] 0.002501 115 0.09% : 0.000002s : 5: match.depend_value_elim 94.73% : 0.002369s : 71: match.inline 2.31% : 0.000058s : 2: match.list_to_tuple_eliminator_ 1.44% : 0.000036s : 24: match.switch_simplify 1.43% : 0.000036s : 13: match.tuple_list_get_item_eliminator ------[predicate.] 0.001662 9701 1.19% : 0.000020s : 123: predicate.accumulaten_eliminater 0.34% : 0.000006s : 17: predicate.ad_related_special_op_eliminate 0.50% : 0.000008s : 58: predicate.addn_check_dump 1.17% : 0.000019s : 123: predicate.addn_zero_filter 1.10% : 0.000018s : 123: predicate.adjust_all_reduce_mul_add 2.22% : 0.000037s : 181: predicate.arithmetic_simplify 1.21% : 0.000020s : 123: predicate.cast_eliminate 0.32% : 0.000005s : 34: predicate.check_bprop_eliminate 0.50% : 0.000008s : 58: predicate.compare_switch_simplify 0.08% : 0.000001s : 17: predicate.const_output_eliminate 0.57% : 0.000009s : 61: predicate.depend_value_elim 1.17% : 0.000019s : 123: predicate.dict_get_item_const_eliminator 1.28% : 0.000021s : 123: predicate.dict_get_item_eliminator 1.10% : 0.000018s : 123: predicate.dict_set_item_eliminator 0.35% : 0.000006s : 34: predicate.dumpgradient_eliminate 0.09% : 0.000001s : 17: predicate.elim_not_effective 0.21% : 0.000003s : 17: predicate.elim_shapecalc_of_broadcastargs 1.27% : 0.000021s : 140: predicate.environ_add_const_eliminate 1.21% : 0.000020s : 140: predicate.environ_get_add_eliminate 1.17% : 0.000020s : 140: predicate.environ_get_depend_swap 1.82% : 0.000030s : 198: predicate.environ_get_eliminate 1.19% : 0.000020s : 140: predicate.environ_get_set_eliminate 1.87% : 0.000031s : 209: predicate.exchange_switch_depend_value 2.63% : 0.000044s : 209: predicate.float_depend_g_call 0.51% : 0.000008s : 58: predicate.float_environ_get_switch 0.70% : 0.000012s : 75: predicate.float_tuple_getitem_switch 0.07% : 0.000001s : 17: predicate.fold_const_symbol 0.34% : 0.000006s : 35: predicate.get_grad_eliminate 0.10% : 0.000002s : 17: predicate.graph_param_transform 0.49% : 0.000008s : 58: predicate.incorporate_call 0.44% : 0.000007s : 58: predicate.incorporate_call_switch 5.73% : 0.000095s : 459: predicate.inline 0.39% : 0.000006s : 35: predicate.inline_without_move 0.15% : 0.000002s : 35: predicate.j_node_and_user_rematch 0.46% : 0.000008s : 36: predicate.less_batch_normalization 1.96% : 0.000033s : 172: predicate.list_to_tuple_eliminator_ 2.59% : 0.000043s : 295: predicate.load_eliminater 0.34% : 0.000006s : 17: predicate.loop_unroll_after_grad 3.17% : 0.000053s : 314: predicate.loop_unroll_before_grad 1.47% : 0.000024s : 157: predicate.make_slice_get_slice_eliminator 0.50% : 0.000008s : 58: predicate.merge_addn 0.31% : 0.000005s : 34: predicate.micro_step_allgather_replace 0.29% : 0.000005s : 34: predicate.mini_step_allgather_replace 1.22% : 0.000020s : 123: predicate.minmaximum_grad 0.42% : 0.000007s : 17: predicate.mutable_eliminate 0.20% : 0.000003s : 17: predicate.opt_reshape 0.17% : 0.000003s : 17: predicate.parallel_virtual_node 3.09% : 0.000051s : 209: predicate.partial_defer_inline 1.73% : 0.000029s : 155: predicate.partial_eliminate 1.11% : 0.000018s : 123: predicate.print_const_string_wrapper 0.48% : 0.000008s : 53: predicate.reduce_all_const_elim 1.57% : 0.000026s : 123: predicate.reduce_eliminate 2.55% : 0.000042s : 295: predicate.redundant_stop_gradient_eliminater 0.15% : 0.000003s : 35: predicate.remove_not_recompute_node 1.17% : 0.000019s : 172: predicate.replace_applicator 0.17% : 0.000003s : 35: predicate.replace_old_param 0.11% : 0.000002s : 17: predicate.reset_defer_inline 1.36% : 0.000023s : 123: predicate.reshape_eliminate 0.36% : 0.000006s : 34: predicate.row_tensor_add_zeros_like 0.21% : 0.000003s : 17: predicate.row_tensor_eliminate 0.52% : 0.000009s : 34: predicate.same_eliminate 0.21% : 0.000003s : 40: predicate.set_cell_output_no_recompute 0.37% : 0.000006s : 35: predicate.shard_identity_eliminate 0.32% : 0.000005s : 34: predicate.special_op_eliminate 0.60% : 0.000010s : 58: predicate.specialize_transform 0.46% : 0.000008s : 34: predicate.split_environ_get_set_with_tuple_value 0.35% : 0.000006s : 35: predicate.stack_unstack_eliminate 0.17% : 0.000003s : 17: predicate.switch_call_monad_eliminater 2.13% : 0.000035s : 209: predicate.switch_defer_inline 2.25% : 0.000037s : 243: predicate.switch_layer_defer_inline 6.51% : 0.000108s : 646: predicate.switch_simplify 1.18% : 0.000020s : 123: predicate.tile_eliminate 1.11% : 0.000018s : 123: predicate.transpose_eliminate 4.76% : 0.000079s : 157: predicate.tuple_list_convert_item_index_to_positive 1.70% : 0.000028s : 157: predicate.tuple_list_get_item_const_eliminator 1.57% : 0.000026s : 157: predicate.tuple_list_get_item_depend_reorder 3.11% : 0.000052s : 228: predicate.tuple_list_get_item_eliminator 1.64% : 0.000027s : 157: predicate.tuple_list_get_set_item_eliminator 2.18% : 0.000036s : 215: predicate.tuple_list_set_item_eliminator 1.55% : 0.000026s : 170: predicate.tuple_to_list_eliminator_ 2.50% : 0.000041s : 295: predicate.updatestate_pure_node_eliminater 3.05% : 0.000051s : 353: predicate.updatestate_useless_node_eliminater 0.16% : 0.000003s : 17: predicate.value_based_eliminate 0.32% : 0.000005s : 35: predicate.virtual_dataset_eliminate 0.34% : 0.000006s : 35: predicate.virtual_output_eliminate 0.16% : 0.000003s : 17: predicate.virtual_view_grad_eliminate 0.18% : 0.000003s : 17: predicate.zero_like_fill_zero ------[func_graph_cloner_run.] 0.047486 135 73.03% : 0.034680s : 55: func_graph_cloner_run.FuncGraphClonerGraph 26.97% : 0.012805s : 80: func_graph_cloner_run.FuncGraphSpecializer ------[meta_graph.] 0.000000 0 ------[manager.] 0.000000 0 ------[pynative] 0.000000 0 ------[others.] 1.089908 209 0.00% : 0.000004s : 1: ForceFp32Comm 0.48% : 0.005223s : 1: add_attr 0.48% : 0.005205s : 1: add_attr_with_inline 0.00% : 0.000005s : 1: add_comm_op_reuse_tag 0.01% : 0.000157s : 1: add_recomputation 0.00% : 0.000006s : 1: assign_add_opt 0.22% : 0.002404s : 1: auto_monad 0.01% : 0.000066s : 1: auto_monad_reorder 0.00% : 0.000005s : 1: begin_end_overlap_inline 0.00% : 0.000006s : 1: bias_add_comm_swap 0.08% : 0.000854s : 1: bootstrap 0.00% : 0.000045s : 1: cconv 0.00% : 0.000004s : 1: comm_op_add_attrs 0.00% : 0.000042s : 1: control_data_broadcast_order 0.00% : 0.000029s : 1: convert_after_rewriter 0.01% : 0.000072s : 1: cse_after_recomputation 0.00% : 0.000005s : 1: dataset_repeat_opt 0.00% : 0.000006s : 1: detach_backward 0.00% : 0.000025s : 1: environ_conv 0.02% : 0.000216s : 1: event_method 0.00% : 0.000006s : 1: full_micro_interleaved_order_control 0.00% : 0.000006s : 1: get_jit_bprop_graph 0.01% : 0.000063s : 1: graph_reusing 0.00% : 0.000005s : 1: grouped_pairwise_exchange_alltoall 0.00% : 0.000005s : 1: handle_group_info 0.00% : 0.000007s : 1: inline 0.00% : 0.000006s : 1: insert-virtual-dataset 0.00% : 0.000004s : 1: interleave_parallel_branches 0.00% : 0.000004s : 1: interleave_split_concat_branches 0.00% : 0.000006s : 1: label_fine_grained_interleaved_index 0.00% : 0.000008s : 1: label_micro_interleaved_index 0.05% : 0.000550s : 1: loop_unroll 0.00% : 0.000005s : 1: merge_cast_opt 0.00% : 0.000006s : 1: micro_interleaved_order_control 0.07% : 0.000793s : 1: mutable_eliminate 0.00% : 0.000015s : 1: offloading_packed_experts 0.00% : 0.000036s : 1: opt.transform.loop_unroll_optimizer 0.00% : 0.000042s : 1: opt.transform.mutable_eliminate 5.11% : 0.055682s : 95: opt.transform.opt_a 0.01% : 0.000126s : 1: opt.transform.opt_after_cconv 0.01% : 0.000073s : 1: opt.transform.opt_after_jit_grad 0.05% : 0.000544s : 28: opt.transform.opt_b 0.02% : 0.000214s : 2: opt.transform.opt_trans_graph 0.02% : 0.000168s : 4: opt.transform.symbol_engine_opt 11.66% : 0.127070s : 1: opt_a 0.03% : 0.000300s : 1: opt_after_cconv 0.06% : 0.000680s : 1: opt_after_jit_grad 0.07% : 0.000775s : 1: opt_b 12.21% : 0.133076s : 1: optimize 0.00% : 0.000046s : 1: optimize_parallel_all_gather_comm 0.00% : 0.000018s : 1: order_py_execute_after_rewriter 0.00% : 0.000054s : 1: overlap_grad_flash_sp 0.00% : 0.000004s : 1: overlap_grad_matmul_and_grad_allreduce 0.00% : 0.000012s : 1: overlap_grad_ring_attention 0.00% : 0.000006s : 1: overlap_opt_shard_grad_in_pipeline 0.00% : 0.000004s : 1: overlap_opt_shard_in_pipeline 0.00% : 0.000006s : 1: overlap_param_gather 0.00% : 0.000004s : 1: overlap_recompute_allgather_and_fa_grad 0.00% : 0.000014s : 1: overlap_recompute_and_grad_model_parallel 0.00% : 0.000006s : 1: overlap_recompute_comm 0.00% : 0.000009s : 1: parallel-infer-symbol 0.00% : 0.000004s : 1: parallel-infer-symbol-second 0.00% : 0.000005s : 1: partial_unused_args_eliminate 0.00% : 0.000005s : 1: pipeline_parallel_scheduler 0.00% : 0.000005s : 1: pipeline_split 0.02% : 0.000253s : 1: pre_auto_parallel 0.02% : 0.000226s : 1: py_interpret_to_execute 0.00% : 0.000047s : 1: py_interpret_to_execute_after_opt_a 0.00% : 0.000004s : 1: remove_cast_before_assign_add 0.01% : 0.000104s : 1: remove_dup_value 0.55% : 0.005970s : 1: renormalize.infer 0.54% : 0.005895s : 1: renormalize.specialize 0.00% : 0.000007s : 1: reorder_send_recv_between_fp_bp 0.00% : 0.000009s : 1: rewriter_after_jit_bprop_graph 0.08% : 0.000898s : 1: rewriter_after_opt_a 0.07% : 0.000774s : 1: rewriter_before_opt_a 0.00% : 0.000007s : 1: slice_cell_reuse_recomputed_activation 0.00% : 0.000005s : 1: slice_recompute_activation 0.00% : 0.000004s : 1: split_layernorm_comm 0.00% : 0.000005s : 1: split_matmul_comm_elemetwise 0.00% : 0.000021s : 1: swap_dp_allreduce_reducescatter 0.05% : 0.000509s : 1: symbol_engine_optimizer 0.02% : 0.000251s : 1: tuple_transform 67.90% : 0.740060s : 1: type_inference . [hook] pytest_runtest_teardown:test_matmul_qkv_bf16_0[True-mstype0-k_n_shape1-1] tests/st/infer/ops/test_internal_ops/test_matmul_split.py::test_matmul_qkv_bf16_0[True-mstype0-k_n_shape1-1],max_mem:140.0M [WARNING] ME(152078:281472851701552,MainProcess):2026-01-29-17:43:31.317.68 [mindspore/context.py:1334] For 'context.set_context', the parameter 'device_target' will be deprecated and removed in a future version. Please use the api mindspore.set_device() instead. TotalTime = 1.48259, [21] [bootstrap]: 0.00073214 [type_inference]: 1.26787 [event_method]: 0.00021189 [auto_monad]: 0.0007428 [graph_reusing]: 4.852e-05 [inline]: 3.11999e-06 [add_attr]: 0.00640519, [1] [add_attr_with_inline]: 0.00638286, [1] [Cycle 1]: 0.00041419, [2] [tag_attr]: 0.00027095 [meta_addattr_fg_expand]: 7.097e-05 [parallel-infer-symbol]: 4.50001e-06 [pre_auto_parallel]: 0.00031148 [insert-virtual-dataset]: 2.94999e-06 [parallel-infer-symbol-second]: 1.20999e-06 [dataset_repeat_opt]: 2.22001e-06 [pipeline_split]: 1.76e-06 [optimize]: 0.204974, [53] [py_interpret_to_execute]: 0.00030406 [rewriter_before_opt_a]: 0.00084602 [opt_a]: 0.198431, [2] [Cycle 1]: 0.196255, [45] [expand_dump_flag]: 2.87e-05 [switch_simplify]: 0.0008523 [loop_unroll]: 0.00038246 [a_1]: 0.0343688 [with_stream_mark]: 6.846e-05 [recompute_prepare]: 4.413e-05 [updatestate_depend_eliminate]: 2.384e-05 [updatestate_assign_eliminate]: 1.559e-05 [updatestate_loads_eliminate]: 2.076e-05 [parameter_eliminate]: 3.29001e-06 [a_2]: 0.00078966 [accelerated_algorithm]: 6.904e-05 [shard]: 3.06001e-06 [meta_shard_fg_expand]: 2.893e-05 [shard_inline]: 2.378e-05 [merge_send_recv]: 2.057e-05 [auto_parallel]: 2.304e-05 [parallel]: 3.655e-05 [flash_sp]: 1.618e-05 [merge_comm]: 0.0120878 [allreduce_fusion]: 4.151e-05 [matmul_add_comm_reduction]: 2.964e-05 [allreduce_slice_to_reducescatter]: 1.19e-06 [virtual_shard_identity]: 5.879e-05 [virtual_dataset]: 2.609e-05 [get_grad_eliminate_]: 2.291e-05 [virtual_output]: 2.24e-05 [merge_forward]: 1.474e-05 [cell_reuse_recompute_pass]: 3.93001e-06 [offload_activation]: 2.446e-05 [cell_reuse_handle_not_recompute_node_pass]: 6.407e-05 [merge_recompute_call_nodes]: 2.07999e-06 [before_grad]: 4.169e-05 [set_forward_comm_id_for_comm_node_pass]: 1.517e-05 [meta_fg_expand]: 3.869e-05 [flash_sp_send_recv_attached]: 1.391e-05 [receive_attached]: 2.79001e-06 [after_resolve]: 3.257e-05 [a_after_grad]: 3.706e-05 [renormalize]: 0.143953 [add_forward_monad_depend]: 2.723e-05 [auto_monad_grad]: 3.41001e-06 [auto_monad_eliminator]: 9.744e-05 [cse]: 0.0020206 [a_3]: 0.00017809 [Cycle 2]: 0.00215549, [45] [expand_dump_flag]: 3.46999e-06 [switch_simplify]: 2.728e-05 [loop_unroll]: 2.219e-05 [a_1]: 0.0006975 [with_stream_mark]: 3.862e-05 [recompute_prepare]: 2.227e-05 [updatestate_depend_eliminate]: 1.552e-05 [updatestate_assign_eliminate]: 1.155e-05 [updatestate_loads_eliminate]: 1.439e-05 [parameter_eliminate]: 2.53e-06 [a_2]: 0.00031051 [accelerated_algorithm]: 2.858e-05 [shard]: 2.17999e-06 [meta_shard_fg_expand]: 8.58001e-06 [shard_inline]: 2.055e-05 [merge_send_recv]: 2.18e-05 [auto_parallel]: 2.025e-05 [parallel]: 1.136e-05 [flash_sp]: 4.75001e-06 [merge_comm]: 1.306e-05 [allreduce_fusion]: 1.272e-05 [matmul_add_comm_reduction]: 2.457e-05 [allreduce_slice_to_reducescatter]: 1.20001e-06 [virtual_shard_identity]: 2.318e-05 [virtual_dataset]: 2.038e-05 [get_grad_eliminate_]: 2.141e-05 [virtual_output]: 1.939e-05 [merge_forward]: 1.299e-05 [cell_reuse_recompute_pass]: 3.46999e-06 [offload_activation]: 2.586e-05 [cell_reuse_handle_not_recompute_node_pass]: 4.086e-05 [merge_recompute_call_nodes]: 1.45001e-06 [before_grad]: 3.433e-05 [set_forward_comm_id_for_comm_node_pass]: 1.407e-05 [meta_fg_expand]: 9.69e-06 [flash_sp_send_recv_attached]: 1.87999e-06 [receive_attached]: 2.96999e-06 [after_resolve]: 2.702e-05 [a_after_grad]: 3.224e-05 [renormalize]: 8.00064e-08 [add_forward_monad_depend]: 3.09001e-06 [auto_monad_grad]: 2.49001e-06 [auto_monad_eliminator]: 6.31e-05 [cse]: 8.704e-05 [a_3]: 0.00013747 [py_interpret_to_execute_after_opt_a]: 3.72e-05 [slice_cell_reuse_recomputed_activation]: 2.31e-06 [rewriter_after_opt_a]: 0.00086964 [convert_after_rewriter]: 2.584e-05 [order_py_execute_after_rewriter]: 1.486e-05 [mutable_eliminate]: 0.00085713 [opt_b]: 0.00083671, [1] [Cycle 1]: 0.00082543, [7] [b_1]: 0.00055729 [b_2]: 2.388e-05 [updatestate_depend_eliminate]: 2.067e-05 [updatestate_assign_eliminate]: 1.271e-05 [updatestate_loads_eliminate]: 1.357e-05 [renormalize]: 9.39996e-07 [cse]: 9.683e-05 [optimize_parallel_all_gather_comm]: 4.287e-05 [overlap_param_gather]: 2.44001e-06 [cconv]: 4.416e-05 [loop_unroll]: 0.00062464 [opt_after_cconv]: 0.00031957, [1] [Cycle 1]: 0.00031008, [7] [c_1]: 0.00013064 [parameter_eliminate]: 6.86999e-06 [updatestate_depend_eliminate]: 2.075e-05 [updatestate_assign_eliminate]: 1.178e-05 [updatestate_loads_eliminate]: 1.355e-05 [cse]: 8.172e-05 [renormalize]: 6.69999e-07 [remove_dup_value]: 9.958e-05 [tuple_transform]: 0.00027478, [1] [Cycle 1]: 0.00026779, [4] [d_1]: 0.00021081 [none_parameter_eliminate]: 3.38999e-06 [renormalize]: 2.50002e-07 [switch_simplify]: 2.414e-05 [partial_unused_args_eliminate]: 2.67001e-06 [add_recomputation]: 0.00016516 [cse_after_recomputation]: 8.158e-05, [1] [Cycle 1]: 7.345e-05, [1] [cse]: 6.426e-05 [environ_conv]: 2.546e-05 [swap_dp_allreduce_reducescatter]: 1.958e-05 [bias_add_comm_swap]: 3.41001e-06 [label_micro_interleaved_index]: 7.08998e-06 [label_fine_grained_interleaved_index]: 2.61e-06 [merge_cast_opt]: 1.37e-06 [slice_recompute_activation]: 2.53003e-06 [micro_interleaved_order_control]: 2.88e-06 [assign_add_opt]: 1.61998e-06 [ForceFp32Comm]: 8.80013e-07 [remove_cast_before_assign_add]: 1.61998e-06 [full_micro_interleaved_order_control]: 2.19001e-06 [reorder_send_recv_between_fp_bp]: 2.79999e-06 [comm_op_add_attrs]: 1.32e-06 [add_comm_op_reuse_tag]: 1.02e-06 [interleave_split_concat_branches]: 1.20999e-06 [interleave_parallel_branches]: 1.05001e-06 [overlap_opt_shard_in_pipeline]: 1.24e-06 [overlap_opt_shard_grad_in_pipeline]: 2.68e-06 [control_data_broadcast_order]: 4.31e-05 [grouped_pairwise_exchange_alltoall]: 1.56002e-06 [offloading_packed_experts]: 1.164e-05 [overlap_recompute_and_grad_model_parallel]: 1.087e-05 [overlap_grad_matmul_and_grad_allreduce]: 1.32999e-06 [overlap_recompute_allgather_and_fa_grad]: 1.34e-06 [overlap_recompute_comm]: 2.22001e-06 [overlap_grad_ring_attention]: 9.83998e-06 [overlap_grad_flash_sp]: 5.111e-05 [begin_end_overlap_inline]: 6.60017e-07 [split_matmul_comm_elemetwise]: 2.45002e-06 [split_layernorm_comm]: 1.89e-06 [handle_group_info]: 1.00001e-06 [symbol_engine_optimizer]: 0.00053517, [1] [Cycle 1]: 0.00052766, [6] [build]: 0.00028891 [elim_shapecalc]: 4.09e-05 [elim_not_effective]: 5.195e-05 [opt_reshape]: 3.451e-05 [fold_const_symbol]: 5.886e-05 [renormalize]: 4.00003e-07 [detach_backward]: 2.81e-06 [pipeline_parallel_scheduler]: 1.70001e-06 [auto_monad_reorder]: 6.626e-05 [get_jit_bprop_graph]: 2.46e-06 [rewriter_after_jit_bprop_graph]: 5.42001e-06 [opt_after_jit_grad]: 0.00081684 [validate]: 0.00010906 Sums bootstrap : 0.000732s : 0.05% type_inference : 1.267873s : 85.97% event_method : 0.000212s : 0.01% auto_monad : 0.000743s : 0.05% graph_reusing : 0.000049s : 0.00% inline : 0.000003s : 0.00% add_attr.add_attr_with_inline.tag_attr : 0.000271s : 0.02% add_attr.add_attr_with_inline.meta_addattr_fg_expand : 0.000071s : 0.00% parallel-infer-symbol : 0.000005s : 0.00% pre_auto_parallel : 0.000311s : 0.02% insert-virtual-dataset : 0.000003s : 0.00% parallel-infer-symbol-second : 0.000001s : 0.00% dataset_repeat_opt : 0.000002s : 0.00% pipeline_split : 0.000002s : 0.00% optimize.py_interpret_to_execute : 0.000304s : 0.02% optimize.rewriter_before_opt_a : 0.000846s : 0.06% optimize.opt_a.expand_dump_flag : 0.000032s : 0.00% optimize.opt_a.switch_simplify : 0.000880s : 0.06% optimize.opt_a.loop_unroll : 0.000405s : 0.03% optimize.opt_a.a_1 : 0.035066s : 2.38% optimize.opt_a.with_stream_mark : 0.000107s : 0.01% optimize.opt_a.recompute_prepare : 0.000066s : 0.00% optimize.opt_a.updatestate_depend_eliminate : 0.000039s : 0.00% optimize.opt_a.updatestate_assign_eliminate : 0.000027s : 0.00% optimize.opt_a.updatestate_loads_eliminate : 0.000035s : 0.00% optimize.opt_a.parameter_eliminate : 0.000006s : 0.00% optimize.opt_a.a_2 : 0.001100s : 0.07% optimize.opt_a.accelerated_algorithm : 0.000098s : 0.01% optimize.opt_a.shard : 0.000005s : 0.00% optimize.opt_a.meta_shard_fg_expand : 0.000038s : 0.00% optimize.opt_a.shard_inline : 0.000044s : 0.00% optimize.opt_a.merge_send_recv : 0.000042s : 0.00% optimize.opt_a.auto_parallel : 0.000043s : 0.00% optimize.opt_a.parallel : 0.000048s : 0.00% optimize.opt_a.flash_sp : 0.000021s : 0.00% optimize.opt_a.merge_comm : 0.012101s : 0.82% optimize.opt_a.allreduce_fusion : 0.000054s : 0.00% optimize.opt_a.matmul_add_comm_reduction : 0.000054s : 0.00% optimize.opt_a.allreduce_slice_to_reducescatter : 0.000002s : 0.00% optimize.opt_a.virtual_shard_identity : 0.000082s : 0.01% optimize.opt_a.virtual_dataset : 0.000046s : 0.00% optimize.opt_a.get_grad_eliminate_ : 0.000044s : 0.00% optimize.opt_a.virtual_output : 0.000042s : 0.00% optimize.opt_a.merge_forward : 0.000028s : 0.00% optimize.opt_a.cell_reuse_recompute_pass : 0.000007s : 0.00% optimize.opt_a.offload_activation : 0.000050s : 0.00% optimize.opt_a.cell_reuse_handle_not_recompute_node_pass : 0.000105s : 0.01% optimize.opt_a.merge_recompute_call_nodes : 0.000004s : 0.00% optimize.opt_a.before_grad : 0.000076s : 0.01% optimize.opt_a.set_forward_comm_id_for_comm_node_pass : 0.000029s : 0.00% optimize.opt_a.meta_fg_expand : 0.000048s : 0.00% optimize.opt_a.flash_sp_send_recv_attached : 0.000016s : 0.00% optimize.opt_a.receive_attached : 0.000006s : 0.00% optimize.opt_a.after_resolve : 0.000060s : 0.00% optimize.opt_a.a_after_grad : 0.000069s : 0.00% optimize.opt_a.renormalize : 0.143953s : 9.76% optimize.opt_a.add_forward_monad_depend : 0.000030s : 0.00% optimize.opt_a.auto_monad_grad : 0.000006s : 0.00% optimize.opt_a.auto_monad_eliminator : 0.000161s : 0.01% optimize.opt_a.cse : 0.002108s : 0.14% optimize.opt_a.a_3 : 0.000316s : 0.02% optimize.py_interpret_to_execute_after_opt_a : 0.000037s : 0.00% optimize.slice_cell_reuse_recomputed_activation : 0.000002s : 0.00% optimize.rewriter_after_opt_a : 0.000870s : 0.06% optimize.convert_after_rewriter : 0.000026s : 0.00% optimize.order_py_execute_after_rewriter : 0.000015s : 0.00% optimize.mutable_eliminate : 0.000857s : 0.06% optimize.opt_b.b_1 : 0.000557s : 0.04% optimize.opt_b.b_2 : 0.000024s : 0.00% optimize.opt_b.updatestate_depend_eliminate : 0.000021s : 0.00% optimize.opt_b.updatestate_assign_eliminate : 0.000013s : 0.00% optimize.opt_b.updatestate_loads_eliminate : 0.000014s : 0.00% optimize.opt_b.renormalize : 0.000001s : 0.00% optimize.opt_b.cse : 0.000097s : 0.01% optimize.optimize_parallel_all_gather_comm : 0.000043s : 0.00% optimize.overlap_param_gather : 0.000002s : 0.00% optimize.cconv : 0.000044s : 0.00% optimize.loop_unroll : 0.000625s : 0.04% optimize.opt_after_cconv.c_1 : 0.000131s : 0.01% optimize.opt_after_cconv.parameter_eliminate : 0.000007s : 0.00% optimize.opt_after_cconv.updatestate_depend_eliminate : 0.000021s : 0.00% optimize.opt_after_cconv.updatestate_assign_eliminate : 0.000012s : 0.00% optimize.opt_after_cconv.updatestate_loads_eliminate : 0.000014s : 0.00% optimize.opt_after_cconv.cse : 0.000082s : 0.01% optimize.opt_after_cconv.renormalize : 0.000001s : 0.00% optimize.remove_dup_value : 0.000100s : 0.01% optimize.tuple_transform.d_1 : 0.000211s : 0.01% optimize.tuple_transform.none_parameter_eliminate : 0.000003s : 0.00% optimize.tuple_transform.renormalize : 0.000000s : 0.00% optimize.tuple_transform.switch_simplify : 0.000024s : 0.00% optimize.partial_unused_args_eliminate : 0.000003s : 0.00% optimize.add_recomputation : 0.000165s : 0.01% optimize.cse_after_recomputation.cse : 0.000064s : 0.00% optimize.environ_conv : 0.000025s : 0.00% optimize.swap_dp_allreduce_reducescatter : 0.000020s : 0.00% optimize.bias_add_comm_swap : 0.000003s : 0.00% optimize.label_micro_interleaved_index : 0.000007s : 0.00% optimize.label_fine_grained_interleaved_index : 0.000003s : 0.00% optimize.merge_cast_opt : 0.000001s : 0.00% optimize.slice_recompute_activation : 0.000003s : 0.00% optimize.micro_interleaved_order_control : 0.000003s : 0.00% optimize.assign_add_opt : 0.000002s : 0.00% optimize.ForceFp32Comm : 0.000001s : 0.00% optimize.remove_cast_before_assign_add : 0.000002s : 0.00% optimize.full_micro_interleaved_order_control : 0.000002s : 0.00% optimize.reorder_send_recv_between_fp_bp : 0.000003s : 0.00% optimize.comm_op_add_attrs : 0.000001s : 0.00% optimize.add_comm_op_reuse_tag : 0.000001s : 0.00% optimize.interleave_split_concat_branches : 0.000001s : 0.00% optimize.interleave_parallel_branches : 0.000001s : 0.00% optimize.overlap_opt_shard_in_pipeline : 0.000001s : 0.00% optimize.overlap_opt_shard_grad_in_pipeline : 0.000003s : 0.00% optimize.control_data_broadcast_order : 0.000043s : 0.00% optimize.grouped_pairwise_exchange_alltoall : 0.000002s : 0.00% optimize.offloading_packed_experts : 0.000012s : 0.00% optimize.overlap_recompute_and_grad_model_parallel : 0.000011s : 0.00% optimize.overlap_grad_matmul_and_grad_allreduce : 0.000001s : 0.00% optimize.overlap_recompute_allgather_and_fa_grad : 0.000001s : 0.00% optimize.overlap_recompute_comm : 0.000002s : 0.00% optimize.overlap_grad_ring_attention : 0.000010s : 0.00% optimize.overlap_grad_flash_sp : 0.000051s : 0.00% optimize.begin_end_overlap_inline : 0.000001s : 0.00% optimize.split_matmul_comm_elemetwise : 0.000002s : 0.00% optimize.split_layernorm_comm : 0.000002s : 0.00% optimize.handle_group_info : 0.000001s : 0.00% optimize.symbol_engine_optimizer.build : 0.000289s : 0.02% optimize.symbol_engine_optimizer.elim_shapecalc : 0.000041s : 0.00% optimize.symbol_engine_optimizer.elim_not_effective : 0.000052s : 0.00% optimize.symbol_engine_optimizer.opt_reshape : 0.000035s : 0.00% optimize.symbol_engine_optimizer.fold_const_symbol : 0.000059s : 0.00% optimize.symbol_engine_optimizer.renormalize : 0.000000s : 0.00% detach_backward : 0.000003s : 0.00% pipeline_parallel_scheduler : 0.000002s : 0.00% auto_monad_reorder : 0.000066s : 0.00% get_jit_bprop_graph : 0.000002s : 0.00% rewriter_after_jit_bprop_graph : 0.000005s : 0.00% opt_after_jit_grad : 0.000817s : 0.06% validate : 0.000109s : 0.01% Time group info: ------[substitution.] 0.004212 569 0.40% : 0.000017s : 8: substitution.depend_value_elim 0.34% : 0.000014s : 14: substitution.elim_not_effective 0.43% : 0.000018s : 12: substitution.float_tuple_getitem_switch 0.65% : 0.000027s : 14: substitution.fold_const_symbol 0.36% : 0.000015s : 17: substitution.graph_param_transform 71.52% : 0.003013s : 71: substitution.inline 0.32% : 0.000013s : 29: substitution.j_node_and_user_rematch 0.96% : 0.000040s : 6: substitution.less_batch_normalization 1.32% : 0.000055s : 2: substitution.list_to_tuple_eliminator_ 0.15% : 0.000006s : 6: substitution.load_eliminater 0.86% : 0.000036s : 21: substitution.minmaximum_grad 0.21% : 0.000009s : 2: substitution.opt_reshape 0.52% : 0.000022s : 29: substitution.remove_not_recompute_node 0.24% : 0.000010s : 4: substitution.replace_old_param 1.82% : 0.000077s : 18: substitution.reshape_eliminate 1.23% : 0.000052s : 24: substitution.switch_simplify 4.07% : 0.000172s : 51: substitution.tuple_list_convert_item_index_to_positive 2.24% : 0.000094s : 51: substitution.tuple_list_get_item_const_eliminator 2.86% : 0.000121s : 51: substitution.tuple_list_get_item_depend_reorder 5.55% : 0.000234s : 73: substitution.tuple_list_get_item_eliminator 2.82% : 0.000119s : 51: substitution.tuple_list_get_set_item_eliminator 0.62% : 0.000026s : 6: substitution.updatestate_pure_node_eliminater 0.52% : 0.000022s : 9: substitution.updatestate_useless_node_eliminater ------[type_inference.] 1.267483 2 85.72% : 1.086472s : 1: type_inference.infer 14.28% : 0.181010s : 1: type_inference.specialize ------[replace.] 0.001322 115 3.64% : 0.000048s : 5: replace.depend_value_elim 60.27% : 0.000797s : 71: replace.inline 3.10% : 0.000041s : 2: replace.list_to_tuple_eliminator_ 23.59% : 0.000312s : 24: replace.switch_simplify 9.39% : 0.000124s : 13: replace.tuple_list_get_item_eliminator ------[match.] 0.003090 115 0.07% : 0.000002s : 5: match.depend_value_elim 95.80% : 0.002961s : 71: match.inline 1.70% : 0.000052s : 2: match.list_to_tuple_eliminator_ 1.22% : 0.000038s : 24: match.switch_simplify 1.21% : 0.000037s : 13: match.tuple_list_get_item_eliminator ------[predicate.] 0.013210 9701 0.15% : 0.000020s : 123: predicate.accumulaten_eliminater 0.05% : 0.000007s : 17: predicate.ad_related_special_op_eliminate 0.06% : 0.000008s : 58: predicate.addn_check_dump 0.15% : 0.000020s : 123: predicate.addn_zero_filter 0.13% : 0.000018s : 123: predicate.adjust_all_reduce_mul_add 0.28% : 0.000037s : 181: predicate.arithmetic_simplify 0.16% : 0.000021s : 123: predicate.cast_eliminate 0.04% : 0.000005s : 34: predicate.check_bprop_eliminate 0.06% : 0.000008s : 58: predicate.compare_switch_simplify 0.01% : 0.000001s : 17: predicate.const_output_eliminate 0.07% : 0.000010s : 61: predicate.depend_value_elim 0.15% : 0.000020s : 123: predicate.dict_get_item_const_eliminator 0.17% : 0.000022s : 123: predicate.dict_get_item_eliminator 0.15% : 0.000019s : 123: predicate.dict_set_item_eliminator 0.05% : 0.000007s : 34: predicate.dumpgradient_eliminate 0.01% : 0.000001s : 17: predicate.elim_not_effective 0.03% : 0.000003s : 17: predicate.elim_shapecalc_of_broadcastargs 0.17% : 0.000022s : 140: predicate.environ_add_const_eliminate 0.15% : 0.000020s : 140: predicate.environ_get_add_eliminate 0.16% : 0.000021s : 140: predicate.environ_get_depend_swap 0.22% : 0.000029s : 198: predicate.environ_get_eliminate 0.15% : 0.000020s : 140: predicate.environ_get_set_eliminate 0.23% : 0.000031s : 209: predicate.exchange_switch_depend_value 0.38% : 0.000050s : 209: predicate.float_depend_g_call 0.06% : 0.000008s : 58: predicate.float_environ_get_switch 0.08% : 0.000011s : 75: predicate.float_tuple_getitem_switch 0.01% : 0.000001s : 17: predicate.fold_const_symbol 0.04% : 0.000006s : 35: predicate.get_grad_eliminate 0.01% : 0.000002s : 17: predicate.graph_param_transform 0.06% : 0.000008s : 58: predicate.incorporate_call 0.06% : 0.000007s : 58: predicate.incorporate_call_switch 88.12% : 0.011641s : 459: predicate.inline 0.06% : 0.000008s : 35: predicate.inline_without_move 0.02% : 0.000002s : 35: predicate.j_node_and_user_rematch 0.06% : 0.000007s : 36: predicate.less_batch_normalization 0.21% : 0.000028s : 172: predicate.list_to_tuple_eliminator_ 0.33% : 0.000044s : 295: predicate.load_eliminater 0.06% : 0.000007s : 17: predicate.loop_unroll_after_grad 0.49% : 0.000065s : 314: predicate.loop_unroll_before_grad 0.19% : 0.000025s : 157: predicate.make_slice_get_slice_eliminator 0.06% : 0.000008s : 58: predicate.merge_addn 0.04% : 0.000005s : 34: predicate.micro_step_allgather_replace 0.04% : 0.000005s : 34: predicate.mini_step_allgather_replace 0.14% : 0.000019s : 123: predicate.minmaximum_grad 0.06% : 0.000007s : 17: predicate.mutable_eliminate 0.02% : 0.000003s : 17: predicate.opt_reshape 0.02% : 0.000003s : 17: predicate.parallel_virtual_node 0.49% : 0.000065s : 209: predicate.partial_defer_inline 0.19% : 0.000025s : 155: predicate.partial_eliminate 0.14% : 0.000019s : 123: predicate.print_const_string_wrapper 0.06% : 0.000008s : 53: predicate.reduce_all_const_elim 0.21% : 0.000027s : 123: predicate.reduce_eliminate 0.32% : 0.000042s : 295: predicate.redundant_stop_gradient_eliminater 0.02% : 0.000003s : 35: predicate.remove_not_recompute_node 0.13% : 0.000018s : 172: predicate.replace_applicator 0.02% : 0.000003s : 35: predicate.replace_old_param 0.02% : 0.000002s : 17: predicate.reset_defer_inline 0.17% : 0.000022s : 123: predicate.reshape_eliminate 0.04% : 0.000005s : 34: predicate.row_tensor_add_zeros_like 0.02% : 0.000003s : 17: predicate.row_tensor_eliminate 0.06% : 0.000008s : 34: predicate.same_eliminate 0.03% : 0.000003s : 40: predicate.set_cell_output_no_recompute 0.06% : 0.000008s : 35: predicate.shard_identity_eliminate 0.04% : 0.000006s : 34: predicate.special_op_eliminate 0.07% : 0.000009s : 58: predicate.specialize_transform 0.05% : 0.000006s : 34: predicate.split_environ_get_set_with_tuple_value 0.05% : 0.000006s : 35: predicate.stack_unstack_eliminate 0.02% : 0.000003s : 17: predicate.switch_call_monad_eliminater 0.26% : 0.000034s : 209: predicate.switch_defer_inline 0.30% : 0.000040s : 243: predicate.switch_layer_defer_inline 0.86% : 0.000114s : 646: predicate.switch_simplify 0.15% : 0.000020s : 123: predicate.tile_eliminate 0.14% : 0.000019s : 123: predicate.transpose_eliminate 0.21% : 0.000028s : 157: predicate.tuple_list_convert_item_index_to_positive 0.23% : 0.000030s : 157: predicate.tuple_list_get_item_const_eliminator 0.21% : 0.000027s : 157: predicate.tuple_list_get_item_depend_reorder 0.42% : 0.000055s : 228: predicate.tuple_list_get_item_eliminator 0.20% : 0.000027s : 157: predicate.tuple_list_get_set_item_eliminator 0.28% : 0.000037s : 215: predicate.tuple_list_set_item_eliminator 0.21% : 0.000028s : 170: predicate.tuple_to_list_eliminator_ 0.32% : 0.000042s : 295: predicate.updatestate_pure_node_eliminater 0.42% : 0.000056s : 353: predicate.updatestate_useless_node_eliminater 0.02% : 0.000003s : 17: predicate.value_based_eliminate 0.05% : 0.000006s : 35: predicate.virtual_dataset_eliminate 0.04% : 0.000006s : 35: predicate.virtual_output_eliminate 0.02% : 0.000003s : 17: predicate.virtual_view_grad_eliminate 0.02% : 0.000003s : 17: predicate.zero_like_fill_zero ------[func_graph_cloner_run.] 0.206517 135 33.97% : 0.070155s : 55: func_graph_cloner_run.FuncGraphClonerGraph 66.03% : 0.136362s : 80: func_graph_cloner_run.FuncGraphSpecializer ------[meta_graph.] 0.000000 0 ------[manager.] 0.000000 0 ------[pynative] 0.000000 0 ------[others.] 1.877231 209 0.00% : 0.000004s : 1: ForceFp32Comm 0.34% : 0.006414s : 1: add_attr 0.34% : 0.006391s : 1: add_attr_with_inline 0.00% : 0.000005s : 1: add_comm_op_reuse_tag 0.01% : 0.000173s : 1: add_recomputation 0.00% : 0.000006s : 1: assign_add_opt 0.04% : 0.000760s : 1: auto_monad 0.00% : 0.000072s : 1: auto_monad_reorder 0.00% : 0.000005s : 1: begin_end_overlap_inline 0.00% : 0.000006s : 1: bias_add_comm_swap 0.04% : 0.000791s : 1: bootstrap 0.00% : 0.000048s : 1: cconv 0.00% : 0.000004s : 1: comm_op_add_attrs 0.00% : 0.000048s : 1: control_data_broadcast_order 0.00% : 0.000032s : 1: convert_after_rewriter 0.00% : 0.000085s : 1: cse_after_recomputation 0.00% : 0.000005s : 1: dataset_repeat_opt 0.00% : 0.000006s : 1: detach_backward 0.00% : 0.000029s : 1: environ_conv 0.01% : 0.000226s : 1: event_method 0.00% : 0.000007s : 1: full_micro_interleaved_order_control 0.00% : 0.000007s : 1: get_jit_bprop_graph 0.00% : 0.000055s : 1: graph_reusing 0.00% : 0.000005s : 1: grouped_pairwise_exchange_alltoall 0.00% : 0.000005s : 1: handle_group_info 0.00% : 0.000007s : 1: inline 0.00% : 0.000007s : 1: insert-virtual-dataset 0.00% : 0.000004s : 1: interleave_parallel_branches 0.00% : 0.000004s : 1: interleave_split_concat_branches 0.00% : 0.000006s : 1: label_fine_grained_interleaved_index 0.00% : 0.000010s : 1: label_micro_interleaved_index 0.03% : 0.000639s : 1: loop_unroll 0.00% : 0.000006s : 1: merge_cast_opt 0.00% : 0.000006s : 1: micro_interleaved_order_control 0.05% : 0.000872s : 1: mutable_eliminate 0.00% : 0.000015s : 1: offloading_packed_experts 0.00% : 0.000041s : 1: opt.transform.loop_unroll_optimizer 0.00% : 0.000047s : 1: opt.transform.mutable_eliminate 2.04% : 0.038338s : 95: opt.transform.opt_a 0.01% : 0.000129s : 1: opt.transform.opt_after_cconv 0.00% : 0.000080s : 1: opt.transform.opt_after_jit_grad 0.03% : 0.000540s : 28: opt.transform.opt_b 0.01% : 0.000231s : 2: opt.transform.opt_trans_graph 0.01% : 0.000180s : 4: opt.transform.symbol_engine_opt 10.57% : 0.198436s : 1: opt_a 0.02% : 0.000324s : 1: opt_after_cconv 0.04% : 0.000833s : 1: opt_after_jit_grad 0.04% : 0.000841s : 1: opt_b 10.92% : 0.204981s : 1: optimize 0.00% : 0.000048s : 1: optimize_parallel_all_gather_comm 0.00% : 0.000019s : 1: order_py_execute_after_rewriter 0.00% : 0.000057s : 1: overlap_grad_flash_sp 0.00% : 0.000004s : 1: overlap_grad_matmul_and_grad_allreduce 0.00% : 0.000013s : 1: overlap_grad_ring_attention 0.00% : 0.000006s : 1: overlap_opt_shard_grad_in_pipeline 0.00% : 0.000004s : 1: overlap_opt_shard_in_pipeline 0.00% : 0.000007s : 1: overlap_param_gather 0.00% : 0.000004s : 1: overlap_recompute_allgather_and_fa_grad 0.00% : 0.000014s : 1: overlap_recompute_and_grad_model_parallel 0.00% : 0.000007s : 1: overlap_recompute_comm 0.00% : 0.000012s : 1: parallel-infer-symbol 0.00% : 0.000004s : 1: parallel-infer-symbol-second 0.00% : 0.000006s : 1: partial_unused_args_eliminate 0.00% : 0.000005s : 1: pipeline_parallel_scheduler 0.00% : 0.000005s : 1: pipeline_split 0.02% : 0.000324s : 1: pre_auto_parallel 0.02% : 0.000319s : 1: py_interpret_to_execute 0.00% : 0.000042s : 1: py_interpret_to_execute_after_opt_a 0.00% : 0.000004s : 1: remove_cast_before_assign_add 0.01% : 0.000108s : 1: remove_dup_value 0.37% : 0.007036s : 1: renormalize.infer 7.29% : 0.136895s : 1: renormalize.specialize 0.00% : 0.000007s : 1: reorder_send_recv_between_fp_bp 0.00% : 0.000009s : 1: rewriter_after_jit_bprop_graph 0.05% : 0.000882s : 1: rewriter_after_opt_a 0.05% : 0.000861s : 1: rewriter_before_opt_a 0.00% : 0.000005s : 1: slice_cell_reuse_recomputed_activation 0.00% : 0.000005s : 1: slice_recompute_activation 0.00% : 0.000005s : 1: split_layernorm_comm 0.00% : 0.000005s : 1: split_matmul_comm_elemetwise 0.00% : 0.000023s : 1: swap_dp_allreduce_reducescatter 0.03% : 0.000539s : 1: symbol_engine_optimizer 0.01% : 0.000279s : 1: tuple_transform 67.54% : 1.267905s : 1: type_inference . [hook] pytest_runtest_teardown:test_matmul_qkv_bf16_0[True-mstype0-k_n_shape1-16] tests/st/infer/ops/test_internal_ops/test_matmul_split.py::test_matmul_qkv_bf16_0[True-mstype0-k_n_shape1-16],max_mem:140.0M [WARNING] ME(152078:281472851701552,MainProcess):2026-01-29-17:43:50.292.651 [mindspore/context.py:1334] For 'context.set_context', the parameter 'device_target' will be deprecated and removed in a future version. Please use the api mindspore.set_device() instead. TotalTime = 0.578832, [21] [bootstrap]: 0.00078118 [type_inference]: 0.477853 [event_method]: 0.00022392 [auto_monad]: 0.00080728 [graph_reusing]: 5.294e-05 [inline]: 2.86999e-06 [add_attr]: 0.0176279, [1] [add_attr_with_inline]: 0.0176082, [1] [Cycle 1]: 0.00030142, [2] [tag_attr]: 0.00020082 [meta_addattr_fg_expand]: 5.101e-05 [parallel-infer-symbol]: 3.68e-06 [pre_auto_parallel]: 0.00025211 [insert-virtual-dataset]: 2.41998e-06 [parallel-infer-symbol-second]: 1.15999e-06 [dataset_repeat_opt]: 2.37001e-06 [pipeline_split]: 1.54e-06 [optimize]: 0.0799742, [53] [py_interpret_to_execute]: 0.00025141 [rewriter_before_opt_a]: 0.00078742 [opt_a]: 0.0733611, [2] [Cycle 1]: 0.071046, [45] [expand_dump_flag]: 1.526e-05 [switch_simplify]: 0.00081452 [loop_unroll]: 0.00031961 [a_1]: 0.0500848 [with_stream_mark]: 6.226e-05 [recompute_prepare]: 4.153e-05 [updatestate_depend_eliminate]: 2.092e-05 [updatestate_assign_eliminate]: 1.602e-05 [updatestate_loads_eliminate]: 2.007e-05 [parameter_eliminate]: 2.49999e-06 [a_2]: 0.00081725 [accelerated_algorithm]: 7.62e-05 [shard]: 3.23e-06 [meta_shard_fg_expand]: 2.874e-05 [shard_inline]: 2.403e-05 [merge_send_recv]: 2.167e-05 [auto_parallel]: 2.233e-05 [parallel]: 3.435e-05 [flash_sp]: 1.516e-05 [merge_comm]: 1.315e-05 [allreduce_fusion]: 1.29e-05 [matmul_add_comm_reduction]: 2.388e-05 [allreduce_slice_to_reducescatter]: 1.07e-06 [virtual_shard_identity]: 4.455e-05 [virtual_dataset]: 2.493e-05 [get_grad_eliminate_]: 2.213e-05 [virtual_output]: 2.225e-05 [merge_forward]: 1.551e-05 [cell_reuse_recompute_pass]: 2.79001e-06 [offload_activation]: 2.6e-05 [cell_reuse_handle_not_recompute_node_pass]: 5.162e-05 [merge_recompute_call_nodes]: 1.81003e-06 [before_grad]: 3.912e-05 [set_forward_comm_id_for_comm_node_pass]: 1.522e-05 [meta_fg_expand]: 1.688e-05 [flash_sp_send_recv_attached]: 6.63998e-06 [receive_attached]: 2.31e-06 [after_resolve]: 2.997e-05 [a_after_grad]: 3.751e-05 [renormalize]: 0.0160897 [add_forward_monad_depend]: 1.993e-05 [auto_monad_grad]: 2.96999e-06 [auto_monad_eliminator]: 9.366e-05 [cse]: 0.00134853 [a_3]: 0.00018547 [Cycle 2]: 0.00229861, [45] [expand_dump_flag]: 3.48e-06 [switch_simplify]: 2.614e-05 [loop_unroll]: 2.182e-05 [a_1]: 0.00069132 [with_stream_mark]: 3.951e-05 [recompute_prepare]: 2.728e-05 [updatestate_depend_eliminate]: 1.613e-05 [updatestate_assign_eliminate]: 1.169e-05 [updatestate_loads_eliminate]: 1.403e-05 [parameter_eliminate]: 3.06001e-06 [a_2]: 0.00031847 [accelerated_algorithm]: 3.446e-05 [shard]: 2.37999e-06 [meta_shard_fg_expand]: 9.15001e-06 [shard_inline]: 2.167e-05 [merge_send_recv]: 1.995e-05 [auto_parallel]: 2.11e-05 [parallel]: 1.107e-05 [flash_sp]: 4.60001e-06 [merge_comm]: 1.377e-05 [allreduce_fusion]: 1.285e-05 [matmul_add_comm_reduction]: 2.429e-05 [allreduce_slice_to_reducescatter]: 1.09e-06 [virtual_shard_identity]: 2.873e-05 [virtual_dataset]: 2.241e-05 [get_grad_eliminate_]: 2.204e-05 [virtual_output]: 1.999e-05 [merge_forward]: 1.351e-05 [cell_reuse_recompute_pass]: 3.40998e-06 [offload_activation]: 2.589e-05 [cell_reuse_handle_not_recompute_node_pass]: 6.227e-05 [merge_recompute_call_nodes]: 1.65001e-06 [before_grad]: 3.749e-05 [set_forward_comm_id_for_comm_node_pass]: 1.561e-05 [meta_fg_expand]: 1.042e-05 [flash_sp_send_recv_attached]: 1.67001e-06 [receive_attached]: 2.83998e-06 [after_resolve]: 3.108e-05 [a_after_grad]: 3.745e-05 [renormalize]: 9.00181e-08 [add_forward_monad_depend]: 4.83001e-06 [auto_monad_grad]: 2.90998e-06 [auto_monad_eliminator]: 7.156e-05 [cse]: 9.43e-05 [a_3]: 0.0001505 [py_interpret_to_execute_after_opt_a]: 3.454e-05 [slice_cell_reuse_recomputed_activation]: 2.48e-06 [rewriter_after_opt_a]: 0.0008791 [convert_after_rewriter]: 2.709e-05 [order_py_execute_after_rewriter]: 1.431e-05 [mutable_eliminate]: 0.00088651 [opt_b]: 0.00079371, [1] [Cycle 1]: 0.0007825, [7] [b_1]: 0.00055902 [b_2]: 2.549e-05 [updatestate_depend_eliminate]: 2.101e-05 [updatestate_assign_eliminate]: 1.221e-05 [updatestate_loads_eliminate]: 1.481e-05 [renormalize]: 1.74e-06 [cse]: 9.708e-05 [optimize_parallel_all_gather_comm]: 4.189e-05 [overlap_param_gather]: 2.13998e-06 [cconv]: 4.416e-05 [loop_unroll]: 0.00088152 [opt_after_cconv]: 0.00031828, [1] [Cycle 1]: 0.00030784, [7] [c_1]: 0.00013113 [parameter_eliminate]: 6.58e-06 [updatestate_depend_eliminate]: 2.214e-05 [updatestate_assign_eliminate]: 1.093e-05 [updatestate_loads_eliminate]: 1.305e-05 [cse]: 8.139e-05 [renormalize]: 1.40999e-06 [remove_dup_value]: 9.778e-05 [tuple_transform]: 0.00026086, [1] [Cycle 1]: 0.00025495, [4] [d_1]: 0.00020424 [none_parameter_eliminate]: 2.27999e-06 [renormalize]: 3.89991e-07 [switch_simplify]: 2.382e-05 [partial_unused_args_eliminate]: 2.21e-06 [add_recomputation]: 0.00015548 [cse_after_recomputation]: 7.326e-05, [1] [Cycle 1]: 6.696e-05, [1] [cse]: 5.963e-05 [environ_conv]: 2.383e-05 [swap_dp_allreduce_reducescatter]: 1.999e-05 [bias_add_comm_swap]: 3.55998e-06 [label_micro_interleaved_index]: 6.61999e-06 [label_fine_grained_interleaved_index]: 2.81e-06 [merge_cast_opt]: 1.54e-06 [slice_recompute_activation]: 2.37001e-06 [micro_interleaved_order_control]: 2.32001e-06 [assign_add_opt]: 1.79e-06 [ForceFp32Comm]: 9.00007e-07 [remove_cast_before_assign_add]: 1.07e-06 [full_micro_interleaved_order_control]: 2.09e-06 [reorder_send_recv_between_fp_bp]: 2.74999e-06 [comm_op_add_attrs]: 1.22e-06 [add_comm_op_reuse_tag]: 1.00001e-06 [interleave_split_concat_branches]: 1.15001e-06 [interleave_parallel_branches]: 1.03001e-06 [overlap_opt_shard_in_pipeline]: 1.19003e-06 [overlap_opt_shard_grad_in_pipeline]: 1.92999e-06 [control_data_broadcast_order]: 3.974e-05 [grouped_pairwise_exchange_alltoall]: 1.52001e-06 [offloading_packed_experts]: 1.086e-05 [overlap_recompute_and_grad_model_parallel]: 1.089e-05 [overlap_grad_matmul_and_grad_allreduce]: 1.23002e-06 [overlap_recompute_allgather_and_fa_grad]: 1.32999e-06 [overlap_recompute_comm]: 2.52001e-06 [overlap_grad_ring_attention]: 9.37001e-06 [overlap_grad_flash_sp]: 4.934e-05 [begin_end_overlap_inline]: 7.00005e-07 [split_matmul_comm_elemetwise]: 2.39001e-06 [split_layernorm_comm]: 1.60001e-06 [handle_group_info]: 1.30001e-06 [symbol_engine_optimizer]: 0.00052043, [1] [Cycle 1]: 0.00051325, [6] [build]: 0.00028231 [elim_shapecalc]: 3.569e-05 [elim_not_effective]: 5.064e-05 [opt_reshape]: 3.441e-05 [fold_const_symbol]: 5.775e-05 [renormalize]: 9.5999e-07 [detach_backward]: 3.28998e-06 [pipeline_parallel_scheduler]: 1.49e-06 [auto_monad_reorder]: 6.239e-05 [get_jit_bprop_graph]: 2.08002e-06 [rewriter_after_jit_bprop_graph]: 5.34e-06 [opt_after_jit_grad]: 0.00079339 [validate]: 0.0001143 Sums bootstrap : 0.000781s : 0.14% type_inference : 0.477853s : 85.35% event_method : 0.000224s : 0.04% auto_monad : 0.000807s : 0.14% graph_reusing : 0.000053s : 0.01% inline : 0.000003s : 0.00% add_attr.add_attr_with_inline.tag_attr : 0.000201s : 0.04% add_attr.add_attr_with_inline.meta_addattr_fg_expand : 0.000051s : 0.01% parallel-infer-symbol : 0.000004s : 0.00% pre_auto_parallel : 0.000252s : 0.05% insert-virtual-dataset : 0.000002s : 0.00% parallel-infer-symbol-second : 0.000001s : 0.00% dataset_repeat_opt : 0.000002s : 0.00% pipeline_split : 0.000002s : 0.00% optimize.py_interpret_to_execute : 0.000251s : 0.04% optimize.rewriter_before_opt_a : 0.000787s : 0.14% optimize.opt_a.expand_dump_flag : 0.000019s : 0.00% optimize.opt_a.switch_simplify : 0.000841s : 0.15% optimize.opt_a.loop_unroll : 0.000341s : 0.06% optimize.opt_a.a_1 : 0.050776s : 9.07% optimize.opt_a.with_stream_mark : 0.000102s : 0.02% optimize.opt_a.recompute_prepare : 0.000069s : 0.01% optimize.opt_a.updatestate_depend_eliminate : 0.000037s : 0.01% optimize.opt_a.updatestate_assign_eliminate : 0.000028s : 0.00% optimize.opt_a.updatestate_loads_eliminate : 0.000034s : 0.01% optimize.opt_a.parameter_eliminate : 0.000006s : 0.00% optimize.opt_a.a_2 : 0.001136s : 0.20% optimize.opt_a.accelerated_algorithm : 0.000111s : 0.02% optimize.opt_a.shard : 0.000006s : 0.00% optimize.opt_a.meta_shard_fg_expand : 0.000038s : 0.01% optimize.opt_a.shard_inline : 0.000046s : 0.01% optimize.opt_a.merge_send_recv : 0.000042s : 0.01% optimize.opt_a.auto_parallel : 0.000043s : 0.01% optimize.opt_a.parallel : 0.000045s : 0.01% optimize.opt_a.flash_sp : 0.000020s : 0.00% optimize.opt_a.merge_comm : 0.000027s : 0.00% optimize.opt_a.allreduce_fusion : 0.000026s : 0.00% optimize.opt_a.matmul_add_comm_reduction : 0.000048s : 0.01% optimize.opt_a.allreduce_slice_to_reducescatter : 0.000002s : 0.00% optimize.opt_a.virtual_shard_identity : 0.000073s : 0.01% optimize.opt_a.virtual_dataset : 0.000047s : 0.01% optimize.opt_a.get_grad_eliminate_ : 0.000044s : 0.01% optimize.opt_a.virtual_output : 0.000042s : 0.01% optimize.opt_a.merge_forward : 0.000029s : 0.01% optimize.opt_a.cell_reuse_recompute_pass : 0.000006s : 0.00% optimize.opt_a.offload_activation : 0.000052s : 0.01% optimize.opt_a.cell_reuse_handle_not_recompute_node_pass : 0.000114s : 0.02% optimize.opt_a.merge_recompute_call_nodes : 0.000003s : 0.00% optimize.opt_a.before_grad : 0.000077s : 0.01% optimize.opt_a.set_forward_comm_id_for_comm_node_pass : 0.000031s : 0.01% optimize.opt_a.meta_fg_expand : 0.000027s : 0.00% optimize.opt_a.flash_sp_send_recv_attached : 0.000008s : 0.00% optimize.opt_a.receive_attached : 0.000005s : 0.00% optimize.opt_a.after_resolve : 0.000061s : 0.01% optimize.opt_a.a_after_grad : 0.000075s : 0.01% optimize.opt_a.renormalize : 0.016090s : 2.87% optimize.opt_a.add_forward_monad_depend : 0.000025s : 0.00% optimize.opt_a.auto_monad_grad : 0.000006s : 0.00% optimize.opt_a.auto_monad_eliminator : 0.000165s : 0.03% optimize.opt_a.cse : 0.001443s : 0.26% optimize.opt_a.a_3 : 0.000336s : 0.06% optimize.py_interpret_to_execute_after_opt_a : 0.000035s : 0.01% optimize.slice_cell_reuse_recomputed_activation : 0.000002s : 0.00% optimize.rewriter_after_opt_a : 0.000879s : 0.16% optimize.convert_after_rewriter : 0.000027s : 0.00% optimize.order_py_execute_after_rewriter : 0.000014s : 0.00% optimize.mutable_eliminate : 0.000887s : 0.16% optimize.opt_b.b_1 : 0.000559s : 0.10% optimize.opt_b.b_2 : 0.000025s : 0.00% optimize.opt_b.updatestate_depend_eliminate : 0.000021s : 0.00% optimize.opt_b.updatestate_assign_eliminate : 0.000012s : 0.00% optimize.opt_b.updatestate_loads_eliminate : 0.000015s : 0.00% optimize.opt_b.renormalize : 0.000002s : 0.00% optimize.opt_b.cse : 0.000097s : 0.02% optimize.optimize_parallel_all_gather_comm : 0.000042s : 0.01% optimize.overlap_param_gather : 0.000002s : 0.00% optimize.cconv : 0.000044s : 0.01% optimize.loop_unroll : 0.000882s : 0.16% optimize.opt_after_cconv.c_1 : 0.000131s : 0.02% optimize.opt_after_cconv.parameter_eliminate : 0.000007s : 0.00% optimize.opt_after_cconv.updatestate_depend_eliminate : 0.000022s : 0.00% optimize.opt_after_cconv.updatestate_assign_eliminate : 0.000011s : 0.00% optimize.opt_after_cconv.updatestate_loads_eliminate : 0.000013s : 0.00% optimize.opt_after_cconv.cse : 0.000081s : 0.01% optimize.opt_after_cconv.renormalize : 0.000001s : 0.00% optimize.remove_dup_value : 0.000098s : 0.02% optimize.tuple_transform.d_1 : 0.000204s : 0.04% optimize.tuple_transform.none_parameter_eliminate : 0.000002s : 0.00% optimize.tuple_transform.renormalize : 0.000000s : 0.00% optimize.tuple_transform.switch_simplify : 0.000024s : 0.00% optimize.partial_unused_args_eliminate : 0.000002s : 0.00% optimize.add_recomputation : 0.000155s : 0.03% optimize.cse_after_recomputation.cse : 0.000060s : 0.01% optimize.environ_conv : 0.000024s : 0.00% optimize.swap_dp_allreduce_reducescatter : 0.000020s : 0.00% optimize.bias_add_comm_swap : 0.000004s : 0.00% optimize.label_micro_interleaved_index : 0.000007s : 0.00% optimize.label_fine_grained_interleaved_index : 0.000003s : 0.00% optimize.merge_cast_opt : 0.000002s : 0.00% optimize.slice_recompute_activation : 0.000002s : 0.00% optimize.micro_interleaved_order_control : 0.000002s : 0.00% optimize.assign_add_opt : 0.000002s : 0.00% optimize.ForceFp32Comm : 0.000001s : 0.00% optimize.remove_cast_before_assign_add : 0.000001s : 0.00% optimize.full_micro_interleaved_order_control : 0.000002s : 0.00% optimize.reorder_send_recv_between_fp_bp : 0.000003s : 0.00% optimize.comm_op_add_attrs : 0.000001s : 0.00% optimize.add_comm_op_reuse_tag : 0.000001s : 0.00% optimize.interleave_split_concat_branches : 0.000001s : 0.00% optimize.interleave_parallel_branches : 0.000001s : 0.00% optimize.overlap_opt_shard_in_pipeline : 0.000001s : 0.00% optimize.overlap_opt_shard_grad_in_pipeline : 0.000002s : 0.00% optimize.control_data_broadcast_order : 0.000040s : 0.01% optimize.grouped_pairwise_exchange_alltoall : 0.000002s : 0.00% optimize.offloading_packed_experts : 0.000011s : 0.00% optimize.overlap_recompute_and_grad_model_parallel : 0.000011s : 0.00% optimize.overlap_grad_matmul_and_grad_allreduce : 0.000001s : 0.00% optimize.overlap_recompute_allgather_and_fa_grad : 0.000001s : 0.00% optimize.overlap_recompute_comm : 0.000003s : 0.00% optimize.overlap_grad_ring_attention : 0.000009s : 0.00% optimize.overlap_grad_flash_sp : 0.000049s : 0.01% optimize.begin_end_overlap_inline : 0.000001s : 0.00% optimize.split_matmul_comm_elemetwise : 0.000002s : 0.00% optimize.split_layernorm_comm : 0.000002s : 0.00% optimize.handle_group_info : 0.000001s : 0.00% optimize.symbol_engine_optimizer.build : 0.000282s : 0.05% optimize.symbol_engine_optimizer.elim_shapecalc : 0.000036s : 0.01% optimize.symbol_engine_optimizer.elim_not_effective : 0.000051s : 0.01% optimize.symbol_engine_optimizer.opt_reshape : 0.000034s : 0.01% optimize.symbol_engine_optimizer.fold_const_symbol : 0.000058s : 0.01% optimize.symbol_engine_optimizer.renormalize : 0.000001s : 0.00% detach_backward : 0.000003s : 0.00% pipeline_parallel_scheduler : 0.000001s : 0.00% auto_monad_reorder : 0.000062s : 0.01% get_jit_bprop_graph : 0.000002s : 0.00% rewriter_after_jit_bprop_graph : 0.000005s : 0.00% opt_after_jit_grad : 0.000793s : 0.14% validate : 0.000114s : 0.02% Time group info: ------[substitution.] 0.003884 569 0.41% : 0.000016s : 8: substitution.depend_value_elim 0.33% : 0.000013s : 14: substitution.elim_not_effective 0.50% : 0.000019s : 12: substitution.float_tuple_getitem_switch 0.69% : 0.000027s : 14: substitution.fold_const_symbol 0.38% : 0.000015s : 17: substitution.graph_param_transform 69.63% : 0.002704s : 71: substitution.inline 0.38% : 0.000015s : 29: substitution.j_node_and_user_rematch 1.01% : 0.000039s : 6: substitution.less_batch_normalization 1.52% : 0.000059s : 2: substitution.list_to_tuple_eliminator_ 0.18% : 0.000007s : 6: substitution.load_eliminater 0.93% : 0.000036s : 21: substitution.minmaximum_grad 0.22% : 0.000008s : 2: substitution.opt_reshape 0.55% : 0.000022s : 29: substitution.remove_not_recompute_node 0.23% : 0.000009s : 4: substitution.replace_old_param 1.76% : 0.000069s : 18: substitution.reshape_eliminate 1.21% : 0.000047s : 24: substitution.switch_simplify 4.16% : 0.000162s : 51: substitution.tuple_list_convert_item_index_to_positive 2.33% : 0.000090s : 51: substitution.tuple_list_get_item_const_eliminator 3.04% : 0.000118s : 51: substitution.tuple_list_get_item_depend_reorder 6.37% : 0.000247s : 73: substitution.tuple_list_get_item_eliminator 2.95% : 0.000114s : 51: substitution.tuple_list_get_set_item_eliminator 0.63% : 0.000025s : 6: substitution.updatestate_pure_node_eliminater 0.59% : 0.000023s : 9: substitution.updatestate_useless_node_eliminater ------[type_inference.] 0.477458 2 94.99% : 0.453523s : 1: type_inference.infer 5.01% : 0.023935s : 1: type_inference.specialize ------[replace.] 0.001208 115 5.09% : 0.000061s : 5: replace.depend_value_elim 57.33% : 0.000693s : 71: replace.inline 3.72% : 0.000045s : 2: replace.list_to_tuple_eliminator_ 24.46% : 0.000296s : 24: replace.switch_simplify 9.41% : 0.000114s : 13: replace.tuple_list_get_item_eliminator ------[match.] 0.002794 115 0.08% : 0.000002s : 5: match.depend_value_elim 95.03% : 0.002655s : 71: match.inline 2.02% : 0.000056s : 2: match.list_to_tuple_eliminator_ 1.21% : 0.000034s : 24: match.switch_simplify 1.66% : 0.000046s : 13: match.tuple_list_get_item_eliminator ------[predicate.] 0.001643 9701 1.18% : 0.000019s : 123: predicate.accumulaten_eliminater 0.34% : 0.000006s : 17: predicate.ad_related_special_op_eliminate 0.50% : 0.000008s : 58: predicate.addn_check_dump 1.14% : 0.000019s : 123: predicate.addn_zero_filter 1.10% : 0.000018s : 123: predicate.adjust_all_reduce_mul_add 2.48% : 0.000041s : 181: predicate.arithmetic_simplify 1.17% : 0.000019s : 123: predicate.cast_eliminate 0.31% : 0.000005s : 34: predicate.check_bprop_eliminate 0.49% : 0.000008s : 58: predicate.compare_switch_simplify 0.08% : 0.000001s : 17: predicate.const_output_eliminate 0.68% : 0.000011s : 61: predicate.depend_value_elim 1.23% : 0.000020s : 123: predicate.dict_get_item_const_eliminator 1.32% : 0.000022s : 123: predicate.dict_get_item_eliminator 1.12% : 0.000018s : 123: predicate.dict_set_item_eliminator 0.40% : 0.000007s : 34: predicate.dumpgradient_eliminate 0.09% : 0.000002s : 17: predicate.elim_not_effective 0.21% : 0.000003s : 17: predicate.elim_shapecalc_of_broadcastargs 1.22% : 0.000020s : 140: predicate.environ_add_const_eliminate 1.19% : 0.000020s : 140: predicate.environ_get_add_eliminate 1.20% : 0.000020s : 140: predicate.environ_get_depend_swap 1.84% : 0.000030s : 198: predicate.environ_get_eliminate 1.20% : 0.000020s : 140: predicate.environ_get_set_eliminate 1.90% : 0.000031s : 209: predicate.exchange_switch_depend_value 2.84% : 0.000047s : 209: predicate.float_depend_g_call 0.49% : 0.000008s : 58: predicate.float_environ_get_switch 0.70% : 0.000011s : 75: predicate.float_tuple_getitem_switch 0.07% : 0.000001s : 17: predicate.fold_const_symbol 0.40% : 0.000007s : 35: predicate.get_grad_eliminate 0.10% : 0.000002s : 17: predicate.graph_param_transform 0.54% : 0.000009s : 58: predicate.incorporate_call 0.45% : 0.000007s : 58: predicate.incorporate_call_switch 5.90% : 0.000097s : 459: predicate.inline 0.47% : 0.000008s : 35: predicate.inline_without_move 0.15% : 0.000002s : 35: predicate.j_node_and_user_rematch 0.54% : 0.000009s : 36: predicate.less_batch_normalization 1.75% : 0.000029s : 172: predicate.list_to_tuple_eliminator_ 2.58% : 0.000042s : 295: predicate.load_eliminater 0.37% : 0.000006s : 17: predicate.loop_unroll_after_grad 3.09% : 0.000051s : 314: predicate.loop_unroll_before_grad 1.66% : 0.000027s : 157: predicate.make_slice_get_slice_eliminator 0.51% : 0.000008s : 58: predicate.merge_addn 0.32% : 0.000005s : 34: predicate.micro_step_allgather_replace 0.33% : 0.000005s : 34: predicate.mini_step_allgather_replace 1.17% : 0.000019s : 123: predicate.minmaximum_grad 0.46% : 0.000007s : 17: predicate.mutable_eliminate 0.22% : 0.000004s : 17: predicate.opt_reshape 0.18% : 0.000003s : 17: predicate.parallel_virtual_node 3.58% : 0.000059s : 209: predicate.partial_defer_inline 1.56% : 0.000026s : 155: predicate.partial_eliminate 1.29% : 0.000021s : 123: predicate.print_const_string_wrapper 0.47% : 0.000008s : 53: predicate.reduce_all_const_elim 1.81% : 0.000030s : 123: predicate.reduce_eliminate 2.62% : 0.000043s : 295: predicate.redundant_stop_gradient_eliminater 0.22% : 0.000004s : 35: predicate.remove_not_recompute_node 1.12% : 0.000018s : 172: predicate.replace_applicator 0.18% : 0.000003s : 35: predicate.replace_old_param 0.11% : 0.000002s : 17: predicate.reset_defer_inline 1.35% : 0.000022s : 123: predicate.reshape_eliminate 0.33% : 0.000005s : 34: predicate.row_tensor_add_zeros_like 0.22% : 0.000004s : 17: predicate.row_tensor_eliminate 0.56% : 0.000009s : 34: predicate.same_eliminate 0.22% : 0.000004s : 40: predicate.set_cell_output_no_recompute 0.41% : 0.000007s : 35: predicate.shard_identity_eliminate 0.35% : 0.000006s : 34: predicate.special_op_eliminate 0.60% : 0.000010s : 58: predicate.specialize_transform 0.43% : 0.000007s : 34: predicate.split_environ_get_set_with_tuple_value 0.44% : 0.000007s : 35: predicate.stack_unstack_eliminate 0.18% : 0.000003s : 17: predicate.switch_call_monad_eliminater 2.25% : 0.000037s : 209: predicate.switch_defer_inline 2.33% : 0.000038s : 243: predicate.switch_layer_defer_inline 6.50% : 0.000107s : 646: predicate.switch_simplify 1.33% : 0.000022s : 123: predicate.tile_eliminate 1.14% : 0.000019s : 123: predicate.transpose_eliminate 1.78% : 0.000029s : 157: predicate.tuple_list_convert_item_index_to_positive 1.75% : 0.000029s : 157: predicate.tuple_list_get_item_const_eliminator 1.64% : 0.000027s : 157: predicate.tuple_list_get_item_depend_reorder 3.18% : 0.000052s : 228: predicate.tuple_list_get_item_eliminator 1.61% : 0.000026s : 157: predicate.tuple_list_get_set_item_eliminator 2.30% : 0.000038s : 215: predicate.tuple_list_set_item_eliminator 1.67% : 0.000027s : 170: predicate.tuple_to_list_eliminator_ 2.49% : 0.000041s : 295: predicate.updatestate_pure_node_eliminater 3.08% : 0.000051s : 353: predicate.updatestate_useless_node_eliminater 0.17% : 0.000003s : 17: predicate.value_based_eliminate 0.37% : 0.000006s : 35: predicate.virtual_dataset_eliminate 0.33% : 0.000005s : 35: predicate.virtual_output_eliminate 0.16% : 0.000003s : 17: predicate.virtual_view_grad_eliminate 0.20% : 0.000003s : 17: predicate.zero_like_fill_zero ------[func_graph_cloner_run.] 0.020823 135 47.47% : 0.009885s : 55: func_graph_cloner_run.FuncGraphClonerGraph 52.53% : 0.010938s : 80: func_graph_cloner_run.FuncGraphSpecializer ------[meta_graph.] 0.000000 0 ------[manager.] 0.000000 0 ------[pynative] 0.000000 0 ------[others.] 0.747497 209 0.00% : 0.000004s : 1: ForceFp32Comm 2.36% : 0.017637s : 1: add_attr 2.36% : 0.017614s : 1: add_attr_with_inline 0.00% : 0.000005s : 1: add_comm_op_reuse_tag 0.02% : 0.000162s : 1: add_recomputation 0.00% : 0.000006s : 1: assign_add_opt 0.11% : 0.000829s : 1: auto_monad 0.01% : 0.000069s : 1: auto_monad_reorder 0.00% : 0.000005s : 1: begin_end_overlap_inline 0.00% : 0.000006s : 1: bias_add_comm_swap 0.11% : 0.000834s : 1: bootstrap 0.01% : 0.000048s : 1: cconv 0.00% : 0.000004s : 1: comm_op_add_attrs 0.01% : 0.000044s : 1: control_data_broadcast_order 0.00% : 0.000034s : 1: convert_after_rewriter 0.01% : 0.000077s : 1: cse_after_recomputation 0.00% : 0.000005s : 1: dataset_repeat_opt 0.00% : 0.000007s : 1: detach_backward 0.00% : 0.000028s : 1: environ_conv 0.03% : 0.000238s : 1: event_method 0.00% : 0.000006s : 1: full_micro_interleaved_order_control 0.00% : 0.000006s : 1: get_jit_bprop_graph 0.01% : 0.000061s : 1: graph_reusing 0.00% : 0.000004s : 1: grouped_pairwise_exchange_alltoall 0.00% : 0.000005s : 1: handle_group_info 0.00% : 0.000007s : 1: inline 0.00% : 0.000006s : 1: insert-virtual-dataset 0.00% : 0.000004s : 1: interleave_parallel_branches 0.00% : 0.000004s : 1: interleave_split_concat_branches 0.00% : 0.000006s : 1: label_fine_grained_interleaved_index 0.00% : 0.000010s : 1: label_micro_interleaved_index 0.12% : 0.000898s : 1: loop_unroll 0.00% : 0.000006s : 1: merge_cast_opt 0.00% : 0.000005s : 1: micro_interleaved_order_control 0.12% : 0.000903s : 1: mutable_eliminate 0.00% : 0.000015s : 1: offloading_packed_experts 0.01% : 0.000042s : 1: opt.transform.loop_unroll_optimizer 0.01% : 0.000047s : 1: opt.transform.mutable_eliminate 7.23% : 0.054011s : 95: opt.transform.opt_a 0.02% : 0.000130s : 1: opt.transform.opt_after_cconv 0.01% : 0.000078s : 1: opt.transform.opt_after_jit_grad 0.07% : 0.000542s : 28: opt.transform.opt_b 0.03% : 0.000225s : 2: opt.transform.opt_trans_graph 0.02% : 0.000172s : 4: opt.transform.symbol_engine_opt 9.81% : 0.073366s : 1: opt_a 0.04% : 0.000322s : 1: opt_after_cconv 0.11% : 0.000810s : 1: opt_after_jit_grad 0.11% : 0.000798s : 1: opt_b 10.70% : 0.079981s : 1: optimize 0.01% : 0.000047s : 1: optimize_parallel_all_gather_comm 0.00% : 0.000018s : 1: order_py_execute_after_rewriter 0.01% : 0.000054s : 1: overlap_grad_flash_sp 0.00% : 0.000004s : 1: overlap_grad_matmul_and_grad_allreduce 0.00% : 0.000012s : 1: overlap_grad_ring_attention 0.00% : 0.000006s : 1: overlap_opt_shard_grad_in_pipeline 0.00% : 0.000004s : 1: overlap_opt_shard_in_pipeline 0.00% : 0.000006s : 1: overlap_param_gather 0.00% : 0.000004s : 1: overlap_recompute_allgather_and_fa_grad 0.00% : 0.000014s : 1: overlap_recompute_and_grad_model_parallel 0.00% : 0.000007s : 1: overlap_recompute_comm 0.00% : 0.000010s : 1: parallel-infer-symbol 0.00% : 0.000004s : 1: parallel-infer-symbol-second 0.00% : 0.000006s : 1: partial_unused_args_eliminate 0.00% : 0.000005s : 1: pipeline_parallel_scheduler 0.00% : 0.000005s : 1: pipeline_split 0.03% : 0.000261s : 1: pre_auto_parallel 0.03% : 0.000261s : 1: py_interpret_to_execute 0.01% : 0.000039s : 1: py_interpret_to_execute_after_opt_a 0.00% : 0.000004s : 1: remove_cast_before_assign_add 0.01% : 0.000103s : 1: remove_dup_value 0.99% : 0.007416s : 1: renormalize.infer 1.16% : 0.008649s : 1: renormalize.specialize 0.00% : 0.000007s : 1: reorder_send_recv_between_fp_bp 0.00% : 0.000009s : 1: rewriter_after_jit_bprop_graph 0.12% : 0.000894s : 1: rewriter_after_opt_a 0.11% : 0.000798s : 1: rewriter_before_opt_a 0.00% : 0.000006s : 1: slice_cell_reuse_recomputed_activation 0.00% : 0.000005s : 1: slice_recompute_activation 0.00% : 0.000004s : 1: split_layernorm_comm 0.00% : 0.000005s : 1: split_matmul_comm_elemetwise 0.00% : 0.000024s : 1: swap_dp_allreduce_reducescatter 0.07% : 0.000524s : 1: symbol_engine_optimizer 0.04% : 0.000266s : 1: tuple_transform 63.93% : 0.477890s : 1: type_inference . [hook] pytest_runtest_teardown:test_matmul_qkv_bf16_0[True-mstype0-k_n_shape2-1] tests/st/infer/ops/test_internal_ops/test_matmul_split.py::test_matmul_qkv_bf16_0[True-mstype0-k_n_shape2-1],max_mem:196.0M [WARNING] ME(152078:281472851701552,MainProcess):2026-01-29-17:44:08.994.498 [mindspore/context.py:1334] For 'context.set_context', the parameter 'device_target' will be deprecated and removed in a future version. Please use the api mindspore.set_device() instead. TotalTime = 0.827583, [21] [bootstrap]: 0.00068445 [type_inference]: 0.707362 [event_method]: 0.00022478 [auto_monad]: 0.00076289 [graph_reusing]: 4.816e-05 [inline]: 5.44998e-06 [add_attr]: 0.0114449, [1] [add_attr_with_inline]: 0.0114271, [1] [Cycle 1]: 0.00031327, [2] [tag_attr]: 0.00020213 [meta_addattr_fg_expand]: 5.052e-05 [parallel-infer-symbol]: 3.91999e-06 [pre_auto_parallel]: 0.00025443 [insert-virtual-dataset]: 2.61e-06 [parallel-infer-symbol-second]: 1.15999e-06 [dataset_repeat_opt]: 2.43998e-06 [pipeline_split]: 2.06e-06 [optimize]: 0.105695, [53] [py_interpret_to_execute]: 0.00023083 [rewriter_before_opt_a]: 0.00076571 [opt_a]: 0.0994253, [2] [Cycle 1]: 0.0970498, [45] [expand_dump_flag]: 1.616e-05 [switch_simplify]: 0.0146833 [loop_unroll]: 0.00046625 [a_1]: 0.0178241 [with_stream_mark]: 6.362e-05 [recompute_prepare]: 4.255e-05 [updatestate_depend_eliminate]: 2.191e-05 [updatestate_assign_eliminate]: 1.577e-05 [updatestate_loads_eliminate]: 2.009e-05 [parameter_eliminate]: 2.88e-06 [a_2]: 0.00080376 [accelerated_algorithm]: 7.134e-05 [shard]: 2.57001e-06 [meta_shard_fg_expand]: 2.653e-05 [shard_inline]: 2.484e-05 [merge_send_recv]: 2.178e-05 [auto_parallel]: 2.202e-05 [parallel]: 3.687e-05 [flash_sp]: 1.57e-05 [merge_comm]: 1.445e-05 [allreduce_fusion]: 1.287e-05 [matmul_add_comm_reduction]: 2.322e-05 [allreduce_slice_to_reducescatter]: 8.10018e-07 [virtual_shard_identity]: 2.859e-05 [virtual_dataset]: 2.367e-05 [get_grad_eliminate_]: 2.178e-05 [virtual_output]: 2.125e-05 [merge_forward]: 1.378e-05 [cell_reuse_recompute_pass]: 3.06999e-06 [offload_activation]: 2.42e-05 [cell_reuse_handle_not_recompute_node_pass]: 4.859e-05 [merge_recompute_call_nodes]: 1.58002e-06 [before_grad]: 3.993e-05 [set_forward_comm_id_for_comm_node_pass]: 1.421e-05 [meta_fg_expand]: 1.796e-05 [flash_sp_send_recv_attached]: 6.53998e-06 [receive_attached]: 2.41998e-06 [after_resolve]: 3.166e-05 [a_after_grad]: 3.712e-05 [renormalize]: 0.0386732 [add_forward_monad_depend]: 1.929e-05 [auto_monad_grad]: 3.25e-06 [auto_monad_eliminator]: 9.682e-05 [cse]: 0.0026111 [a_3]: 0.0205478 [Cycle 2]: 0.00235618, [45] [expand_dump_flag]: 4.55999e-06 [switch_simplify]: 3.437e-05 [loop_unroll]: 2.331e-05 [a_1]: 0.00069303 [with_stream_mark]: 3.818e-05 [recompute_prepare]: 2.624e-05 [updatestate_depend_eliminate]: 1.743e-05 [updatestate_assign_eliminate]: 1.247e-05 [updatestate_loads_eliminate]: 1.504e-05 [parameter_eliminate]: 3.34001e-06 [a_2]: 0.00035173 [accelerated_algorithm]: 3.61e-05 [shard]: 3.09001e-06 [meta_shard_fg_expand]: 1.238e-05 [shard_inline]: 2.366e-05 [merge_send_recv]: 2.211e-05 [auto_parallel]: 2.234e-05 [parallel]: 1.157e-05 [flash_sp]: 4.58001e-06 [merge_comm]: 1.388e-05 [allreduce_fusion]: 1.314e-05 [matmul_add_comm_reduction]: 2.694e-05 [allreduce_slice_to_reducescatter]: 7.79983e-07 [virtual_shard_identity]: 2.749e-05 [virtual_dataset]: 2.294e-05 [get_grad_eliminate_]: 2.38e-05 [virtual_output]: 2.254e-05 [merge_forward]: 1.355e-05 [cell_reuse_recompute_pass]: 3.74002e-06 [offload_activation]: 2.744e-05 [cell_reuse_handle_not_recompute_node_pass]: 4.923e-05 [merge_recompute_call_nodes]: 1.98002e-06 [before_grad]: 4.073e-05 [set_forward_comm_id_for_comm_node_pass]: 1.564e-05 [meta_fg_expand]: 1.118e-05 [flash_sp_send_recv_attached]: 1.96e-06 [receive_attached]: 2.44001e-06 [after_resolve]: 3.034e-05 [a_after_grad]: 3.344e-05 [renormalize]: 8.00064e-08 [add_forward_monad_depend]: 5.01002e-06 [auto_monad_grad]: 2.69999e-06 [auto_monad_eliminator]: 7.163e-05 [cse]: 0.00010282 [a_3]: 0.00014313 [py_interpret_to_execute_after_opt_a]: 3.658e-05 [slice_cell_reuse_recomputed_activation]: 2.85998e-06 [rewriter_after_opt_a]: 0.0008796 [convert_after_rewriter]: 2.65e-05 [order_py_execute_after_rewriter]: 1.477e-05 [mutable_eliminate]: 0.00084713 [opt_b]: 0.00079636, [1] [Cycle 1]: 0.00078596, [7] [b_1]: 0.00055437 [b_2]: 2.438e-05 [updatestate_depend_eliminate]: 2.129e-05 [updatestate_assign_eliminate]: 1.285e-05 [updatestate_loads_eliminate]: 1.462e-05 [renormalize]: 1.19e-06 [cse]: 0.00010895 [optimize_parallel_all_gather_comm]: 4.121e-05 [overlap_param_gather]: 4.38001e-06 [cconv]: 4.17e-05 [loop_unroll]: 0.00056918 [opt_after_cconv]: 0.00029296, [1] [Cycle 1]: 0.00028458, [7] [c_1]: 0.00012384 [parameter_eliminate]: 4.76002e-06 [updatestate_depend_eliminate]: 1.888e-05 [updatestate_assign_eliminate]: 1.063e-05 [updatestate_loads_eliminate]: 1.319e-05 [cse]: 7.443e-05 [renormalize]: 7.50006e-07 [remove_dup_value]: 9.245e-05 [tuple_transform]: 0.0002525, [1] [Cycle 1]: 0.00024647, [4] [d_1]: 0.00019556 [none_parameter_eliminate]: 2.44999e-06 [renormalize]: 3.7998e-07 [switch_simplify]: 2.452e-05 [partial_unused_args_eliminate]: 1.96e-06 [add_recomputation]: 0.00015266 [cse_after_recomputation]: 7.159e-05, [1] [Cycle 1]: 6.498e-05, [1] [cse]: 5.824e-05 [environ_conv]: 2.287e-05 [swap_dp_allreduce_reducescatter]: 1.776e-05 [bias_add_comm_swap]: 2.98e-06 [label_micro_interleaved_index]: 4.56002e-06 [label_fine_grained_interleaved_index]: 2.47001e-06 [merge_cast_opt]: 1.35001e-06 [slice_recompute_activation]: 2.39001e-06 [micro_interleaved_order_control]: 2.49999e-06 [assign_add_opt]: 1.19998e-06 [ForceFp32Comm]: 8.50006e-07 [remove_cast_before_assign_add]: 1.05001e-06 [full_micro_interleaved_order_control]: 1.95001e-06 [reorder_send_recv_between_fp_bp]: 2.98e-06 [comm_op_add_attrs]: 1.17e-06 [add_comm_op_reuse_tag]: 9.79984e-07 [interleave_split_concat_branches]: 1.49e-06 [interleave_parallel_branches]: 1.06002e-06 [overlap_opt_shard_in_pipeline]: 5.12e-06 [overlap_opt_shard_grad_in_pipeline]: 1.96e-06 [control_data_broadcast_order]: 3.871e-05 [grouped_pairwise_exchange_alltoall]: 1.37999e-06 [offloading_packed_experts]: 1.067e-05 [overlap_recompute_and_grad_model_parallel]: 1.042e-05 [overlap_grad_matmul_and_grad_allreduce]: 1.37e-06 [overlap_recompute_allgather_and_fa_grad]: 1.32999e-06 [overlap_recompute_comm]: 2.51998e-06 [overlap_grad_ring_attention]: 9.53002e-06 [overlap_grad_flash_sp]: 4.808e-05 [begin_end_overlap_inline]: 4.89992e-07 [split_matmul_comm_elemetwise]: 1.95001e-06 [split_layernorm_comm]: 1.69e-06 [handle_group_info]: 9.39996e-07 [symbol_engine_optimizer]: 0.00062817, [1] [Cycle 1]: 0.00062249, [6] [build]: 0.00037996 [elim_shapecalc]: 3.645e-05 [elim_not_effective]: 5.157e-05 [opt_reshape]: 3.382e-05 [fold_const_symbol]: 6.883e-05 [renormalize]: 5.89993e-07 [detach_backward]: 2.99999e-06 [pipeline_parallel_scheduler]: 1.49e-06 [auto_monad_reorder]: 6.143e-05 [get_jit_bprop_graph]: 1.89999e-06 [rewriter_after_jit_bprop_graph]: 4.97e-06 [opt_after_jit_grad]: 0.00064801 [validate]: 0.00010624 Sums bootstrap : 0.000684s : 0.08% type_inference : 0.707362s : 86.82% event_method : 0.000225s : 0.03% auto_monad : 0.000763s : 0.09% graph_reusing : 0.000048s : 0.01% inline : 0.000005s : 0.00% add_attr.add_attr_with_inline.tag_attr : 0.000202s : 0.02% add_attr.add_attr_with_inline.meta_addattr_fg_expand : 0.000051s : 0.01% parallel-infer-symbol : 0.000004s : 0.00% pre_auto_parallel : 0.000254s : 0.03% insert-virtual-dataset : 0.000003s : 0.00% parallel-infer-symbol-second : 0.000001s : 0.00% dataset_repeat_opt : 0.000002s : 0.00% pipeline_split : 0.000002s : 0.00% optimize.py_interpret_to_execute : 0.000231s : 0.03% optimize.rewriter_before_opt_a : 0.000766s : 0.09% optimize.opt_a.expand_dump_flag : 0.000021s : 0.00% optimize.opt_a.switch_simplify : 0.014718s : 1.81% optimize.opt_a.loop_unroll : 0.000490s : 0.06% optimize.opt_a.a_1 : 0.018517s : 2.27% optimize.opt_a.with_stream_mark : 0.000102s : 0.01% optimize.opt_a.recompute_prepare : 0.000069s : 0.01% optimize.opt_a.updatestate_depend_eliminate : 0.000039s : 0.00% optimize.opt_a.updatestate_assign_eliminate : 0.000028s : 0.00% optimize.opt_a.updatestate_loads_eliminate : 0.000035s : 0.00% optimize.opt_a.parameter_eliminate : 0.000006s : 0.00% optimize.opt_a.a_2 : 0.001155s : 0.14% optimize.opt_a.accelerated_algorithm : 0.000107s : 0.01% optimize.opt_a.shard : 0.000006s : 0.00% optimize.opt_a.meta_shard_fg_expand : 0.000039s : 0.00% optimize.opt_a.shard_inline : 0.000048s : 0.01% optimize.opt_a.merge_send_recv : 0.000044s : 0.01% optimize.opt_a.auto_parallel : 0.000044s : 0.01% optimize.opt_a.parallel : 0.000048s : 0.01% optimize.opt_a.flash_sp : 0.000020s : 0.00% optimize.opt_a.merge_comm : 0.000028s : 0.00% optimize.opt_a.allreduce_fusion : 0.000026s : 0.00% optimize.opt_a.matmul_add_comm_reduction : 0.000050s : 0.01% optimize.opt_a.allreduce_slice_to_reducescatter : 0.000002s : 0.00% optimize.opt_a.virtual_shard_identity : 0.000056s : 0.01% optimize.opt_a.virtual_dataset : 0.000047s : 0.01% optimize.opt_a.get_grad_eliminate_ : 0.000046s : 0.01% optimize.opt_a.virtual_output : 0.000044s : 0.01% optimize.opt_a.merge_forward : 0.000027s : 0.00% optimize.opt_a.cell_reuse_recompute_pass : 0.000007s : 0.00% optimize.opt_a.offload_activation : 0.000052s : 0.01% optimize.opt_a.cell_reuse_handle_not_recompute_node_pass : 0.000098s : 0.01% optimize.opt_a.merge_recompute_call_nodes : 0.000004s : 0.00% optimize.opt_a.before_grad : 0.000081s : 0.01% optimize.opt_a.set_forward_comm_id_for_comm_node_pass : 0.000030s : 0.00% optimize.opt_a.meta_fg_expand : 0.000029s : 0.00% optimize.opt_a.flash_sp_send_recv_attached : 0.000008s : 0.00% optimize.opt_a.receive_attached : 0.000005s : 0.00% optimize.opt_a.after_resolve : 0.000062s : 0.01% optimize.opt_a.a_after_grad : 0.000071s : 0.01% optimize.opt_a.renormalize : 0.038673s : 4.75% optimize.opt_a.add_forward_monad_depend : 0.000024s : 0.00% optimize.opt_a.auto_monad_grad : 0.000006s : 0.00% optimize.opt_a.auto_monad_eliminator : 0.000168s : 0.02% optimize.opt_a.cse : 0.002714s : 0.33% optimize.opt_a.a_3 : 0.020691s : 2.54% optimize.py_interpret_to_execute_after_opt_a : 0.000037s : 0.00% optimize.slice_cell_reuse_recomputed_activation : 0.000003s : 0.00% optimize.rewriter_after_opt_a : 0.000880s : 0.11% optimize.convert_after_rewriter : 0.000026s : 0.00% optimize.order_py_execute_after_rewriter : 0.000015s : 0.00% optimize.mutable_eliminate : 0.000847s : 0.10% optimize.opt_b.b_1 : 0.000554s : 0.07% optimize.opt_b.b_2 : 0.000024s : 0.00% optimize.opt_b.updatestate_depend_eliminate : 0.000021s : 0.00% optimize.opt_b.updatestate_assign_eliminate : 0.000013s : 0.00% optimize.opt_b.updatestate_loads_eliminate : 0.000015s : 0.00% optimize.opt_b.renormalize : 0.000001s : 0.00% optimize.opt_b.cse : 0.000109s : 0.01% optimize.optimize_parallel_all_gather_comm : 0.000041s : 0.01% optimize.overlap_param_gather : 0.000004s : 0.00% optimize.cconv : 0.000042s : 0.01% optimize.loop_unroll : 0.000569s : 0.07% optimize.opt_after_cconv.c_1 : 0.000124s : 0.02% optimize.opt_after_cconv.parameter_eliminate : 0.000005s : 0.00% optimize.opt_after_cconv.updatestate_depend_eliminate : 0.000019s : 0.00% optimize.opt_after_cconv.updatestate_assign_eliminate : 0.000011s : 0.00% optimize.opt_after_cconv.updatestate_loads_eliminate : 0.000013s : 0.00% optimize.opt_after_cconv.cse : 0.000074s : 0.01% optimize.opt_after_cconv.renormalize : 0.000001s : 0.00% optimize.remove_dup_value : 0.000092s : 0.01% optimize.tuple_transform.d_1 : 0.000196s : 0.02% optimize.tuple_transform.none_parameter_eliminate : 0.000002s : 0.00% optimize.tuple_transform.renormalize : 0.000000s : 0.00% optimize.tuple_transform.switch_simplify : 0.000025s : 0.00% optimize.partial_unused_args_eliminate : 0.000002s : 0.00% optimize.add_recomputation : 0.000153s : 0.02% optimize.cse_after_recomputation.cse : 0.000058s : 0.01% optimize.environ_conv : 0.000023s : 0.00% optimize.swap_dp_allreduce_reducescatter : 0.000018s : 0.00% optimize.bias_add_comm_swap : 0.000003s : 0.00% optimize.label_micro_interleaved_index : 0.000005s : 0.00% optimize.label_fine_grained_interleaved_index : 0.000002s : 0.00% optimize.merge_cast_opt : 0.000001s : 0.00% optimize.slice_recompute_activation : 0.000002s : 0.00% optimize.micro_interleaved_order_control : 0.000002s : 0.00% optimize.assign_add_opt : 0.000001s : 0.00% optimize.ForceFp32Comm : 0.000001s : 0.00% optimize.remove_cast_before_assign_add : 0.000001s : 0.00% optimize.full_micro_interleaved_order_control : 0.000002s : 0.00% optimize.reorder_send_recv_between_fp_bp : 0.000003s : 0.00% optimize.comm_op_add_attrs : 0.000001s : 0.00% optimize.add_comm_op_reuse_tag : 0.000001s : 0.00% optimize.interleave_split_concat_branches : 0.000001s : 0.00% optimize.interleave_parallel_branches : 0.000001s : 0.00% optimize.overlap_opt_shard_in_pipeline : 0.000005s : 0.00% optimize.overlap_opt_shard_grad_in_pipeline : 0.000002s : 0.00% optimize.control_data_broadcast_order : 0.000039s : 0.00% optimize.grouped_pairwise_exchange_alltoall : 0.000001s : 0.00% optimize.offloading_packed_experts : 0.000011s : 0.00% optimize.overlap_recompute_and_grad_model_parallel : 0.000010s : 0.00% optimize.overlap_grad_matmul_and_grad_allreduce : 0.000001s : 0.00% optimize.overlap_recompute_allgather_and_fa_grad : 0.000001s : 0.00% optimize.overlap_recompute_comm : 0.000003s : 0.00% optimize.overlap_grad_ring_attention : 0.000010s : 0.00% optimize.overlap_grad_flash_sp : 0.000048s : 0.01% optimize.begin_end_overlap_inline : 0.000000s : 0.00% optimize.split_matmul_comm_elemetwise : 0.000002s : 0.00% optimize.split_layernorm_comm : 0.000002s : 0.00% optimize.handle_group_info : 0.000001s : 0.00% optimize.symbol_engine_optimizer.build : 0.000380s : 0.05% optimize.symbol_engine_optimizer.elim_shapecalc : 0.000036s : 0.00% optimize.symbol_engine_optimizer.elim_not_effective : 0.000052s : 0.01% optimize.symbol_engine_optimizer.opt_reshape : 0.000034s : 0.00% optimize.symbol_engine_optimizer.fold_const_symbol : 0.000069s : 0.01% optimize.symbol_engine_optimizer.renormalize : 0.000001s : 0.00% detach_backward : 0.000003s : 0.00% pipeline_parallel_scheduler : 0.000001s : 0.00% auto_monad_reorder : 0.000061s : 0.01% get_jit_bprop_graph : 0.000002s : 0.00% rewriter_after_jit_bprop_graph : 0.000005s : 0.00% opt_after_jit_grad : 0.000648s : 0.08% validate : 0.000106s : 0.01% Time group info: ------[substitution.] 0.004520 569 0.39% : 0.000018s : 8: substitution.depend_value_elim 0.32% : 0.000014s : 14: substitution.elim_not_effective 0.41% : 0.000019s : 12: substitution.float_tuple_getitem_switch 0.82% : 0.000037s : 14: substitution.fold_const_symbol 0.32% : 0.000015s : 17: substitution.graph_param_transform 71.77% : 0.003244s : 71: substitution.inline 0.32% : 0.000014s : 29: substitution.j_node_and_user_rematch 0.89% : 0.000040s : 6: substitution.less_batch_normalization 1.71% : 0.000077s : 2: substitution.list_to_tuple_eliminator_ 0.16% : 0.000007s : 6: substitution.load_eliminater 0.72% : 0.000033s : 21: substitution.minmaximum_grad 0.18% : 0.000008s : 2: substitution.opt_reshape 0.45% : 0.000020s : 29: substitution.remove_not_recompute_node 0.18% : 0.000008s : 4: substitution.replace_old_param 1.64% : 0.000074s : 18: substitution.reshape_eliminate 1.71% : 0.000077s : 24: substitution.switch_simplify 3.72% : 0.000168s : 51: substitution.tuple_list_convert_item_index_to_positive 2.20% : 0.000099s : 51: substitution.tuple_list_get_item_const_eliminator 2.88% : 0.000130s : 51: substitution.tuple_list_get_item_depend_reorder 5.40% : 0.000244s : 73: substitution.tuple_list_get_item_eliminator 2.70% : 0.000122s : 51: substitution.tuple_list_get_set_item_eliminator 0.59% : 0.000026s : 6: substitution.updatestate_pure_node_eliminater 0.52% : 0.000023s : 9: substitution.updatestate_useless_node_eliminater ------[type_inference.] 0.705761 2 84.26% : 0.594705s : 1: type_inference.infer 15.74% : 0.111057s : 1: type_inference.specialize ------[replace.] 0.001414 115 3.92% : 0.000055s : 5: replace.depend_value_elim 56.26% : 0.000795s : 71: replace.inline 3.59% : 0.000051s : 2: replace.list_to_tuple_eliminator_ 27.31% : 0.000386s : 24: replace.switch_simplify 8.91% : 0.000126s : 13: replace.tuple_list_get_item_eliminator ------[match.] 0.003373 115 0.06% : 0.000002s : 5: match.depend_value_elim 94.78% : 0.003197s : 71: match.inline 2.19% : 0.000074s : 2: match.list_to_tuple_eliminator_ 1.73% : 0.000058s : 24: match.switch_simplify 1.24% : 0.000042s : 13: match.tuple_list_get_item_eliminator ------[predicate.] 0.001770 9701 1.12% : 0.000020s : 123: predicate.accumulaten_eliminater 0.30% : 0.000005s : 17: predicate.ad_related_special_op_eliminate 0.45% : 0.000008s : 58: predicate.addn_check_dump 1.08% : 0.000019s : 123: predicate.addn_zero_filter 1.12% : 0.000020s : 123: predicate.adjust_all_reduce_mul_add 2.35% : 0.000042s : 181: predicate.arithmetic_simplify 1.15% : 0.000020s : 123: predicate.cast_eliminate 0.29% : 0.000005s : 34: predicate.check_bprop_eliminate 0.46% : 0.000008s : 58: predicate.compare_switch_simplify 0.08% : 0.000001s : 17: predicate.const_output_eliminate 0.60% : 0.000011s : 61: predicate.depend_value_elim 1.21% : 0.000021s : 123: predicate.dict_get_item_const_eliminator 1.28% : 0.000023s : 123: predicate.dict_get_item_eliminator 1.03% : 0.000018s : 123: predicate.dict_set_item_eliminator 0.38% : 0.000007s : 34: predicate.dumpgradient_eliminate 0.08% : 0.000001s : 17: predicate.elim_not_effective 0.20% : 0.000003s : 17: predicate.elim_shapecalc_of_broadcastargs 1.25% : 0.000022s : 140: predicate.environ_add_const_eliminate 1.45% : 0.000026s : 140: predicate.environ_get_add_eliminate 1.28% : 0.000023s : 140: predicate.environ_get_depend_swap 1.66% : 0.000029s : 198: predicate.environ_get_eliminate 1.14% : 0.000020s : 140: predicate.environ_get_set_eliminate 1.85% : 0.000033s : 209: predicate.exchange_switch_depend_value 2.74% : 0.000048s : 209: predicate.float_depend_g_call 0.47% : 0.000008s : 58: predicate.float_environ_get_switch 0.67% : 0.000012s : 75: predicate.float_tuple_getitem_switch 0.08% : 0.000001s : 17: predicate.fold_const_symbol 0.35% : 0.000006s : 35: predicate.get_grad_eliminate 0.08% : 0.000002s : 17: predicate.graph_param_transform 0.47% : 0.000008s : 58: predicate.incorporate_call 0.42% : 0.000007s : 58: predicate.incorporate_call_switch 5.68% : 0.000101s : 459: predicate.inline 0.40% : 0.000007s : 35: predicate.inline_without_move 0.14% : 0.000003s : 35: predicate.j_node_and_user_rematch 0.47% : 0.000008s : 36: predicate.less_batch_normalization 1.58% : 0.000028s : 172: predicate.list_to_tuple_eliminator_ 2.35% : 0.000042s : 295: predicate.load_eliminater 0.32% : 0.000006s : 17: predicate.loop_unroll_after_grad 5.88% : 0.000104s : 314: predicate.loop_unroll_before_grad 1.47% : 0.000026s : 157: predicate.make_slice_get_slice_eliminator 0.48% : 0.000008s : 58: predicate.merge_addn 0.30% : 0.000005s : 34: predicate.micro_step_allgather_replace 0.89% : 0.000016s : 34: predicate.mini_step_allgather_replace 1.16% : 0.000020s : 123: predicate.minmaximum_grad 0.38% : 0.000007s : 17: predicate.mutable_eliminate 0.18% : 0.000003s : 17: predicate.opt_reshape 0.17% : 0.000003s : 17: predicate.parallel_virtual_node 3.62% : 0.000064s : 209: predicate.partial_defer_inline 1.41% : 0.000025s : 155: predicate.partial_eliminate 1.05% : 0.000019s : 123: predicate.print_const_string_wrapper 0.47% : 0.000008s : 53: predicate.reduce_all_const_elim 1.54% : 0.000027s : 123: predicate.reduce_eliminate 2.36% : 0.000042s : 295: predicate.redundant_stop_gradient_eliminater 0.16% : 0.000003s : 35: predicate.remove_not_recompute_node 1.01% : 0.000018s : 172: predicate.replace_applicator 0.19% : 0.000003s : 35: predicate.replace_old_param 0.11% : 0.000002s : 17: predicate.reset_defer_inline 1.20% : 0.000021s : 123: predicate.reshape_eliminate 0.33% : 0.000006s : 34: predicate.row_tensor_add_zeros_like 0.18% : 0.000003s : 17: predicate.row_tensor_eliminate 0.59% : 0.000010s : 34: predicate.same_eliminate 0.25% : 0.000004s : 40: predicate.set_cell_output_no_recompute 0.38% : 0.000007s : 35: predicate.shard_identity_eliminate 0.33% : 0.000006s : 34: predicate.special_op_eliminate 0.58% : 0.000010s : 58: predicate.specialize_transform 0.45% : 0.000008s : 34: predicate.split_environ_get_set_with_tuple_value 0.33% : 0.000006s : 35: predicate.stack_unstack_eliminate 0.16% : 0.000003s : 17: predicate.switch_call_monad_eliminater 2.01% : 0.000036s : 209: predicate.switch_defer_inline 2.18% : 0.000039s : 243: predicate.switch_layer_defer_inline 8.34% : 0.000148s : 646: predicate.switch_simplify 1.17% : 0.000021s : 123: predicate.tile_eliminate 1.11% : 0.000020s : 123: predicate.transpose_eliminate 1.65% : 0.000029s : 157: predicate.tuple_list_convert_item_index_to_positive 1.79% : 0.000032s : 157: predicate.tuple_list_get_item_const_eliminator 1.51% : 0.000027s : 157: predicate.tuple_list_get_item_depend_reorder 2.95% : 0.000052s : 228: predicate.tuple_list_get_item_eliminator 1.66% : 0.000029s : 157: predicate.tuple_list_get_set_item_eliminator 2.03% : 0.000036s : 215: predicate.tuple_list_set_item_eliminator 1.49% : 0.000026s : 170: predicate.tuple_to_list_eliminator_ 2.41% : 0.000043s : 295: predicate.updatestate_pure_node_eliminater 2.93% : 0.000052s : 353: predicate.updatestate_useless_node_eliminater 0.16% : 0.000003s : 17: predicate.value_based_eliminate 0.35% : 0.000006s : 35: predicate.virtual_dataset_eliminate 0.32% : 0.000006s : 35: predicate.virtual_output_eliminate 0.16% : 0.000003s : 17: predicate.virtual_view_grad_eliminate 0.17% : 0.000003s : 17: predicate.zero_like_fill_zero ------[func_graph_cloner_run.] 0.018667 135 53.82% : 0.010046s : 55: func_graph_cloner_run.FuncGraphClonerGraph 46.18% : 0.008621s : 80: func_graph_cloner_run.FuncGraphSpecializer ------[meta_graph.] 0.000000 0 ------[manager.] 0.000000 0 ------[pynative] 0.000000 0 ------[others.] 1.040449 209 0.00% : 0.000003s : 1: ForceFp32Comm 1.10% : 0.011453s : 1: add_attr 1.10% : 0.011433s : 1: add_attr_with_inline 0.00% : 0.000004s : 1: add_comm_op_reuse_tag 0.02% : 0.000159s : 1: add_recomputation 0.00% : 0.000005s : 1: assign_add_opt 0.07% : 0.000780s : 1: auto_monad 0.01% : 0.000066s : 1: auto_monad_reorder 0.00% : 0.000004s : 1: begin_end_overlap_inline 0.00% : 0.000007s : 1: bias_add_comm_swap 0.07% : 0.000731s : 1: bootstrap 0.00% : 0.000046s : 1: cconv 0.00% : 0.000004s : 1: comm_op_add_attrs 0.00% : 0.000043s : 1: control_data_broadcast_order 0.00% : 0.000032s : 1: convert_after_rewriter 0.01% : 0.000075s : 1: cse_after_recomputation 0.00% : 0.000006s : 1: dataset_repeat_opt 0.00% : 0.000006s : 1: detach_backward 0.00% : 0.000026s : 1: environ_conv 0.02% : 0.000239s : 1: event_method 0.00% : 0.000006s : 1: full_micro_interleaved_order_control 0.00% : 0.000006s : 1: get_jit_bprop_graph 0.01% : 0.000058s : 1: graph_reusing 0.00% : 0.000004s : 1: grouped_pairwise_exchange_alltoall 0.00% : 0.000005s : 1: handle_group_info 0.00% : 0.000010s : 1: inline 0.00% : 0.000007s : 1: insert-virtual-dataset 0.00% : 0.000004s : 1: interleave_parallel_branches 0.00% : 0.000004s : 1: interleave_split_concat_branches 0.00% : 0.000006s : 1: label_fine_grained_interleaved_index 0.00% : 0.000007s : 1: label_micro_interleaved_index 0.06% : 0.000582s : 1: loop_unroll 0.00% : 0.000005s : 1: merge_cast_opt 0.00% : 0.000005s : 1: micro_interleaved_order_control 0.08% : 0.000861s : 1: mutable_eliminate 0.00% : 0.000014s : 1: offloading_packed_experts 0.00% : 0.000036s : 1: opt.transform.loop_unroll_optimizer 0.00% : 0.000046s : 1: opt.transform.mutable_eliminate 5.39% : 0.056099s : 95: opt.transform.opt_a 0.01% : 0.000122s : 1: opt.transform.opt_after_cconv 0.01% : 0.000076s : 1: opt.transform.opt_after_jit_grad 0.05% : 0.000538s : 28: opt.transform.opt_b 0.02% : 0.000217s : 2: opt.transform.opt_trans_graph 0.02% : 0.000185s : 4: opt.transform.symbol_engine_opt 9.56% : 0.099430s : 1: opt_a 0.03% : 0.000297s : 1: opt_after_cconv 0.06% : 0.000660s : 1: opt_after_jit_grad 0.08% : 0.000801s : 1: opt_b 10.16% : 0.105702s : 1: optimize 0.00% : 0.000046s : 1: optimize_parallel_all_gather_comm 0.00% : 0.000018s : 1: order_py_execute_after_rewriter 0.01% : 0.000052s : 1: overlap_grad_flash_sp 0.00% : 0.000004s : 1: overlap_grad_matmul_and_grad_allreduce 0.00% : 0.000013s : 1: overlap_grad_ring_attention 0.00% : 0.000006s : 1: overlap_opt_shard_grad_in_pipeline 0.00% : 0.000009s : 1: overlap_opt_shard_in_pipeline 0.00% : 0.000009s : 1: overlap_param_gather 0.00% : 0.000004s : 1: overlap_recompute_allgather_and_fa_grad 0.00% : 0.000013s : 1: overlap_recompute_and_grad_model_parallel 0.00% : 0.000007s : 1: overlap_recompute_comm 0.00% : 0.000010s : 1: parallel-infer-symbol 0.00% : 0.000005s : 1: parallel-infer-symbol-second 0.00% : 0.000006s : 1: partial_unused_args_eliminate 0.00% : 0.000005s : 1: pipeline_parallel_scheduler 0.00% : 0.000005s : 1: pipeline_split 0.03% : 0.000265s : 1: pre_auto_parallel 0.02% : 0.000240s : 1: py_interpret_to_execute 0.00% : 0.000042s : 1: py_interpret_to_execute_after_opt_a 0.00% : 0.000004s : 1: remove_cast_before_assign_add 0.01% : 0.000097s : 1: remove_dup_value 2.92% : 0.030368s : 1: renormalize.infer 0.80% : 0.008284s : 1: renormalize.specialize 0.00% : 0.000007s : 1: reorder_send_recv_between_fp_bp 0.00% : 0.000008s : 1: rewriter_after_jit_bprop_graph 0.09% : 0.000895s : 1: rewriter_after_opt_a 0.07% : 0.000777s : 1: rewriter_before_opt_a 0.00% : 0.000006s : 1: slice_cell_reuse_recomputed_activation 0.00% : 0.000005s : 1: slice_recompute_activation 0.00% : 0.000004s : 1: split_layernorm_comm 0.00% : 0.000005s : 1: split_matmul_comm_elemetwise 0.00% : 0.000021s : 1: swap_dp_allreduce_reducescatter 0.06% : 0.000632s : 1: symbol_engine_optimizer 0.02% : 0.000257s : 1: tuple_transform 67.99% : 0.707406s : 1: type_inference . [hook] pytest_runtest_teardown:test_matmul_qkv_bf16_0[True-mstype0-k_n_shape2-16] tests/st/infer/ops/test_internal_ops/test_matmul_split.py::test_matmul_qkv_bf16_0[True-mstype0-k_n_shape2-16],max_mem:196.0M [WARNING] ME(152078:281472851701552,MainProcess):2026-01-29-17:44:25.615.746 [mindspore/context.py:1334] For 'context.set_context', the parameter 'device_target' will be deprecated and removed in a future version. Please use the api mindspore.set_device() instead. TotalTime = 0.71471, [21] [bootstrap]: 0.0209602 [type_inference]: 0.5998 [event_method]: 0.00021009 [auto_monad]: 0.00069179 [graph_reusing]: 4.734e-05 [inline]: 3.22002e-06 [add_attr]: 0.00503903, [1] [add_attr_with_inline]: 0.00502128, [1] [Cycle 1]: 0.00033641, [2] [tag_attr]: 0.00021697 [meta_addattr_fg_expand]: 6.067e-05 [parallel-infer-symbol]: 3.90998e-06 [pre_auto_parallel]: 0.00027367 [insert-virtual-dataset]: 2.94999e-06 [parallel-infer-symbol-second]: 1.22e-06 [dataset_repeat_opt]: 2.34001e-06 [pipeline_split]: 1.81e-06 [optimize]: 0.0863707, [53] [py_interpret_to_execute]: 0.00026823 [rewriter_before_opt_a]: 0.0209555 [opt_a]: 0.0299653, [2] [Cycle 1]: 0.027811, [45] [expand_dump_flag]: 3.699e-05 [switch_simplify]: 0.0009267 [loop_unroll]: 0.00032533 [a_1]: 0.00940092 [with_stream_mark]: 5.797e-05 [recompute_prepare]: 3.861e-05 [updatestate_depend_eliminate]: 2.032e-05 [updatestate_assign_eliminate]: 1.714e-05 [updatestate_loads_eliminate]: 1.898e-05 [parameter_eliminate]: 3.7e-06 [a_2]: 0.00083888 [accelerated_algorithm]: 6.222e-05 [shard]: 2.76e-06 [meta_shard_fg_expand]: 2.162e-05 [shard_inline]: 2.225e-05 [merge_send_recv]: 2.07e-05 [auto_parallel]: 2.04e-05 [parallel]: 3.099e-05 [flash_sp]: 1.453e-05 [merge_comm]: 1.411e-05 [allreduce_fusion]: 1.26e-05 [matmul_add_comm_reduction]: 2.124e-05 [allreduce_slice_to_reducescatter]: 1.07e-06 [virtual_shard_identity]: 2.371e-05 [virtual_dataset]: 2.293e-05 [get_grad_eliminate_]: 2.079e-05 [virtual_output]: 2.238e-05 [merge_forward]: 1.303e-05 [cell_reuse_recompute_pass]: 2.26e-06 [offload_activation]: 2.298e-05 [cell_reuse_handle_not_recompute_node_pass]: 4.536e-05 [merge_recompute_call_nodes]: 1.55999e-06 [before_grad]: 3.862e-05 [set_forward_comm_id_for_comm_node_pass]: 1.464e-05 [meta_fg_expand]: 1.595e-05 [flash_sp_send_recv_attached]: 5.26002e-06 [receive_attached]: 2.11998e-06 [after_resolve]: 3.016e-05 [a_after_grad]: 3.612e-05 [renormalize]: 0.013801 [add_forward_monad_depend]: 1.924e-05 [auto_monad_grad]: 3.41001e-06 [auto_monad_eliminator]: 8.582e-05 [cse]: 0.00106782 [a_3]: 0.00017422 [Cycle 2]: 0.00213208, [45] [expand_dump_flag]: 2.99999e-06 [switch_simplify]: 2.777e-05 [loop_unroll]: 2.193e-05 [a_1]: 0.00068323 [with_stream_mark]: 3.499e-05 [recompute_prepare]: 2.408e-05 [updatestate_depend_eliminate]: 1.558e-05 [updatestate_assign_eliminate]: 1.219e-05 [updatestate_loads_eliminate]: 1.399e-05 [parameter_eliminate]: 2.74999e-06 [a_2]: 0.00031565 [accelerated_algorithm]: 3.095e-05 [shard]: 3.07002e-06 [meta_shard_fg_expand]: 7.56999e-06 [shard_inline]: 2.104e-05 [merge_send_recv]: 1.977e-05 [auto_parallel]: 1.968e-05 [parallel]: 1.017e-05 [flash_sp]: 5.07999e-06 [merge_comm]: 1.334e-05 [allreduce_fusion]: 1.212e-05 [matmul_add_comm_reduction]: 2.363e-05 [allreduce_slice_to_reducescatter]: 1.22999e-06 [virtual_shard_identity]: 2.396e-05 [virtual_dataset]: 1.99e-05 [get_grad_eliminate_]: 2.092e-05 [virtual_output]: 2.016e-05 [merge_forward]: 1.221e-05 [cell_reuse_recompute_pass]: 3.2e-06 [offload_activation]: 2.55e-05 [cell_reuse_handle_not_recompute_node_pass]: 4.06e-05 [merge_recompute_call_nodes]: 2.09999e-06 [before_grad]: 3.533e-05 [set_forward_comm_id_for_comm_node_pass]: 1.366e-05 [meta_fg_expand]: 9.49999e-06 [flash_sp_send_recv_attached]: 2.27999e-06 [receive_attached]: 2.41e-06 [after_resolve]: 2.726e-05 [a_after_grad]: 3.204e-05 [renormalize]: 8.00064e-08 [add_forward_monad_depend]: 2.44999e-06 [auto_monad_grad]: 2.37001e-06 [auto_monad_eliminator]: 5.705e-05 [cse]: 8.091e-05 [a_3]: 0.00013483 [py_interpret_to_execute_after_opt_a]: 3.297e-05 [slice_cell_reuse_recomputed_activation]: 2.81999e-06 [rewriter_after_opt_a]: 0.00087488 [convert_after_rewriter]: 2.346e-05 [order_py_execute_after_rewriter]: 1.44e-05 [mutable_eliminate]: 0.00079115 [opt_b]: 0.0302882, [1] [Cycle 1]: 0.0302774, [7] [b_1]: 0.0299851 [b_2]: 3.194e-05 [updatestate_depend_eliminate]: 2.892e-05 [updatestate_assign_eliminate]: 1.297e-05 [updatestate_loads_eliminate]: 1.586e-05 [renormalize]: 8.30012e-07 [cse]: 0.00011997 [optimize_parallel_all_gather_comm]: 4.773e-05 [overlap_param_gather]: 2.19001e-06 [cconv]: 4.548e-05 [loop_unroll]: 0.00093374 [opt_after_cconv]: 0.00032918, [1] [Cycle 1]: 0.00031631, [7] [c_1]: 0.00013746 [parameter_eliminate]: 6.29999e-06 [updatestate_depend_eliminate]: 2.154e-05 [updatestate_assign_eliminate]: 1.13e-05 [updatestate_loads_eliminate]: 1.367e-05 [cse]: 8.424e-05 [renormalize]: 9.10019e-07 [remove_dup_value]: 0.00010542 [tuple_transform]: 0.0003304, [1] [Cycle 1]: 0.0003227, [4] [d_1]: 0.00026975 [none_parameter_eliminate]: 3.47002e-06 [renormalize]: 2.00002e-07 [switch_simplify]: 2.437e-05 [partial_unused_args_eliminate]: 2.32999e-06 [add_recomputation]: 0.00015864 [cse_after_recomputation]: 8.224e-05, [1] [Cycle 1]: 7.392e-05, [1] [cse]: 6.415e-05 [environ_conv]: 2.612e-05 [swap_dp_allreduce_reducescatter]: 2.049e-05 [bias_add_comm_swap]: 3.63e-06 [label_micro_interleaved_index]: 6.23002e-06 [label_fine_grained_interleaved_index]: 2.68e-06 [merge_cast_opt]: 1.65001e-06 [slice_recompute_activation]: 2.29999e-06 [micro_interleaved_order_control]: 2.63998e-06 [assign_add_opt]: 1.74998e-06 [ForceFp32Comm]: 1.14e-06 [remove_cast_before_assign_add]: 1.05001e-06 [full_micro_interleaved_order_control]: 2.18002e-06 [reorder_send_recv_between_fp_bp]: 2.63e-06 [comm_op_add_attrs]: 1.30999e-06 [add_comm_op_reuse_tag]: 1.05001e-06 [interleave_split_concat_branches]: 1.19998e-06 [interleave_parallel_branches]: 1.02998e-06 [overlap_opt_shard_in_pipeline]: 1.27e-06 [overlap_opt_shard_grad_in_pipeline]: 2.19001e-06 [control_data_broadcast_order]: 3.792e-05 [grouped_pairwise_exchange_alltoall]: 1.77001e-06 [offloading_packed_experts]: 1.094e-05 [overlap_recompute_and_grad_model_parallel]: 1.113e-05 [overlap_grad_matmul_and_grad_allreduce]: 1.48002e-06 [overlap_recompute_allgather_and_fa_grad]: 1.44998e-06 [overlap_recompute_comm]: 2.16e-06 [overlap_grad_ring_attention]: 9.28002e-06 [overlap_grad_flash_sp]: 4.952e-05 [begin_end_overlap_inline]: 5.00004e-07 [split_matmul_comm_elemetwise]: 2.58998e-06 [split_layernorm_comm]: 1.73002e-06 [handle_group_info]: 1.02e-06 [symbol_engine_optimizer]: 0.00056218, [1] [Cycle 1]: 0.00055515, [6] [build]: 0.00029565 [elim_shapecalc]: 5.861e-05 [elim_not_effective]: 5.472e-05 [opt_reshape]: 3.604e-05 [fold_const_symbol]: 6.006e-05 [renormalize]: 2.50002e-07 [detach_backward]: 2.64999e-06 [pipeline_parallel_scheduler]: 1.50001e-06 [auto_monad_reorder]: 6.935e-05 [get_jit_bprop_graph]: 2.29999e-06 [rewriter_after_jit_bprop_graph]: 6.94999e-06 [opt_after_jit_grad]: 0.00085288 [validate]: 0.00011534 Sums bootstrap : 0.020960s : 2.96% type_inference : 0.599800s : 84.67% event_method : 0.000210s : 0.03% auto_monad : 0.000692s : 0.10% graph_reusing : 0.000047s : 0.01% inline : 0.000003s : 0.00% add_attr.add_attr_with_inline.tag_attr : 0.000217s : 0.03% add_attr.add_attr_with_inline.meta_addattr_fg_expand : 0.000061s : 0.01% parallel-infer-symbol : 0.000004s : 0.00% pre_auto_parallel : 0.000274s : 0.04% insert-virtual-dataset : 0.000003s : 0.00% parallel-infer-symbol-second : 0.000001s : 0.00% dataset_repeat_opt : 0.000002s : 0.00% pipeline_split : 0.000002s : 0.00% optimize.py_interpret_to_execute : 0.000268s : 0.04% optimize.rewriter_before_opt_a : 0.020956s : 2.96% optimize.opt_a.expand_dump_flag : 0.000040s : 0.01% optimize.opt_a.switch_simplify : 0.000954s : 0.13% optimize.opt_a.loop_unroll : 0.000347s : 0.05% optimize.opt_a.a_1 : 0.010084s : 1.42% optimize.opt_a.with_stream_mark : 0.000093s : 0.01% optimize.opt_a.recompute_prepare : 0.000063s : 0.01% optimize.opt_a.updatestate_depend_eliminate : 0.000036s : 0.01% optimize.opt_a.updatestate_assign_eliminate : 0.000029s : 0.00% optimize.opt_a.updatestate_loads_eliminate : 0.000033s : 0.00% optimize.opt_a.parameter_eliminate : 0.000006s : 0.00% optimize.opt_a.a_2 : 0.001155s : 0.16% optimize.opt_a.accelerated_algorithm : 0.000093s : 0.01% optimize.opt_a.shard : 0.000006s : 0.00% optimize.opt_a.meta_shard_fg_expand : 0.000029s : 0.00% optimize.opt_a.shard_inline : 0.000043s : 0.01% optimize.opt_a.merge_send_recv : 0.000040s : 0.01% optimize.opt_a.auto_parallel : 0.000040s : 0.01% optimize.opt_a.parallel : 0.000041s : 0.01% optimize.opt_a.flash_sp : 0.000020s : 0.00% optimize.opt_a.merge_comm : 0.000027s : 0.00% optimize.opt_a.allreduce_fusion : 0.000025s : 0.00% optimize.opt_a.matmul_add_comm_reduction : 0.000045s : 0.01% optimize.opt_a.allreduce_slice_to_reducescatter : 0.000002s : 0.00% optimize.opt_a.virtual_shard_identity : 0.000048s : 0.01% optimize.opt_a.virtual_dataset : 0.000043s : 0.01% optimize.opt_a.get_grad_eliminate_ : 0.000042s : 0.01% optimize.opt_a.virtual_output : 0.000043s : 0.01% optimize.opt_a.merge_forward : 0.000025s : 0.00% optimize.opt_a.cell_reuse_recompute_pass : 0.000005s : 0.00% optimize.opt_a.offload_activation : 0.000048s : 0.01% optimize.opt_a.cell_reuse_handle_not_recompute_node_pass : 0.000086s : 0.01% optimize.opt_a.merge_recompute_call_nodes : 0.000004s : 0.00% optimize.opt_a.before_grad : 0.000074s : 0.01% optimize.opt_a.set_forward_comm_id_for_comm_node_pass : 0.000028s : 0.00% optimize.opt_a.meta_fg_expand : 0.000025s : 0.00% optimize.opt_a.flash_sp_send_recv_attached : 0.000008s : 0.00% optimize.opt_a.receive_attached : 0.000005s : 0.00% optimize.opt_a.after_resolve : 0.000057s : 0.01% optimize.opt_a.a_after_grad : 0.000068s : 0.01% optimize.opt_a.renormalize : 0.013801s : 1.95% optimize.opt_a.add_forward_monad_depend : 0.000022s : 0.00% optimize.opt_a.auto_monad_grad : 0.000006s : 0.00% optimize.opt_a.auto_monad_eliminator : 0.000143s : 0.02% optimize.opt_a.cse : 0.001149s : 0.16% optimize.opt_a.a_3 : 0.000309s : 0.04% optimize.py_interpret_to_execute_after_opt_a : 0.000033s : 0.00% optimize.slice_cell_reuse_recomputed_activation : 0.000003s : 0.00% optimize.rewriter_after_opt_a : 0.000875s : 0.12% optimize.convert_after_rewriter : 0.000023s : 0.00% optimize.order_py_execute_after_rewriter : 0.000014s : 0.00% optimize.mutable_eliminate : 0.000791s : 0.11% optimize.opt_b.b_1 : 0.029985s : 4.23% optimize.opt_b.b_2 : 0.000032s : 0.00% optimize.opt_b.updatestate_depend_eliminate : 0.000029s : 0.00% optimize.opt_b.updatestate_assign_eliminate : 0.000013s : 0.00% optimize.opt_b.updatestate_loads_eliminate : 0.000016s : 0.00% optimize.opt_b.renormalize : 0.000001s : 0.00% optimize.opt_b.cse : 0.000120s : 0.02% optimize.optimize_parallel_all_gather_comm : 0.000048s : 0.01% optimize.overlap_param_gather : 0.000002s : 0.00% optimize.cconv : 0.000045s : 0.01% optimize.loop_unroll : 0.000934s : 0.13% optimize.opt_after_cconv.c_1 : 0.000137s : 0.02% optimize.opt_after_cconv.parameter_eliminate : 0.000006s : 0.00% optimize.opt_after_cconv.updatestate_depend_eliminate : 0.000022s : 0.00% optimize.opt_after_cconv.updatestate_assign_eliminate : 0.000011s : 0.00% optimize.opt_after_cconv.updatestate_loads_eliminate : 0.000014s : 0.00% optimize.opt_after_cconv.cse : 0.000084s : 0.01% optimize.opt_after_cconv.renormalize : 0.000001s : 0.00% optimize.remove_dup_value : 0.000105s : 0.01% optimize.tuple_transform.d_1 : 0.000270s : 0.04% optimize.tuple_transform.none_parameter_eliminate : 0.000003s : 0.00% optimize.tuple_transform.renormalize : 0.000000s : 0.00% optimize.tuple_transform.switch_simplify : 0.000024s : 0.00% optimize.partial_unused_args_eliminate : 0.000002s : 0.00% optimize.add_recomputation : 0.000159s : 0.02% optimize.cse_after_recomputation.cse : 0.000064s : 0.01% optimize.environ_conv : 0.000026s : 0.00% optimize.swap_dp_allreduce_reducescatter : 0.000020s : 0.00% optimize.bias_add_comm_swap : 0.000004s : 0.00% optimize.label_micro_interleaved_index : 0.000006s : 0.00% optimize.label_fine_grained_interleaved_index : 0.000003s : 0.00% optimize.merge_cast_opt : 0.000002s : 0.00% optimize.slice_recompute_activation : 0.000002s : 0.00% optimize.micro_interleaved_order_control : 0.000003s : 0.00% optimize.assign_add_opt : 0.000002s : 0.00% optimize.ForceFp32Comm : 0.000001s : 0.00% optimize.remove_cast_before_assign_add : 0.000001s : 0.00% optimize.full_micro_interleaved_order_control : 0.000002s : 0.00% optimize.reorder_send_recv_between_fp_bp : 0.000003s : 0.00% optimize.comm_op_add_attrs : 0.000001s : 0.00% optimize.add_comm_op_reuse_tag : 0.000001s : 0.00% optimize.interleave_split_concat_branches : 0.000001s : 0.00% optimize.interleave_parallel_branches : 0.000001s : 0.00% optimize.overlap_opt_shard_in_pipeline : 0.000001s : 0.00% optimize.overlap_opt_shard_grad_in_pipeline : 0.000002s : 0.00% optimize.control_data_broadcast_order : 0.000038s : 0.01% optimize.grouped_pairwise_exchange_alltoall : 0.000002s : 0.00% optimize.offloading_packed_experts : 0.000011s : 0.00% optimize.overlap_recompute_and_grad_model_parallel : 0.000011s : 0.00% optimize.overlap_grad_matmul_and_grad_allreduce : 0.000001s : 0.00% optimize.overlap_recompute_allgather_and_fa_grad : 0.000001s : 0.00% optimize.overlap_recompute_comm : 0.000002s : 0.00% optimize.overlap_grad_ring_attention : 0.000009s : 0.00% optimize.overlap_grad_flash_sp : 0.000050s : 0.01% optimize.begin_end_overlap_inline : 0.000001s : 0.00% optimize.split_matmul_comm_elemetwise : 0.000003s : 0.00% optimize.split_layernorm_comm : 0.000002s : 0.00% optimize.handle_group_info : 0.000001s : 0.00% optimize.symbol_engine_optimizer.build : 0.000296s : 0.04% optimize.symbol_engine_optimizer.elim_shapecalc : 0.000059s : 0.01% optimize.symbol_engine_optimizer.elim_not_effective : 0.000055s : 0.01% optimize.symbol_engine_optimizer.opt_reshape : 0.000036s : 0.01% optimize.symbol_engine_optimizer.fold_const_symbol : 0.000060s : 0.01% optimize.symbol_engine_optimizer.renormalize : 0.000000s : 0.00% detach_backward : 0.000003s : 0.00% pipeline_parallel_scheduler : 0.000002s : 0.00% auto_monad_reorder : 0.000069s : 0.01% get_jit_bprop_graph : 0.000002s : 0.00% rewriter_after_jit_bprop_graph : 0.000007s : 0.00% opt_after_jit_grad : 0.000853s : 0.12% validate : 0.000115s : 0.02% Time group info: ------[substitution.] 0.003815 569 0.46% : 0.000017s : 8: substitution.depend_value_elim 0.37% : 0.000014s : 14: substitution.elim_not_effective 0.52% : 0.000020s : 12: substitution.float_tuple_getitem_switch 0.73% : 0.000028s : 14: substitution.fold_const_symbol 0.41% : 0.000016s : 17: substitution.graph_param_transform 71.45% : 0.002726s : 71: substitution.inline 0.35% : 0.000013s : 29: substitution.j_node_and_user_rematch 0.92% : 0.000035s : 6: substitution.less_batch_normalization 1.38% : 0.000053s : 2: substitution.list_to_tuple_eliminator_ 0.19% : 0.000007s : 6: substitution.load_eliminater 0.80% : 0.000030s : 21: substitution.minmaximum_grad 0.26% : 0.000010s : 2: substitution.opt_reshape 0.50% : 0.000019s : 29: substitution.remove_not_recompute_node 0.21% : 0.000008s : 4: substitution.replace_old_param 1.74% : 0.000066s : 18: substitution.reshape_eliminate 1.51% : 0.000058s : 24: substitution.switch_simplify 3.96% : 0.000151s : 51: substitution.tuple_list_convert_item_index_to_positive 2.08% : 0.000079s : 51: substitution.tuple_list_get_item_const_eliminator 2.76% : 0.000105s : 51: substitution.tuple_list_get_item_depend_reorder 5.42% : 0.000207s : 73: substitution.tuple_list_get_item_eliminator 2.64% : 0.000101s : 51: substitution.tuple_list_get_set_item_eliminator 0.74% : 0.000028s : 6: substitution.updatestate_pure_node_eliminater 0.61% : 0.000023s : 9: substitution.updatestate_useless_node_eliminater ------[type_inference.] 0.599415 2 91.96% : 0.551212s : 1: type_inference.infer 8.04% : 0.048203s : 1: type_inference.specialize ------[replace.] 0.001134 115 3.57% : 0.000040s : 5: replace.depend_value_elim 54.50% : 0.000618s : 71: replace.inline 3.59% : 0.000041s : 2: replace.list_to_tuple_eliminator_ 29.48% : 0.000334s : 24: replace.switch_simplify 8.87% : 0.000101s : 13: replace.tuple_list_get_item_eliminator ------[match.] 0.002811 115 0.09% : 0.000003s : 5: match.depend_value_elim 95.44% : 0.002683s : 71: match.inline 1.79% : 0.000050s : 2: match.list_to_tuple_eliminator_ 1.54% : 0.000043s : 24: match.switch_simplify 1.13% : 0.000032s : 13: match.tuple_list_get_item_eliminator ------[predicate.] 0.001688 9701 1.09% : 0.000018s : 123: predicate.accumulaten_eliminater 0.34% : 0.000006s : 17: predicate.ad_related_special_op_eliminate 0.51% : 0.000009s : 58: predicate.addn_check_dump 1.15% : 0.000019s : 123: predicate.addn_zero_filter 1.01% : 0.000017s : 123: predicate.adjust_all_reduce_mul_add 2.31% : 0.000039s : 181: predicate.arithmetic_simplify 1.11% : 0.000019s : 123: predicate.cast_eliminate 0.30% : 0.000005s : 34: predicate.check_bprop_eliminate 0.49% : 0.000008s : 58: predicate.compare_switch_simplify 0.09% : 0.000002s : 17: predicate.const_output_eliminate 0.62% : 0.000011s : 61: predicate.depend_value_elim 1.13% : 0.000019s : 123: predicate.dict_get_item_const_eliminator 1.23% : 0.000021s : 123: predicate.dict_get_item_eliminator 1.07% : 0.000018s : 123: predicate.dict_set_item_eliminator 0.40% : 0.000007s : 34: predicate.dumpgradient_eliminate 0.09% : 0.000002s : 17: predicate.elim_not_effective 0.20% : 0.000003s : 17: predicate.elim_shapecalc_of_broadcastargs 1.27% : 0.000021s : 140: predicate.environ_add_const_eliminate 1.20% : 0.000020s : 140: predicate.environ_get_add_eliminate 1.31% : 0.000022s : 140: predicate.environ_get_depend_swap 1.71% : 0.000029s : 198: predicate.environ_get_eliminate 1.17% : 0.000020s : 140: predicate.environ_get_set_eliminate 1.90% : 0.000032s : 209: predicate.exchange_switch_depend_value 2.50% : 0.000042s : 209: predicate.float_depend_g_call 0.47% : 0.000008s : 58: predicate.float_environ_get_switch 0.72% : 0.000012s : 75: predicate.float_tuple_getitem_switch 0.08% : 0.000001s : 17: predicate.fold_const_symbol 0.33% : 0.000006s : 35: predicate.get_grad_eliminate 0.09% : 0.000002s : 17: predicate.graph_param_transform 0.49% : 0.000008s : 58: predicate.incorporate_call 0.44% : 0.000007s : 58: predicate.incorporate_call_switch 5.56% : 0.000094s : 459: predicate.inline 0.41% : 0.000007s : 35: predicate.inline_without_move 0.14% : 0.000002s : 35: predicate.j_node_and_user_rematch 0.48% : 0.000008s : 36: predicate.less_batch_normalization 1.66% : 0.000028s : 172: predicate.list_to_tuple_eliminator_ 2.57% : 0.000043s : 295: predicate.load_eliminater 0.50% : 0.000008s : 17: predicate.loop_unroll_after_grad 3.32% : 0.000056s : 314: predicate.loop_unroll_before_grad 1.61% : 0.000027s : 157: predicate.make_slice_get_slice_eliminator 0.51% : 0.000009s : 58: predicate.merge_addn 0.28% : 0.000005s : 34: predicate.micro_step_allgather_replace 0.29% : 0.000005s : 34: predicate.mini_step_allgather_replace 1.11% : 0.000019s : 123: predicate.minmaximum_grad 0.31% : 0.000005s : 17: predicate.mutable_eliminate 0.23% : 0.000004s : 17: predicate.opt_reshape 0.18% : 0.000003s : 17: predicate.parallel_virtual_node 3.16% : 0.000053s : 209: predicate.partial_defer_inline 1.46% : 0.000025s : 155: predicate.partial_eliminate 1.11% : 0.000019s : 123: predicate.print_const_string_wrapper 0.49% : 0.000008s : 53: predicate.reduce_all_const_elim 1.50% : 0.000025s : 123: predicate.reduce_eliminate 2.58% : 0.000044s : 295: predicate.redundant_stop_gradient_eliminater 0.15% : 0.000003s : 35: predicate.remove_not_recompute_node 1.06% : 0.000018s : 172: predicate.replace_applicator 0.16% : 0.000003s : 35: predicate.replace_old_param 0.11% : 0.000002s : 17: predicate.reset_defer_inline 1.27% : 0.000021s : 123: predicate.reshape_eliminate 0.28% : 0.000005s : 34: predicate.row_tensor_add_zeros_like 0.20% : 0.000003s : 17: predicate.row_tensor_eliminate 0.42% : 0.000007s : 34: predicate.same_eliminate 0.21% : 0.000003s : 40: predicate.set_cell_output_no_recompute 0.34% : 0.000006s : 35: predicate.shard_identity_eliminate 0.33% : 0.000006s : 34: predicate.special_op_eliminate 0.60% : 0.000010s : 58: predicate.specialize_transform 0.38% : 0.000006s : 34: predicate.split_environ_get_set_with_tuple_value 0.35% : 0.000006s : 35: predicate.stack_unstack_eliminate 0.16% : 0.000003s : 17: predicate.switch_call_monad_eliminater 2.04% : 0.000034s : 209: predicate.switch_defer_inline 2.26% : 0.000038s : 243: predicate.switch_layer_defer_inline 8.62% : 0.000146s : 646: predicate.switch_simplify 1.18% : 0.000020s : 123: predicate.tile_eliminate 1.09% : 0.000018s : 123: predicate.transpose_eliminate 1.67% : 0.000028s : 157: predicate.tuple_list_convert_item_index_to_positive 1.75% : 0.000029s : 157: predicate.tuple_list_get_item_const_eliminator 1.60% : 0.000027s : 157: predicate.tuple_list_get_item_depend_reorder 2.95% : 0.000050s : 228: predicate.tuple_list_get_item_eliminator 1.64% : 0.000028s : 157: predicate.tuple_list_get_set_item_eliminator 2.22% : 0.000037s : 215: predicate.tuple_list_set_item_eliminator 2.34% : 0.000040s : 170: predicate.tuple_to_list_eliminator_ 2.57% : 0.000043s : 295: predicate.updatestate_pure_node_eliminater 4.59% : 0.000078s : 353: predicate.updatestate_useless_node_eliminater 0.18% : 0.000003s : 17: predicate.value_based_eliminate 0.32% : 0.000005s : 35: predicate.virtual_dataset_eliminate 0.33% : 0.000006s : 35: predicate.virtual_output_eliminate 0.16% : 0.000003s : 17: predicate.virtual_view_grad_eliminate 0.18% : 0.000003s : 17: predicate.zero_like_fill_zero ------[func_graph_cloner_run.] 0.018204 135 55.62% : 0.010126s : 55: func_graph_cloner_run.FuncGraphClonerGraph 44.38% : 0.008078s : 80: func_graph_cloner_run.FuncGraphSpecializer ------[meta_graph.] 0.000000 0 ------[manager.] 0.000000 0 ------[pynative] 0.000000 0 ------[others.] 0.863702 209 0.00% : 0.000004s : 1: ForceFp32Comm 0.58% : 0.005047s : 1: add_attr 0.58% : 0.005028s : 1: add_attr_with_inline 0.00% : 0.000005s : 1: add_comm_op_reuse_tag 0.02% : 0.000165s : 1: add_recomputation 0.00% : 0.000006s : 1: assign_add_opt 0.08% : 0.000706s : 1: auto_monad 0.01% : 0.000075s : 1: auto_monad_reorder 0.00% : 0.000005s : 1: begin_end_overlap_inline 0.00% : 0.000007s : 1: bias_add_comm_swap 2.43% : 0.021010s : 1: bootstrap 0.01% : 0.000050s : 1: cconv 0.00% : 0.000004s : 1: comm_op_add_attrs 0.00% : 0.000042s : 1: control_data_broadcast_order 0.00% : 0.000028s : 1: convert_after_rewriter 0.01% : 0.000086s : 1: cse_after_recomputation 0.00% : 0.000006s : 1: dataset_repeat_opt 0.00% : 0.000006s : 1: detach_backward 0.00% : 0.000030s : 1: environ_conv 0.03% : 0.000223s : 1: event_method 0.00% : 0.000006s : 1: full_micro_interleaved_order_control 0.00% : 0.000007s : 1: get_jit_bprop_graph 0.01% : 0.000054s : 1: graph_reusing 0.00% : 0.000005s : 1: grouped_pairwise_exchange_alltoall 0.00% : 0.000005s : 1: handle_group_info 0.00% : 0.000007s : 1: inline 0.00% : 0.000007s : 1: insert-virtual-dataset 0.00% : 0.000004s : 1: interleave_parallel_branches 0.00% : 0.000004s : 1: interleave_split_concat_branches 0.00% : 0.000006s : 1: label_fine_grained_interleaved_index 0.00% : 0.000009s : 1: label_micro_interleaved_index 0.11% : 0.000947s : 1: loop_unroll 0.00% : 0.000006s : 1: merge_cast_opt 0.00% : 0.000006s : 1: micro_interleaved_order_control 0.09% : 0.000801s : 1: mutable_eliminate 0.00% : 0.000014s : 1: offloading_packed_experts 0.01% : 0.000048s : 1: opt.transform.loop_unroll_optimizer 0.00% : 0.000038s : 1: opt.transform.mutable_eliminate 1.54% : 0.013328s : 95: opt.transform.opt_a 0.02% : 0.000136s : 1: opt.transform.opt_after_cconv 0.01% : 0.000081s : 1: opt.transform.opt_after_jit_grad 3.47% : 0.029940s : 28: opt.transform.opt_b 0.03% : 0.000290s : 2: opt.transform.opt_trans_graph 0.02% : 0.000202s : 4: opt.transform.symbol_engine_opt 3.47% : 0.029972s : 1: opt_a 0.04% : 0.000334s : 1: opt_after_cconv 0.10% : 0.000867s : 1: opt_after_jit_grad 3.51% : 0.030294s : 1: opt_b 10.00% : 0.086379s : 1: optimize 0.01% : 0.000052s : 1: optimize_parallel_all_gather_comm 0.00% : 0.000018s : 1: order_py_execute_after_rewriter 0.01% : 0.000053s : 1: overlap_grad_flash_sp 0.00% : 0.000004s : 1: overlap_grad_matmul_and_grad_allreduce 0.00% : 0.000013s : 1: overlap_grad_ring_attention 0.00% : 0.000006s : 1: overlap_opt_shard_grad_in_pipeline 0.00% : 0.000004s : 1: overlap_opt_shard_in_pipeline 0.00% : 0.000007s : 1: overlap_param_gather 0.00% : 0.000005s : 1: overlap_recompute_allgather_and_fa_grad 0.00% : 0.000014s : 1: overlap_recompute_and_grad_model_parallel 0.00% : 0.000006s : 1: overlap_recompute_comm 0.00% : 0.000010s : 1: parallel-infer-symbol 0.00% : 0.000005s : 1: parallel-infer-symbol-second 0.00% : 0.000006s : 1: partial_unused_args_eliminate 0.00% : 0.000005s : 1: pipeline_parallel_scheduler 0.00% : 0.000005s : 1: pipeline_split 0.03% : 0.000284s : 1: pre_auto_parallel 0.03% : 0.000280s : 1: py_interpret_to_execute 0.00% : 0.000038s : 1: py_interpret_to_execute_after_opt_a 0.00% : 0.000004s : 1: remove_cast_before_assign_add 0.01% : 0.000112s : 1: remove_dup_value 0.73% : 0.006304s : 1: renormalize.infer 0.87% : 0.007479s : 1: renormalize.specialize 0.00% : 0.000007s : 1: reorder_send_recv_between_fp_bp 0.00% : 0.000010s : 1: rewriter_after_jit_bprop_graph 0.10% : 0.000884s : 1: rewriter_after_opt_a 2.43% : 0.020983s : 1: rewriter_before_opt_a 0.00% : 0.000006s : 1: slice_cell_reuse_recomputed_activation 0.00% : 0.000005s : 1: slice_recompute_activation 0.00% : 0.000004s : 1: split_layernorm_comm 0.00% : 0.000006s : 1: split_matmul_comm_elemetwise 0.00% : 0.000025s : 1: swap_dp_allreduce_reducescatter 0.07% : 0.000566s : 1: symbol_engine_optimizer 0.04% : 0.000335s : 1: tuple_transform 69.45% : 0.599833s : 1: type_inference . [hook] pytest_runtest_teardown:test_matmul_qkv_bf16_0[True-mstype0-k_n_shape3-1] tests/st/infer/ops/test_internal_ops/test_matmul_split.py::test_matmul_qkv_bf16_0[True-mstype0-k_n_shape3-1],max_mem:208.0M [WARNING] ME(152078:281472851701552,MainProcess):2026-01-29-17:44:37.438.549 [mindspore/context.py:1334] For 'context.set_context', the parameter 'device_target' will be deprecated and removed in a future version. Please use the api mindspore.set_device() instead. TotalTime = 0.976141, [21] [bootstrap]: 0.00071433 [type_inference]: 0.893995 [event_method]: 0.00021296 [auto_monad]: 0.00076307 [graph_reusing]: 5.395e-05 [inline]: 3.41001e-06 [add_attr]: 0.0053607, [1] [add_attr_with_inline]: 0.00534055, [1] [Cycle 1]: 0.00032896, [2] [tag_attr]: 0.00022055 [meta_addattr_fg_expand]: 5.573e-05 [parallel-infer-symbol]: 4.53001e-06 [pre_auto_parallel]: 0.0204388 [insert-virtual-dataset]: 3.13998e-06 [parallel-infer-symbol-second]: 3.56999e-06 [dataset_repeat_opt]: 3.08998e-06 [pipeline_split]: 1.93002e-06 [optimize]: 0.0531656, [53] [py_interpret_to_execute]: 0.00032759 [rewriter_before_opt_a]: 0.00086052 [opt_a]: 0.0463706, [2] [Cycle 1]: 0.0440637, [45] [expand_dump_flag]: 2.951e-05 [switch_simplify]: 0.00086419 [loop_unroll]: 0.00032409 [a_1]: 0.0111403 [with_stream_mark]: 6.167e-05 [recompute_prepare]: 4.07e-05 [updatestate_depend_eliminate]: 2.043e-05 [updatestate_assign_eliminate]: 1.749e-05 [updatestate_loads_eliminate]: 1.87e-05 [parameter_eliminate]: 2.50002e-06 [a_2]: 0.00087073 [accelerated_algorithm]: 7.032e-05 [shard]: 2.21e-06 [meta_shard_fg_expand]: 2.324e-05 [shard_inline]: 2.516e-05 [merge_send_recv]: 2.057e-05 [auto_parallel]: 1.964e-05 [parallel]: 3.402e-05 [flash_sp]: 1.4e-05 [merge_comm]: 1.385e-05 [allreduce_fusion]: 1.294e-05 [matmul_add_comm_reduction]: 2.185e-05 [allreduce_slice_to_reducescatter]: 9.69972e-07 [virtual_shard_identity]: 2.663e-05 [virtual_dataset]: 2.333e-05 [get_grad_eliminate_]: 2.133e-05 [virtual_output]: 2.14e-05 [merge_forward]: 1.319e-05 [cell_reuse_recompute_pass]: 2.01003e-06 [offload_activation]: 2.55e-05 [cell_reuse_handle_not_recompute_node_pass]: 4.855e-05 [merge_recompute_call_nodes]: 1.43002e-06 [before_grad]: 4.139e-05 [set_forward_comm_id_for_comm_node_pass]: 1.425e-05 [meta_fg_expand]: 1.72e-05 [flash_sp_send_recv_attached]: 5.12999e-06 [receive_attached]: 2.26e-06 [after_resolve]: 3.186e-05 [a_after_grad]: 3.779e-05 [renormalize]: 0.0276211 [add_forward_monad_depend]: 2.602e-05 [auto_monad_grad]: 3.48999e-06 [auto_monad_eliminator]: 9.424e-05 [cse]: 0.00167897 [a_3]: 0.00018509 [Cycle 2]: 0.00228626, [45] [expand_dump_flag]: 3.54002e-06 [switch_simplify]: 2.788e-05 [loop_unroll]: 2.258e-05 [a_1]: 0.0006631 [with_stream_mark]: 4.306e-05 [recompute_prepare]: 2.415e-05 [updatestate_depend_eliminate]: 1.668e-05 [updatestate_assign_eliminate]: 1.288e-05 [updatestate_loads_eliminate]: 1.478e-05 [parameter_eliminate]: 2.93e-06 [a_2]: 0.00032669 [accelerated_algorithm]: 3.313e-05 [shard]: 2.80002e-06 [meta_shard_fg_expand]: 1.118e-05 [shard_inline]: 2.149e-05 [merge_send_recv]: 2.28e-05 [auto_parallel]: 2.112e-05 [parallel]: 1.247e-05 [flash_sp]: 4.67e-06 [merge_comm]: 1.404e-05 [allreduce_fusion]: 1.25e-05 [matmul_add_comm_reduction]: 2.526e-05 [allreduce_slice_to_reducescatter]: 9.29984e-07 [virtual_shard_identity]: 2.558e-05 [virtual_dataset]: 2.07e-05 [get_grad_eliminate_]: 2.202e-05 [virtual_output]: 1.963e-05 [merge_forward]: 1.345e-05 [cell_reuse_recompute_pass]: 3.52997e-06 [offload_activation]: 2.647e-05 [cell_reuse_handle_not_recompute_node_pass]: 4.511e-05 [merge_recompute_call_nodes]: 1.54e-06 [before_grad]: 3.696e-05 [set_forward_comm_id_for_comm_node_pass]: 1.424e-05 [meta_fg_expand]: 1.008e-05 [flash_sp_send_recv_attached]: 2.66999e-06 [receive_attached]: 3.13e-06 [after_resolve]: 2.832e-05 [a_after_grad]: 3.356e-05 [renormalize]: 8.9989e-08 [add_forward_monad_depend]: 3.86999e-06 [auto_monad_grad]: 3.02002e-06 [auto_monad_eliminator]: 6.715e-05 [cse]: 0.00015172 [a_3]: 0.00014273 [py_interpret_to_execute_after_opt_a]: 8.82e-05 [slice_cell_reuse_recomputed_activation]: 2.64999e-06 [rewriter_after_opt_a]: 0.00090046 [convert_after_rewriter]: 2.961e-05 [order_py_execute_after_rewriter]: 1.407e-05 [mutable_eliminate]: 0.00086815 [opt_b]: 0.00080235, [1] [Cycle 1]: 0.00079139, [7] [b_1]: 0.00056643 [b_2]: 2.43e-05 [updatestate_depend_eliminate]: 2.054e-05 [updatestate_assign_eliminate]: 1.157e-05 [updatestate_loads_eliminate]: 1.517e-05 [renormalize]: 1.20001e-06 [cse]: 0.00010609 [optimize_parallel_all_gather_comm]: 4.748e-05 [overlap_param_gather]: 2.46e-06 [cconv]: 4.522e-05 [loop_unroll]: 0.00064684 [opt_after_cconv]: 0.00031041, [1] [Cycle 1]: 0.00030089, [7] [c_1]: 0.00012999 [parameter_eliminate]: 5.70001e-06 [updatestate_depend_eliminate]: 1.907e-05 [updatestate_assign_eliminate]: 1.094e-05 [updatestate_loads_eliminate]: 1.346e-05 [cse]: 8.056e-05 [renormalize]: 8.2e-07 [remove_dup_value]: 0.00010128 [tuple_transform]: 0.0003583, [1] [Cycle 1]: 0.00035189, [4] [d_1]: 0.00029934 [none_parameter_eliminate]: 2.64001e-06 [renormalize]: 4.19997e-07 [switch_simplify]: 2.508e-05 [partial_unused_args_eliminate]: 2.44999e-06 [add_recomputation]: 0.00017146 [cse_after_recomputation]: 7.521e-05, [1] [Cycle 1]: 6.762e-05, [1] [cse]: 5.904e-05 [environ_conv]: 2.712e-05 [swap_dp_allreduce_reducescatter]: 2.054e-05 [bias_add_comm_swap]: 3.38e-06 [label_micro_interleaved_index]: 7.48999e-06 [label_fine_grained_interleaved_index]: 3.04999e-06 [merge_cast_opt]: 1.69e-06 [slice_recompute_activation]: 2.46e-06 [micro_interleaved_order_control]: 3.10002e-06 [assign_add_opt]: 1.62001e-06 [ForceFp32Comm]: 8.59989e-07 [remove_cast_before_assign_add]: 1.33002e-06 [full_micro_interleaved_order_control]: 2.27001e-06 [reorder_send_recv_between_fp_bp]: 2.92002e-06 [comm_op_add_attrs]: 1.22e-06 [add_comm_op_reuse_tag]: 1.07e-06 [interleave_split_concat_branches]: 1.40999e-06 [interleave_parallel_branches]: 1.05999e-06 [overlap_opt_shard_in_pipeline]: 1.28002e-06 [overlap_opt_shard_grad_in_pipeline]: 2.12999e-06 [control_data_broadcast_order]: 4.036e-05 [grouped_pairwise_exchange_alltoall]: 1.67001e-06 [offloading_packed_experts]: 1.197e-05 [overlap_recompute_and_grad_model_parallel]: 1.087e-05 [overlap_grad_matmul_and_grad_allreduce]: 1.30001e-06 [overlap_recompute_allgather_and_fa_grad]: 1.34998e-06 [overlap_recompute_comm]: 2.63e-06 [overlap_grad_ring_attention]: 9.81998e-06 [overlap_grad_flash_sp]: 5.486e-05 [begin_end_overlap_inline]: 5.39992e-07 [split_matmul_comm_elemetwise]: 2.70002e-06 [split_layernorm_comm]: 2.46e-06 [handle_group_info]: 1.18001e-06 [symbol_engine_optimizer]: 0.00056528, [1] [Cycle 1]: 0.0005582, [6] [build]: 0.00030875 [elim_shapecalc]: 3.784e-05 [elim_not_effective]: 5.556e-05 [opt_reshape]: 3.856e-05 [fold_const_symbol]: 6.677e-05 [renormalize]: 4.60015e-07 [detach_backward]: 2.72001e-06 [pipeline_parallel_scheduler]: 2.12001e-06 [auto_monad_reorder]: 6.53e-05 [get_jit_bprop_graph]: 2.19001e-06 [rewriter_after_jit_bprop_graph]: 6.91999e-06 [opt_after_jit_grad]: 0.0009035 [validate]: 0.00013878 Sums bootstrap : 0.000714s : 0.07% type_inference : 0.893995s : 92.22% event_method : 0.000213s : 0.02% auto_monad : 0.000763s : 0.08% graph_reusing : 0.000054s : 0.01% inline : 0.000003s : 0.00% add_attr.add_attr_with_inline.tag_attr : 0.000221s : 0.02% add_attr.add_attr_with_inline.meta_addattr_fg_expand : 0.000056s : 0.01% parallel-infer-symbol : 0.000005s : 0.00% pre_auto_parallel : 0.020439s : 2.11% insert-virtual-dataset : 0.000003s : 0.00% parallel-infer-symbol-second : 0.000004s : 0.00% dataset_repeat_opt : 0.000003s : 0.00% pipeline_split : 0.000002s : 0.00% optimize.py_interpret_to_execute : 0.000328s : 0.03% optimize.rewriter_before_opt_a : 0.000861s : 0.09% optimize.opt_a.expand_dump_flag : 0.000033s : 0.00% optimize.opt_a.switch_simplify : 0.000892s : 0.09% optimize.opt_a.loop_unroll : 0.000347s : 0.04% optimize.opt_a.a_1 : 0.011803s : 1.22% optimize.opt_a.with_stream_mark : 0.000105s : 0.01% optimize.opt_a.recompute_prepare : 0.000065s : 0.01% optimize.opt_a.updatestate_depend_eliminate : 0.000037s : 0.00% optimize.opt_a.updatestate_assign_eliminate : 0.000030s : 0.00% optimize.opt_a.updatestate_loads_eliminate : 0.000033s : 0.00% optimize.opt_a.parameter_eliminate : 0.000005s : 0.00% optimize.opt_a.a_2 : 0.001197s : 0.12% optimize.opt_a.accelerated_algorithm : 0.000103s : 0.01% optimize.opt_a.shard : 0.000005s : 0.00% optimize.opt_a.meta_shard_fg_expand : 0.000034s : 0.00% optimize.opt_a.shard_inline : 0.000047s : 0.00% optimize.opt_a.merge_send_recv : 0.000043s : 0.00% optimize.opt_a.auto_parallel : 0.000041s : 0.00% optimize.opt_a.parallel : 0.000046s : 0.00% optimize.opt_a.flash_sp : 0.000019s : 0.00% optimize.opt_a.merge_comm : 0.000028s : 0.00% optimize.opt_a.allreduce_fusion : 0.000025s : 0.00% optimize.opt_a.matmul_add_comm_reduction : 0.000047s : 0.00% optimize.opt_a.allreduce_slice_to_reducescatter : 0.000002s : 0.00% optimize.opt_a.virtual_shard_identity : 0.000052s : 0.01% optimize.opt_a.virtual_dataset : 0.000044s : 0.00% optimize.opt_a.get_grad_eliminate_ : 0.000043s : 0.00% optimize.opt_a.virtual_output : 0.000041s : 0.00% optimize.opt_a.merge_forward : 0.000027s : 0.00% optimize.opt_a.cell_reuse_recompute_pass : 0.000006s : 0.00% optimize.opt_a.offload_activation : 0.000052s : 0.01% optimize.opt_a.cell_reuse_handle_not_recompute_node_pass : 0.000094s : 0.01% optimize.opt_a.merge_recompute_call_nodes : 0.000003s : 0.00% optimize.opt_a.before_grad : 0.000078s : 0.01% optimize.opt_a.set_forward_comm_id_for_comm_node_pass : 0.000028s : 0.00% optimize.opt_a.meta_fg_expand : 0.000027s : 0.00% optimize.opt_a.flash_sp_send_recv_attached : 0.000008s : 0.00% optimize.opt_a.receive_attached : 0.000005s : 0.00% optimize.opt_a.after_resolve : 0.000060s : 0.01% optimize.opt_a.a_after_grad : 0.000071s : 0.01% optimize.opt_a.renormalize : 0.027621s : 2.85% optimize.opt_a.add_forward_monad_depend : 0.000030s : 0.00% optimize.opt_a.auto_monad_grad : 0.000007s : 0.00% optimize.opt_a.auto_monad_eliminator : 0.000161s : 0.02% optimize.opt_a.cse : 0.001831s : 0.19% optimize.opt_a.a_3 : 0.000328s : 0.03% optimize.py_interpret_to_execute_after_opt_a : 0.000088s : 0.01% optimize.slice_cell_reuse_recomputed_activation : 0.000003s : 0.00% optimize.rewriter_after_opt_a : 0.000900s : 0.09% optimize.convert_after_rewriter : 0.000030s : 0.00% optimize.order_py_execute_after_rewriter : 0.000014s : 0.00% optimize.mutable_eliminate : 0.000868s : 0.09% optimize.opt_b.b_1 : 0.000566s : 0.06% optimize.opt_b.b_2 : 0.000024s : 0.00% optimize.opt_b.updatestate_depend_eliminate : 0.000021s : 0.00% optimize.opt_b.updatestate_assign_eliminate : 0.000012s : 0.00% optimize.opt_b.updatestate_loads_eliminate : 0.000015s : 0.00% optimize.opt_b.renormalize : 0.000001s : 0.00% optimize.opt_b.cse : 0.000106s : 0.01% optimize.optimize_parallel_all_gather_comm : 0.000047s : 0.00% optimize.overlap_param_gather : 0.000002s : 0.00% optimize.cconv : 0.000045s : 0.00% optimize.loop_unroll : 0.000647s : 0.07% optimize.opt_after_cconv.c_1 : 0.000130s : 0.01% optimize.opt_after_cconv.parameter_eliminate : 0.000006s : 0.00% optimize.opt_after_cconv.updatestate_depend_eliminate : 0.000019s : 0.00% optimize.opt_after_cconv.updatestate_assign_eliminate : 0.000011s : 0.00% optimize.opt_after_cconv.updatestate_loads_eliminate : 0.000013s : 0.00% optimize.opt_after_cconv.cse : 0.000081s : 0.01% optimize.opt_after_cconv.renormalize : 0.000001s : 0.00% optimize.remove_dup_value : 0.000101s : 0.01% optimize.tuple_transform.d_1 : 0.000299s : 0.03% optimize.tuple_transform.none_parameter_eliminate : 0.000003s : 0.00% optimize.tuple_transform.renormalize : 0.000000s : 0.00% optimize.tuple_transform.switch_simplify : 0.000025s : 0.00% optimize.partial_unused_args_eliminate : 0.000002s : 0.00% optimize.add_recomputation : 0.000171s : 0.02% optimize.cse_after_recomputation.cse : 0.000059s : 0.01% optimize.environ_conv : 0.000027s : 0.00% optimize.swap_dp_allreduce_reducescatter : 0.000021s : 0.00% optimize.bias_add_comm_swap : 0.000003s : 0.00% optimize.label_micro_interleaved_index : 0.000007s : 0.00% optimize.label_fine_grained_interleaved_index : 0.000003s : 0.00% optimize.merge_cast_opt : 0.000002s : 0.00% optimize.slice_recompute_activation : 0.000002s : 0.00% optimize.micro_interleaved_order_control : 0.000003s : 0.00% optimize.assign_add_opt : 0.000002s : 0.00% optimize.ForceFp32Comm : 0.000001s : 0.00% optimize.remove_cast_before_assign_add : 0.000001s : 0.00% optimize.full_micro_interleaved_order_control : 0.000002s : 0.00% optimize.reorder_send_recv_between_fp_bp : 0.000003s : 0.00% optimize.comm_op_add_attrs : 0.000001s : 0.00% optimize.add_comm_op_reuse_tag : 0.000001s : 0.00% optimize.interleave_split_concat_branches : 0.000001s : 0.00% optimize.interleave_parallel_branches : 0.000001s : 0.00% optimize.overlap_opt_shard_in_pipeline : 0.000001s : 0.00% optimize.overlap_opt_shard_grad_in_pipeline : 0.000002s : 0.00% optimize.control_data_broadcast_order : 0.000040s : 0.00% optimize.grouped_pairwise_exchange_alltoall : 0.000002s : 0.00% optimize.offloading_packed_experts : 0.000012s : 0.00% optimize.overlap_recompute_and_grad_model_parallel : 0.000011s : 0.00% optimize.overlap_grad_matmul_and_grad_allreduce : 0.000001s : 0.00% optimize.overlap_recompute_allgather_and_fa_grad : 0.000001s : 0.00% optimize.overlap_recompute_comm : 0.000003s : 0.00% optimize.overlap_grad_ring_attention : 0.000010s : 0.00% optimize.overlap_grad_flash_sp : 0.000055s : 0.01% optimize.begin_end_overlap_inline : 0.000001s : 0.00% optimize.split_matmul_comm_elemetwise : 0.000003s : 0.00% optimize.split_layernorm_comm : 0.000002s : 0.00% optimize.handle_group_info : 0.000001s : 0.00% optimize.symbol_engine_optimizer.build : 0.000309s : 0.03% optimize.symbol_engine_optimizer.elim_shapecalc : 0.000038s : 0.00% optimize.symbol_engine_optimizer.elim_not_effective : 0.000056s : 0.01% optimize.symbol_engine_optimizer.opt_reshape : 0.000039s : 0.00% optimize.symbol_engine_optimizer.fold_const_symbol : 0.000067s : 0.01% optimize.symbol_engine_optimizer.renormalize : 0.000000s : 0.00% detach_backward : 0.000003s : 0.00% pipeline_parallel_scheduler : 0.000002s : 0.00% auto_monad_reorder : 0.000065s : 0.01% get_jit_bprop_graph : 0.000002s : 0.00% rewriter_after_jit_bprop_graph : 0.000007s : 0.00% opt_after_jit_grad : 0.000903s : 0.09% validate : 0.000139s : 0.01% Time group info: ------[substitution.] 0.003769 569 0.47% : 0.000018s : 8: substitution.depend_value_elim 0.36% : 0.000014s : 14: substitution.elim_not_effective 0.51% : 0.000019s : 12: substitution.float_tuple_getitem_switch 0.84% : 0.000031s : 14: substitution.fold_const_symbol 0.45% : 0.000017s : 17: substitution.graph_param_transform 68.87% : 0.002596s : 71: substitution.inline 0.35% : 0.000013s : 29: substitution.j_node_and_user_rematch 1.02% : 0.000038s : 6: substitution.less_batch_normalization 1.64% : 0.000062s : 2: substitution.list_to_tuple_eliminator_ 0.18% : 0.000007s : 6: substitution.load_eliminater 1.02% : 0.000038s : 21: substitution.minmaximum_grad 0.25% : 0.000009s : 2: substitution.opt_reshape 0.53% : 0.000020s : 29: substitution.remove_not_recompute_node 0.22% : 0.000008s : 4: substitution.replace_old_param 1.79% : 0.000067s : 18: substitution.reshape_eliminate 1.37% : 0.000051s : 24: substitution.switch_simplify 4.26% : 0.000160s : 51: substitution.tuple_list_convert_item_index_to_positive 2.42% : 0.000091s : 51: substitution.tuple_list_get_item_const_eliminator 3.09% : 0.000116s : 51: substitution.tuple_list_get_item_depend_reorder 6.19% : 0.000233s : 73: substitution.tuple_list_get_item_eliminator 2.94% : 0.000111s : 51: substitution.tuple_list_get_set_item_eliminator 0.64% : 0.000024s : 6: substitution.updatestate_pure_node_eliminater 0.60% : 0.000023s : 9: substitution.updatestate_useless_node_eliminater ------[type_inference.] 0.893616 2 92.88% : 0.829964s : 1: type_inference.infer 7.12% : 0.063652s : 1: type_inference.specialize ------[replace.] 0.001292 115 9.01% : 0.000116s : 5: replace.depend_value_elim 55.50% : 0.000717s : 71: replace.inline 3.32% : 0.000043s : 2: replace.list_to_tuple_eliminator_ 23.69% : 0.000306s : 24: replace.switch_simplify 8.48% : 0.000110s : 13: replace.tuple_list_get_item_eliminator ------[match.] 0.002683 115 0.11% : 0.000003s : 5: match.depend_value_elim 94.96% : 0.002547s : 71: match.inline 2.17% : 0.000058s : 2: match.list_to_tuple_eliminator_ 1.40% : 0.000038s : 24: match.switch_simplify 1.36% : 0.000036s : 13: match.tuple_list_get_item_eliminator ------[predicate.] 0.001609 9701 1.13% : 0.000018s : 123: predicate.accumulaten_eliminater 0.54% : 0.000009s : 17: predicate.ad_related_special_op_eliminate 0.50% : 0.000008s : 58: predicate.addn_check_dump 1.11% : 0.000018s : 123: predicate.addn_zero_filter 1.03% : 0.000017s : 123: predicate.adjust_all_reduce_mul_add 2.35% : 0.000038s : 181: predicate.arithmetic_simplify 1.23% : 0.000020s : 123: predicate.cast_eliminate 0.30% : 0.000005s : 34: predicate.check_bprop_eliminate 0.52% : 0.000008s : 58: predicate.compare_switch_simplify 0.08% : 0.000001s : 17: predicate.const_output_eliminate 0.71% : 0.000011s : 61: predicate.depend_value_elim 1.22% : 0.000020s : 123: predicate.dict_get_item_const_eliminator 1.25% : 0.000020s : 123: predicate.dict_get_item_eliminator 1.14% : 0.000018s : 123: predicate.dict_set_item_eliminator 0.42% : 0.000007s : 34: predicate.dumpgradient_eliminate 0.09% : 0.000001s : 17: predicate.elim_not_effective 0.22% : 0.000004s : 17: predicate.elim_shapecalc_of_broadcastargs 1.22% : 0.000020s : 140: predicate.environ_add_const_eliminate 1.28% : 0.000021s : 140: predicate.environ_get_add_eliminate 1.26% : 0.000020s : 140: predicate.environ_get_depend_swap 1.82% : 0.000029s : 198: predicate.environ_get_eliminate 1.21% : 0.000019s : 140: predicate.environ_get_set_eliminate 1.87% : 0.000030s : 209: predicate.exchange_switch_depend_value 2.73% : 0.000044s : 209: predicate.float_depend_g_call 0.51% : 0.000008s : 58: predicate.float_environ_get_switch 0.73% : 0.000012s : 75: predicate.float_tuple_getitem_switch 0.09% : 0.000001s : 17: predicate.fold_const_symbol 0.34% : 0.000005s : 35: predicate.get_grad_eliminate 0.12% : 0.000002s : 17: predicate.graph_param_transform 0.51% : 0.000008s : 58: predicate.incorporate_call 0.47% : 0.000008s : 58: predicate.incorporate_call_switch 5.99% : 0.000096s : 459: predicate.inline 0.47% : 0.000008s : 35: predicate.inline_without_move 0.15% : 0.000002s : 35: predicate.j_node_and_user_rematch 0.49% : 0.000008s : 36: predicate.less_batch_normalization 1.73% : 0.000028s : 172: predicate.list_to_tuple_eliminator_ 2.70% : 0.000043s : 295: predicate.load_eliminater 0.42% : 0.000007s : 17: predicate.loop_unroll_after_grad 3.19% : 0.000051s : 314: predicate.loop_unroll_before_grad 1.64% : 0.000026s : 157: predicate.make_slice_get_slice_eliminator 0.54% : 0.000009s : 58: predicate.merge_addn 0.30% : 0.000005s : 34: predicate.micro_step_allgather_replace 0.29% : 0.000005s : 34: predicate.mini_step_allgather_replace 1.14% : 0.000018s : 123: predicate.minmaximum_grad 0.43% : 0.000007s : 17: predicate.mutable_eliminate 0.27% : 0.000004s : 17: predicate.opt_reshape 0.18% : 0.000003s : 17: predicate.parallel_virtual_node 3.55% : 0.000057s : 209: predicate.partial_defer_inline 1.51% : 0.000024s : 155: predicate.partial_eliminate 1.11% : 0.000018s : 123: predicate.print_const_string_wrapper 0.50% : 0.000008s : 53: predicate.reduce_all_const_elim 1.63% : 0.000026s : 123: predicate.reduce_eliminate 2.70% : 0.000043s : 295: predicate.redundant_stop_gradient_eliminater 0.16% : 0.000003s : 35: predicate.remove_not_recompute_node 1.10% : 0.000018s : 172: predicate.replace_applicator 0.18% : 0.000003s : 35: predicate.replace_old_param 0.11% : 0.000002s : 17: predicate.reset_defer_inline 1.39% : 0.000022s : 123: predicate.reshape_eliminate 0.33% : 0.000005s : 34: predicate.row_tensor_add_zeros_like 0.20% : 0.000003s : 17: predicate.row_tensor_eliminate 0.51% : 0.000008s : 34: predicate.same_eliminate 0.22% : 0.000004s : 40: predicate.set_cell_output_no_recompute 0.36% : 0.000006s : 35: predicate.shard_identity_eliminate 0.37% : 0.000006s : 34: predicate.special_op_eliminate 0.62% : 0.000010s : 58: predicate.specialize_transform 0.47% : 0.000008s : 34: predicate.split_environ_get_set_with_tuple_value 0.42% : 0.000007s : 35: predicate.stack_unstack_eliminate 0.18% : 0.000003s : 17: predicate.switch_call_monad_eliminater 2.14% : 0.000034s : 209: predicate.switch_defer_inline 2.31% : 0.000037s : 243: predicate.switch_layer_defer_inline 6.78% : 0.000109s : 646: predicate.switch_simplify 1.18% : 0.000019s : 123: predicate.tile_eliminate 1.13% : 0.000018s : 123: predicate.transpose_eliminate 1.83% : 0.000029s : 157: predicate.tuple_list_convert_item_index_to_positive 1.80% : 0.000029s : 157: predicate.tuple_list_get_item_const_eliminator 1.60% : 0.000026s : 157: predicate.tuple_list_get_item_depend_reorder 3.30% : 0.000053s : 228: predicate.tuple_list_get_item_eliminator 1.63% : 0.000026s : 157: predicate.tuple_list_get_set_item_eliminator 2.24% : 0.000036s : 215: predicate.tuple_list_set_item_eliminator 1.60% : 0.000026s : 170: predicate.tuple_to_list_eliminator_ 2.54% : 0.000041s : 295: predicate.updatestate_pure_node_eliminater 3.15% : 0.000051s : 353: predicate.updatestate_useless_node_eliminater 0.18% : 0.000003s : 17: predicate.value_based_eliminate 0.33% : 0.000005s : 35: predicate.virtual_dataset_eliminate 0.32% : 0.000005s : 35: predicate.virtual_output_eliminate 0.18% : 0.000003s : 17: predicate.virtual_view_grad_eliminate 0.19% : 0.000003s : 17: predicate.zero_like_fill_zero ------[func_graph_cloner_run.] 0.016692 135 54.67% : 0.009125s : 55: func_graph_cloner_run.FuncGraphClonerGraph 45.33% : 0.007566s : 80: func_graph_cloner_run.FuncGraphSpecializer ------[meta_graph.] 0.000000 0 ------[manager.] 0.000000 0 ------[pynative] 0.000000 0 ------[others.] 1.078448 209 0.00% : 0.000004s : 1: ForceFp32Comm 0.50% : 0.005368s : 1: add_attr 0.50% : 0.005347s : 1: add_attr_with_inline 0.00% : 0.000005s : 1: add_comm_op_reuse_tag 0.02% : 0.000178s : 1: add_recomputation 0.00% : 0.000006s : 1: assign_add_opt 0.07% : 0.000782s : 1: auto_monad 0.01% : 0.000071s : 1: auto_monad_reorder 0.00% : 0.000005s : 1: begin_end_overlap_inline 0.00% : 0.000007s : 1: bias_add_comm_swap 0.07% : 0.000768s : 1: bootstrap 0.00% : 0.000049s : 1: cconv 0.00% : 0.000004s : 1: comm_op_add_attrs 0.00% : 0.000044s : 1: control_data_broadcast_order 0.00% : 0.000035s : 1: convert_after_rewriter 0.01% : 0.000079s : 1: cse_after_recomputation 0.00% : 0.000006s : 1: dataset_repeat_opt 0.00% : 0.000007s : 1: detach_backward 0.00% : 0.000031s : 1: environ_conv 0.02% : 0.000226s : 1: event_method 0.00% : 0.000006s : 1: full_micro_interleaved_order_control 0.00% : 0.000008s : 1: get_jit_bprop_graph 0.01% : 0.000061s : 1: graph_reusing 0.00% : 0.000005s : 1: grouped_pairwise_exchange_alltoall 0.00% : 0.000005s : 1: handle_group_info 0.00% : 0.000007s : 1: inline 0.00% : 0.000010s : 1: insert-virtual-dataset 0.00% : 0.000004s : 1: interleave_parallel_branches 0.00% : 0.000004s : 1: interleave_split_concat_branches 0.00% : 0.000006s : 1: label_fine_grained_interleaved_index 0.00% : 0.000011s : 1: label_micro_interleaved_index 0.06% : 0.000659s : 1: loop_unroll 0.00% : 0.000006s : 1: merge_cast_opt 0.00% : 0.000006s : 1: micro_interleaved_order_control 0.08% : 0.000882s : 1: mutable_eliminate 0.00% : 0.000015s : 1: offloading_packed_experts 0.00% : 0.000040s : 1: opt.transform.loop_unroll_optimizer 0.00% : 0.000051s : 1: opt.transform.mutable_eliminate 1.40% : 0.015112s : 95: opt.transform.opt_a 0.01% : 0.000129s : 1: opt.transform.opt_after_cconv 0.01% : 0.000094s : 1: opt.transform.opt_after_jit_grad 0.05% : 0.000549s : 28: opt.transform.opt_b 0.03% : 0.000321s : 2: opt.transform.opt_trans_graph 0.02% : 0.000192s : 4: opt.transform.symbol_engine_opt 4.30% : 0.046376s : 1: opt_a 0.03% : 0.000315s : 1: opt_after_cconv 0.09% : 0.000922s : 1: opt_after_jit_grad 0.07% : 0.000807s : 1: opt_b 4.93% : 0.053174s : 1: optimize 0.00% : 0.000053s : 1: optimize_parallel_all_gather_comm 0.00% : 0.000018s : 1: order_py_execute_after_rewriter 0.01% : 0.000060s : 1: overlap_grad_flash_sp 0.00% : 0.000004s : 1: overlap_grad_matmul_and_grad_allreduce 0.00% : 0.000013s : 1: overlap_grad_ring_attention 0.00% : 0.000006s : 1: overlap_opt_shard_grad_in_pipeline 0.00% : 0.000004s : 1: overlap_opt_shard_in_pipeline 0.00% : 0.000008s : 1: overlap_param_gather 0.00% : 0.000004s : 1: overlap_recompute_allgather_and_fa_grad 0.00% : 0.000014s : 1: overlap_recompute_and_grad_model_parallel 0.00% : 0.000007s : 1: overlap_recompute_comm 0.00% : 0.000010s : 1: parallel-infer-symbol 0.00% : 0.000007s : 1: parallel-infer-symbol-second 0.00% : 0.000006s : 1: partial_unused_args_eliminate 0.00% : 0.000006s : 1: pipeline_parallel_scheduler 0.00% : 0.000005s : 1: pipeline_split 1.90% : 0.020470s : 1: pre_auto_parallel 0.03% : 0.000340s : 1: py_interpret_to_execute 0.01% : 0.000097s : 1: py_interpret_to_execute_after_opt_a 0.00% : 0.000004s : 1: remove_cast_before_assign_add 0.01% : 0.000107s : 1: remove_dup_value 1.88% : 0.020239s : 1: renormalize.infer 0.68% : 0.007356s : 1: renormalize.specialize 0.00% : 0.000007s : 1: reorder_send_recv_between_fp_bp 0.00% : 0.000011s : 1: rewriter_after_jit_bprop_graph 0.08% : 0.000914s : 1: rewriter_after_opt_a 0.08% : 0.000873s : 1: rewriter_before_opt_a 0.00% : 0.000006s : 1: slice_cell_reuse_recomputed_activation 0.00% : 0.000005s : 1: slice_recompute_activation 0.00% : 0.000006s : 1: split_layernorm_comm 0.00% : 0.000006s : 1: split_matmul_comm_elemetwise 0.00% : 0.000024s : 1: swap_dp_allreduce_reducescatter 0.05% : 0.000569s : 1: symbol_engine_optimizer 0.03% : 0.000363s : 1: tuple_transform 82.90% : 0.894026s : 1: type_inference . [hook] pytest_runtest_teardown:test_matmul_qkv_bf16_0[True-mstype0-k_n_shape3-16] tests/st/infer/ops/test_internal_ops/test_matmul_split.py::test_matmul_qkv_bf16_0[True-mstype0-k_n_shape3-16],max_mem:208.0M =============================== warnings summary =============================== ../../../../../../../../../../usr/local/Ascend/cann-8.5.0/python/site-packages/tbe/dsl/classifier/transdata/transdata_classifier.py:222 /usr/local/Ascend/cann-8.5.0/python/site-packages/tbe/dsl/classifier/transdata/transdata_classifier.py:222: DeprecationWarning: invalid escape sequence \B """ ../../../../../../../../../../usr/local/Ascend/cann-8.5.0/python/site-packages/tbe/dsl/unify_schedule/vector/transdata/common/graph/transdata_graph_info.py:143 /usr/local/Ascend/cann-8.5.0/python/site-packages/tbe/dsl/unify_schedule/vector/transdata/common/graph/transdata_graph_info.py:143: DeprecationWarning: invalid escape sequence \c """ ../../../../../../../../../../usr/local/Ascend/cann-8.5.0/python/site-packages/tbe/dsl/unify_schedule/vector/transdata/common/graph/transdata_graph_info.py:170 /usr/local/Ascend/cann-8.5.0/python/site-packages/tbe/dsl/unify_schedule/vector/transdata/common/graph/transdata_graph_info.py:170: DeprecationWarning: invalid escape sequence \c """ ../../../../../../../../anaconda3/envs/ci39/lib/python3.9/site-packages/numpy/core/getlimits.py:549 /home/jenkins/anaconda3/envs/ci39/lib/python3.9/site-packages/numpy/core/getlimits.py:549: UserWarning: The value of the smallest subnormal for type is zero. setattr(self, word, getattr(machar, word).flat[0]) ../../../../../../../../anaconda3/envs/ci39/lib/python3.9/site-packages/numpy/core/getlimits.py:89 /home/jenkins/anaconda3/envs/ci39/lib/python3.9/site-packages/numpy/core/getlimits.py:89: UserWarning: The value of the smallest subnormal for type is zero. return self._float_to_str(self.smallest_subnormal) ../../../../../../../../anaconda3/envs/ci39/lib/python3.9/site-packages/numpy/core/getlimits.py:549 /home/jenkins/anaconda3/envs/ci39/lib/python3.9/site-packages/numpy/core/getlimits.py:549: UserWarning: The value of the smallest subnormal for type is zero. setattr(self, word, getattr(machar, word).flat[0]) ../../../../../../../../anaconda3/envs/ci39/lib/python3.9/site-packages/numpy/core/getlimits.py:89 /home/jenkins/anaconda3/envs/ci39/lib/python3.9/site-packages/numpy/core/getlimits.py:89: UserWarning: The value of the smallest subnormal for type is zero. return self._float_to_str(self.smallest_subnormal) ../../../../../../../../anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/batchnorm_fold2.py:57 /home/jenkins/anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/batchnorm_fold2.py:57: DeprecationWarning: te_fusion.fusion_manager.fusion_manager.register is deprecated,please replace it with tbe.common.register.register_op_compute @fusion_manager.register("batchnorm_fold2") ../../../../../../../../anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/batchnorm_fold2_grad.py:56 /home/jenkins/anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/batchnorm_fold2_grad.py:56: DeprecationWarning: te_fusion.fusion_manager.fusion_manager.register is deprecated,please replace it with tbe.common.register.register_op_compute @fusion_manager.register("batchnorm_fold2_grad") ../../../../../../../../anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/batchnorm_fold2_grad_reduce.py:48 /home/jenkins/anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/batchnorm_fold2_grad_reduce.py:48: DeprecationWarning: te_fusion.fusion_manager.fusion_manager.register is deprecated,please replace it with tbe.common.register.register_op_compute @fusion_manager.register("batchnorm_fold2_grad_reduce") ../../../../../../../../anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/correction_mul.py:51 /home/jenkins/anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/correction_mul.py:51: DeprecationWarning: te_fusion.fusion_manager.fusion_manager.register is deprecated,please replace it with tbe.common.register.register_op_compute @fusion_manager.register("correction_mul") ../../../../../../../../anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/correction_mul_grad.py:51 /home/jenkins/anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/correction_mul_grad.py:51: DeprecationWarning: te_fusion.fusion_manager.fusion_manager.register is deprecated,please replace it with tbe.common.register.register_op_compute @fusion_manager.register("correction_mul_grad") ../../../../../../../../anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/correction_mul_grad.py:143 /home/jenkins/anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/correction_mul_grad.py:143: DeprecationWarning: te_fusion.fusion_manager.fusion_manager.register is deprecated,please replace it with tbe.common.register.register_op_compute @fusion_manager.register("correction_mul_grad_reduce") ../../../../../../../../anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perlayer.py:50 /home/jenkins/anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perlayer.py:50: DeprecationWarning: te_fusion.fusion_manager.fusion_manager.register is deprecated,please replace it with tbe.common.register.register_op_compute @fusion_manager.register("fake_learned_scale_quant_perlayer") ../../../../../../../../anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perlayer_grad.py:92 /home/jenkins/anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perlayer_grad.py:92: DeprecationWarning: te_fusion.fusion_manager.fusion_manager.register is deprecated,please replace it with tbe.common.register.register_op_compute @fusion_manager.register("fake_learned_scale_quant_perlayer_grad_d") ../../../../../../../../anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perlayer_grad_reduce.py:49 /home/jenkins/anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perlayer_grad_reduce.py:49: DeprecationWarning: te_fusion.fusion_manager.fusion_manager.register is deprecated,please replace it with tbe.common.register.register_op_compute @fusion_manager.register("fake_learned_scale_quant_perlayer_grad_d_reduce") ../../../../../../../../anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perchannel.py:50 /home/jenkins/anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perchannel.py:50: DeprecationWarning: te_fusion.fusion_manager.fusion_manager.register is deprecated,please replace it with tbe.common.register.register_op_compute @fusion_manager.register("fake_learned_scale_quant_perchannel") ../../../../../../../../anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perchannel_grad.py:91 /home/jenkins/anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perchannel_grad.py:91: DeprecationWarning: te_fusion.fusion_manager.fusion_manager.register is deprecated,please replace it with tbe.common.register.register_op_compute @fusion_manager.register("fake_learned_scale_quant_perchannel_grad_d") ../../../../../../../../anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perchannel_grad_reduce.py:48 /home/jenkins/anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perchannel_grad_reduce.py:48: DeprecationWarning: te_fusion.fusion_manager.fusion_manager.register is deprecated,please replace it with tbe.common.register.register_op_compute @fusion_manager.register("fake_learned_scale_quant_perchannel_grad_d_reduce") ../../../../../../../../anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/fake_quant_perchannel.py:52 /home/jenkins/anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/fake_quant_perchannel.py:52: DeprecationWarning: te_fusion.fusion_manager.fusion_manager.register is deprecated,please replace it with tbe.common.register.register_op_compute @fusion_manager.register("fake_quant_perchannel") ../../../../../../../../anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/fake_quant_perchannel_grad.py:81 /home/jenkins/anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/fake_quant_perchannel_grad.py:81: DeprecationWarning: te_fusion.fusion_manager.fusion_manager.register is deprecated,please replace it with tbe.common.register.register_op_compute @fusion_manager.register("fake_quant_perchannel_grad") ../../../../../../../../anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/fake_quant_perlayer.py:54 /home/jenkins/anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/fake_quant_perlayer.py:54: DeprecationWarning: te_fusion.fusion_manager.fusion_manager.register is deprecated,please replace it with tbe.common.register.register_op_compute @fusion_manager.register("fake_quant_per_layer") ../../../../../../../../anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/fake_quant_perlayer_grad.py:81 /home/jenkins/anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/fake_quant_perlayer_grad.py:81: DeprecationWarning: te_fusion.fusion_manager.fusion_manager.register is deprecated,please replace it with tbe.common.register.register_op_compute @fusion_manager.register("fake_quant_per_layer_grad") ../../../../../../../../anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/minmax_update_perchannel.py:50 /home/jenkins/anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/minmax_update_perchannel.py:50: DeprecationWarning: te_fusion.fusion_manager.fusion_manager.register is deprecated,please replace it with tbe.common.register.register_op_compute @fusion_manager.register("minmax_update_perchannel") ../../../../../../../../anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/minmax_update_perlayer.py:50 /home/jenkins/anaconda3/envs/ci39/lib/python3.9/site-packages/mindspore/ops/_op_impl/_custom_op/minmax_update_perlayer.py:50: DeprecationWarning: te_fusion.fusion_manager.fusion_manager.register is deprecated,please replace it with tbe.common.register.register_op_compute @fusion_manager.register("minmax_update_perlayer") -- Docs: https://docs.pytest.org/en/stable/warnings.html ================= 16 passed, 25 warnings in 501.29s (0:08:21) ==================