enable ut test for xpu devices (#11712)

Co-authored-by: jundu <jun.du@intel.com> Co-authored-by: Gao, Pengfei <pengfei.gao@intel.com>
2026-02-03 11:15:14 -08:00
parent 0a6925639b
commit 495290aefd
20 changed files with 237 additions and 151 deletions
@@ -16,12 +16,13 @@ from sglang.srt.layers.quantization.awq_triton import (
    awq_dequantize_triton,
    awq_gemm_triton,
 )
+from sglang.srt.utils import get_device
 from sglang.test.ci.ci_register import register_amd_ci
 from sglang.test.test_utils import CustomTestCase

 register_amd_ci(est_time=2, suite="stage-a-test-1-amd")

-device = "cuda"
+device = get_device()


 def reverse_awq_order(t: torch.Tensor) -> torch.Tensor: