CMakeLists.txt 1.3 KB
Newer Older
1
cc_library(
Y
Yan Chunwei 已提交
2
  ir_graph_build_pass
3 4 5
  SRCS ir_graph_build_pass.cc
  DEPS analysis_pass argument ir_pass_manager)
cc_library(
Y
Yan Chunwei 已提交
6
  ir_analysis_pass
7 8 9 10 11 12
  SRCS ir_analysis_pass.cc
  DEPS analysis_pass argument ir_pass_manager)
cc_library(
  memory_optim_pass
  SRCS memory_optimize_pass.cc
  DEPS analysis_pass zero_copy_tensor)
13 14 15
cc_library(
  convert_to_mixed_precision
  SRCS convert_to_mixed_precision.cc
16
  DEPS analysis_pass ir_graph_build_pass auto_mixed_precision_pass)
17
cc_library(
Y
Yan Chunwei 已提交
18
  ir_params_sync_among_devices_pass
19 20 21 22 23 24 25
  SRCS ir_params_sync_among_devices_pass.cc
  DEPS analysis_pass argument ir_pass_manager)
cc_library(
  ir_graph_to_program_pass
  SRCS ir_graph_to_program_pass.cc
  DEPS analysis_pass graph_to_program_pass)
cc_library(
Z
Zhaolong Xing 已提交
26
  adjust_cudnn_workspace_size_pass
27 28 29
  SRCS adjust_cudnn_workspace_size_pass.cc
  DEPS analysis_pass graph_to_program_pass)
cc_library(
30
  inference_op_replace_pass
31 32 33 34 35 36 37 38 39 40 41
  SRCS inference_op_replace_pass.cc
  DEPS analysis_pass graph_to_program_pass)

cc_library(
  analysis_passes
  SRCS passes.cc
  DEPS ir_graph_build_pass
       ir_analysis_pass
       ir_params_sync_among_devices_pass
       adjust_cudnn_workspace_size_pass
       memory_optim_pass
42
       convert_to_mixed_precision
43
       inference_op_replace_pass
44
       ir_graph_to_program_pass)
45

46 47 48
set(analysis_deps
    ${analysis_deps} analysis_passes subgraph_detector
    CACHE INTERNAL "")