Merge pull request PaddlePaddle#29 from carlushuang/0.15.0_1b7e38

fix enforce workspace size, the limit should bigger than user request
sabreshao · Sep 25, 2018 · 61a3a75 · 61a3a75
2 parents bdd96b7 + bf87e64
commit 61a3a75
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddle/fluid/operators/conv_cudnn_op.cu.cc b/paddle/fluid/operators/conv_cudnn_op.cu.cc
@@ -123,7 +123,7 @@ class CUDNNConvOpKernel : public framework::OpKernel<T> {
     PADDLE_ENFORCE(platform::dynload::miopenConvolutionForwardGetWorkSpaceSize(
         handle, cudnn_filter_desc, cudnn_input_desc, cudnn_conv_desc,
         cudnn_output_desc, &workspace_size_in_bytes));
-    PADDLE_ENFORCE_LE(workspace_size_limit, workspace_size_in_bytes,
+    PADDLE_ENFORCE_LE(workspace_size_in_bytes, workspace_size_limit,
                       "workspace_size to be allocated exceeds the limit");
     // Allocate on GPU memory
     platform::CUDAPlace gpu = boost::get<platform::CUDAPlace>(ctx.GetPlace());