[CINN] Refine cmake for pass in cinn (#60683)

* refine cmake for pass in cinn * add dependency in cmake * add dependency in cmake
PaddlePaddle · Jan 10, 2024 · 2d9d46a · 2d9d46a
1 parent a2c1580
commit 2d9d46a
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 67 deletions.
diff --git a/paddle/cinn/hlir/dialect/operator/transforms/CMakeLists.txt b/paddle/cinn/hlir/dialect/operator/transforms/CMakeLists.txt
@@ -1,50 +1,19 @@
-add_subdirectory(group_merge)
-
 if(NOT CINN_ONLY)
 
-  cinn_cc_library(
-    pd_to_cinn_pass
-    SRCS
-    pd_to_cinn_pass.cc
-    DEPS
-    drr
-    cinn_op_dialect
-    op_dialect_vjp)
-
-  cinn_cc_library(
-    add_broadcast_to_elementwise_pass
-    SRCS
-    add_broadcast_to_elementwise_pass.cc
-    DEPS
-    pir
-    cinn_op_dialect
-    op_dialect_vjp)
-
-  cinn_cc_library(
-    fuse_shape_ops_into_generate_shape_op_pass
-    SRCS
-    fuse_shape_ops_into_generate_shape_op_pass.cc
-    DEPS
-    pir
-    cinn_op_dialect
-    op_dialect_vjp)
+  file(GLOB_RECURSE cinn_transforms_srcs "*.cc")
 
-  cinn_cc_library(
-    fully_insert_broadcast_pass
-    SRCS
-    fully_insert_broadcast_pass.cc
-    DEPS
-    pir
-    cinn_op_dialect
-    op_dialect_vjp)
+  set(cinn_transforms_deps
+      pir
+      drr
+      op_dialect
+      cinn_op_dialect
+      op_dialect_vjp
+      cinn_runtime_dialect
+      pir_compiler)
 
-  cinn_cc_library(
-    split_generate_shape_into_shape_ops_pass
-    SRCS
-    split_generate_shape_into_shape_ops_pass.cc
-    DEPS
-    pir
-    cinn_op_dialect
-    op_dialect_vjp)
+  cc_library(
+    cinn_transforms
+    SRCS ${cinn_transforms_srcs}
+    DEPS ${cinn_transforms_deps})
 
 endif()
diff --git a/paddle/cinn/hlir/dialect/operator/transforms/group_merge/CMakeLists.txt b/paddle/cinn/hlir/dialect/operator/transforms/group_merge/CMakeLists.txt
diff --git a/paddle/fluid/pybind/CMakeLists.txt b/paddle/fluid/pybind/CMakeLists.txt
@@ -50,16 +50,8 @@ set(PYBIND_DEPS
     auto_parallel)
 
 if(WITH_CINN)
-  set(PYBIND_DEPS
-      ${PYBIND_DEPS}
-      pir_transforms
-      op_with_group_merge_pass
-      add_broadcast_to_elementwise_pass
-      pd_to_cinn_pass
-      sub_graph_checker
-      fully_insert_broadcast_pass
-      fuse_shape_ops_into_generate_shape_op_pass
-      split_generate_shape_into_shape_ops_pass)
+  set(PYBIND_DEPS ${PYBIND_DEPS} pir_transforms cinn_transforms
+                  sub_graph_checker)
 endif()
 
 if(WITH_PSCORE)