[Fix] Skip autocast_simple_backward_bf16 on platform without avx2_vnni_2 support (#5571) (#5578)

* [Fix] Skip autocast_simple_backward_bf16 on Arrow lake and Lunar lake

* fix format

Author: gaopengff

tests/gpu/examples/test_autocast.py

@@ -1,5 +1,4 @@
 import unittest
-import platform
 import torch
 from torch.testing._internal.common_utils import TestCase
 
@@ -9,13 +8,6 @@
 dpcpp_device = torch.device("xpu")
 checking_atol = 1e-2
 checking_rtol = 3e-2
-cpu_name = platform.processor()
-skipIfNoAvx2Vnni2 = unittest.skipIf(
-    # Arrow lake and Lunar lake
-    cpu_name == "Intel64 Family 6 Model 198 Stepping 2, GenuineIntel"
-    or cpu_name == "Intel64 Family 6 Model 189 Stepping 1, GenuineIntel",
-    "Skip on Arrow/Lunar lake because they do not support avx2_vnni_2",
-)
 
 
 class TestNet(torch.nn.Module):
@@ -66,7 +58,10 @@ def test_autocast_simple_forward_fp16(self):
         print(y_xpu.to("cpu"))
         self.assertEqual(y_xpu.dtype, torch.float16)
 
-    @skipIfNoAvx2Vnni2
+    @unittest.skipIf(
+        not torch.cpu._is_vnni_supported(),
+        "Test requires CPU with avx2_vnni_2",
+    )
     def test_autocast_simple_backward_bf16(self):
         model = TestNet()
         x = torch.ones([2, 3, 8, 6], dtype=torch.float)