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

uanu2002 · web-flow · commit ce3b4e1d21cb · 2024-08-02T14:12:07.000+08:00
* fix error message

* fix commom::error

* fix commom::error
diff --git a/paddle/cinn/adt/schedule_dim.cc b/paddle/cinn/adt/schedule_dim.cc
@@ -52,7 +52,11 @@ std::shared_ptr<IndexExprInferContext> InitIndexExprInferContext(
     const List<Iterator>& input_iterators) {
   std::unordered_map<Variable, const Value> init_var2value;
   for (const auto& iterator : *input_iterators) {
-    CHECK(init_var2value.emplace(iterator, iterator).second);
+    PADDLE_ENFORCE_EQ(
+        init_var2value.emplace(iterator, iterator).second,
+        true,
+        ::common::errors::InvalidArgument(
+            "Insertion failed in init_var2value map. The key already exists."));
   }
 
   return std::make_shared<IndexExprInferContext>(init_var2value);
diff --git a/paddle/cinn/adt/schedule_dim.h b/paddle/cinn/adt/schedule_dim.h
@@ -20,7 +20,7 @@
 #include "paddle/cinn/adt/equation_value.h"
 #include "paddle/cinn/adt/equation_variable.h"
 #include "paddle/cinn/adt/schedule_descriptor.h"
-
+#include "paddle/common/enforce.h"
 namespace cinn::adt {
 
 DEFINE_ADT_TAG(tReduced);
diff --git a/paddle/cinn/backends/codegen_c_x86.cc b/paddle/cinn/backends/codegen_c_x86.cc
@@ -33,7 +33,13 @@ void CodeGenCX86::Visit(const ir::Div *op) {
 void CodeGenCX86::Visit(const ir::Load *op) {
   Expr dense_strided_ramp = detail::StridedRampBase(op->index(), 1);
   if (dense_strided_ramp.defined()) {  // Loading a continuous Ramp address.
-    CHECK(op->type().is_vector());
+    PADDLE_ENFORCE_EQ(
+        op->type().is_vector(),
+        true,
+        ::common::errors::InvalidArgument(
+            "The operation type is expected to be a vector, but it is not. "
+            "Please check the operation type and ensure it is correctly set to "
+            "a vector."));
 
     int bits = op->type().bits() * op->type().lanes();
     if (SupportsAVX512() && bits == 512) {
diff --git a/paddle/cinn/backends/codegen_c_x86.h b/paddle/cinn/backends/codegen_c_x86.h
@@ -97,8 +97,13 @@ class CodeGenCX86 : public CodeGenC {
     auto index = op->index();
     auto *ramp_n = index.template As<ir::Ramp>();
     if (ramp_n) {
-      CHECK(!ramp_n->base.template As<ir::Ramp>())
-          << "base of a Ramp node should not be Ramp type";
+      PADDLE_ENFORCE_EQ(
+          !ramp_n->base.template As<ir::Ramp>(),
+          true,
+          ::common::errors::InvalidArgument(
+              "The base of a Ramp node should not be of Ramp type. "
+              "Please ensure that the base is correctly set to a non-Ramp "
+              "type."));
       IrPrinter::Visit(ramp_n->base);
     } else {
       IrPrinter::Visit(op->index());
diff --git a/paddle/cinn/backends/codegen_cuda_generate_test.cc b/paddle/cinn/backends/codegen_cuda_generate_test.cc
@@ -35,6 +35,7 @@
 #include "paddle/cinn/lang/lower.h"
 #include "paddle/cinn/optim/ir_simplify.h"
 #include "paddle/cinn/utils/timer.h"
+#include "paddle/common/enforce.h"
 
 namespace cinn {
 namespace backends {
@@ -57,7 +58,12 @@ void __launch_bounds__(200) elementwise_mul(const float* __restrict__ A, const f
 }
   )ROC";
   std::ofstream file(cuda_source_name);
-  CHECK(file.is_open()) << "failed to open file " << cuda_source_name;
+  PADDLE_ENFORCE_EQ(file.is_open(),
+                    true,
+                    ::common::errors::Unavailable(
+                        "Failed to open file: %s. Please check if the file "
+                        "path is correct and the file is accessible.",
+                        cuda_source_name));
   file << CodeGenCUDA_Dev::GetSourceHeader();
   file << cuda_source_code;
   file.close();
diff --git a/paddle/cinn/backends/codegen_debug_test.cc b/paddle/cinn/backends/codegen_debug_test.cc
@@ -22,6 +22,7 @@
 #include "paddle/cinn/backends/nvrtc/nvrtc_util.h"
 #include "paddle/cinn/common/context.h"
 #include "paddle/cinn/runtime/cuda/cuda_module.h"
+#include "paddle/common/enforce.h"
 
 namespace cinn {
 namespace backends {
@@ -42,7 +43,11 @@ using runtime::cuda::CUDAModule;
  */
 template <typename T>
 CUdeviceptr CreateCudaMemory(const std::vector<int>& shape, const T* data) {
-  CHECK(!shape.empty()) << "Couldn't create CUDA memory for empty shape";
+  PADDLE_ENFORCE_EQ(!shape.empty(),
+                    true,
+                    ::common::errors::InvalidArgument(
+                        "Couldn't create CUDA memory for empty shape. Please "
+                        "ensure the shape is not empty."));
   CUDA_CALL(cudaDeviceSynchronize());
 
   int numel = 1;
diff --git a/paddle/cinn/backends/codegen_device_util.cc b/paddle/cinn/backends/codegen_device_util.cc
@@ -17,7 +17,7 @@
 #include "paddle/cinn/backends/cuda_util.h"
 #include "paddle/cinn/common/cas.h"
 #include "paddle/cinn/ir/ir_mutator.h"
-
+#include "paddle/common/enforce.h"
 PD_DECLARE_bool(cinn_bucket_compile);
 namespace cinn {
 namespace backends {
@@ -194,7 +194,11 @@ void detail::CollectBucketStrategyHostFunctionVisitor::ProcessLoweredFunc(
     ir::Expr func, ir::Expr predicate) {
   VLOG(4) << "Process Lowered Func" << func;
   ir::_LoweredFunc_ *func_node = func.as_lowered_func();
-  CHECK(func_node);
+  PADDLE_ENFORCE_NOT_NULL(
+      func_node,
+      ::common::errors::InvalidArgument(
+          "The provided function could not be cast to a lowered function. "
+          "Please ensure the function is valid."));
   if (!func_node->cuda_axis_info.valid()) {
     func_node->cuda_axis_info.set_valid(true);
   }
diff --git a/paddle/cinn/backends/extern_func_jit_register.cc b/paddle/cinn/backends/extern_func_jit_register.cc
@@ -24,7 +24,12 @@ void RegisterExternFunctionHelper(const std::string &fn_name,
                                   Target target,
                                   void *fn_ptr) {
   ExternFunctionProtoRegistry::Global().Register(fn_name, fn_proto.release());
-  CHECK(ExternFunctionProtoRegistry::Global().Lookup(fn_name));
+  PADDLE_ENFORCE_NOT_NULL(ExternFunctionProtoRegistry::Global().Lookup(fn_name),
+                          ::common::errors::NotFound(
+                              "The function prototype for '%s' was not found "
+                              "in the ExternFunctionProtoRegistry. Please "
+                              "ensure the function name is correct.",
+                              fn_name));
 
   ExternFunctionEmitterRegistry::Global().Register(
       ExternFuncID{TargetToBackendRepr(target), fn_name.c_str()}, fn_name);
diff --git a/paddle/cinn/backends/extern_func_jit_register.h b/paddle/cinn/backends/extern_func_jit_register.h
@@ -32,7 +32,7 @@
 #include "paddle/cinn/backends/llvm/llvm_util.h"
 #include "paddle/cinn/backends/llvm/runtime_symbol_registry.h"
 #include "paddle/cinn/common/macros.h"
-
+#include "paddle/common/enforce.h"
 /**
  * Helper to register an external function into CINN, including the prototype,
  * the function address.
diff --git a/paddle/cinn/common/axis.cc b/paddle/cinn/common/axis.cc
@@ -19,7 +19,7 @@
 #include "paddle/cinn/poly/dim.h"
 #include "paddle/cinn/poly/domain.h"
 #include "paddle/cinn/poly/stage.h"
-
+#include "paddle/common/enforce.h"
 namespace cinn {
 namespace common {
 
@@ -69,7 +69,12 @@ std::vector<ir::Var> GenDefaultAxis(int naxis) {
   std::vector<ir::Var> axis;
   for (int i = 0; i < naxis; i++) {
     axis.emplace_back(cinn::common::axis_name(i));
-    CHECK(axis.back()->type().valid());
+    PADDLE_ENFORCE_EQ(axis.back()->type().valid(),
+                      true,
+                      ::common::errors::InvalidArgument(
+                          "The type of axis at index %d is invalid. Please "
+                          "ensure each axis has a valid type.",
+                          i));
   }
   return axis;
 }
diff --git a/paddle/cinn/common/axis.h b/paddle/cinn/common/axis.h
@@ -19,7 +19,7 @@
 #include <string>
 #include <utility>
 #include <vector>
-
+#include "paddle/common/enforce.h"
 namespace cinn {
 namespace ir {
 
diff --git a/paddle/cinn/common/scc_walker.h b/paddle/cinn/common/scc_walker.h
@@ -24,7 +24,7 @@
 #include <vector>
 
 #include "paddle/cinn/common/dfs_walker.h"
-
+#include "paddle/common/enforce.h"
 namespace cinn {
 namespace common {
 
@@ -81,7 +81,11 @@ class SccWalker final {
       SccHandler(scc);
       // Update node2root outside dfs visitor.
       for (NodeType node : scc) {
-        CHECK(node2root.emplace(node, root).second);
+        PADDLE_ENFORCE_EQ(node2root.emplace(node, root).second,
+                          true,
+                          ::common::errors::AlreadyExists(
+                              "Failed to insert the node into node2root. The "
+                              "node may already exist."));
       }
     }
   }
diff --git a/paddle/cinn/common/test_helper.h b/paddle/cinn/common/test_helper.h
@@ -20,7 +20,7 @@
 #include "paddle/cinn/backends/llvm/codegen_llvm.h"
 #include "paddle/cinn/backends/llvm/simple_jit.h"
 #include "paddle/cinn/cinn.h"
-
+#include "paddle/common/enforce.h"
 namespace cinn {
 namespace common {
 
@@ -105,7 +105,11 @@ struct ArgsBuilder {
   }
 
   std::vector<cinn_pod_value_t> Build() {
-    CHECK(!data_.empty());
+    PADDLE_ENFORCE_EQ(!data_.empty(),
+                      true,
+                      ::common::errors::InvalidArgument(
+                          "The data_ container is empty. Please "
+                          "ensure it contains valid data."));
     return data_;
   }
 
diff --git a/paddle/cinn/common/type.h b/paddle/cinn/common/type.h
@@ -23,7 +23,7 @@
 #include "paddle/cinn/common/float16_bfloat16_utils.h"
 #include "paddle/cinn/common/macros.h"
 #include "paddle/cinn/runtime/cinn_runtime.h"
-
+#include "paddle/common/enforce.h"
 //! Much of the concepts are borrowed from Halide project.
 
 namespace cinn {
@@ -182,10 +182,12 @@ inline Type Float(int bits,
                   int lanes = 1,
                   Type::specific_type_t st = Type::specific_type_t::None) {
   if (bits == 16) {
-    CHECK(st == Type::specific_type_t::FP16 ||
-          st == Type::specific_type_t::BF16)
-        << "When creating a 16 bits Float, the specific_type_t must be FP16 or "
-           "BF16.";
+    PADDLE_ENFORCE_EQ((st == Type::specific_type_t::FP16 ||
+                       st == Type::specific_type_t::BF16),
+                      true,
+                      ::common::errors::InvalidArgument(
+                          "When creating a 16-bit Float, the specific_type_t "
+                          "must be FP16 or BF16."));
   }
   return Type(Type::type_t ::Float, bits, lanes, st);
 }
