update

Author: zxd1997066

test/quantization/quantize_/workflows/int8/test_int8_tensor.py

@@ -31,7 +31,6 @@
 from torchao.testing.utils import TorchAOIntegrationTestCase, skip_if_xpu
 from torchao.utils import (
     get_available_devices,
-    get_current_accelerator_device,
     torch_version_at_least,
 )
 
@@ -227,7 +226,6 @@ def test_dequantization_accuracy(self, config, device):
             f"Dequantization error is too high to get a SQNR of {compute_error(dequantized, weight_fp)}"
         )
 
-    @unittest.skipIf(not torch.accelerator.is_available(), "Need GPU available")
     @unittest.skipIf(
         not torch_version_at_least("2.7.0"), "torch 2.6.0 and below has custom fx pass"
     )
@@ -236,7 +234,7 @@ def test_available_gpu_kernels(self):
         torch.compiler.reset()
 
         M, K, N = 128, 256, 512
-        device = get_available_devices()
+        device = "xpu" if torch.xpu.is_available() else "cuda"
         m = torch.nn.Sequential(
             torch.nn.Linear(K, N, device=device, dtype=torch.bfloat16)
         )