CMakeLists.txt 553 字节
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
caoying03 已提交
3
      im2col.cu DEPS cblas device_context operator)
C
caoying03 已提交
4
    nv_library(softmax_function SRCS softmax.cc softmax.cu
C
caoying03 已提交
5
      DEPS operator)
Q
qijun 已提交
6
else()
C
caoying03 已提交
7
    cc_library(math_function SRCS math_function.cc im2col.cc
C
caoying03 已提交
8
      DEPS cblas device_context operator)
C
caoying03 已提交
9
    cc_library(softmax_function SRCS softmax.cc DEPS operator)
Q
qijun 已提交
10
endif()
Q
qijun 已提交
11

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