Refine the gpu code.

8 years ago · e967645c15
parent f7be9cb97a
commit e967645c15
2 changed files with 236 additions and 219 deletions
--- a/paddle/operators/math/CMakeLists.txt
+++ b/paddle/operators/math/CMakeLists.txt
@ -1,6 +1,7 @@

 if(WITH_GPU)
-    nv_library(math_function SRCS math_function.cc math_function.cu im2col.cc DEPS cblas device_context)
+    nv_library(math_function SRCS math_function.cc math_function.cu im2col.cc 
+    im2col.cu DEPS cblas device_context)
 else()
    cc_library(math_function SRCS math_function.cc im2col.cc DEPS cblas device_context)
 endif()
--- a/paddle/operators/math/im2col.cu
+++ b/paddle/operators/math/im2col.cu