CMakeLists.txt 1.1 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)
Y
Yan Chunwei 已提交
8 9 10 11 12

cc_library(analysis_passes SRCS passes.cc DEPS
  ir_graph_build_pass
  ir_analysis_pass
  ir_params_sync_among_devices_pass
Z
Zhaolong Xing 已提交
13
  adjust_cudnn_workspace_size_pass
Y
Yan Chunwei 已提交
14
  memory_optim_pass
15
  inference_op_replace_pass
Y
Yan Chunwei 已提交
16 17
  ir_graph_to_program_pass
)
18 19 20

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