diff --git a/paddle/fluid/pir/drr/CMakeLists.txt b/paddle/fluid/pir/drr/CMakeLists.txt
index c1b524dda69a6..0e27d7f45886a 100644
--- a/paddle/fluid/pir/drr/CMakeLists.txt
+++ b/paddle/fluid/pir/drr/CMakeLists.txt
@@ -62,4 +62,4 @@ add_custom_command(
 cc_library(
   drr
   SRCS ${DRR_SRCS} ${op_creator_file}
-  DEPS pd_op_dialect pir)
+  DEPS pd_op_dialect pir_pattern_rewrite pir_pass)
diff --git a/test/cpp/pir/pattern_rewrite/CMakeLists.txt b/test/cpp/pir/pattern_rewrite/CMakeLists.txt
index 2e507c7f20705..b050881bc296c 100644
--- a/test/cpp/pir/pattern_rewrite/CMakeLists.txt
+++ b/test/cpp/pir/pattern_rewrite/CMakeLists.txt
@@ -9,12 +9,12 @@ endif()
 cc_test_old(pattern_rewrite_test SRCS pattern_rewrite_test.cc DEPS
             ${PATTERN_REWRITE_TEST_DEPS})
 
-cc_test_old(drr_test SRCS drr_test.cc DEPS gtest drr)
-cc_test_old(drr_fuse_linear_test SRCS drr_fuse_linear_test.cc DEPS gtest drr)
-cc_test_old(drr_same_type_binding_test SRCS drr_same_type_binding_test.cc DEPS
-            gtest drr)
+paddle_test(drr_test SRCS drr_test.cc DEPS drr)
+paddle_test(drr_fuse_linear_test SRCS drr_fuse_linear_test.cc DEPS drr)
+paddle_test(drr_same_type_binding_test SRCS drr_same_type_binding_test.cc DEPS
+            drr)
 
-cc_test_old(drr_attention_fuse_test SRCS drr_attention_fuse_test.cc DEPS
+paddle_test(drr_attention_fuse_test SRCS drr_attention_fuse_test.cc DEPS
             ${PATTERN_REWRITE_TEST_DEPS} drr)
 
 set_tests_properties(