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
+  for (auto useGpu : {false, true}) {
    testLayerGrad(config,
                  "hsigmoid",
                  100,
                  /* trans */ false, /* useGpu */
-                false);
+                  useGpu);
+  }
 }

 TEST(Layer, multi_cross) {