refine the device

Author: zxd1997066

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

@@ -230,12 +230,12 @@ def test_dequantization_accuracy(self, config, device):
     @unittest.skipIf(
         not torch_version_at_least("2.7.0"), "torch 2.6.0 and below has custom fx pass"
     )
-    @common_utils.parametrize("device", get_available_devices())
-    def test_available_gpu_kernels(self, device):
+    def test_available_gpu_kernels(self):
         """Check which GPU kernels are used"""
         torch.compiler.reset()
 
         M, K, N = 128, 256, 512
+        device = get_current_accelerator_device()
         m = torch.nn.Sequential(
             torch.nn.Linear(K, N, device=device, dtype=torch.bfloat16)
         )
@@ -286,7 +286,7 @@ class TestInt8StaticQuant(TorchAOIntegrationTestCase):
     @common_utils.parametrize(
         "act_mapping_type", [MappingType.SYMMETRIC, MappingType.ASYMMETRIC]
     )
-    @common_utils.parametrize("device", get_current_accelerator_device)
+    @common_utils.parametrize("device", get_available_devices())
     def test_static_activation_per_row_int8_weight(
         self, granularity, dtype, act_mapping_type, device
     ):