CMakeLists.txt 1.1 KB
Newer Older
1 2 3
cc_library(cinn_cache_key SRCS cinn_cache_key.cc DEPS boost graph graph_helper lod_tensor proto_desc)
cc_library(cinn_compiled_object SRCS cinn_compiled_object.cc DEPS feed_fetch_method graph lod_tensor proto_desc)
cc_library(cinn_runner SRCS cinn_runner.cc DEPS cinn_cache_key cinn_compiled_object feed_fetch_method graph lod_tensor scope)
J
jiangcheng 已提交
4
cc_library(build_cinn_pass SRCS build_cinn_pass.cc DEPS pass subgraph_detector)
5

6 7
if (WITH_CINN)
  cc_library(transform_desc SRCS transform_desc.cc DEPS proto_desc cinn)
J
jiangcheng 已提交
8 9
  cc_library(cinn_graph_symbolization SRCS cinn_graph_symbolization.cc DEPS lod_tensor graph graph_helper transform_desc cinn)

10
  cc_test(test_transform_desc SRCS transform_desc_test.cc DEPS transform_desc)
J
jiangcheng 已提交
11
  cc_test(test_cinn_graph_symbolization SRCS cinn_graph_symbolization_test.cc DEPS cinn_graph_symbolization)
12 13
endif()

14 15 16
cc_test(cinn_cache_key_test SRCS cinn_cache_key_test.cc DEPS cinn_cache_key)
cc_test(cinn_runner_test SRCS cinn_runner_test.cc DEPS cinn_runner proto_desc)
cc_test(cinn_compiled_object_test SRCS cinn_compiled_object_test.cc DEPS cinn_compiled_object)
J
jiangcheng 已提交
17
cc_test(test_build_cinn_pass SRCS build_cinn_pass_test.cc DEPS build_cinn_pass)