upstream pulled

PaddlePaddle · Nov 9, 2021 · ff3b94e · ff3b94e
1 parent ff4c010
commit ff3b94e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/paddle/fluid/tests/unittests/mkldnn/test_pool2d_bf16_mkldnn_op.py b/python/paddle/fluid/tests/unittests/mkldnn/test_pool2d_bf16_mkldnn_op.py
@@ -17,14 +17,15 @@
 import unittest
 import numpy as np
 import paddle.fluid.core as core
-from paddle.fluid.tests.unittests.op_test import OpTest, skip_check_grad_ci, convert_float_to_uint16
+from paddle.fluid.tests.unittests.op_test import OpTest, OpTestTool, skip_check_grad_ci, convert_float_to_uint16
 from paddle.fluid.tests.unittests.test_pool2d_op import TestPool2D_Op_Mixin, TestPool2D_Op, max_pool2D_forward_naive
 from paddle.fluid.tests.unittests.npu.test_pool2d_op_npu import pool2d_backward_navie as pool2d_backward_naive
 from paddle import enable_static
 
 
 @unittest.skipIf(not core.supports_bfloat16(),
                  "place does not support BF16 evaluation")
+@OpTestTool.skip_if_not_cpu_bf16()
 class TestPoolBf16MklDNNOpGradMixin(TestPool2D_Op_Mixin):
     def init_kernel_type(self):
         self.use_mkldnn = True