CMakeLists.txt 1.3 KB
Newer Older
1 2
cc_library(ir_graph_build_pass SRCS ir_graph_build_pass.cc DEPS analysis_pass argument ir_pass_manager)
cc_library(ir_analysis_pass SRCS ir_analysis_pass.cc DEPS analysis_pass argument ir_pass_manager)
3
cc_library(memory_optim_pass SRCS memory_optimize_pass.cc DEPS analysis_pass zero_copy_tensor)
N
nhzlx 已提交
4
cc_library(ir_params_sync_among_devices_pass SRCS ir_params_sync_among_devices_pass.cc DEPS analysis_pass argument ir_pass_manager)
Y
Yan Chunwei 已提交
5
cc_library(ir_graph_to_program_pass SRCS ir_graph_to_program_pass.cc DEPS analysis_pass graph_to_program_pass)
Z
Zhaolong Xing 已提交
6
cc_library(adjust_cudnn_workspace_size_pass SRCS adjust_cudnn_workspace_size_pass.cc DEPS analysis_pass graph_to_program_pass)
7
cc_library(inference_op_replace_pass SRCS inference_op_replace_pass.cc DEPS analysis_pass graph_to_program_pass)
8 9 10 11 12
IF(WITH_TESTING)
  cc_library(ir_graph_clean_pass SRCS ir_graph_clean_pass.cc DEPS analysis_pass gtest)
ELSE()
  cc_library(ir_graph_clean_pass SRCS ir_graph_clean_pass.cc DEPS analysis_pass)
ENDIF()
Y
Yan Chunwei 已提交
13 14 15 16 17

cc_library(analysis_passes SRCS passes.cc DEPS
  ir_graph_build_pass
  ir_analysis_pass
  ir_params_sync_among_devices_pass
Z
Zhaolong Xing 已提交
18
  adjust_cudnn_workspace_size_pass
Y
Yan Chunwei 已提交
19
  memory_optim_pass
20
  inference_op_replace_pass
Y
Yan Chunwei 已提交
21
  ir_graph_to_program_pass
22
  ir_graph_clean_pass
Y
Yan Chunwei 已提交
23
)
24 25 26

set(analysis_deps ${analysis_deps}
        analysis_passes
T
tensor-tang 已提交
27
        subgraph_detector
28
        CACHE INTERNAL "")