【Error Message BUAA No. 19】 将报错信息修改为 PADDLE_ENFORCE_EQ 的形式 (#66613)

MufanColin · web-flow · commit f122a00c16d7 · 2024-07-31T16:27:30.000+08:00
* 将 CHECK_EQ 统一替换为 PADDLE_ENFORCE_EQ

* Replaced it-&gt;dump()

* 优化了报错信息
diff --git a/test/cpp/pir/cinn/add_broadcast_to_elementwise_test.cc b/test/cpp/pir/cinn/add_broadcast_to_elementwise_test.cc
@@ -122,13 +122,29 @@ TEST(PatternRewrite, broadcast_elementwise) {
                 .block()
                 ->begin();
 
-  CHECK_EQ(it->isa<paddle::dialect::FullOp>(), true);
+  PADDLE_ENFORCE_EQ(it->isa<paddle::dialect::FullOp>(),
+                    true,
+                    phi::errors::PreconditionNotMet(
+                        "Expected FullOp but found different operation type: " +
+                        std::string(it->name())));
   it++;
-  CHECK_EQ(it->isa<paddle::dialect::FullOp>(), true);
+  PADDLE_ENFORCE_EQ(it->isa<paddle::dialect::FullOp>(),
+                    true,
+                    phi::errors::PreconditionNotMet(
+                        "Expected FullOp but found different operation type: " +
+                        std::string(it->name())));
   it++;
-  CHECK_EQ(it->isa<paddle::dialect::FullOp>(), true);
+  PADDLE_ENFORCE_EQ(it->isa<paddle::dialect::FullOp>(),
+                    true,
+                    phi::errors::PreconditionNotMet(
+                        "Expected FullOp but found different operation type: " +
+                        std::string(it->name())));
   it++;
-  CHECK_EQ(it->isa<paddle::dialect::AddOp>(), true);
+  PADDLE_ENFORCE_EQ(it->isa<paddle::dialect::AddOp>(),
+                    true,
+                    phi::errors::PreconditionNotMet(
+                        "Expected AddOp but found different operation type: " +
+                        std::string(it->name())));
 }
 
 TEST(PatternRewrite, broadcast_elementwise_both) {
@@ -151,15 +167,35 @@ TEST(PatternRewrite, broadcast_elementwise_both) {
                 .block()
                 ->begin();
 
-  CHECK_EQ(it->isa<paddle::dialect::FullOp>(), true);
+  PADDLE_ENFORCE_EQ(it->isa<paddle::dialect::FullOp>(),
+                    true,
+                    phi::errors::PreconditionNotMet(
+                        "Expected FullOp but found different operation type: " +
+                        std::string(it->name())));
   it++;
-  CHECK_EQ(it->isa<paddle::dialect::FullOp>(), true);
+  PADDLE_ENFORCE_EQ(it->isa<paddle::dialect::FullOp>(),
+                    true,
+                    phi::errors::PreconditionNotMet(
+                        "Expected FullOp but found different operation type: " +
+                        std::string(it->name())));
   it++;
-  CHECK_EQ(it->isa<paddle::dialect::FullOp>(), true);
+  PADDLE_ENFORCE_EQ(it->isa<paddle::dialect::FullOp>(),
+                    true,
+                    phi::errors::PreconditionNotMet(
+                        "Expected FullOp but found different operation type: " +
+                        std::string(it->name())));
   it++;
-  CHECK_EQ(it->isa<paddle::dialect::FullOp>(), true);
+  PADDLE_ENFORCE_EQ(it->isa<paddle::dialect::FullOp>(),
+                    true,
+                    phi::errors::PreconditionNotMet(
+                        "Expected FullOp but found different operation type: " +
+                        std::string(it->name())));
   it++;
-  CHECK_EQ(it->isa<paddle::dialect::AddOp>(), true);
+  PADDLE_ENFORCE_EQ(it->isa<paddle::dialect::AddOp>(),
+                    true,
+                    phi::errors::PreconditionNotMet(
+                        "Expected AddOp but found different operation type: " +
+                        std::string(it->name())));
 }
 
 TEST(PatternRewrite, broadcast_elementwise_sub_both) {
@@ -182,13 +218,34 @@ TEST(PatternRewrite, broadcast_elementwise_sub_both) {
                 .block()
                 ->begin();
 
-  CHECK_EQ(it->isa<paddle::dialect::FullOp>(), true);
+  PADDLE_ENFORCE_EQ(it->isa<paddle::dialect::FullOp>(),
+                    true,
+                    phi::errors::PreconditionNotMet(
+                        "Expected FullOp but found different operation type: " +
+                        std::string(it->name())));
   it++;
-  CHECK_EQ(it->isa<paddle::dialect::FullOp>(), true);
+  PADDLE_ENFORCE_EQ(it->isa<paddle::dialect::FullOp>(),
+                    true,
+                    phi::errors::PreconditionNotMet(
+                        "Expected FullOp but found different operation type: " +
+                        std::string(it->name())));
   it++;
-  CHECK_EQ(it->isa<paddle::dialect::FullOp>(), true);
+  PADDLE_ENFORCE_EQ(it->isa<paddle::dialect::FullOp>(),
+                    true,
+                    phi::errors::PreconditionNotMet(
+                        "Expected FullOp but found different operation type: " +
+                        std::string(it->name())));
   it++;
-  CHECK_EQ(it->isa<paddle::dialect::FullOp>(), true);
+  PADDLE_ENFORCE_EQ(it->isa<paddle::dialect::FullOp>(),
+                    true,
+                    phi::errors::PreconditionNotMet(
+                        "Expected FullOp but found different operation type: " +
+                        std::string(it->name())));
   it++;
-  CHECK_EQ(it->isa<paddle::dialect::SubtractOp>(), true);
+  PADDLE_ENFORCE_EQ(
+      it->isa<paddle::dialect::SubtractOp>(),
+      true,
+      phi::errors::PreconditionNotMet(
+          "Expected SubtractOp but found different operation type: " +
+          std::string(it->name())));
 }
