CMakeLists.txt 929 字节
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 8 9 10
if (WITH_CINN)
  cc_library(transform_desc SRCS transform_desc.cc DEPS proto_desc cinn)
  cc_test(test_transform_desc SRCS transform_desc_test.cc DEPS transform_desc)
endif()

11 12 13
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 已提交
14
cc_test(test_build_cinn_pass SRCS build_cinn_pass_test.cc DEPS build_cinn_pass)