diff --git a/test/cpp/ir/shape_dialect/CMakeLists.txt b/test/cpp/ir/shape_dialect/CMakeLists.txt
index 7b959a283b20e5aa472e37b249df7e3e5f6b8a6b..ae3e3d63d52bd2167c6f43433ac76ff1f4e26981 100644
--- a/test/cpp/ir/shape_dialect/CMakeLists.txt
+++ b/test/cpp/ir/shape_dialect/CMakeLists.txt
@@ -1 +1,8 @@
-cc_test_old(assist_struct_test SRCS assist_struct_test.cc DEPS ir gtest)
+cc_test_old(
+  symbolic_op_test
+  SRCS
+  symbolic_op_test.cc
+  DEPS
+  pd_dialect
+  ir
+  gtest)
diff --git a/test/cpp/ir/shape_dialect/assist_struct_test.cc b/test/cpp/ir/shape_dialect/symbolic_op_test.cc
similarity index 97%
rename from test/cpp/ir/shape_dialect/assist_struct_test.cc
rename to test/cpp/ir/shape_dialect/symbolic_op_test.cc
index 2d5216104f5897a980cd6336d7cf429f2cfe138d..5dfcf19c22f34df240c309660c0ed0c5968e4449 100644
--- a/test/cpp/ir/shape_dialect/assist_struct_test.cc
+++ b/test/cpp/ir/shape_dialect/symbolic_op_test.cc
@@ -14,6 +14,7 @@
 
 #include <gtest/gtest.h>
 #include <map>
+#include "paddle/fluid/ir/dialect/paddle_dialect/ir/pd_dialect.h"
 #include "paddle/fluid/ir/dialect/paddle_dialect/ir/pd_type.h"
 #include "paddle/ir/core/block.h"
 #include "paddle/ir/core/builder.h"
@@ -97,6 +98,7 @@ TEST(assist_struct_test, symbolic_dim_mgr) {
   ir::IrContext *ctx = ir::IrContext::Instance();
   ir::Program program(ctx);
   ctx->GetOrRegisterDialect<ir::dialect::ShapeDialect>();
+  ctx->GetOrRegisterDialect<paddle::dialect::PaddleDialect>();
 
   ir::SymbolicDimMgr symDimMgr(program.module_op());
   ir::dialect::SymbolicDim symDimS0 = symDimMgr.newSymbolicDim();