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

Add 'NPU' to the runtime exception message in get_device (#14225)

Browse Source 
Co-authored-by: ronnie_zheng <zl19940307@163.com>
This commit is contained in:
Raul Torres
2025-12-04 17:34:31 +03:00
committed by GitHub
co-authored by ronnie_zheng
parent 079ac237da
commit 29a2d4b59f
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/srt/utils/common.py
+1 -1
View File
@@ -1840,7 +1840,7 @@ def get_device(device_id: Optional[int] = None) -> str:
"Habana frameworks detected, but failed to import 'habana_frameworks.torch.hpu'." "Habana frameworks detected, but failed to import 'habana_frameworks.torch.hpu'."
) )
raise RuntimeError("No accelerator (CUDA, XPU, HPU) is available.") raise RuntimeError("No accelerator (CUDA, XPU, HPU, NPU) is available.")
@lru_cache(maxsize=1) @lru_cache(maxsize=1)