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

uanu2002 · web-flow · commit 7df4b3ed06f5 · 2024-07-31T14:48:13.000+08:00
* fix error message

* fix error message

* fix repeat.cc
diff --git a/paddle/cinn/hlir/op/contrib/repeat.cc b/paddle/cinn/hlir/op/contrib/repeat.cc
@@ -45,12 +45,27 @@ std::vector<ir::Tensor> Repeat(const ir::Tensor &tensor,
                                int axis,
                                const std::string &output_name) {
   int ndim = static_cast<int>(tensor->shape.size());
-  CHECK(-ndim - 1 <= axis && axis <= ndim)
-      << "repeat only accepts `axis` in [-data.ndim - 1, data.ndim]"
-      << ", but got axis = " << axis << ", and data.ndim = " << ndim;
-  CHECK(repeats >= 1) << "repeat only accepts `repeats >= 1`"
-                      << ", but got repeats = " << repeats;
-
+  PADDLE_ENFORCE_EQ(-ndim - 1 <= axis && axis <= ndim,
+                    true,
+                    phi::errors::InvalidArgument(
+                        "The value of `axis` is out of the valid range. "
+                        "Repeat only accepts `axis` in the range [-data.ndim - "
+                        "1, data.ndim], "
+                        "but got axis = %d, and data.ndim = %d. "
+                        "Please check your input and ensure `axis` is within "
+                        "the valid range.",
+                        axis,
+                        ndim));
+
+  PADDLE_ENFORCE_GE(
+      repeats,
+      1,
+      phi::errors::InvalidArgument(
+          "The value of `repeats` is less than 1. "
+          "Repeat only accepts `repeats >= 1`, but got repeats = %d. "
+          "Please check your input and ensure `repeats` is greater than or "
+          "equal to 1.",
+          repeats));
   if (axis < 0) {
     // Calculate offset from last dimension
     axis += ndim;
@@ -96,22 +111,48 @@ std::shared_ptr<framework::OpStrategy> StrategyForRepeat(
       axis = absl::get<int>(iter.second);
     }
   }
-
-  CHECK(repeats >= 1) << "repeat only accepts `repeats >= 1`"
-                      << ", but got repeats = " << repeats;
-
+  PADDLE_ENFORCE_GE(
+      repeats,
+      1,
+      phi::errors::InvalidArgument(
+          "The value of `repeats` is less than 1. "
+          "Repeat only accepts `repeats >= 1`, but got repeats = %d. "
+          "Please check your input and ensure `repeats` is greater than or "
+          "equal to 1.",
+          repeats));
   framework::CINNCompute repeat_compute([=](lang::Args args,
                                             lang::RetValue *ret) {
-    CHECK(!args.empty())
-        << "The input arguments of Repeat compute is empty! Please check.\n";
-    CINNValuePack pack_args = args[0];
-    PADDLE_ENFORCE_GE(pack_args.size(),
-                      1U,
+    PADDLE_ENFORCE_EQ(!args.empty(),
+                      true,
                       phi::errors::InvalidArgument(
-                          "at least 1 input tensors for Repeat compute\n"));
+                          "The input arguments of Repeat compute is empty. "
+                          "Please check your input arguments and ensure they "
+                          "are not empty."));
+
+    CINNValuePack pack_args = args[0];
+
+    PADDLE_ENFORCE_GE(
+        pack_args.size(),
+        1U,
+        phi::errors::InvalidArgument(
+            "At least 1 input tensor is required for Repeat compute, "
+            "but got %d input tensors. Please check your input.",
+            pack_args.size()));
+
     Expr A = pack_args[0];
-    CHECK(A.as_tensor());
-    CHECK(!output_shapes.empty());
+    PADDLE_ENFORCE_NOT_NULL(
+        A.as_tensor(),
+        phi::errors::InvalidArgument(
+            "The first argument in pack_args is null "
+            "Please ensure the first argument is a valid tensor."));
+
+    PADDLE_ENFORCE_EQ(
+        !output_shapes.empty(),
+        true,
+        phi::errors::InvalidArgument(
+            "The output shapes are empty. "
+            "Please ensure the output shapes are correctly specified."));
+
     auto tensor_A = A.as_tensor_ref();
     VLOG(3) << "A shape: " << utils::Join(tensor_A->shape, ", ")
             << ", output_shapes: " << utils::Join(output_shapes[0], ", ");
@@ -120,7 +161,13 @@ std::shared_ptr<framework::OpStrategy> StrategyForRepeat(
     std::string tensor_name = pack_args[1].operator std::string();
 
     std::vector<ir::Tensor> out = Repeat(tensor_A, repeats, axis, tensor_name);
-    CHECK(out.size() == 1U) << "The size of Repeat's output should be 1";
+    PADDLE_ENFORCE_EQ(
+        out.size(),
+        1U,
+        phi::errors::InvalidArgument(
+            "The size of Repeat's output should be 1, but got %d. "
+            "Please check your Repeat function implementation.",
+            out.size()));
 
     std::vector<cinn::common::CINNValue> res;
     for (auto &t : out) {
@@ -132,8 +179,13 @@ std::shared_ptr<framework::OpStrategy> StrategyForRepeat(
 
   framework::CINNSchedule repeat_schedule([=](lang::Args args,
                                               lang::RetValue *ret) {
-    CHECK(!args.empty())
-        << "The input argument of repeat schedule is empty! Please check.\n";
+    PADDLE_ENFORCE_EQ(
+        !args.empty(),
+        true,
+        phi::errors::InvalidArgument(
+            "The input argument of repeat schedule is empty. "
+            "Please check your input arguments and ensure they are "
+            "not empty."));
     cinn::common::CINNValuePack arg_pack = args[0];
     std::vector<Expr> vec_ast;
     for (int i = 0; i < arg_pack.size(); i++) {
@@ -142,7 +194,12 @@ std::shared_ptr<framework::OpStrategy> StrategyForRepeat(
         vec_ast.emplace_back(temp);
       }
     }
-    CHECK(!vec_ast.empty());
+    PADDLE_ENFORCE_EQ(
+        !vec_ast.empty(),
+        true,
+        phi::errors::InvalidArgument(
+            "The vector of AST expressions is empty. "
+            "Please ensure there are valid expressions in the argument pack."));
     ir::ModuleExpr mod_expr(vec_ast);
     ir::IRSchedule ir_sch(mod_expr);
     ir_sch.MergeExprs();
diff --git a/paddle/cinn/ir/ir_printer.cc b/paddle/cinn/ir/ir_printer.cc
@@ -321,7 +321,11 @@ void IrPrinter::Visit(const _Module_ *x) {}
 void IrPrinter::Visit(const _Var_ *x) { str_ += x->name; }
 void IrPrinter::Visit(const Alloc *x) {
   auto *buffer = x->destination.As<ir::_Buffer_>();
-  CHECK(buffer);
+  PADDLE_ENFORCE_NOT_NULL(
+      buffer,
+      phi::errors::InvalidArgument("The destination is not a valid buffer. "
+                                   "Please ensure that `x->destination` is "
+                                   "properly assigned to a buffer."));
   str_ += "alloc(";
   str_ += buffer->name;
   str_ += ", ";
@@ -340,7 +344,11 @@ void IrPrinter::Visit(const Select *x) {
 void IrPrinter::Visit(const Load *x) {
   if (x->is_addr_tensor()) {
     auto *tensor = x->tensor.As<ir::_Tensor_>();
-    CHECK(tensor);
+    PADDLE_ENFORCE_NOT_NULL(
+        tensor,
+        phi::errors::InvalidArgument("The tensor is not valid. "
+                                     "Please ensure that `x->tensor` is "
+                                     "properly assigned to a tensor."));
     str_ += tensor->name;
   } else if (x->is_addr_scalar()) {
     Visit(x->tensor);
@@ -359,7 +367,12 @@ void IrPrinter::Visit(const Load *x) {
 void IrPrinter::Visit(const Store *x) {
   if (x->is_addr_tensor()) {
     auto *tensor_node = x->tensor.As<ir::_Tensor_>();
-    CHECK(tensor_node);
+    PADDLE_ENFORCE_NOT_NULL(
+        tensor_node,
+        phi::errors::InvalidArgument("The tensor node is not valid. "
+                                     "Please ensure that `x->tensor` is "
+                                     "properly assigned to a tensor node."));
+
     str_ += tensor_node->name;
   } else if (x->is_addr_scalar()) {
     Visit(x->tensor);
@@ -378,7 +391,12 @@ void IrPrinter::Visit(const Store *x) {
 }
 void IrPrinter::Visit(const Free *x) {
   auto *buffer = x->destination.As<ir::_Buffer_>();
-  CHECK(buffer);
+  PADDLE_ENFORCE_NOT_NULL(
+      buffer,
+      phi::errors::InvalidArgument("The destination is not a valid buffer. "
+                                   "Please ensure that `x->destination` is "
+                                   "properly assigned to a buffer."));
+
   str_ += "free(";
   str_ += buffer->name;
   str_ += ")";
@@ -434,7 +452,13 @@ void IrPrinter::Visit(const _LoweredFunc_ *f) {
   Visit(f->body);
 }
 void IrPrinter::Visit(const Let *f) {
-  CHECK(f->type().valid());
+  PADDLE_ENFORCE_EQ(
+      f->type().valid(),
+      true,
+      phi::errors::InvalidArgument(
+          "The type of `f` is not valid. "
+          "Please ensure that `f->type()` returns a valid type."));
+
   str_ += f->type().to_string();
   str_ += " ";
   Visit(f->symbol);
@@ -543,7 +567,11 @@ void IrPrinter::Visit(const PrimitiveNode *x) {
 
 void IrPrinter::Visit(const _BufferRange_ *x) {
   auto *buffer = x->buffer.As<ir::_Buffer_>();
-  CHECK(buffer);
+  PADDLE_ENFORCE_NOT_NULL(
+      buffer,
+      phi::errors::InvalidArgument(
+          "The buffer is not valid. "
+          "Please ensure that `x->buffer` is properly assigned to a buffer."));
   str_ += buffer->name;
   str_ += "[";
   for (std::size_t i = 0; i < x->ranges.size(); i++) {
diff --git a/test/cpp/inference/api/analyzer_capi_gpu_tester.cc b/test/cpp/inference/api/analyzer_capi_gpu_tester.cc
@@ -33,7 +33,11 @@ TEST(PD_AnalysisConfig, use_gpu) {
   PD_DisableGpu(config);
   PD_SetCpuMathLibraryNumThreads(config, 10);
   int num_thread = PD_CpuMathLibraryNumThreads(config);
-  CHECK(10 == num_thread) << "NO";
+  PADDLE_ENFORCE_EQ(10,
+                    num_thread,
+                    phi::errors::InvalidArgument("The num of thread should be"
+                                                 "equal to 10, but got %d.",
+                                                 num_thread));
   PD_SwitchSpecifyInputNames(config, true);
   PD_SwitchIrDebug(config, true);
   PD_SetModel(config, model_dir.c_str(), nullptr);
@@ -42,32 +46,95 @@ TEST(PD_AnalysisConfig, use_gpu) {
   LOG(INFO) << model_dir_;
   PD_EnableUseGpu(config, 100, 0);
   bool use_gpu = PD_UseGpu(config);
-  CHECK(use_gpu) << "NO";
+  PADDLE_ENFORCE_EQ(use_gpu,
+                    true,
+                    phi::errors::InvalidArgument(
+                        "GPU is not enabled. "
+                        "The configuration indicates that GPU should be used, "
+                        "but it is currently disabled. "
+                        "Please check your configuration settings and ensure "
+                        "that GPU is properly enabled."));
   int device = PD_GpuDeviceId(config);
-  CHECK(0 == device) << "NO";
+  PADDLE_ENFORCE_EQ(
+      device,
+      0,
+      phi::errors::InvalidArgument("The device ID is incorrect. "
+                                   "Expected device ID is 0, but received %d. "
+                                   "Please check your device configuration and "
+                                   "ensure the correct device ID is used.",
+                                   device));
   int init_size = PD_MemoryPoolInitSizeMb(config);
-  CHECK(100 == init_size) << "NO";
+  PADDLE_ENFORCE_EQ(init_size,
+                    100,
+                    phi::errors::InvalidArgument(
+                        "The initial size of the memory pool is incorrect. "
+                        "Expected size is 100 MB, but received %d MB. "
+                        "Please check your configuration settings and ensure "
+                        "the correct memory pool size is set.",
+                        init_size));
   float frac = PD_FractionOfGpuMemoryForPool(config);
   LOG(INFO) << frac;
   PD_EnableCUDNN(config);
   bool cudnn = PD_CudnnEnabled(config);
-  CHECK(cudnn) << "NO";
+  PADDLE_ENFORCE_EQ(cudnn,
+                    true,
+                    phi::errors::InvalidArgument(
+                        "cuDNN is not enabled. "
+                        "The configuration indicates that cuDNN should be "
+                        "enabled, but it is currently disabled. "
+                        "Please check your configuration settings and ensure "
+                        "that cuDNN is properly enabled."));
   PD_SwitchIrOptim(config, true);
   bool ir_optim = PD_IrOptim(config);
-  CHECK(ir_optim) << "NO";
+  PADDLE_ENFORCE_EQ(ir_optim,
+                    true,
+                    phi::errors::InvalidArgument(
+                        "IR optimization is not enabled. "
+                        "The configuration indicates that IR optimization "
+                        "should be enabled, but it is currently disabled. "
+                        "Please check your configuration settings and ensure "
+                        "that IR optimization is properly enabled."));
   PD_EnableTensorRtEngine(
       config, 1 << 20, 1, 3, Precision::kFloat32, false, false);
   bool trt_enable = PD_TensorrtEngineEnabled(config);
-  CHECK(trt_enable) << "NO";
+  PADDLE_ENFORCE_EQ(trt_enable,
+                    true,
+                    phi::errors::InvalidArgument(
+                        "TensorRT engine is not enabled. "
+                        "The configuration indicates that TensorRT engine "
+                        "should be enabled, but it is currently disabled. "
+                        "Please check your configuration settings and ensure "
+                        "that TensorRT engine is properly enabled."));
   PD_EnableMemoryOptim(config);
   bool memory_optim_enable = PD_MemoryOptimEnabled(config);
-  CHECK(memory_optim_enable) << "NO";
+  PADDLE_ENFORCE_EQ(memory_optim_enable,
+                    true,
+                    phi::errors::InvalidArgument(
+                        "Memory optimization is not enabled. "
+                        "The configuration indicates that memory optimization "
+                        "should be enabled, but it is currently disabled. "
+                        "Please check your configuration settings and ensure "
+                        "that memory optimization is properly enabled."));
   PD_EnableProfile(config);
   bool profiler_enable = PD_ProfileEnabled(config);
-  CHECK(profiler_enable) << "NO";
+  PADDLE_ENFORCE_EQ(profiler_enable,
+                    true,
+                    phi::errors::InvalidArgument(
+                        "Profiler is not enabled. "
+                        "The configuration indicates that the profiler should "
+                        "be enabled, but it is currently disabled. "
+                        "Please check your configuration settings and ensure "
+                        "that the profiler is properly enabled."));
   PD_SetInValid(config);
   bool is_valid = PD_IsValid(config);
-  CHECK(!is_valid) << "NO";
+  PADDLE_ENFORCE_EQ(
+      is_valid,
+      true,
+      phi::errors::InvalidArgument(
+          "Configuration is not valid. "
+          "The configuration should be valid, but it is currently invalid. "
+          "Please check your configuration settings and ensure they are "
+          "correct."));
   PD_DeleteAnalysisConfig(config);
 }
 
@@ -77,7 +144,14 @@ TEST(PD_AnalysisConfig, trt_int8) {
   PD_EnableUseGpu(config, 100, 0);
   PD_EnableTensorRtEngine(config, 1 << 20, 1, 3, Precision::kInt8, false, true);
   bool trt_enable = PD_TensorrtEngineEnabled(config);
-  CHECK(trt_enable) << "NO";
+  PADDLE_ENFORCE_EQ(trt_enable,
+                    true,
+                    phi::errors::InvalidArgument(
+                        "TensorRT engine is not enabled. "
+                        "The configuration indicates that TensorRT engine "
+                        "should be enabled, but it is currently disabled. "
+                        "Please check your configuration settings and ensure "
+                        "that TensorRT engine is properly enabled."));
   PD_DeleteAnalysisConfig(config);
 }
 
@@ -88,7 +162,14 @@ TEST(PD_AnalysisConfig, trt_fp16) {
   PD_EnableTensorRtEngine(
       config, 1 << 20, 1, 3, Precision::kHalf, false, false);
   bool trt_enable = PD_TensorrtEngineEnabled(config);
-  CHECK(trt_enable) << "NO";
+  PADDLE_ENFORCE_EQ(trt_enable,
+                    true,
+                    phi::errors::InvalidArgument(
+                        "TensorRT engine is not enabled. "
+                        "The configuration indicates that TensorRT engine "
+                        "should be enabled, but it is currently disabled. "
+                        "Please check your configuration settings and ensure "
+                        "that TensorRT engine is properly enabled."));
   PD_Predictor *predictor = PD_NewPredictor(config);
   PD_DeletePredictor(predictor);
   PD_DeleteAnalysisConfig(config);
