add gpu test in test_LayerGrad

7 years ago · 3022a79082
parent e6b540eacf
commit 3022a79082
1 changed files with 7 additions and 6 deletions
--- 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) {