add gpu test in test_LayerGrad

3022a790 · peterzhang2029 · e6b540ea · 3022a790
隐藏空白更改
内联并排

Showing with 7 addition and 6 deletion

paddle/gserver/tests/test_LayerGrad.cpp paddle/gserver/tests/test_LayerGrad.cpp +7 -6

未找到文件。
--- a/paddle/gserver/tests/test_LayerGrad.cpp
+++ b/paddle/gserver/tests/test_LayerGrad.cpp
@@ -681,12 +681,13 @@ TEST(Layer, hsigmoidLayer) {
  config.layerConfig.add_inputs();
  config.layerConfig.add_inputs();

-  // Not support GPU now
-  testLayerGrad(config,
-                "hsigmoid",
-                100,
-                /* trans */ false, /* useGpu */
-                false);
+  for (auto useGpu : {false, true}) {
+    testLayerGrad(config,
+                  "hsigmoid",
+                  100,
+                  /* trans */ false, /* useGpu */
+                  useGpu);
+  }
 }

 TEST(Layer, multi_cross) {