CMakeLists.txt 561 字节
Newer Older
T
tensor-tang 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16
message(STATUS "compile with lite host kernels")

cc_library(feed_compute_host SRCS feed_compute.cc DEPS ${lite_kernel_deps})
cc_library(fetch_compute_host SRCS fetch_compute.cc DEPS ${lite_kernel_deps})
cc_library(reshape_compute_host SRCS reshape_compute.cc DEPS ${lite_kernel_deps} reshape_op_lite)

lite_cc_test(test_reshape_compute_host SRCS reshape_compute_test.cc DEPS reshape_compute_host)

set(host_kernels
    feed_compute_host
    fetch_compute_host
    reshape_compute_host
    )

set(host_kernels "${host_kernels}" CACHE INTERNAL "host kernels")