优化了一个报错信息 (#66502)

Wizard-ZP · web-flow · commit e75a6b6f567a · 2024-07-29T14:27:12.000+08:00
diff --git a/paddle/cinn/hlir/op/contrib/assert_true.cc b/paddle/cinn/hlir/op/contrib/assert_true.cc
@@ -40,22 +40,31 @@ std::shared_ptr<framework::OpStrategy> StrategyForAssertTrue(
     const std::vector<Type> &out_type,
     const std::vector<std::vector<int>> &output_shapes,
     const Target &target) {
-  framework::CINNCompute assert_true_compute([=](lang::Args args,
-                                                 lang::RetValue *ret) {
-    CHECK(!args.empty())
-        << "The input argument of assert_true is empty! Please check.";
-    CINNValuePack pack_args = args[0];
-    CHECK_GE(pack_args.size(), 2U)
-        << "Two input tensors are required for the computation of assert_true.";
-    Expr a_expr = pack_args[0];
-    Expr b_expr = pack_args[1];
-    ir::Tensor a = a_expr.as_tensor_ref();
-    ir::Tensor b = b_expr.as_tensor_ref();
-    std::string tensor_name = "assert_true_out";
-    auto out = pe::Identity(b, tensor_name).front();
-    std::vector<CINNValue> res{CINNValue(out)};
-    *ret = CINNValuePack{res};
-  });
+  framework::CINNCompute assert_true_compute(
+      [=](lang::Args args, lang::RetValue *ret) {
+        PADDLE_ENFORCE_EQ(
+            !args.empty(),
+            true,
+            phi::errors::InvalidArgument("The input argument of assert_true is "
+                                         "empty! Please check."));
+        CINNValuePack pack_args = args[0];
+        PADDLE_ENFORCE_GE(
+            pack_args.size(),
+            2U,
+            phi::errors::InvalidArgument("Two input tensors are required "
+                                         "for the computation of assert_true, "
+                                         "but received size %d.",
+                                         pack_args.size()));
+
+        Expr a_expr = pack_args[0];
+        Expr b_expr = pack_args[1];
+        ir::Tensor a = a_expr.as_tensor_ref();
+        ir::Tensor b = b_expr.as_tensor_ref();
+        std::string tensor_name = "assert_true_out";
+        auto out = pe::Identity(b, tensor_name).front();
+        std::vector<CINNValue> res{CINNValue(out)};
+        *ret = CINNValuePack{res};
+      });
   auto strategy = std::make_shared<framework::OpStrategy>();
   strategy->AddImpl(assert_true_compute,
                     GetElementwiseScheduleFunc(output_shapes, target),
