CMakeLists.txt 2.4 KB
Newer Older
1 2 3 4
pass_library(
  build_cinn_pass
  base
  DEPS
S
sunli 已提交
5
  cinn_subgraph_detector
6 7 8 9 10
  subgraph_detector
  cinn_compiler
  errors
  enforce)

11 12 13
cc_library(
  cinn_cache_key
  SRCS cinn_cache_key.cc
R
Ruibiao Chen 已提交
14
  DEPS graph graph_helper lod_tensor proto_desc)
S
sunli 已提交
15 16 17 18
cc_library(
  cinn_subgraph_detector
  SRCS cinn_subgraph_detector.cc
  DEPS graph graph_helper subgraph_detector lod_tensor proto_desc)
19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40
cc_library(
  transform_desc
  SRCS transform_desc.cc
  DEPS proto_desc cinn)
cc_library(
  transform_type
  SRCS transform_type.cc
  DEPS errors enforce cinn)
cc_library(
  cinn_graph_symbolization
  SRCS cinn_graph_symbolization.cc
  DEPS lod_tensor graph transform_desc cinn)
cc_library(
  cinn_compiler
  SRCS cinn_compiler.cc
  DEPS framework_proto
       graph
       lod_tensor
       cinn_cache_key
       cinn_graph_symbolization
       cinn
       cinn_launch_context)
41

42
if(WITH_TESTING)
43
  cc_test_old(cinn_lib_test SRCS cinn_lib_test.cc DEPS cinn)
H
Huihuang Zheng 已提交
44
  set_tests_properties(cinn_lib_test PROPERTIES LABELS "RUN_TYPE=CINN")
45

46 47
  cc_test_old(cinn_cache_key_test SRCS cinn_cache_key_test.cc DEPS
              cinn_cache_key)
H
Huihuang Zheng 已提交
48
  set_tests_properties(cinn_cache_key_test PROPERTIES LABELS "RUN_TYPE=CINN")
49

50
  cc_test_old(
51
    build_cinn_pass_test
52 53 54 55 56 57 58 59
    SRCS
    build_cinn_pass_test.cc
    DEPS
    build_cinn_pass
    cinn_compiler
    op_registry
    mul_op
    activation_op
60 61
    elementwise_add_op
    generated_op)
H
Huihuang Zheng 已提交
62
  set_tests_properties(build_cinn_pass_test PROPERTIES LABELS "RUN_TYPE=CINN")
63

64 65
  cc_test_old(transform_desc_test SRCS transform_desc_test.cc DEPS
              transform_desc)
H
Huihuang Zheng 已提交
66
  set_tests_properties(transform_desc_test PROPERTIES LABELS "RUN_TYPE=CINN")
67

68 69
  cc_test_old(transform_type_test SRCS transform_type_test.cc DEPS
              transform_type)
70 71
  set_tests_properties(transform_type_test PROPERTIES LABELS "RUN_TYPE=CINN")

72 73
  cc_test_old(cinn_graph_symbolization_test SRCS
              cinn_graph_symbolization_test.cc DEPS cinn_graph_symbolization)
74 75
  set_tests_properties(cinn_graph_symbolization_test PROPERTIES LABELS
                                                                "RUN_TYPE=CINN")
76

77
  cc_test_old(
78
    cinn_compiler_test
79 80 81 82 83 84 85 86 87 88 89
    SRCS
    cinn_compiler_test.cc
    DEPS
    cinn_compiler
    place
    proto_desc
    graph_viz_pass
    build_cinn_pass
    cinn
    mul_op
    activation_op
90 91
    elementwise_add_op
    generated_op)
H
Huihuang Zheng 已提交
92 93
  set_tests_properties(cinn_compiler_test PROPERTIES LABELS "RUN_TYPE=CINN")
endif()