【Error Message No.32,33 BUAA】Update error massage of paddle\fluid\framework\multi_trainer.cc & paddle\fluid\inference\tensorrt\convert\test_custom_plugin_creater.cc & paddle\fluid\inference\tensorrt\plugin\group_norm_op_plugin.cu (#66632)

Fripping · web-flow · commit 50121d2076bc · 2024-07-29T09:56:50.000+08:00
* update error message

* update error message

* Delete paddle/fluid/framework/.ipynb_checkpoints directory

* Delete test/cpp/eager/task_tests/.ipynb_checkpoints directory

* Delete paddle/fluid/inference/tensorrt/plugin/.ipynb_checkpoints directory

* Delete paddle/fluid/inference/tensorrt/convert/.ipynb_checkpoints directory

* Update forward_autograd_test.cc
diff --git a/paddle/fluid/framework/multi_trainer.cc b/paddle/fluid/framework/multi_trainer.cc
@@ -139,7 +139,13 @@ void MultiTrainer::InitTrainerEnv(const ProgramDesc& main_program,
   // multi thread load
   auto pool = GetThreadPool(thread_num_);
   std::vector<std::future<void>> wait_futures;
-  CHECK_EQ(static_cast<int>(pool.size()), thread_num_);
+  PADDLE_ENFORCE_EQ(
+      static_cast<int>(pool.size()),
+      thread_num_,
+      phi::errors::InvalidArgument("static_cast<int>(pool.size()) is invalid, "
+                                   "expected %d but received %d",
+                                   thread_num_,
+                                   static_cast<int>(pool.size())));
   for (int i = 0; i < thread_num_; ++i) {
     wait_futures.emplace_back(pool[i]->Run([this, i, &main_program, &place]() {
 #ifdef PADDLE_WITH_HETERPS
@@ -232,7 +238,13 @@ void MultiTrainer::Run() {
   VLOG(3) << "Going to run";
   auto pool = GetThreadPool(thread_num_);
   std::vector<std::future<void>> wait_futures;
-  CHECK_EQ(static_cast<int>(pool.size()), thread_num_);
+  PADDLE_ENFORCE_EQ(
+      static_cast<int>(pool.size()),
+      thread_num_,
+      phi::errors::InvalidArgument("static_cast<int>(pool.size()) is invalid, "
+                                   "expected %d but received %d",
+                                   thread_num_,
+                                   static_cast<int>(pool.size())));
   for (int i = 0; i < thread_num_; ++i) {
     if (!debug_) {
       wait_futures.emplace_back(
diff --git a/paddle/fluid/inference/tensorrt/convert/test_custom_plugin_creater.cc b/paddle/fluid/inference/tensorrt/convert/test_custom_plugin_creater.cc
@@ -111,7 +111,11 @@ TEST(CustomPluginCreater, StaticShapePlugin) {
   auto &custom_plugin_tell = OpTeller::Global().GetCustomPluginTeller();
 
   framework::OpDesc custom_op(*op_desc, nullptr);
-  CHECK_EQ((*custom_plugin_tell)(custom_op, false, false), true);
+  PADDLE_ENFORCE_EQ(
+      (*custom_plugin_tell)(custom_op, false, true),
+      true,
+      phi::errors::InvalidArgument(
+          "(*custom_plugin_tell)(custom_op, false, true) is False."));
 
   OpTeller::Global().SetOpConverterType(&custom_op,
                                         OpConverterType::CustomPluginCreater);
@@ -193,7 +197,11 @@ TEST(CustomPluginCreater, DynamicShapePlugin) {
   auto &custom_plugin_tell = OpTeller::Global().GetCustomPluginTeller();
 
   framework::OpDesc custom_op(*op_desc, nullptr);
-  CHECK_EQ((*custom_plugin_tell)(custom_op, false, true), true);
+  PADDLE_ENFORCE_EQ(
+      (*custom_plugin_tell)(custom_op, false, true),
+      true,
+      phi::errors::InvalidArgument(
+          "(*custom_plugin_tell)(custom_op, false, true) is False."));
 
   OpTeller::Global().SetOpConverterType(&custom_op,
                                         OpConverterType::CustomPluginCreater);
diff --git a/paddle/fluid/inference/tensorrt/plugin/group_norm_op_plugin.cu b/paddle/fluid/inference/tensorrt/plugin/group_norm_op_plugin.cu
@@ -783,8 +783,19 @@ int GroupNormPluginDynamic::enqueue(
       params_.invDHWC =
           1.F / static_cast<float>(params_.dhw * params_.cPerGroup);
       params_.groupsPerBlock = cPerBlock / params_.cPerGroup;
-      CHECK_EQ(cPerBlock % params_.cPerGroup, 0);
-      CHECK_EQ(params_.cPerGroup % 2, 0);
+      PADDLE_ENFORCE_EQ(cPerBlock % params_.cPerGroup,
+                        0,
+                        phi::errors::InvalidArgument(
+                            "cPerBlock should be multiple of params_.cPerGroup"
+                            "now cPerBlock is %d, params_.cPerGroup is %d",
+                            cPerBlock,
+                            params_.cPerGroup));
+      PADDLE_ENFORCE_EQ(
+          params_.cPerGroup % 2,
+          0,
+          phi::errors::InvalidArgument(
+              "params_.cPerGroup should be a even number, but received %d",
+              params_.cPerGroup));
       params_.eps = eps_;
       params_.dqScaleIn = input_desc[0].scale;
       params_.inv_qScale = 1.f / output_desc[0].scale;
