Fix the error function/GemmConvOp.

34960920 · dangqingqing · 0fce0fe6 · 34960920
显示空白变更内容
内联并排

Showing with 5 addition and 5 deletion

paddle/function/GemmConvOp.cpp paddle/function/GemmConvOp.cpp +5 -5

未找到文件。
--- a/paddle/function/GemmConvOp.cpp
+++ b/paddle/function/GemmConvOp.cpp
@@ -126,6 +126,11 @@ public:
      inputData += inputChannels * inputHeight * inputWidth;
      outputData += outputChannels * outputHeight * outputWidth;
    }
+#ifdef PADDLE_MOBILE_INFERENCE
+    if (Device == DEVICE_TYPE_CPU) {
+      delete memory_;
+    }
+#endif
  }
 };

@@ -233,11 +238,6 @@ public:
      inputGrad += inputChannels * inputHeight * inputWidth;
      outputGrad += outputChannels * outputHeight * outputWidth;
    }
-#ifdef PADDLE_MOBILE_INFERENCE
-    if (Device == DEVICE_TYPE_CPU) {
-      delete memory_;
-    }
-#endif
  }
 };