Skip to content

Commit 3022a79

Browse files
add gpu test in test_LayerGrad
1 parent e6b540e commit 3022a79

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

paddle/gserver/tests/test_LayerGrad.cpp

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -681,12 +681,13 @@ TEST(Layer, hsigmoidLayer) {
681681
config.layerConfig.add_inputs();
682682
config.layerConfig.add_inputs();
683683

684-
// Not support GPU now
685-
testLayerGrad(config,
686-
"hsigmoid",
687-
100,
688-
/* trans */ false, /* useGpu */
689-
false);
684+
for (auto useGpu : {false, true}) {
685+
testLayerGrad(config,
686+
"hsigmoid",
687+
100,
688+
/* trans */ false, /* useGpu */
689+
useGpu);
690+
}
690691
}
691692

692693
TEST(Layer, multi_cross) {

0 commit comments

Comments
 (0)