Disable elementwise_sub_mkldnn test on GPU

AshburnLee · Sep 21, 2021 · 314f214 · 314f214
1 parent 557ff38
commit 314f214
Showing 1 changed file with 5 additions and 10 deletions.
diff --git a/python/paddle/fluid/tests/unittests/mkldnn/test_elementwise_sub_mkldnn_op.py b/python/paddle/fluid/tests/unittests/mkldnn/test_elementwise_sub_mkldnn_op.py
@@ -17,25 +17,20 @@
 import numpy as np
 from paddle.fluid.tests.unittests.test_elementwise_sub_op import TestElementwiseSubOp
 from paddle import enable_static
+from paddle.fluid.tests.unittests.op_test import OpTestTool
+from paddle.fluid.framework import _current_expected_place
+import paddle.fluid.core as core
 
 
+@OpTestTool.skip_if(not (isinstance(_current_expected_place(), core.CPUPlace)),
+                    "GPU is not supported")
 class TestMKLDNNElementwiseSubOp(TestElementwiseSubOp):
     def init_kernel_type(self):
         self.use_mkldnn = True
 
     def init_dtype(self):
         self.dtype = np.float32
 
-    # # TODO(piotrekobiIntel): Enable when grad is ready
-    # def test_check_grad_normal(self):
-    #     pass
-
-    # def test_check_grad_ingore_x(self):
-    #     pass
-
-    # def test_check_grad_ingore_y(self):
-    #     pass
-
 
 class TestMKLDNNElementwiseSubOp2(TestMKLDNNElementwiseSubOp):
     def init_input_output(self):