minor change to convTransLayer test in test_LayerGrad

44912099 · wangyang59 · 53e1629a · 44912099
隐藏空白更改
内联并排

Showing with 3 addition and 4 deletion

paddle/gserver/tests/test_LayerGrad.cpp paddle/gserver/tests/test_LayerGrad.cpp +3 -4

未找到文件。
--- 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);
+  for (auto useGpu : {false, true}) {
-#ifndef PADDLE_ONLY_CPU
+    testConvTransLayer("exconvt", /* trans= */ false, /* useGpu= */ useGpu);
-  testConvTransLayer("exconvt", /* trans= */ false, /* useGpu= */ true);
+  }
-#endif
 }
 TEST(Layer, blockExpandLayer) {