CMakeLists.txt 1.1 KB
Newer Older
Q
qijun 已提交
1
if(WITH_GPU)
C
chengduoZH 已提交
2
    nv_library(math_function SRCS math_function.cc math_function.cu im2col.cc im2col.cu DEPS cblas device_context operator)
3
    nv_test(math_function_test SRCS math_function_test.cc DEPS math_function tensor)
4
    nv_library(softmax SRCS softmax.cc softmax.cu DEPS operator)
5
    nv_library(cross_entropy SRCS cross_entropy.cc cross_entropy.cu DEPS operator)
C
chengduoZH 已提交
6
    nv_library(pooling SRCS pooling.cc pooling.cu DEPS device_context)
C
chengduoZH 已提交
7
    nv_library(vol2col SRCS vol2col.cc vol2col.cu DEPS device_context)
Q
qijun 已提交
8
else()
C
chengduoZH 已提交
9
    cc_library(math_function SRCS math_function.cc im2col.cc DEPS cblas device_context operator)
10
    cc_test(math_function_test SRCS math_function_test.cc DEPS math_function tensor)
11 12
    cc_library(softmax SRCS softmax.cc DEPS operator)
    cc_library(cross_entropy SRCS cross_entropy.cc DEPS operator)
C
chengduoZH 已提交
13
    cc_library(pooling SRCS pooling.cc DEPS device_context)
C
chengduoZH 已提交
14
    cc_library(vol2col SRCS vol2col.cc DEPS device_context)
Q
qijun 已提交
15
endif()
Q
qijun 已提交
16

H
hedaoyuan 已提交
17
cc_test(im2col_test SRCS im2col_test.cc DEPS math_function tensor)
C
chengduoZH 已提交
18
cc_test(vol2col_test SRCS vol2col_test.cc DEPS vol2col tensor)