[XPU][Tests] Enable test_per_token_group_quant_int8 on XPU (#55681)

Signed-off-by: pmanczak <pawel.manczak@intel.com>
2026-09-11 15:59:13 +08:00
parent fadfe1c7d4
commit 6376c601ed
1 changed files with 5 additions and 2 deletions
@@ -413,9 +413,12 @@ def test_per_token_group_quant_fp8_packed_large_mn():

@pytest.mark.parametrize("shape", [(32, 128), (64, 256), (16, 512)])
@pytest.mark.parametrize("group_size", [64, 128])
-@pytest.mark.skipif(not torch.cuda.is_available(), reason="CUDA not available")
+@pytest.mark.skipif(
+    not (current_platform.is_cuda_alike() or current_platform.is_xpu()),
+    reason="Only test on CUDA/ROCm/XPU.",
+)
 def test_per_token_group_quant_int8(shape, group_size: int):
-    device = "cuda"
+    device = current_platform.device_type

    torch.manual_seed(42)
    num_tokens, hidden_dim = shape