CMakeLists.txt 993 字节
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)
Y
Yan Chunwei 已提交
7

C
Chunwei 已提交
8
cc_library(analysis_passes SRCS use_passes.cc DEPS
Y
Yan Chunwei 已提交
9 10 11
  ir_graph_build_pass
  ir_analysis_pass
  ir_params_sync_among_devices_pass
Z
Zhaolong Xing 已提交
12
  adjust_cudnn_workspace_size_pass
Y
Yan Chunwei 已提交
13 14 15
  memory_optim_pass
  ir_graph_to_program_pass
)
16 17 18

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