【Error Message BUAA No.35】fix error message (#66860)

uanu2002 · web-flow · commit 52adacd624ca · 2024-08-02T10:03:59.000+08:00
* fix error message

* fix cinn_runtime.cc

* fix error message

* fix codestyle

* fix codestyle

* restore runtime
diff --git a/paddle/cinn/common/broadcast_tree.cc b/paddle/cinn/common/broadcast_tree.cc
@@ -93,7 +93,11 @@ bool SearchBroadcastImpl(const symbol::Broadcast<symbol::DimExpr>& variadic,
                          const DoEachT& DoEach) {
   const auto& operands = *(variadic.operands);
   for (const auto& operand : operands) {
-    CHECK(!operand.isa<int64_t>());
+    PADDLE_ENFORCE_EQ(
+        !operand.isa<int64_t>(),
+        true,
+        phi::errors::InvalidArgument("Invalid operand type. Expected operand "
+                                     "not to be of type int64_t."));
     if (SearchBroadcast(operand, DoEach)) return true;
   }
   return DoEach(variadic);
@@ -255,8 +259,13 @@ std::optional<symbol::Broadcastable<symbol::DimExpr>> GetFirstCstrBroadcastable(
       }
     }
     if (lhs_symbol.has_value() && rhs_symbol.has_value()) {
-      CHECK(lhs_symbol != rhs_symbol)
-          << lhs_symbol.value() << " != " << rhs_symbol.value();
+      PADDLE_ENFORCE_NE(lhs_symbol,
+                        rhs_symbol,
+                        phi::errors::InvalidArgument(
+                            "Symbols should not be equal. "
+                            "Received lhs_symbol = %s, rhs_symbol = %s.",
+                            lhs_symbol.value(),
+                            rhs_symbol.value()));
       ret = symbol::Broadcastable<symbol::DimExpr>{lhs_symbol.value(),
                                                    rhs_symbol.value()};
       return true;
@@ -294,7 +303,10 @@ std::optional<symbol::Broadcastable<symbol::DimExpr>> GetFirstCstrBroadcastable(
                       2,
                       ::common::errors::InvalidArgument(
                           "The operands size should be greater than 2."));
-    CHECK(operands->at(0) != operands->at(1));
+    PADDLE_ENFORCE_NE(
+        operands->at(0),
+        operands->at(1),
+        phi::errors::InvalidArgument("Operands should not be equal. "));
     ret = symbol::Broadcastable<symbol::DimExpr>{operands->at(0),
                                                  operands->at(1)};
     return true;
diff --git a/paddle/cinn/ir/schedule/impl/compute_location.cc b/paddle/cinn/ir/schedule/impl/compute_location.cc
@@ -132,7 +132,11 @@ void DyScheduleImpl::SimpleComputeAt(const Expr& block, const Expr& loop) {
     VLOG(3) << i << "-th block_loop:\n" << block_loops[i];
     std::optional<bool> prove_eq = analyzer.ProveEQ(
         loops[i].As<ir::For>()->extent, block_loops[i].As<ir::For>()->extent);
-    CHECK(prove_eq.has_value() && prove_eq.value());
+    PADDLE_ENFORCE_EQ(
+        prove_eq.has_value() && prove_eq.value(),
+        true,
+        phi::errors::InvalidArgument("The proof result should have a value and "
+                                     "the proof result is false."));
     if (!prove_eq.has_value() || prove_eq.value() == false) {
       os << "Extent of loop in Expr Param(loop) and extent of loop in Expr "
             "Param(block) should be equal correspondingly!\n";
@@ -226,8 +230,14 @@ void DyScheduleImpl::ReverseComputeAt(const Expr& block,
   CINN_IR_SCHEDULE_BEGIN();
   std::string primitive = "ReverseComputeAt";
   std::ostringstream os;
-  CHECK(block.As<ir::ScheduleBlockRealize>());
-  CHECK(loop.As<ir::For>());
+  PADDLE_ENFORCE_NOT_NULL(
+      block.As<ir::ScheduleBlockRealize>(),
+      phi::errors::InvalidArgument(
+          "The block argument must be of type ScheduleBlockRealize."));
+
+  PADDLE_ENFORCE_NOT_NULL(
+      loop.As<ir::For>(),
+      phi::errors::InvalidArgument("The loop argument must be of type For."));
   Expr root = this->GetRootBlock(block);
   auto producers = GetProducers(block, root);
   auto consumers = GetConsumers(block, root);
diff --git a/paddle/cinn/runtime/cpu/host_intrinsics.cc b/paddle/cinn/runtime/cpu/host_intrinsics.cc
@@ -28,7 +28,12 @@
 extern "C" {
 
 void __cinn_host_tanh_v(const cinn_buffer_t* x, cinn_buffer_t* out) {
-  CINN_CHECK_EQ(x->num_elements(), out->num_elements());
+  PADDLE_ENFORCE_EQ(
+      x->num_elements(),
+      out->num_elements(),
+      phi::errors::InvalidArgument(
+          "The number of elements in input buffer (x) must be equal to the "
+          "number of elements in output buffer (out)."));
   int xn = x->num_elements();
   auto* x_data = reinterpret_cast<float*>(x->memory);
   auto* out_data = reinterpret_cast<float*>(out->memory);
diff --git a/paddle/fluid/distributed/ps/service/ps_graph_client.cc b/paddle/fluid/distributed/ps/service/ps_graph_client.cc
@@ -113,15 +113,13 @@ ::std::future<int32_t> PsGraphClient::PullSparsePtr(
       auto it = keys2rank_vec[shard].find(k);
       if (it != keys2rank_vec[shard].end()) {
         rank = it->second;
-        /*
-        int real = rank;
-        int expect = (k / 8) % 2;
-        CHECK(real == expect);
-        */
       } else {
         // Should not happen
         VLOG(0) << "PullSparsePtr, miss key " << k << " rank=" << _rank_id;
-        CHECK(it != keys2rank_vec[shard].end());
+        PADDLE_ENFORCE_NE(it,
+                          keys2rank_vec[shard].end(),
+                          phi::errors::InvalidArgument(
+                              "The key was not found in the expected shard."));
       }
     } else {
       rank = ps_wrapper->PartitionKeyForRank(k);
@@ -216,7 +214,11 @@ ::std::future<int32_t> PsGraphClient::PullSparseKey(
         rank = it->second;
       } else {
         VLOG(0) << "PullSparseKey, miss key " << k << " rank=" << _rank_id;
-        CHECK(it != keys2rank_vec[shard].end());
+        PADDLE_ENFORCE_NE(
+            it,
+            keys2rank_vec[shard].end(),
+            phi::errors::InvalidArgument(
+                "The key was not found in the expected shard.", k));
       }
     } else {
       rank = ps_wrapper->PartitionKeyForRank(k);
