CMakeLists.txt 759 字节
Newer Older
Q
qijun 已提交
1
if(WITH_GPU)
C
caoying03 已提交
2
    nv_library(math_function SRCS math_function.cc math_function.cu im2col.cc
C
chengduoZH 已提交
3
      im2col.cu pooling.cc pooling.cu DEPS cblas device_context operator)
C
caoying03 已提交
4
    nv_library(softmax_function SRCS softmax.cc softmax.cu
C
caoying03 已提交
5
      DEPS operator)
6 7
    nv_library(cross_entropy_function SRCS cross_entropy.cc cross_entropy.cu
      DEPS operator)
Q
qijun 已提交
8
else()
C
chengduoZH 已提交
9
    cc_library(math_function SRCS math_function.cc im2col.cc pooling.cc
C
caoying03 已提交
10
      DEPS cblas device_context operator)
C
caoying03 已提交
11
    cc_library(softmax_function SRCS softmax.cc DEPS operator)
12
    cc_library(cross_entropy_function SRCS cross_entropy.cc DEPS operator)
Q
qijun 已提交
13
endif()
Q
qijun 已提交
14

Q
qijun 已提交
15
nv_test(math_function_test SRCS math_function_test.cc DEPS math_function tensor)
H
hedaoyuan 已提交
16
cc_test(im2col_test SRCS im2col_test.cc DEPS math_function tensor)