fix code

PaddlePaddle · Jan 12, 2024 · 9d5998c · 9d5998c
1 parent 4db0c0a
commit 9d5998c
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 14 deletions.
diff --git a/paddle/fluid/primitive/base/decomp_trans.cc b/paddle/fluid/primitive/base/decomp_trans.cc
@@ -282,12 +282,11 @@ void DecompProgram::decomp_program() {
   for (size_t i = 0; i < src_vars_.size(); i++) {
     orig_vars_dict[src_vars_[i]] = static_cast<int>(i);
   }
-  if (VLOG_IS_ON(4)) {
-    std::ostringstream orig_prog_stream;
-    program_->Print(orig_prog_stream);
-    std::cout << "[Prim] Origin program before decomp :\n"
-              << orig_prog_stream.str();
-  }
+  std::ostringstream orig_prog_stream;
+  program_->Print(orig_prog_stream);
+  VLOG(4) << "[Prim] Origin program before decomp :\n"
+          << orig_prog_stream.str();
+
   if (!paddle::prim::PrimCommonUtils::IsFwdPrimEnabled()) {
     return;
   }
@@ -339,12 +338,9 @@ void DecompProgram::decomp_program() {
   }
   auto& builder = *(paddle::dialect::ApiBuilder::Instance().GetBuilder());
   builder.SetInsertionPointToBlockEnd(block);
-  if (VLOG_IS_ON(4)) {
-    std::ostringstream decomp_prog_stream;
-    program_->Print(decomp_prog_stream);
-    std::cout << "[Prim] New program after decomp :\n"
-              << decomp_prog_stream.str();
-  }
+  std::ostringstream decomp_prog_stream;
+  program_->Print(decomp_prog_stream);
+  VLOG(4) << "[Prim] New program after decomp :\n" << decomp_prog_stream.str();
   dst_vars_ = tar_vars;
   return;
 }

diff --git a/test/legacy_test/CMakeLists.txt b/test/legacy_test/CMakeLists.txt
@@ -995,8 +995,8 @@ else()
   set_tests_properties(test_conv3d_transpose_part2_op PROPERTIES TIMEOUT 120)
   set_tests_properties(test_conv3d_op PROPERTIES TIMEOUT 120)
   set_tests_properties(test_norm_op PROPERTIES TIMEOUT 150)
-  set_tests_properties(test_batch_norm_op_prim_nchw PROPERTIES TIMEOUT 300)
-  set_tests_properties(test_batch_norm_op_prim_nhwc PROPERTIES TIMEOUT 300)
+  set_tests_properties(test_batch_norm_op_prim_nchw PROPERTIES TIMEOUT 250)
+  set_tests_properties(test_batch_norm_op_prim_nhwc PROPERTIES TIMEOUT 250)
   set_tests_properties(test_layer_norm_op PROPERTIES TIMEOUT 250)
   set_tests_properties(test_pool3d_op PROPERTIES TIMEOUT 150)
 endif()