Fix the error function/GemmConvOp.

PaddlePaddle · Dec 15, 2017 · 3496092 · 3496092
1 parent 0fce0fe
commit 3496092
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/paddle/function/GemmConvOp.cpp b/paddle/function/GemmConvOp.cpp
@@ -126,6 +126,11 @@ class GemmConvFunction : public ConvFunctionBase {
       inputData += inputChannels * inputHeight * inputWidth;
       outputData += outputChannels * outputHeight * outputWidth;
     }
+#ifdef PADDLE_MOBILE_INFERENCE
+    if (Device == DEVICE_TYPE_CPU) {
+      delete memory_;
+    }
+#endif
   }
 };
 
@@ -233,11 +238,6 @@ class GemmConvGradInputFunction : public ConvFunctionBase {
       inputGrad += inputChannels * inputHeight * inputWidth;
       outputGrad += outputChannels * outputHeight * outputWidth;
     }
-#ifdef PADDLE_MOBILE_INFERENCE
-    if (Device == DEVICE_TYPE_CPU) {
-      delete memory_;
-    }
-#endif
   }
 };