From 86bdb2f33fa9e9e806e8248b14a172ce4e0557c6 Mon Sep 17 00:00:00 2001
From: wanghaoshuang <wanghaoshuang@baidu.com>
Date: Wed, 5 Jul 2017 10:36:22 +0800
Subject: [PATCH] fix crop function test

---
 paddle/function/CropOpTest.cpp | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/paddle/function/CropOpTest.cpp b/paddle/function/CropOpTest.cpp
index 71d9b058128..dcba972e106 100644
--- a/paddle/function/CropOpTest.cpp
+++ b/paddle/function/CropOpTest.cpp
@@ -34,8 +34,10 @@ TEST(Crop, real) {
             TensorShape outDims{numSamples, 2, 3, 3};
             compare.addInputs(
                 BufferArg(VALUE_TYPE_FLOAT, test_grad ? outDims : inDims));
-            compare.addOutputs(BufferArg(
-                VALUE_TYPE_FLOAT, test_grad ? inDims : outDims, ASSIGN_TO));
+            compare.addOutputs(BufferArg(VALUE_TYPE_FLOAT,
+                                         test_grad ? inDims : outDims,
+                                         tes_grad ? ADD_TO : ASSIGN_TO),
+                               test_grad ? ADD_TO : ASSIGN_TO);
             compare.run();
           }
         }
-- 
GitLab