From 449120991ef9d4deb5e9635be3017824a0370a29 Mon Sep 17 00:00:00 2001
From: wangyang59 <wangyang59@baidu.com>
Date: Tue, 8 Nov 2016 15:02:49 -0800
Subject: [PATCH] minor change to convTransLayer test in test_LayerGrad

---
 paddle/gserver/tests/test_LayerGrad.cpp | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/paddle/gserver/tests/test_LayerGrad.cpp b/paddle/gserver/tests/test_LayerGrad.cpp
index 02a30719f9..7b6e6fd399 100644
--- a/paddle/gserver/tests/test_LayerGrad.cpp
+++ b/paddle/gserver/tests/test_LayerGrad.cpp
@@ -349,10 +349,9 @@ void testConvTransLayer(const string& type, bool trans, bool useGpu) {
 }
 
 TEST(Layer, convTransLayer) {
-  testConvTransLayer("exconvt", /* trans= */ false, /* useGpu= */ false);
-#ifndef PADDLE_ONLY_CPU
-  testConvTransLayer("exconvt", /* trans= */ false, /* useGpu= */ true);
-#endif
+  for (auto useGpu : {false, true}) {
+    testConvTransLayer("exconvt", /* trans= */ false, /* useGpu= */ useGpu);
+  }
 }
 
 TEST(Layer, blockExpandLayer) {
-- 
GitLab