【Error Message BUAA】fix error in 5 files. (#66551)

Wizard-ZP · web-flow · commit cf7398606249 · 2024-08-01T10:19:11.000+08:00
* 优化了6个文件

* 取消了ctr_double_accessor.h中的更改
diff --git a/paddle/cinn/hlir/framework/pir/op_lowering_impl.cc b/paddle/cinn/hlir/framework/pir/op_lowering_impl.cc
@@ -1159,7 +1159,6 @@ ir::LoweredFunc OpLowererImpl::GenerateInferShapeFunc(
     const OpLoweringGroupPtr& group,
     const std::vector<ir::Tensor> group_func_arg_tensors,
     const std::vector<ir::Argument> group_func_args) {
-  // CHECK_EQ(group_func_arg_tensors.size(), group_func_args.size());
   std::vector<ir::Expr> ir_bodys;
   int output_tensor_idx = 0;
   for (int tensor_arg_idx = 0; tensor_arg_idx < group_func_arg_tensors.size();
diff --git a/paddle/cinn/operator_fusion/utils.h b/paddle/cinn/operator_fusion/utils.h
@@ -70,8 +70,21 @@ static std::vector<int64_t> GetReduceAxisIdx(pir::Operation* reduce_op) {
     if (axis < 0) {
       axis += input_rank;
     }
-    CHECK_GE(axis, 0);
-    CHECK_LT(axis, input_rank);
+    PADDLE_ENFORCE_GE(
+        axis,
+        0,
+        phi::errors::InvalidArgument(
+            "The 'axis' must be greater than or equal to 0, but received %d.",
+            axis));
+
+    PADDLE_ENFORCE_LT(axis,
+                      input_rank,
+                      phi::errors::InvalidArgument(
+                          "The 'axis' must be less than 'input_rank', but "
+                          "received axis = %d and input_rank = %d.",
+                          axis,
+                          input_rank));
+
     reduce_axis_idx.push_back(axis);
   }
   VLOG(4) << "GetReduceAxisIdx: " << utils::Join(reduce_axis_idx, ",");
diff --git a/paddle/fluid/framework/fleet/fleet_wrapper.cc b/paddle/fluid/framework/fleet/fleet_wrapper.cc
@@ -260,6 +260,7 @@ void FleetWrapper::HeterPushSparseVars(
     show_index = 1;
     click_index = 2;
   }
+
   PADDLE_ENFORCE_GE(grad_dim,
                     0,
                     phi::errors::InvalidArgument(
diff --git a/paddle/fluid/framework/fleet/heter_ps/heter_comm_kernel.cu b/paddle/fluid/framework/fleet/heter_ps/heter_comm_kernel.cu
@@ -504,7 +504,13 @@ void HeterCommKernel::dy_mf_fill_shard_grads(KeyType* d_shard_keys,
   fill_shard_key_kernel<<<grid_size, block_size_, 0, stream>>>(
       d_shard_keys, d_keys, idx, c_len);
 
-  CHECK_EQ(grad_value_size % sizeof(float), 0);
+  PADDLE_ENFORCE_EQ(grad_value_size % sizeof(float),
+                    0,
+                    phi::errors::InvalidArgument(
+                        "The size of 'grad_value_size' must be a multiple of "
+                        "sizeof(float), but received size %d.",
+                        grad_value_size));
+
   size_t N = len * grad_value_size_float;
   grid_size = (N - 1) / block_size_ + 1;
   gather_dvals_by_unit_kernel<<<grid_size, block_size_, 0, stream>>>(
@@ -561,7 +567,13 @@ void HeterCommKernel::dy_mf_fill_dvals(float* d_shard_vals,
                                        size_t val_size,
                                        const StreamType& stream) {
   const size_t val_size_float = val_size / sizeof(float);
-  CHECK_EQ(val_size % sizeof(float), 0);
+  PADDLE_ENFORCE_EQ(
+      val_size % sizeof(float),
+      0,
+      phi::errors::InvalidArgument("The size of 'val_size' must be a multiple "
+                                   "of sizeof(float), but received size %d.",
+                                   val_size));
+
   size_t N = len * val_size_float;
   const int grid_size = (N - 1) / block_size_ + 1;
   // fill by float, d_shard_vals to d_vals
@@ -765,7 +777,13 @@ void HeterCommKernel::scatter_vals(const ValType* d_shard_vals,
                                    size_t value_bytes,
                                    const StreamType& stream) {
   const size_t val_size_unit = size_t(value_bytes / sizeof(ValType));
-  CHECK_EQ(value_bytes % sizeof(ValType), 0);
+  PADDLE_ENFORCE_EQ(value_bytes % sizeof(ValType),
+                    0,
+                    phi::errors::InvalidArgument(
+                        "The size of 'value_bytes' must be a multiple of "
+                        "sizeof(ValType), but received size %d.",
+                        value_bytes));
+
   size_t N = len * val_size_unit;
   const int grid_size = (N - 1) / block_size_ + 1;
   // fill by float, d_shard_vals to d_vals
@@ -899,7 +917,13 @@ void HeterCommKernel::check_valid_values(const int& type,
                                          const size_t& value_bytes,
                                          const StreamType& stream,
                                          bool debug) {
-  CHECK_EQ(value_bytes % sizeof(float), 0);
+  PADDLE_ENFORCE_EQ(value_bytes % sizeof(float),
+                    0,
+                    phi::errors::InvalidArgument(
+                        "The size of 'value_bytes' must be a multiple of "
+                        "sizeof(float), but received size %d.",
+                        value_bytes));
+
   const int grid_size = (N - 1) / block_size_ + 1;
   const int num = static_cast<int>(value_bytes / sizeof(float));
   check_valid_values_kernel<<<grid_size, block_size_, 0, stream>>>(
diff --git a/paddle/fluid/framework/ir/trt_remove_amp_strategy_op_pass.cc b/paddle/fluid/framework/ir/trt_remove_amp_strategy_op_pass.cc
@@ -62,7 +62,12 @@ void TrtRemoveAMPStrategyOpPass::ApplyImpl(Graph *graph) const {
   std::unordered_set<ir::Node *> fp16_vars;
   std::unordered_set<ir::Node *> cast_ops;
   for (auto *op_node : op_nodes) {
-    CHECK_EQ(op_node->IsOp(), true);
+    PADDLE_ENFORCE_EQ(
+        op_node->IsOp(),
+        true,
+        phi::errors::InvalidArgument(
+            "The 'op_node' must be an operator node, but it is not."));
+
     auto *op_desc = op_node->Op();
     if (op_desc->Type() == "cast") {
       auto input_dtype = op_node->inputs[0]->Var()->GetDataType();

Original file line number	Diff line number	Diff line change
`@@ -260,6 +260,7 @@ void FleetWrapper::HeterPushSparseVars(`
`260`	`260`	`show_index = 1;`
`261`	`261`	`click_index = 2;`
`262`	`262`	`}`
	`263`	`+`
`263`	`264`	`PADDLE_ENFORCE_GE(grad_dim,`
`264`	`265`	`0,`
`265`	`266`	`phi::errors::InvalidArgument(`