minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 13 Packages Projects Releases Wiki Activity

[XPU] Re-add intel xpu on triton paths in diffusion platforms (#36654)

Browse Source
This commit is contained in:
Kotthagattu Meher Sai
2026-09-07 13:30:22 +08:00
committed by GitHub
parent e24e31efd9
commit 97dcbf9410
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/kernels/ops/diffusion/common/platform.py
+2 -2
View File
@@ -31,7 +31,7 @@ _CUDA_LIKE = frozenset({"cuda", "hip"})
def platform_key() -> str: def platform_key() -> str:
"""Return the live device family: ``cuda``/``hip``/``npu``/``mps``/``musa``/``cpu``. """Return the live device family: ``cuda``/``hip``/``npu``/``mps``/``musa``/``xpu``/``cpu``.
Deliberately *not* memoized: :func:`select_impl` calls it at module import Deliberately *not* memoized: :func:`select_impl` calls it at module import
time, and latching that first answer would freeze the choice before the time, and latching that first answer would freeze the choice before the
@@ -40,7 +40,7 @@ def platform_key() -> str:
""" """
from sglang.multimodal_gen.runtime.platforms import current_platform from sglang.multimodal_gen.runtime.platforms import current_platform
for name in ("cuda", "hip", "npu", "mps", "musa"): for name in ("cuda", "hip", "npu", "mps", "musa", "xpu"):
if getattr(current_platform, f"is_{name}")(): if getattr(current_platform, f"is_{name}")():
return name return name
return "cpu" return "cpu"
python/sglang/multimodal_gen/runtime/platforms/__init__.py
+1
View File
@@ -201,6 +201,7 @@ def resolve_current_platform_cls_qualname() -> str:
"mps": "sglang.multimodal_gen.runtime.platforms.mps.MpsPlatform", "mps": "sglang.multimodal_gen.runtime.platforms.mps.MpsPlatform",
"npu": "sglang.multimodal_gen.runtime.platforms.npu.NPUPlatformBase", "npu": "sglang.multimodal_gen.runtime.platforms.npu.NPUPlatformBase",
"musa": "sglang.multimodal_gen.runtime.platforms.musa.MusaPlatform", "musa": "sglang.multimodal_gen.runtime.platforms.musa.MusaPlatform",
"xpu": "sglang.multimodal_gen.runtime.platforms.xpu.XpuPlatform",
} }
qualname = forced_map.get(forced_platform.lower()) qualname = forced_map.get(forced_platform.lower())
if qualname is None: if qualname is None: