disable eager compile in inference

xiaoguoguo626807 · Jan 4, 2023 · a60eb1f · a60eb1f
1 parent d4a22b4
commit a60eb1f
Show file tree

Hide file tree

Showing 2 changed files with 47 additions and 45 deletions.
diff --git a/paddle/fluid/eager/tests/performance_tests/CMakeLists.txt b/paddle/fluid/eager/tests/performance_tests/CMakeLists.txt
@@ -1,46 +1,48 @@
-cc_library(
-  performance_benchmark_utils
-  SRCS benchmark_utils.cc
-  DEPS ${eager_deps}
-       ${fluid_deps}
-       ${generated_deps}
-       eager_scale
-       scale_node
-       scale_op
-       matmul_v2_op
-       dygraph_function
-       eager_prim_api)
+if(NOT (NOT WITH_PYTHON AND ON_INFER))
+  cc_library(
+    performance_benchmark_utils
+    SRCS benchmark_utils.cc
+    DEPS ${eager_deps}
+         ${fluid_deps}
+         ${generated_deps}
+         eager_scale
+         scale_node
+         scale_op
+         matmul_v2_op
+         dygraph_function
+         eager_prim_api)
 
-cc_test_old(
-  test_egr_performance_benchmark_eager_cpu
-  SRCS
-  benchmark_eager_cpu.cc
-  DEPS
-  performance_benchmark_utils
-  ${eager_deps}
-  ${fluid_deps})
-cc_test_old(
-  test_egr_performance_benchmark_fluid_cpu
-  SRCS
-  benchmark_fluid_cpu.cc
-  DEPS
-  performance_benchmark_utils
-  ${eager_deps}
-  ${fluid_deps})
+  cc_test_old(
+    test_egr_performance_benchmark_eager_cpu
+    SRCS
+    benchmark_eager_cpu.cc
+    DEPS
+    performance_benchmark_utils
+    ${eager_deps}
+    ${fluid_deps})
+  cc_test_old(
+    test_egr_performance_benchmark_fluid_cpu
+    SRCS
+    benchmark_fluid_cpu.cc
+    DEPS
+    performance_benchmark_utils
+    ${eager_deps}
+    ${fluid_deps})
 
-cc_test_old(
-  test_egr_performance_benchmark_eager_cuda
-  SRCS
-  benchmark_eager_cuda.cc
-  DEPS
-  performance_benchmark_utils
-  ${eager_deps}
-  ${fluid_deps})
-cc_test_old(
-  test_egr_performance_benchmark_fluid_cuda
-  SRCS
-  benchmark_fluid_cuda.cc
-  DEPS
-  performance_benchmark_utils
-  ${eager_deps}
-  ${fluid_deps})
+  cc_test_old(
+    test_egr_performance_benchmark_eager_cuda
+    SRCS
+    benchmark_eager_cuda.cc
+    DEPS
+    performance_benchmark_utils
+    ${eager_deps}
+    ${fluid_deps})
+  cc_test_old(
+    test_egr_performance_benchmark_fluid_cuda
+    SRCS
+    benchmark_fluid_cuda.cc
+    DEPS
+    performance_benchmark_utils
+    ${eager_deps}
+    ${fluid_deps})
+endif()
diff --git a/paddle/fluid/prim/api/CMakeLists.txt b/paddle/fluid/prim/api/CMakeLists.txt
@@ -9,5 +9,5 @@ else()
   cc_library(
     prim_api
     SRCS all.cc
-    DEPS static_utils static_prim_api eager_prim_api)
+    DEPS static_utils static_prim_api)
 endif()