From 1d41a6d45a917010f21088d1917040565649f2a5 Mon Sep 17 00:00:00 2001 From: chengduoZH Date: Wed, 11 Oct 2017 10:11:45 +0800 Subject: [PATCH] update paddle/operators/math/CMakeLists.txt --- paddle/operators/math/CMakeLists.txt | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/paddle/operators/math/CMakeLists.txt b/paddle/operators/math/CMakeLists.txt index 575e89eed80..d32924db85e 100644 --- a/paddle/operators/math/CMakeLists.txt +++ b/paddle/operators/math/CMakeLists.txt @@ -3,13 +3,13 @@ if(WITH_GPU) nv_test(math_function_test SRCS math_function_test.cc DEPS math_function tensor) nv_library(softmax SRCS softmax.cc softmax.cu DEPS operator) nv_library(cross_entropy SRCS cross_entropy.cc cross_entropy.cu DEPS operator) - nv_library(vol2col SRCS vol2col.cc vol2col.cu DEPS cblas device_context operator) + nv_library(vol2col SRCS vol2col.cc vol2col.cu DEPS device_context operator) else() cc_library(math_function SRCS math_function.cc im2col.cc pooling.cc DEPS cblas device_context operator) cc_test(math_function_test SRCS math_function_test.cc DEPS math_function tensor) cc_library(softmax SRCS softmax.cc DEPS operator) cc_library(cross_entropy SRCS cross_entropy.cc DEPS operator) - cc_library(vol2col SRCS vol2col.cc DEPS cblas device_context operator) + cc_library(vol2col SRCS vol2col.cc DEPS device_context operator) endif() -- GitLab