【Error Message No. 27】paddle/cinn/lang/* (#62973)

shuaihehe · web-flow · commit ee570d300c2c · 2024-03-26T10:38:48.000+08:00
* fix

* fix1
diff --git a/paddle/cinn/lang/builtin.cc b/paddle/cinn/lang/builtin.cc
@@ -96,13 +96,17 @@ EXTERN_CALL_IMP(Popc, popc);
 #undef EXTERN_CALL_IMP
 #undef EXTERN_CALL_IMP_NO_VEC
 
-#define EXTERN_BINARY_CALL_IMP(name__, target__)                       \
-  Expr name__(Expr a, Expr b) {                                        \
-    CHECK_EQ(a.type(), b.type())                                       \
-        << #name__ << "'s inputs type not equal, where a:" << a.type() \
-        << " but b:" << b.type();                                      \
-    return ir::Call::Make(                                             \
-        a->type(), #target__, {a, b}, {}, ir::CallType::Extern);       \
+#define EXTERN_BINARY_CALL_IMP(name__, target__)                         \
+  Expr name__(Expr a, Expr b) {                                          \
+    PADDLE_ENFORCE_EQ(                                                   \
+        a.type(),                                                        \
+        b.type(),                                                        \
+        phi::errors::InvalidArgument(#name__ "'s inputs type not equal," \
+                                             "where a:%s but b:%s.",     \
+                                     a.type(),                           \
+                                     b.type()));                         \
+    return ir::Call::Make(                                               \
+        a->type(), #target__, {a, b}, {}, ir::CallType::Extern);         \
   }
 
 EXTERN_BINARY_CALL_IMP(Remainder, mod)
@@ -117,9 +121,13 @@ Expr Zero(const Type& type) { return ir::Zero(type); }
 Expr One(const Type& type) { return ir::One(type); }
 
 Expr FloorDivide(Expr a, Expr b) {
-  CHECK_EQ(a.type(), b.type())
-      << "FloorDivide's inputs type not equal, where a:" << a.type()
-      << " but b:" << b.type();
+  PADDLE_ENFORCE_EQ(a.type(),
+                    b.type(),
+                    phi::errors::InvalidArgument(
+                        "FloorDivide's inputs type not equal, where a:%s "
+                        " but b:%s.",
+                        a.type(),
+                        b.type()));
   if (a.type().is_float()) {
     return Floor(a / b);
   } else if (a.type().is_uint()) {
@@ -136,7 +144,12 @@ Expr FloorDivide(Expr a, Expr b) {
 }
 
 Expr min_value(const Type& type) {
-  CHECK_EQ(type.lanes(), 1);
+  PADDLE_ENFORCE_EQ(
+      type.lanes(),
+      1,
+      phi::errors::InvalidArgument("The value of min type's lanes is incorrect"
+                                   "Expected value is 1, but receive %d. ",
+                                   type.lanes()));
 #define FOR_CASE(type__)                                                     \
   if (type == type_of<type__>()) {                                           \
     return Expr(static_cast<type__>(std::numeric_limits<type__>::lowest())); \
@@ -158,7 +171,12 @@ Expr min_value(const Type& type) {
 }
 
 Expr max_value(const Type& type) {
-  CHECK_EQ(type.lanes(), 1);
+  PADDLE_ENFORCE_EQ(
+      type.lanes(),
+      1,
+      phi::errors::InvalidArgument("The value of max type's lanes is incorrect"
+                                   "Expected value is 1, but receive %d. ",
+                                   type.lanes()));
 
 #define FOR_CASE(type__)                                                  \
   if (type == type_of<type__>()) {                                        \
@@ -183,7 +201,12 @@ Expr max_value(const Type& type) {
 }
 
 Expr Epsilon(const Type& type) {
-  CHECK_EQ(type.lanes(), 1);
+  PADDLE_ENFORCE_EQ(type.lanes(),
+                    1,
+                    phi::errors::InvalidArgument(
+                        "The value of epsilon type's lanes is incorrect"
+                        "Expected value is 1, but receive %d. ",
+                        type.lanes()));
 
 #define FOR_CASE(type__)                                                      \
   if (type == type_of<type__>()) {                                            \
@@ -245,7 +268,12 @@ Expr IsNan(Expr e) {
 }
 
 Expr Infinity(const Type& type) {
-  CHECK_EQ(type.lanes(), 1U);
+  PADDLE_ENFORCE_EQ(type.lanes(),
+                    1U,
+                    phi::errors::InvalidArgument(
+                        "The value of infinity type's lanes is incorrect"
+                        "Expected value is 1, but receive %d. ",
+                        type.lanes()));
   if (type.is_float()) {
     if (type.bits() == 64) {
       return make_const(type, std::numeric_limits<double>::infinity());
diff --git a/paddle/cinn/lang/compute.cc b/paddle/cinn/lang/compute.cc
@@ -47,7 +47,12 @@ ir::Tensor Compute(const std::vector<Expr> &domain,
   return Compute(
       domain,
       [fn](const std::vector<Expr> &axis) -> Expr {
-        CHECK_EQ(axis.size(), 1);
+        PADDLE_ENFORCE_EQ(axis.size(),
+                          1,
+                          phi::errors::InvalidArgument(
+                              "The size of axis vector is incorrect"
+                              "Expected value is 1, but receive %d. ",
+                              axis.size()));
         return fn(axis[0]);
       },
       name,
@@ -61,7 +66,12 @@ ir::Tensor Compute(const std::vector<Expr> &domain,
   return Compute(
       domain,
       [fn](const std::vector<Expr> &axis) -> Expr {
-        CHECK_EQ(axis.size(), 2);
+        PADDLE_ENFORCE_EQ(axis.size(),
+                          2,
+                          phi::errors::InvalidArgument(
+                              "The size of axis vector is incorrect"
+                              "Expected value is 2, but receive %d. ",
+                              axis.size()));
         return fn(axis[0], axis[1]);
       },
       name,
@@ -75,7 +85,12 @@ ir::Tensor Compute(const std::vector<Expr> &domain,
   return Compute(
       domain,
       [fn](const std::vector<Expr> &axis) -> Expr {
-        CHECK_EQ(axis.size(), 3);
+        PADDLE_ENFORCE_EQ(axis.size(),
+                          3,
+                          phi::errors::InvalidArgument(
+                              "The size of axis vector is incorrect"
+                              "Expected value is 3, but receive %d. ",
+                              axis.size()));
         return fn(axis[0], axis[1], axis[2]);
       },
       name,
@@ -89,7 +104,12 @@ ir::Tensor Compute(const std::vector<Expr> &domain,
   return Compute(
       domain,
       [fn](const std::vector<Expr> &axis) -> Expr {
-        CHECK_EQ(axis.size(), 4);
+        PADDLE_ENFORCE_EQ(axis.size(),
+                          4,
+                          phi::errors::InvalidArgument(
+                              "The size of axis vector is incorrect"
+                              "Expected value is 4, but receive %d. ",
+                              axis.size()));
         return fn(axis[0], axis[1], axis[2], axis[3]);
       },
       name,
@@ -103,7 +123,12 @@ ir::Tensor Compute(const std::vector<Expr> &domain,
   return Compute(
       domain,
       [fn](const std::vector<Expr> &axis) -> Expr {
-        CHECK_EQ(axis.size(), 5);
+        PADDLE_ENFORCE_EQ(axis.size(),
+                          5,
+                          phi::errors::InvalidArgument(
+                              "The size of axis vector is incorrect"
+                              "Expected value is 5, but receive %d. ",
+                              axis.size()));
         return fn(axis[0], axis[1], axis[2], axis[3], axis[4]);
       },
       name,
@@ -117,7 +142,12 @@ ir::Tensor Compute(const std::vector<Expr> &domain,
   return Compute(
       domain,
       [fn](const std::vector<Expr> &axis) -> Expr {
-        CHECK_EQ(axis.size(), 6);
+        PADDLE_ENFORCE_EQ(axis.size(),
+                          6,
+                          phi::errors::InvalidArgument(
+                              "The size of axis vector is incorrect"
+                              "Expected value is 6, but receive %d. ",
+                              axis.size()));
         return fn(axis[0], axis[1], axis[2], axis[3], axis[4], axis[5]);
       },
       name,
diff --git a/paddle/cinn/lang/lower.cc b/paddle/cinn/lang/lower.cc
@@ -337,8 +337,11 @@ ir::LoweredFunc LowerToAst(const std::string& name,
                            const Target& target) {
   std::vector<ir::LoweredFunc> result =
       LowerToAstVec(name, tensor_args, tensor_group, target);
-  CHECK_EQ(result.size(), 1UL) << "LowerToAst contains not only 1 LoweredFunc, "
-                                  "use LowerToAstVec instead.";
+  PADDLE_ENFORCE_EQ(result.size(),
+                    1UL,
+                    phi::errors::InvalidArgument(
+                        "LowerToAst contains not only 1 LoweredFunc, "
+                        "use LowerToAstVec instead."));
   return result[0];
 }
 
diff --git a/paddle/cinn/lang/lower_impl.cc b/paddle/cinn/lang/lower_impl.cc
@@ -718,7 +718,13 @@ std::vector<Expr> LowerImpl::GenerateFunctionBody(
   std::unordered_map<std::string, std::vector<Expr>> resized_buffer_cache;
 
   for (auto& group : schedule->groups) {
-    CHECK_GT(group.nodes.size(), 0) << "group is empty";
+    PADDLE_ENFORCE_GT(
+        group.nodes.size(),
+        0,
+        phi::errors::InvalidArgument(
+            "Group is empty"
+            "Expected size of group is larger than 0, but receive %d. ",
+            group.nodes.size()));
     bool all_temp_tensor = true;
     for (auto& node : group.nodes) {
       if (!tensor_map.count(node->id())) {
