[PIR]Fix add broadcast bug polish code (#59756)

phlrain · web-flow · commit eebd31c9c059 · 2023-12-08T10:09:09.000+08:00
* fix pir compiler name id bug

* remove usless code

* remove code

* fix bug

* remove usless code

* fix add broadcat bug
diff --git a/paddle/cinn/hlir/dialect/operator/transforms/add_broadcast_to_elementwise_pass.cc b/paddle/cinn/hlir/dialect/operator/transforms/add_broadcast_to_elementwise_pass.cc
@@ -60,7 +60,7 @@ std::vector<int64_t> GetOutputShape(const phi::DDim& x, const phi::DDim& y) {
 
     vec_res.resize(max_rank);
     for (size_t i = 0; i < max_rank; ++i) {
-      vec_res[i] = GetDimByIndex(x, y, short_align_axis, max_rank);
+      vec_res[i] = GetDimByIndex(x, y, short_align_axis, i);
     }
   }
 
diff --git a/paddle/cinn/hlir/framework/pir/group.h b/paddle/cinn/hlir/framework/pir/group.h
@@ -145,15 +145,6 @@ struct Group {
           group_inputs.insert(value);
           continue;
         }
-
-        // if (std::find(this->input_names.begin(),
-        //               this->input_names.end(),
-        //               CompatibleInfo::ValueName(value)) !=
-        //     this->input_names.end()) {
-        //   // if the input data in group's input_names
-        //   group_inputs.insert(value);
-        //   continue;
-        // }
       }
     }
 
diff --git a/paddle/cinn/hlir/framework/pir/utils.cc b/paddle/cinn/hlir/framework/pir/utils.cc
@@ -38,12 +38,9 @@ const std::unordered_map<std::string, std::string> CompatibleInfo::OP_NAMES = {
     {"pd_op.full", "fill_constant"},
     {"pd_op.sum", "reduce_sum"},
     {"pd_op.max", "reduce_max"},
-    {"pd_op.mean", "reduce_mean"},
     {"pd_op.add", "elementwise_add"},
     {"pd_op.elementwise_pow", "pow"},
     {"pd_op.multiply", "elementwise_mul"},
-    {"pd_op.remainder", "remainder"},
-    {"pd_op.floor_divide", "floor_divide"},
     {"pd_op.maximum", "max"},
     {"pd_op.minimum", "min"},
     {"pd_op.split_with_num", "split"},

Original file line number	Diff line number	Diff line change
`@@ -60,7 +60,7 @@ std::vector<int64_t> GetOutputShape(const phi::DDim& x, const phi::DDim& y) {`
`60`	`60`
`61`	`61`	`vec_res.resize(max_rank);`
`62`	`62`	`for (size_t i = 0; i < max_rank; ++i) {`
`63`		`- vec_res[i] = GetDimByIndex(x, y, short_align_axis, max_rank);`
	`63`	`+ vec_res[i] = GetDimByIndex(x, y, short_align_axis, i);`
`64`	`64`	`}`
`65`	`65`	`}`
`66`	`66`
Original file line number	Diff line number	Diff line change
`@@ -145,15 +145,6 @@ struct Group {`
`145`	`145`	`group_inputs.insert(value);`
`146`	`146`	`continue;`
`147`	`147`	`}`
`148`		`-`
`149`		`- // if (std::find(this->input_names.begin(),`
`150`		`- // this->input_names.end(),`
`151`		`- // CompatibleInfo::ValueName(value)) !=`
`152`		`- // this->input_names.end()) {`
`153`		`- // // if the input data in group's input_names`
`154`		`- // group_inputs.insert(value);`
`155`		`- // continue;`
`156`		`- // }`
`157`	`148`	`}`
`158`	`149`	`}`
`159`	`150`