Replace IR_THROW() with PADDLE_THROW()

zyfncg · Aug 25, 2023 · da03ecb · da03ecb
1 parent 35c9390
commit da03ecb
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 10 deletions.
diff --git a/paddle/fluid/ir/drr/ir_operation_creator.cc b/paddle/fluid/ir/drr/ir_operation_creator.cc
@@ -70,7 +70,7 @@ Operation* CreateOperation(const OpCall& op_call,
         op_call.outputs()[1]->name(),
         std::make_shared<IrValue>(reshape_op->result(1)));
     return reshape_op;
-    
+
   } else if (op_call.name() == "pd.transpose") {
     const auto& inputs = op_call.inputs();
     std::vector<Value> ir_values =
@@ -96,19 +96,16 @@ Operation* CreateOperation(const OpCall& op_call,
 
   } else if (op_call.name() == "pd.full") {
     const auto& inputs = op_call.inputs();
-    std::vector<Value> ir_values = 
+    std::vector<Value> ir_values =
         GetIrValuesByDrrTensors(inputs, *res_match_ctx);
     Operation* full_op = rewriter.Build<paddle::dialect::FullOp>(
-        CreateAttributeMap(op_call, src_match_ctx)
-    );
-    res_match_ctx->BindIrValue(
-      op_call.outputs()[0]->name(),
-      std::make_shared<IrValue>(full_op->result(0)));
+        CreateAttributeMap(op_call, src_match_ctx));
+    res_match_ctx->BindIrValue(op_call.outputs()[0]->name(),
+                               std::make_shared<IrValue>(full_op->result(0)));
     return full_op;
   }
 
-  LOG(ERROR) << "Unknown op " << op_call.name();
-  return nullptr;
+  PADDLE_THROW("Unknown op :" + op_call.name());
 }
 
 }  // namespace drr

diff --git a/paddle/fluid/ir/drr/match_context_impl.h b/paddle/fluid/ir/drr/match_context_impl.h
@@ -79,7 +79,7 @@ struct IrAttrTypeCast<std::vector<int64_t>> {
           attr.dyn_cast<paddle::dialect::IntArrayAttribute>().data().GetData();
       return result;
     }
-    IR_THROW("Dynamic cast failed for IR attribute vector<int64_t>");
+    PADDLE_THROW("Dynamic cast failed for IR attribute vector<int64_t>");
   }
 };