From cdf8d99080c1c36c505cd5dbe7572fe2f71bac6d Mon Sep 17 00:00:00 2001
From: hedaoyuan <hedaoyuan@github.com>
Date: Wed, 28 Jun 2017 16:26:16 +0800
Subject: [PATCH] Bug fix.

---
 paddle/function/nnpack/NNPACKConvOp.cpp     | 4 ++--
 paddle/function/nnpack/NNPACKConvOpTest.cpp | 7 +++++--
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/paddle/function/nnpack/NNPACKConvOp.cpp b/paddle/function/nnpack/NNPACKConvOp.cpp
index 5e4de5546..d75fab040 100644
--- a/paddle/function/nnpack/NNPACKConvOp.cpp
+++ b/paddle/function/nnpack/NNPACKConvOp.cpp
@@ -74,9 +74,9 @@ public:
 
   virtual void check(const BufferArgs& inputs,
                      const BufferArgs& outputs) override {
-    const TensorShape& output = inputs[0].shape();
+    const TensorShape& input = inputs[0].shape();
     const TensorShape& filter = inputs[1].shape();
-    const TensorShape& input = outputs[0].shape();
+    const TensorShape& output = outputs[0].shape();
     checkShape(input, filter, output);
   }
 
diff --git a/paddle/function/nnpack/NNPACKConvOpTest.cpp b/paddle/function/nnpack/NNPACKConvOpTest.cpp
index e7ce61cc6..481801121 100644
--- a/paddle/function/nnpack/NNPACKConvOpTest.cpp
+++ b/paddle/function/nnpack/NNPACKConvOpTest.cpp
@@ -60,12 +60,15 @@ public:
                             << " outputWidth=" << outputSize
                             << " stride=" << stride << " padding=" << padding;
 
+                  std::vector<size_t> paddings = {padding, padding};
+                  std::vector<size_t> strides = {stride, stride};
                   Compare2Function<DEVICE_TYPE_CPU, DEVICE_TYPE_CPU> test(
                       conv1,
                       conv2,
                       FuncConfig()
-                          .set("padding", padding)
-                          .set("stride", stride)
+                          .set("paddings", paddings)
+                          .set("strides", strides)
+                          .set("groups", (size_t)1)
                           .set("algo", algo));
 
                   TensorShape shape0{
-- 
GitLab