CMakeLists.txt 840 字节
Newer Older
S
update  
superjomn 已提交
1
message(STATUS "compile with lite host kernels")
2 3 4 5 6
cc_library(fc_compute_host SRCS fc_compute.cc DEPS ${lite_kernel_deps})
cc_library(relu_compute_host SRCS relu_compute.cc DEPS ${lite_kernel_deps})
cc_library(mul_compute_host SRCS mul_compute.cc DEPS ${lite_kernel_deps})
cc_library(scale_compute_host SRCS scale_compute.cc DEPS ${lite_kernel_deps})
cc_library(feed_compute_host SRCS feed_compute.cc DEPS ${lite_kernel_deps})
7
cc_library(fetch_compute_host SRCS fetch_compute.cc DEPS ${lite_kernel_deps})
S
superjomn 已提交
8

S
update  
superjomn 已提交
9
set(host_kernels
10 11
        feed_compute_host
        fetch_compute_host
S
superjomn 已提交
12
        fc_compute_host
S
superjomn 已提交
13 14 15 16
        relu_compute_host
        mul_compute_host
        scale_compute_host
        )
S
superjomn 已提交
17

S
update  
superjomn 已提交
18 19 20
set(host_kernels "${host_kernels}" CACHE INTERNAL "host kernels")

lite_cc_test(test_fc_compute SRCS fc_compute_test.cc DEPS fc_compute_host fc_op_lite)