CMakeLists.txt 618 字节
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 7 8 9 10 11 12 13 14 15 16 17 18 19

cc_library(fc_compute_arm SRCS fc_compute.cc DEPS ${lite_kernel_deps} eigen3)
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)

set(arm_kernels
    fc_compute_arm
    relu_compute_arm
    mul_compute_arm
    scale_compute_arm
    )

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