From 6fd4816d9f1ae31124139e6a6bfb20b141694e9b Mon Sep 17 00:00:00 2001 From: Zaili Wang <109502517+ZailiWang@users.noreply.github.com> Date: Fri, 19 Sep 2025 02:38:02 +0800 Subject: [PATCH] Fix sgl_kernel import failure on devices other than CUDA (#10610) --- docs/platforms/cpu_server.md | 2 +- sgl-kernel/python/sgl_kernel/__init__.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/platforms/cpu_server.md b/docs/platforms/cpu_server.md index 9e9e407ce..7e14b7a07 100644 --- a/docs/platforms/cpu_server.md +++ b/docs/platforms/cpu_server.md @@ -63,7 +63,7 @@ is required to enable SGLang service with CPU engine. conda create -n sgl-cpu python=3.12 -y conda activate sgl-cpu -# Optional: Set PyTorch CPU as primary pip install channel to avoid installing CUDA version +# Set PyTorch CPU as primary pip install channel to avoid installing the larger CUDA-enabled version and prevent potential runtime issues. pip config set global.index-url https://download.pytorch.org/whl/cpu pip config set global.extra-index-url https://pypi.org/simple diff --git a/sgl-kernel/python/sgl_kernel/__init__.py b/sgl-kernel/python/sgl_kernel/__init__.py index 81456dbbe..37ba4e3a9 100644 --- a/sgl-kernel/python/sgl_kernel/__init__.py +++ b/sgl-kernel/python/sgl_kernel/__init__.py @@ -23,7 +23,7 @@ def _find_cuda_home(): return cuda_home -if torch.version.hip is None: +if torch.version.cuda is not None: cuda_home = Path(_find_cuda_home()) if (cuda_home / "lib").is_dir():