Refine the gpu code.

e967645c · hedaoyuan · f7be9cb9 · e967645c · e967645c
展开全部隐藏空白更改
内联并排

Showing with 236 addition and 219 deletion

paddle/operators/math/CMakeLists.txt paddle/operators/math/CMakeLists.txt +2 -1

paddle/operators/math/im2col.cu paddle/operators/math/im2col.cu +234 -218

未找到文件。
--- a/paddle/operators/math/CMakeLists.txt
+++ b/paddle/operators/math/CMakeLists.txt

 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