CMakeLists.txt 703 字节
Newer Older
1
if(NOT (LITE_WITH_LIGHT_WEIGHT_FRAMEWORK AND LITE_WITH_ARM))
2 3 4
    return()
endif()

S
update  
superjomn 已提交
5
message(STATUS "compile with lite ARM kernels")
6

T
tensor-tang 已提交
7
cc_library(fc_compute_arm SRCS fc_compute.cc DEPS ${lite_kernel_deps} math_arm)
8 9 10 11
cc_library(relu_compute_arm SRCS relu_compute.cc DEPS ${lite_kernel_deps})
cc_library(mul_compute_arm SRCS mul_compute.cc DEPS ${lite_kernel_deps} eigen3)
cc_library(scale_compute_arm SRCS scale_compute.cc DEPS ${lite_kernel_deps} eigen3)

12
lite_cc_test(test_fc_compute_arm SRCS fc_compute_test.cc DEPS fc_compute_arm math_arm)
T
tensor-tang 已提交
13

14 15 16 17
set(arm_kernels
    fc_compute_arm
    relu_compute_arm
    mul_compute_arm
18
    scale_compute_arm)
19 20

set(arm_kernels "${arm_kernels}" CACHE INTERNAL "arm kernels")