【Error Message No. 31 Part2】fix CHECK_* in paddle/cinn/utils -part (#63039)

jinyouzhi · web-flow · commit 9d8b6be4b29b · 2024-03-28T10:32:16.000+08:00
diff --git a/paddle/cinn/utils/multi_threading.cc b/paddle/cinn/utils/multi_threading.cc
@@ -28,8 +28,12 @@ namespace utils {
 
 SequenceDispatcher::SequenceDispatcher(int begin, int end, int step)
     : end_(end), step_(step), index_(begin) {
-  CHECK_LE(begin, end) << StringFormat("begin[%d] > end[%d]", begin, end);
-  CHECK_GT(step, 0) << "step is less than 0";
+  PADDLE_ENFORCE_LE(
+      begin,
+      end,
+      phi::errors::InvalidArgument("begin[%d] > end[%d]", begin, end));
+  PADDLE_ENFORCE_GT(
+      step, 0, phi::errors::InvalidArgument("step is less than 0."));
 }
 
 int SequenceDispatcher::Next() const {
@@ -47,7 +51,10 @@ void parallel_run(const WorkerFuncType& fn,
   if (num_threads == -1 || num_threads > std::thread::hardware_concurrency()) {
     num_threads = std::thread::hardware_concurrency();
   }
-  CHECK_GT(num_threads, 0) << "num_threads should be greater than 0";
+  PADDLE_ENFORCE_GT(
+      num_threads,
+      0,
+      phi::errors::PreconditionNotMet("num_threads should be greater than 0"));
 
   // worker function of a thread
   auto worker = [&fn, &dispatcher](int tid) -> int {
diff --git a/paddle/cinn/utils/multi_threading_test.cc b/paddle/cinn/utils/multi_threading_test.cc
@@ -20,6 +20,8 @@
 #include <memory>
 #include <vector>
 
+#include "paddle/common/enforce.h"
+
 namespace cinn {
 namespace utils {
 
@@ -35,7 +37,8 @@ TEST(JobDispatcher, SequenceDispatcher) {
 TEST(parallel_run, Basic) {
   std::vector<int> results(100, -1);
   auto worker_fn = [&results](int index) {
-    CHECK_LT(index, results.size()) << "index invalid";
+    PADDLE_ENFORCE_LT(
+        index, results.size(), phi::errors::InvalidArgument("invalid index!"));
     results[index] = index;
   };
   // check process every index in the extent of [0, 100) with step 1
diff --git a/paddle/cinn/utils/random_engine.h b/paddle/cinn/utils/random_engine.h
@@ -18,6 +18,7 @@
 #include <stdint.h>
 
 #include <random>
+#include "paddle/common/enforce.h"
 
 namespace cinn {
 namespace utils {
@@ -69,7 +70,10 @@ class LinearRandomEngine {
     if (state == 0) {
       state = 1;
     }
-    CHECK_GE(state, 0) << "Random seed must be greater than 0";
+    PADDLE_ENFORCE_GE(
+        state,
+        0,
+        phi::errors::PreconditionNotMet("Random seed must be greater than 0"));
 
     return state;
   }
@@ -109,7 +113,10 @@ double SampleUniformDouble(double min,
 template <typename T>
 int SampleDiscreteFromDistribution(const std::vector<T>& weights,
                                    LinearRandomEngine::StateType* rand_seed) {
-  CHECK_GT(weights.size(), 0);
+  PADDLE_ENFORCE_GT(
+      weights.size(),
+      0,
+      phi::errors::PreconditionNotMet("Size of target weights is empty."));
   LinearRandomEngine engine(rand_seed);
   std::discrete_distribution<int> dist(weights.begin(), weights.end());
   return dist(engine);
diff --git a/paddle/cinn/utils/sized_multi_set.h b/paddle/cinn/utils/sized_multi_set.h
@@ -19,6 +19,7 @@
 #include <functional>
 #include <memory>
 #include <set>
+#include "paddle/common/enforce.h"
 
 namespace cinn {
 namespace utils {
@@ -55,7 +56,10 @@ class SizedMultiSet {
   }
 
   void Pop() {
-    CHECK_GE(multi_set_.size(), 1UL) << "Call Pop on empty SizedMultiSet";
+    PADDLE_ENFORCE_GE(
+        multi_set_.size(),
+        1UL,
+        phi::errors::PreconditionNotMet("Call Pop on empty SizedMultiSet."));
     if (pop_max_when_full_) {
       multi_set_.erase(--multi_set_.end());
     } else {
