Skip bf16 paged attention UT on unsupported platforms. (#4711)(#4715)

* Skip bf16 paged attention tests on unsupported platforms.

* Fix format

* Correct condition for XeLpg

Author: Kanya-Mo

tests/gpu/examples/test_paged_attention.py

@@ -2,6 +2,7 @@
 import random
 from typing import List, Optional, Tuple
 import intel_extension_for_pytorch as ipex  # noqa
+import pytest
 from torch.testing._internal.common_utils import (
     TestCase,
 )
@@ -310,6 +311,10 @@ def test_fp16(self):
         for version in ["v1", "v2"]:
             self.paged_attention(version, torch.float16)
 
+    @pytest.mark.skipif(
+        not torch.xpu.has_xmx(),
+        reason="Paged_attention: No bf16 support for current gpu arch.",
+    )
     def test_bf16(self):
         for version in ["v1", "v2"]:
             self.paged_attention(version, torch.bfloat16)