disabled tests for GPU

jakpiase · Mar 18, 2022 · 93bb52e · 93bb52e
1 parent 5c9c5a1
commit 93bb52e
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/python/paddle/fluid/tests/unittests/mkldnn/test_softplus_mkldnn_op.py b/python/paddle/fluid/tests/unittests/mkldnn/test_softplus_mkldnn_op.py
@@ -30,8 +30,7 @@ def ref_softplus(x, beta, threshold):
     return out
 
 
-@OpTestTool.skip_if(not (isinstance(_current_expected_place(), core.CPUPlace)),
-                    "GPU is not supported")
+@OpTestTool.skip_if_not_cpu_bf16()
 class TestSoftplusOneDNNOp(OpTest):
     def setUp(self):
         self.op_type = "softplus"