CMakeLists.txt 1.3 KB
Newer Older
Y
Yan Chunwei 已提交
1 2 3 4
if(NOT LITE_WITH_X86)
    return()
endif()

L
liuwei1031 已提交
5 6 7 8 9
cc_library(activation_compute_x86 SRCS activation_compute.cc DEPS ${lite_kernel_deps} activation_op)
cc_library(mean_compute_x86 SRCS mean_compute.cc DEPS ${lite_kernel_deps})
cc_library(fill_constant_compute_x86 SRCS fill_constant_compute.cc DEPS ${lite_kernel_deps})
cc_library(sgd_compute_x86 SRCS sgd_compute.cc DEPS ${lite_kernel_deps})

10 11 12 13 14 15 16 17
cc_library(fc_compute_x86 SRCS fc_compute.cc DEPS ${lite_kernel_deps})
cc_library(mul_compute_x86 SRCS mul_compute.cc DEPS ${lite_kernel_deps})
cc_library(relu_compute_x86 SRCS relu_compute.cc DEPS ${lite_kernel_deps})
cc_library(scale_compute_x86 SRCS scale_compute.cc DEPS ${lite_kernel_deps})
cc_library(elementwise_compute_x86 SRCS elementwise_compute.cc DEPS ${lite_kernel_deps} elementwise_sub_op elementwise_add_op)
cc_library(softmax_compute_x86 SRCS softmax_compute.cc DEPS ${lite_kernel_deps} softmax)
cc_library(dropout_compute_x86 SRCS dropout_compute.cc DEPS ${lite_kernel_deps} )

L
liuwei1031 已提交
18 19 20 21 22 23
set(x86_kernels
    activation_compute_x86
    elementwise_compute_x86
    mean_compute_x86
    fill_constant_compute_x86
    mul_compute_x86
24 25 26 27 28
    relu_compute_x86
    fc_compute_x86
    scale_compute_x86
    softmax_compute_x86 
    dropout_compute_x86
L
liuwei1031 已提交
29 30 31
    )

set(x86_kernels "${x86_kernels}" CACHE INTERNAL "x86 kernels")