added skipping tests for GPU

PaddlePaddle · Sep 23, 2021 · d629217 · d629217
1 parent a685a60
commit d629217
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/python/paddle/fluid/tests/unittests/mkldnn/test_shape_mkldnn_op.py b/python/paddle/fluid/tests/unittests/mkldnn/test_shape_mkldnn_op.py
@@ -16,12 +16,13 @@
 
 import unittest
 import numpy as np
-from paddle.fluid.tests.unittests.op_test import OpTest
+from paddle.fluid.tests.unittests.op_test import OpTest, OpTestTool
 import paddle
 from paddle.fluid import core
 from paddle.fluid.op import Operator
 
 
+@OpTestTool.skip_if_not_cpu_bf16()
 class TestShapeOneDNNOp(OpTest):
     def setUp(self):
         self.op_type = "shape"
@@ -35,7 +36,7 @@ def config(self):
         self.dtype = np.float32
 
     def test_check_output(self):
-        self.check_output()
+        self.check_output_with_place(core.CPUPlace())
 
 
 class TestShape1DOneDNNOp(TestShapeOneDNNOp):