File tree Expand file tree Collapse file tree 1 file changed +0
-19
lines changed
paddle/cinn/hlir/framework/pir Expand file tree Collapse file tree 1 file changed +0
-19
lines changed Original file line number Diff line number Diff line change @@ -85,25 +85,6 @@ std::shared_ptr<GroupInfo> OpLowererImpl::GetGroupInfo(
8585 std::set<std::string>(fusion_group_info.reduce_var_name .begin (),
8686 fusion_group_info.reduce_var_name .end ());
8787
88- for (auto & op : group->output_ops ()) {
89- group_info->direct_output_var_names .insert (ValueName (op->result (0 )));
90- // collect all output tensor.
91- if (op->name () == " cinn_op.yield_store" ) {
92- auto input_var_name = ValueName (op->operand_source (0 ));
93- if (group_info->broadcast_info .count (input_var_name)) {
94- auto base_info = group_info->broadcast_info [input_var_name];
95- base_info.with_constrain = true ;
96- group_info->broadcast_info [ValueName (op->result (0 ))] = base_info;
97- }
98- }
99- for (auto opresult : op->results ()) {
100- if (tensor_map.count (opresult) == 0 ) {
101- continue ;
102- }
103- group_info->direct_output_var_names .insert (ValueName (opresult));
104- }
105- }
106-
10788 for (auto & val : group->output_values ()) {
10889 group_info->direct_output_var_names .insert (ValueName (val));
10990 }
You can’t perform that action at this time.
0 commit comments