【Error Message No. 10 BUAA】rewrite error messageerror message (#66540)

crazyxiaoxi · web-flow · commit 7cd477554838 · 2024-07-29T09:53:42.000+08:00
* error message

* fix some small problem

* fix some small problem

* add )

* phi not platform

* char wrong rerun
diff --git a/test/cpp/eager/data_structure_tests/accumulation_node_test.cc b/test/cpp/eager/data_structure_tests/accumulation_node_test.cc
@@ -73,23 +73,40 @@ TEST(AccumulationNode, SelectedRowsAddToTensor) {
       std::dynamic_pointer_cast<phi::SelectedRows>(ret_et0.impl())
           ->value()
           .data<float>();
-  CHECK_EQ(ret_et0_ptr[0], static_cast<float>(10.0f));
+  PADDLE_ENFORCE_EQ(ret_et0_ptr[0],
+                    static_cast<float>(10.0f),
+                    phi::errors::InvalidArgument(
+                        "The value of the first element of the selected rows "
+                        "should be 10.0f."));
   paddle::small_vector<std::vector<paddle::Tensor>, kSlotSmallVectorSize>
       et1_vec = {{et1}};
   paddle::Tensor ret_et1 = node->operator()(et1_vec)[0][0];
   auto* ret_et1_ptr =
       std::dynamic_pointer_cast<phi::DenseTensor>(ret_et1.impl())
           ->data<float>();
-  CHECK_EQ(ret_et1_ptr[0], static_cast<float>(20.0f));
+  PADDLE_ENFORCE_EQ(ret_et1_ptr[0],
+                    static_cast<float>(20.0f),
+                    phi::errors::InvalidArgument(
+                        "The value of the first element of the dense tensor "
+                        "should be 20.0f"));
   // Check Retain Grad
-  CHECK_EQ(std::dynamic_pointer_cast<phi::SelectedRows>(et0.impl())
-               ->value()
-               .data<float>()[0],
-           static_cast<float>(10.0f));
+  PADDLE_ENFORCE_EQ(
+      std::dynamic_pointer_cast<phi::SelectedRows>(et0.impl())
+          ->value()
+          .data<float>()[0],
+      static_cast<float>(10.0f),
+      phi::errors::InvalidArgument(
+          "The value of the first element of the dense tensor should be "
+          "10.0f"));
+
   paddle::Tensor* grad = EagerUtils::mutable_grad(input_et);
   auto* grad_ptr =
       std::dynamic_pointer_cast<phi::DenseTensor>(grad->impl())->data<float>();
-  CHECK_EQ(grad_ptr[0], static_cast<float>(30.0f));
+  PADDLE_ENFORCE_EQ(grad_ptr[0],
+                    static_cast<float>(30.0f),
+                    phi::errors::InvalidArgument(
+                        "The value of the first element of the dense tensor "
+                        "should be 30.0f"));
 }
 
 TEST(AccumulationNode, SelectedRowsMerge) {
@@ -136,25 +153,40 @@ TEST(AccumulationNode, SelectedRowsMerge) {
       std::dynamic_pointer_cast<phi::SelectedRows>(ret_et0.impl())
           ->value()
           .data<float>();
-  CHECK_EQ(ret_et0_ptr[0], static_cast<float>(10.0f));
+  PADDLE_ENFORCE_EQ(ret_et0_ptr[0],
+                    static_cast<float>(10.0f),
+                    phi::errors::InvalidArgument(
+                        "The value of the first element of the selected rows "
+                        "should be 10.0f."));
   paddle::small_vector<std::vector<paddle::Tensor>, kSlotSmallVectorSize>
       et1_vec = {{et1}};
   paddle::Tensor ret_et1 = node->operator()(et1_vec)[0][0];
   auto* ret_et1_ptr =
       std::dynamic_pointer_cast<phi::SelectedRows>(ret_et1.impl())
           ->value()
           .data<float>();
-  CHECK_EQ(ret_et1_ptr[0], static_cast<float>(20.0f));
+  PADDLE_ENFORCE_EQ(ret_et1_ptr[0],
+                    static_cast<float>(20.0f),
+                    phi::errors::InvalidArgument(
+                        "The value of the first element of the selected rows "
+                        "should be 20.0f."));
   // Check Retain Grad
-  CHECK_EQ(std::dynamic_pointer_cast<phi::SelectedRows>(et0.impl())
-               ->value()
-               .data<float>()[0],
-           static_cast<float>(10.0f));
+  PADDLE_ENFORCE_EQ(std::dynamic_pointer_cast<phi::SelectedRows>(et0.impl())
+                        ->value()
+                        .data<float>()[0],
+                    static_cast<float>(10.0f),
+                    phi::errors::InvalidArgument(
+                        "The value of the first element of the selected rows "
+                        "should be 10.0f."));
   paddle::Tensor* grad = EagerUtils::mutable_grad(input_et);
   auto* grad_ptr = std::dynamic_pointer_cast<phi::SelectedRows>(grad->impl())
                        ->value()
                        .data<float>();
-  CHECK_EQ(grad_ptr[0], static_cast<float>(30.0f));
+  PADDLE_ENFORCE_EQ(grad_ptr[0],
+                    static_cast<float>(30.0f),
+                    phi::errors::InvalidArgument(
+                        "The value of the first element of the selected rows "
+                        "should be 30.0f."));
 }
 
 TEST(AccumulationNode, SelectedRowsAddTensor) {
@@ -203,24 +235,39 @@ TEST(AccumulationNode, SelectedRowsAddTensor) {
       std::dynamic_pointer_cast<phi::SelectedRows>(ret_et0.impl())
           ->value()
           .data<float>();
-  CHECK_EQ(ret_et0_ptr[0], static_cast<float>(10.0f));
+  PADDLE_ENFORCE_EQ(ret_et0_ptr[0],
+                    static_cast<float>(10.0f),
+                    phi::errors::InvalidArgument(
+                        "The value of the first element of the selected rows "
+                        "should be 10.0f."));
   paddle::small_vector<std::vector<paddle::Tensor>, kSlotSmallVectorSize>
       et1_vec = {{et1}};
   paddle::Tensor ret_et1 = node->operator()(et1_vec)[0][0];
   auto* ret_et1_ptr =
       std::dynamic_pointer_cast<phi::SelectedRows>(ret_et1.impl())
           ->value()
           .data<float>();
-  CHECK_EQ(ret_et1_ptr[0], static_cast<float>(20.0f));
+  PADDLE_ENFORCE_EQ(ret_et1_ptr[0],
+                    static_cast<float>(20.0f),
+                    phi::errors::InvalidArgument(
+                        "The value of the first element of the selected rows "
+                        "should be 20.0f"));
   // Check Retain Grad
-  CHECK_EQ(std::dynamic_pointer_cast<phi::SelectedRows>(et0.impl())
-               ->value()
-               .data<float>()[0],
-           static_cast<float>(10.0f));
+  PADDLE_ENFORCE_EQ(std::dynamic_pointer_cast<phi::SelectedRows>(et0.impl())
+                        ->value()
+                        .data<float>()[0],
+                    static_cast<float>(10.0f),
+                    phi::errors::InvalidArgument(
+                        "The value of the first element of the selected rows "
+                        "should be 10.0f"));
   paddle::Tensor* grad = EagerUtils::mutable_grad(input_et);
   auto* grad_ptr =
       std::dynamic_pointer_cast<phi::DenseTensor>(grad->impl())->data<float>();
-  CHECK_EQ(grad_ptr[0], static_cast<float>(30.0f));
+  PADDLE_ENFORCE_EQ(grad_ptr[0],
+                    static_cast<float>(30.0f),
+                    phi::errors::InvalidArgument(
+                        "The value of the first element of the dense tensor "
+                        "should be 30.0f"));
 }
 
 TEST(AccumulationNode, Tensor) {
@@ -276,32 +323,47 @@ TEST(AccumulationNode, Tensor) {
   auto* ret_et0_ptr =
       std::dynamic_pointer_cast<phi::DenseTensor>(ret_et0.impl())
           ->data<phi::dtype::float16>();
-  CHECK_EQ(ret_et0_ptr[0], phi::dtype::float16(10.0f));
-
+  PADDLE_ENFORCE_EQ(ret_et0_ptr[0],
+                    phi::dtype::float16(10.0f),
+                    phi::errors::InvalidArgument(
+                        "The value of the first element of the dense tensor "
+                        "should be 10.0f."));
   paddle::small_vector<std::vector<paddle::Tensor>, kSlotSmallVectorSize>
       et1_vec = {{et1}};
   paddle::Tensor ret_et1 = node->operator()(et1_vec)[0][0];
 
   auto* ret_et1_ptr =
       std::dynamic_pointer_cast<phi::DenseTensor>(ret_et1.impl())
           ->data<phi::dtype::float16>();
-  CHECK_EQ(ret_et1_ptr[0], phi::dtype::float16(20.0f));
-
+  PADDLE_ENFORCE_EQ(ret_et1_ptr[0],
+                    phi::dtype::float16(20.0f),
+                    phi::errors::InvalidArgument(
+                        "The value of the first element of the dense tensor "
+                        "should be 20.0f"));
   // Check Retain Grad
-  CHECK_EQ(std::dynamic_pointer_cast<phi::DenseTensor>(et0.impl())
-               ->data<phi::dtype::float16>()[0],
-           phi::dtype::float16(10.0f));
+  PADDLE_ENFORCE_EQ(std::dynamic_pointer_cast<phi::DenseTensor>(et0.impl())
+                        ->data<phi::dtype::float16>()[0],
+                    phi::dtype::float16(10.0f),
+                    phi::errors::InvalidArgument(
+                        "The value of the first element of the dense tensor "
+                        "should be 10.0f"));
   paddle::Tensor* grad = EagerUtils::mutable_grad(input_et);
   auto* grad_ptr = std::dynamic_pointer_cast<phi::DenseTensor>(grad->impl())
                        ->data<phi::dtype::float16>();
-  CHECK_EQ(grad_ptr[0], phi::dtype::float16(30.0f));
+  PADDLE_ENFORCE_EQ(grad_ptr[0],
+                    phi::dtype::float16(30.0f),
+                    phi::errors::InvalidArgument(
+                        "The value of the first element of the dense tensor "
+                        "should be 30.0f"));
 
   // Reduce Hook case 1: Call RegisterReduceHook and run operator()
   VLOG(6) << "Test Reduce Hook";
-  CHECK_EQ(std::dynamic_pointer_cast<phi::DenseTensor>(et0.impl())
-               ->data<phi::dtype::float16>()[0],
-           phi::dtype::float16(10.0f));
-
+  PADDLE_ENFORCE_EQ(std::dynamic_pointer_cast<phi::DenseTensor>(et0.impl())
+                        ->data<phi::dtype::float16>()[0],
+                    phi::dtype::float16(10.0f),
+                    phi::errors::InvalidArgument(
+                        "The value of the first element of the dense tensor "
+                        "should be 10.0f"));
   auto reduce_hook_1 = [&]() -> void {
     auto* input_et_ptr =
         std::dynamic_pointer_cast<phi::DenseTensor>(input_et.impl())
@@ -318,14 +380,21 @@ TEST(AccumulationNode, Tensor) {
   // Check operator() result, should be 36.0
   auto* _ret_ptr = std::dynamic_pointer_cast<phi::DenseTensor>(_ret.impl())
                        ->data<phi::dtype::float16>();
-  CHECK_EQ(_ret_ptr[0], phi::dtype::float16(10.0f));
+  PADDLE_ENFORCE_EQ(_ret_ptr[0],
+                    phi::dtype::float16(10.0f),
+                    phi::errors::InvalidArgument(
+                        "The value of the first element of the dense tensor "
+                        "should be 10.0f"));
 
   // Check Retain Grad, should be 36.0
   auto* _ret_input_et_ptr =
       std::dynamic_pointer_cast<phi::DenseTensor>(input_et.impl())
           ->data<phi::dtype::float16>();
-  CHECK_EQ(_ret_input_et_ptr[0], phi::dtype::float16(36.0f));
-
+  PADDLE_ENFORCE_EQ(_ret_input_et_ptr[0],
+                    phi::dtype::float16(36.0f),
+                    phi::errors::InvalidArgument(
+                        "The value of the first element of the dense tensor "
+                        "should be 36.0f"));
   // Reduce Hook case 2: Call RegisterReduceHook and ApplyReduceHooks directly
   VLOG(6) << "Test Reduce Hook";
   auto reduce_hook_2 = [&]() -> void {
@@ -339,7 +408,10 @@ TEST(AccumulationNode, Tensor) {
   node->ApplyReduceHooks();
 
   // Check ApplyReduceHooks result
-  CHECK_EQ(std::dynamic_pointer_cast<phi::DenseTensor>(et0.impl())
-               ->data<phi::dtype::float16>()[0],
-           phi::dtype::float16(100.0f));
+  PADDLE_ENFORCE_EQ(std::dynamic_pointer_cast<phi::DenseTensor>(et0.impl())
+                        ->data<phi::dtype::float16>()[0],
+                    phi::dtype::float16(100.0f),
+                    phi::errors::InvalidArgument(
+                        "The value of the first element of the dense tensor "
+                        "should be 100.0f"));
 }
