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

7
cc_library(feed_compute_host SRCS feed_compute.cc DEPS ${lite_kernel_deps})
8
cc_library(fetch_compute_host SRCS fetch_compute.cc DEPS ${lite_kernel_deps})
S
superjomn 已提交
9

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

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