[PIR+CINN]Polish CINN Support Op Name List (#58766)

* [PIR+CINN]Polish CINN Support Op Name List * add multiply
PaddlePaddle · Nov 8, 2023 · 17c4134 · 17c4134
1 parent b67c4cf
commit 17c4134
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 8 deletions.
diff --git a/paddle/cinn/hlir/framework/pir/utils.cc b/paddle/cinn/hlir/framework/pir/utils.cc
@@ -30,6 +30,7 @@ namespace hlir {
 namespace framework {
 namespace pir {
 
+// Mapping PaddleDialect Op into CINN AST Compute register Op
 const std::unordered_map<std::string, std::string> CompatibleInfo::OP_NAMES = {
     {"pd_op.full", "fill_constant"},
     {"pd_op.sum", "reduce_sum"},
@@ -39,8 +40,9 @@ const std::unordered_map<std::string, std::string> CompatibleInfo::OP_NAMES = {
     {"pd_op.divide", "divide"},
     {"cinn_op.broadcast", "broadcast_to"}};
 
+// Tagging PaddleDialect Op with REGITER_OP_MAPPER(OP)
 const std::unordered_set<std::string> CompatibleInfo::CINN_WHITE_OPS = {
-    "subtract"};
+    "subtract", "divide", "broadcast_to", "multiply"};
 
 bool CompatibleInfo::IsSupportCinn(const ::pir::Operation& op) {
   return CINN_WHITE_OPS.find(CompatibleInfo::OpName(op)) !=

diff --git a/paddle/fluid/pir/transforms/build_cinn_pass.cc b/paddle/fluid/pir/transforms/build_cinn_pass.cc
@@ -144,13 +144,6 @@ bool IsSupportCinn(pir::Operation* op) {
   bool registered =
       ::cinn::frontend::OpMapperRegistry::Global()->Find(op_name) != nullptr;
 
-  // TODO(phlrain): cinn fronted op name is not same with name in codegen
-  //                update using a better way define allow op list
-  if (op_name == "subtract" || op_name == "divide" ||
-      op_name == "broadcast_to") {
-    return true;
-  }
-
   OpTransInfo trans_info;
   bool is_support = registered && !trans_info.default_deny_ops().count(op_name);
   // if the op type is registered in CINN and allow_ops is not empty, return