[DRR] replace cc_test_old with cc_test (#58268)

gongshaotian · zyfncg · web-flow · commit a7ea724c0319 · 2023-10-25T16:52:04.000+08:00
* replace cc_test_old with cc_test_old

* set source_source_files_properties

* set pattern_rewrite_test use cc_test_old

* replace cc_test_old by paddle_test for drr tests

* fix bug

* edit patternRewrite

* delete note

---------

Co-authored-by: zyfncg &lt;zhangyunfei07@baidu.com&gt;
diff --git a/paddle/fluid/framework/executor_cache.cc b/paddle/fluid/framework/executor_cache.cc
@@ -543,6 +543,9 @@ std::unique_ptr<::pir::Program> ConstructBackwardIrProgram(
   if (FLAGS_new_ir_apply_inplace_pass) {
     ::pir::PassManager pm(::pir::IrContext::Instance(), 3);
     pm.AddPass(::pir::CreateInplacePass());
+    if (VLOG_IS_ON(6)) {
+      pm.EnableIRPrinting();
+    }
     pm.Run(res.get());
   }
 
diff --git a/paddle/fluid/pybind/pir.cc b/paddle/fluid/pybind/pir.cc
@@ -67,6 +67,8 @@ using pir::Value;
 using pybind11::return_value_policy;
 
 USE_PASS(dead_code_elimination_pass);
+USE_PASS(attention_fuse_pass);
+USE_PASS(fused_gemm_epilogue_pass);
 USE_PASS(fused_dropout_add_pass);
 USE_PASS(inplace_pass);
 
diff --git a/test/cpp/pir/pattern_rewrite/CMakeLists.txt b/test/cpp/pir/pattern_rewrite/CMakeLists.txt
@@ -10,45 +10,25 @@ endif()
 cc_test_old(pattern_rewrite_test SRCS pattern_rewrite_test.cc DEPS
             ${PATTERN_REWRITE_TEST_DEPS})
 
-cc_test_old(
+cc_test(
   drr_test
-  SRCS
-  drr_test.cc
-  DEPS
-  drr
-  gtest
-  pd_op_dialect
-  pir)
-cc_test_old(
+  SRCS drr_test.cc
+  DEPS drr)
+
+cc_test(
   drr_same_type_binding_test
-  SRCS
-  drr_same_type_binding_test.cc
-  DEPS
-  drr
-  gtest
-  pd_op_dialect
-  pir)
+  SRCS drr_same_type_binding_test.cc
+  DEPS drr gtest pd_op_dialect pir)
 
-cc_test_old(
+cc_test(
   drr_fuse_linear_test
-  SRCS
-  drr_fuse_linear_test.cc
-  DEPS
-  fusion_passes
-  drr
-  gtest
-  pd_op_dialect
-  pir)
-cc_test_old(
+  SRCS drr_fuse_linear_test.cc
+  DEPS fusion_passes drr gtest pd_op_dialect pir)
+
+cc_test(
   drr_attention_fuse_test
-  SRCS
-  drr_attention_fuse_test.cc
-  DEPS
-  fusion_passes
-  drr
-  gtest
-  pd_op_dialect
-  pir)
+  SRCS drr_attention_fuse_test.cc
+  DEPS fusion_passes drr gtest pd_op_dialect pir)
 
 set_tests_properties(
   pattern_rewrite_test PROPERTIES ENVIRONMENT

Original file line number	Diff line number	Diff line change
`@@ -543,6 +543,9 @@ std::unique_ptr<::pir::Program> ConstructBackwardIrProgram(`
`543`	`543`	`if (FLAGS_new_ir_apply_inplace_pass) {`
`544`	`544`	`::pir::PassManager pm(::pir::IrContext::Instance(), 3);`
`545`	`545`	`pm.AddPass(::pir::CreateInplacePass());`
	`546`	`+ if (VLOG_IS_ON(6)) {`
	`547`	`+ pm.EnableIRPrinting();`
	`548`	`+ }`
`546`	`549`	`pm.Run(res.get());`
`547`	`550`	`}`
`548`	`551`