[XPU][Tests] Enable test_per_token_group_quant_int8 on XPU (#55681)
Signed-off-by: pmanczak <pawel.manczak@intel.com>
pmanczak committed
Sep 11, 2026 at 09:59 UTC
6376c601ede7c8ff25d2bd89dbaa46297a45db10
1 file changed
+5
-2
tests/kernels/quantization/test_per_token_group_quant.py
+5
-2
@@ -413,9 +413,12 @@ def test_per_token_group_quant_fp8_packed_large_mn():
413
414
@pytest.mark.parametrize("shape", [(32, 128), (64, 256), (16, 512)])
415
@pytest.mark.parametrize("group_size", [64, 128])
416
-@pytest.mark.skipif(not torch.cuda.is_available(), reason="CUDA not available")
416
+@pytest.mark.skipif(
417
+ not (current_platform.is_cuda_alike() or current_platform.is_xpu()),
418
+ reason="Only test on CUDA/ROCm/XPU.",
419
+)
420
def test_per_token_group_quant_int8(shape, group_size: int):
418
- device = "cuda"
421
+ device = current_platform.device_type
422
423
torch.manual_seed(42)
424
num_tokens, hidden_dim = shape