diff --git a/paddle/operators/math/CMakeLists.txt b/paddle/operators/math/CMakeLists.txt index 575e89eed8049ffc1becf8ff76d03d1f15f93487..d32924db85e6c0554e719d61bfbc9e7eaf0a7a3e 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()