diff --git a/paddle/scripts/paddle_build.bat b/paddle/scripts/paddle_build.bat index 3f06a573def7bea041cecc79b3819e8921e8de79..7fd4a5aee109b2a1c3d30c1b3267d721837b5371 100644 --- a/paddle/scripts/paddle_build.bat +++ b/paddle/scripts/paddle_build.bat @@ -447,7 +447,6 @@ test_imperative_static_runner_while^|^ test_optimizer_in_control_flow^|^ test_fuse_bn_act_pass^|^ test_fuse_bn_add_act_pass^|^ -test_activation_mkldnn_op^|^ test_tsm^|^ test_gru_rnn_op^|^ test_rnn_op^|^ diff --git a/python/paddle/fluid/tests/unittests/mkldnn/test_activation_mkldnn_op.py b/python/paddle/fluid/tests/unittests/mkldnn/test_activation_mkldnn_op.py index 63db1b1475d40d60ef7084397dd2d6e386c58024..611f5a9d6d15d843c9f237d02f1e7499f6cfac97 100644 --- a/python/paddle/fluid/tests/unittests/mkldnn/test_activation_mkldnn_op.py +++ b/python/paddle/fluid/tests/unittests/mkldnn/test_activation_mkldnn_op.py @@ -79,6 +79,8 @@ class TestMKLDNNGeluDim2Approx(TestActivation): self.attrs = {"use_mkldnn": True, "approximate": True} +@unittest.skipIf(not core.supports_bfloat16(), + "place does not support BF16 evaluation") class TestMKLDNNGeluBf16Dim2(TestActivation): def setUp(self): self.op_type = "gelu" @@ -98,6 +100,8 @@ class TestMKLDNNGeluBf16Dim2(TestActivation): pass +@unittest.skipIf(not core.supports_bfloat16(), + "place does not support BF16 evaluation") class TestMKLDNNGeluBf16Dim2Approx(TestActivation): def setUp(self): self.op_type = "gelu" @@ -225,6 +229,8 @@ class TestMKLDNNGeluDim4Approx(TestActivation): self.attrs = {"use_mkldnn": True, "approximate": True} +@unittest.skipIf(not core.supports_bfloat16(), + "place does not support BF16 evaluation") class TestMKLDNNGeluBf16Dim4(TestActivation): def setUp(self): self.op_type = "gelu" @@ -244,6 +250,8 @@ class TestMKLDNNGeluBf16Dim4(TestActivation): pass +@unittest.skipIf(not core.supports_bfloat16(), + "place does not support BF16 evaluation") class TestMKLDNNGeluBf16Dim4Approx(TestActivation): def setUp(self): self.op_type = "gelu"