minor change to convTransLayer test in test_LayerGrad

wangyang59 · wangyang59 · commit 449120991ef9 · 2016-11-09T11:43:51.000-08:00
diff --git 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) {

Original file line number	Diff line number	Diff line change
`@@ -349,10 +349,9 @@ void testConvTransLayer(const string& type, bool trans, bool useGpu) {`
`349`	`349`	`}`
`350`	`350`
`351`	`351`	`TEST(Layer, convTransLayer) {`
`352`		`- testConvTransLayer("exconvt", /* trans= / false, / useGpu= */ false);`
`353`		`-#ifndef PADDLE_ONLY_CPU`
`354`		`- testConvTransLayer("exconvt", /* trans= / false, / useGpu= */ true);`
`355`		`-#endif`
	`352`	`+ for (auto useGpu : {false, true}) {`
	`353`	`+ testConvTransLayer("exconvt", /* trans= / false, / useGpu= */ useGpu);`
	`354`	`+ }`
`356`	`355`	`}`
`357`	`356`
`358`	`357`	`TEST(Layer, blockExpandLayer) {`