CMakeLists.txt 559 字节
Newer Older
S
update  
superjomn 已提交
1
message(STATUS "compile with lite host kernels")
S
superjomn 已提交
2

3
cc_library(feed_compute_host SRCS feed_compute.cc DEPS ${lite_kernel_deps})
4
cc_library(fetch_compute_host SRCS fetch_compute.cc DEPS ${lite_kernel_deps})
5 6 7
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)
S
superjomn 已提交
8

S
update  
superjomn 已提交
9
set(host_kernels
S
superjomn 已提交
10 11
    feed_compute_host
    fetch_compute_host
12
    reshape_compute_host
S
superjomn 已提交
13
    )
S
superjomn 已提交
14

Y
Yan Chunwei 已提交
15
set(host_kernels "${host_kernels}" CACHE GLOBAL "host kernels")
T
tensor-tang 已提交
16