CMakeLists.txt 1.9 KB
Newer Older
1
cc_library(cinn_cache_key SRCS cinn_cache_key.cc DEPS boost graph graph_helper lod_tensor proto_desc)
2
cc_library(build_cinn_pass SRCS build_cinn_pass.cc DEPS pass subgraph_detector graph_pattern_detector cinn_compiler errors enforce)
3
cc_library(transform_desc SRCS transform_desc.cc DEPS proto_desc cinn)
4
cc_library(transform_type SRCS transform_type.cc DEPS errors enforce cinn)
5
cc_library(cinn_graph_symbolization SRCS cinn_graph_symbolization.cc DEPS lod_tensor graph transform_desc cinn)
6
cc_library(cinn_compiler SRCS cinn_compiler.cc DEPS framework_proto graph lod_tensor cinn_cache_key cinn_graph_symbolization cinn cinn_launch_context)
7

H
Huihuang Zheng 已提交
8 9 10
if (WITH_TESTING)
  cc_test(cinn_lib_test SRCS cinn_lib_test.cc DEPS cinn)
  set_tests_properties(cinn_lib_test PROPERTIES LABELS "RUN_TYPE=CINN")
11

H
Huihuang Zheng 已提交
12 13
  cc_test(cinn_cache_key_test SRCS cinn_cache_key_test.cc DEPS cinn_cache_key)
  set_tests_properties(cinn_cache_key_test PROPERTIES LABELS "RUN_TYPE=CINN")
14

15
  cc_test(build_cinn_pass_test SRCS build_cinn_pass_test.cc DEPS build_cinn_pass cinn_compiler op_registry mul_op activation_op elementwise_add_op)
H
Huihuang Zheng 已提交
16
  set_tests_properties(build_cinn_pass_test PROPERTIES LABELS "RUN_TYPE=CINN")
17

H
Huihuang Zheng 已提交
18 19
  cc_test(transform_desc_test SRCS transform_desc_test.cc DEPS transform_desc)
  set_tests_properties(transform_desc_test PROPERTIES LABELS "RUN_TYPE=CINN")
20

21 22 23
  cc_test(transform_type_test SRCS transform_type_test.cc DEPS transform_type)
  set_tests_properties(transform_type_test PROPERTIES LABELS "RUN_TYPE=CINN")

H
Huihuang Zheng 已提交
24 25
  cc_test(cinn_graph_symbolization_test SRCS cinn_graph_symbolization_test.cc DEPS cinn_graph_symbolization)
  set_tests_properties(cinn_graph_symbolization_test PROPERTIES LABELS "RUN_TYPE=CINN")
26

27
  cc_test(cinn_compiler_test SRCS cinn_compiler_test.cc DEPS cinn_compiler place proto_desc graph_viz_pass build_cinn_pass cinn mul_op activation_op elementwise_add_op)
H
Huihuang Zheng 已提交
28 29
  set_tests_properties(cinn_compiler_test PROPERTIES LABELS "RUN_TYPE=CINN")
endif()