[BUAA][Error Message]优化了5个文件的报错信息 (#67154)

Wizard-ZP · web-flow · commit fb162427132a · 2024-08-12T14:59:49.000+08:00
* 优化了5个文件的报错信息

* 加入了一个文件 删除了一个文件

* debug

* codestyle

* 去除了一个文件，加入了一个文件

* codestyle

* debug

* code style
diff --git a/paddle/cinn/backends/codegen_cuda_host.cc b/paddle/cinn/backends/codegen_cuda_host.cc
@@ -36,7 +36,10 @@ llvm::Value* CodeGenCudaHost::LowerGPUKernelLauncher(
     const ir::_LoweredFunc_* func) {
   auto body = func->body;
   auto* call_ir = body.As<ir::Call>();
-  CHECK(call_ir);
+  PADDLE_ENFORCE_EQ(
+      call_ir,
+      nullptr,
+      phi::errors::InvalidArgument("The 'call_ir' must be true."));
 
   // Create the function
   // @{
@@ -144,7 +147,12 @@ llvm::Value* CodeGenCudaHost::LowerGPUKernelLauncher(
             b_->getInt8PtrTy(), kvalue, r_arg.as_var()->name + "_ptr_load"));
       } else if (r_arg.as_var()->type().is_cpp_handle() ||
                  r_arg.as_var()->type().is_int(32)) {
-        CHECK(global_args.count(r_arg.as_var()->name));
+        PADDLE_ENFORCE_EQ(
+            global_args.count(r_arg.as_var()->name),
+            1,
+            phi::errors::InvalidArgument(
+                "The argument '%s' must be present in global_args.",
+                r_arg.as_var()->name.c_str()));
         call_args.push_back(global_args[r_arg.as_var()->name]);
       } else {
         CINN_NOT_IMPLEMENTED;
@@ -285,7 +293,12 @@ llvm::Value* CodeGenCudaHost::LowerGPUKernelCall(const ir::Call* call_ir) {
         call_args.push_back(b_->CreateLoad(
             b_->getInt8PtrTy(), kvalue, r_arg.as_var()->name + "_ptr_load"));
       } else if (r_arg.as_var()->type().is_cpp_handle()) {
-        CHECK(global_args.count(r_arg.as_var()->name));
+        PADDLE_ENFORCE_EQ(
+            global_args.count(r_arg.as_var()->name),
+            1,
+            phi::errors::InvalidArgument(
+                "The argument '%s' must be present in global_args.",
+                r_arg.as_var()->name.c_str()));
         call_args.push_back(global_args[r_arg.as_var()->name]);
       } else if (r_arg.as_var()->type().is_int()) {
         call_args.push_back(GetVar(r_arg.as_var()->name, false));
diff --git a/paddle/cinn/backends/codegen_invoke_module.cc b/paddle/cinn/backends/codegen_invoke_module.cc
@@ -71,9 +71,17 @@ llvm::Value* CodeGenInvokeModule::LowerParseArgsValueCall(
       2,
       ::common::errors::InvalidArgument(
           "The number of arguments of ParseArgsValue should be 2"));
-  CHECK(call_ir->read_args[0].is_var() &&
-        call_ir->read_args[0].as_var()->type().is_cpp_handle());
-  CHECK(call_ir->read_args[1].type().is_int(32));
+  PADDLE_ENFORCE_EQ(
+      call_ir->read_args[0].is_var() &&
+          call_ir->read_args[0].as_var()->type().is_cpp_handle(),
+      true,
+      phi::errors::InvalidArgument("The first read argument must be a variable "
+                                   "with a C++ handle type."));
+
+  PADDLE_ENFORCE_EQ(call_ir->read_args[1].type().is_int(32),
+                    true,
+                    phi::errors::InvalidArgument(
+                        "The second read argument must be of type int32."));
   args_type.push_back(CinnTypeToLLVMType(type_of<void*>(), m_));
   args_type.push_back(CinnTypeToLLVMType(type_of<int32_t>(), m_));
 
@@ -94,7 +102,10 @@ llvm::Value* CodeGenSwitchHost::LowerInnerCaseCall(const ir::Call* op) {
                  [](auto& arg) { return std::addressof(arg); });
   // TODO(Hongqing-work): Add check for parameter type
   llvm::Function* call_func = m_->getFunction(op->name);
-  CHECK(call_func) << "Unknown function referenced. [" << op->name << "]";
+  PADDLE_ENFORCE_NOT_NULL(
+      call_func,
+      phi::errors::InvalidArgument("Unknown function referenced. [%s]",
+                                   op->name.c_str()));
   b_->CreateCall(call_func, ll_function_args);
   return nullptr;
 }
diff --git a/paddle/fluid/distributed/ps/table/memory_sparse_table.cc b/paddle/fluid/distributed/ps/table/memory_sparse_table.cc
@@ -83,12 +83,27 @@ int32_t MemorySparseTable::InitializeValue() {
     _shard_merge_rate = _config.has_shard_merge_rate()
                             ? _config.shard_merge_rate()
                             : _shard_merge_rate;
-    CHECK((_m_avg_local_shard_num = static_cast<int>(
-               std::ceil(_avg_local_shard_num * _shard_merge_rate)),
-           _m_avg_local_shard_num <= _avg_local_shard_num));
-    CHECK((_m_real_local_shard_num = static_cast<int>(
-               std::ceil(_real_local_shard_num * _shard_merge_rate)),
-           _m_real_local_shard_num <= _real_local_shard_num));
+    _m_avg_local_shard_num =
+        static_cast<int>(std::ceil(_avg_local_shard_num * _shard_merge_rate));
+    PADDLE_ENFORCE_LE(
+        _m_avg_local_shard_num,
+        _avg_local_shard_num,
+        phi::errors::InvalidArgument(
+            "The calculated '_m_avg_local_shard_num' (%d) must be less than or "
+            "equal to '_avg_local_shard_num' (%d).",
+            _m_avg_local_shard_num,
+            _avg_local_shard_num));
+
+    _m_real_local_shard_num =
+        static_cast<int>(std::ceil(_real_local_shard_num * _shard_merge_rate));
+    PADDLE_ENFORCE_LE(
+        _m_real_local_shard_num,
+        _real_local_shard_num,
+        phi::errors::InvalidArgument(
+            "The calculated '_m_real_local_shard_num' (%d) must be less than "
+            "or equal to '_real_local_shard_num' (%d).",
+            _m_real_local_shard_num,
+            _real_local_shard_num));
 
     uint32_t avg_shard_server_num =
         _sparse_table_shard_num / _avg_local_shard_num;
@@ -948,7 +963,12 @@ std::pair<int64_t, int64_t> MemorySparseTable::PrintTableStat() {
 }
 
 int32_t MemorySparseTable::Pull(TableContext &context) {
-  CHECK(context.value_type == Sparse);
+  PADDLE_ENFORCE_EQ(
+      context.value_type,
+      Sparse,
+      phi::errors::InvalidArgument(
+          "The 'value_type' in context must be 'Sparse', but received %d.",
+          context.value_type));
   if (context.use_ptr) {
     char **pull_values = context.pull_context.ptr_values;
     const uint64_t *keys = context.pull_context.keys;
@@ -962,7 +982,12 @@ int32_t MemorySparseTable::Pull(TableContext &context) {
 }
 
 int32_t MemorySparseTable::Push(TableContext &context) {
-  CHECK(context.value_type == Sparse);
+  PADDLE_ENFORCE_EQ(
+      context.value_type,
+      Sparse,
+      phi::errors::InvalidArgument(
+          "The 'value_type' in context must be 'Sparse', but received %d.",
+          context.value_type));
   if (!context.use_ptr) {
     return PushSparse(
         context.push_context.keys, context.push_context.values, context.num);
diff --git a/paddle/fluid/operators/collective/c_reducescatter_op.cc b/paddle/fluid/operators/collective/c_reducescatter_op.cc
@@ -16,15 +16,24 @@ limitations under the License. */
 
 #include <memory>
 
+#include "paddle/common/enforce.h"
+
 namespace paddle::operators {
 
 class CReduceScatterOp : public framework::OperatorWithKernel {
  public:
   using framework::OperatorWithKernel::OperatorWithKernel;
 
   void InferShape(framework::InferShapeContext *ctx) const override {
-    OP_INOUT_CHECK(ctx->HasInput("X"), "Input", "X", "ReduceScatter");
-    OP_INOUT_CHECK(ctx->HasOutput("Out"), "Output", "X", "ReduceScatter");
+    PADDLE_ENFORCE_EQ(ctx->HasInput("X"),
+                      true,
+                      phi::errors::InvalidArgument(
+                          "The input 'X' for ReduceScatter must be provided."));
+    PADDLE_ENFORCE_EQ(
+        ctx->HasOutput("Out"),
+        true,
+        phi::errors::InvalidArgument(
+            "The output 'Out' for ReduceScatter must be provided."));
     int nranks = ctx->Attrs().Get<int>("nranks");
     phi::DDim dim = ctx->GetInputDim("X");
     if (dim[0] > 0 || dim[0] < -1) {
diff --git a/paddle/fluid/operators/collective/c_split_op.cc b/paddle/fluid/operators/collective/c_split_op.cc
@@ -14,6 +14,8 @@ limitations under the License. */
 
 #include "paddle/fluid/operators/collective/c_split_op.h"
 
+#include "paddle/common/enforce.h"
+
 namespace paddle {
 namespace operators {
 
@@ -22,8 +24,14 @@ class CSplitOp : public framework::OperatorWithKernel {
   using framework::OperatorWithKernel::OperatorWithKernel;
 
   void InferShape(framework::InferShapeContext* ctx) const override {
-    OP_INOUT_CHECK(ctx->HasInput("X"), "Input", "X", "c_split");
-    OP_INOUT_CHECK(ctx->HasOutput("Out"), "Output", "Out", "c_split");
+    PADDLE_ENFORCE_EQ(ctx->HasInput("X"),
+                      true,
+                      phi::errors::InvalidArgument(
+                          "The input 'X' for c_split must be provided."));
+    PADDLE_ENFORCE_EQ(ctx->HasOutput("Out"),
+                      true,
+                      phi::errors::InvalidArgument(
+                          "The output 'Out' for c_split must be provided."));
     int nranks = ctx->Attrs().Get<int>("nranks");
     int rank = ctx->Attrs().Get<int>("rank");
     int ring_id = ctx->Attrs().Get<int>("ring_id");
