Fix sgl_kernel import failure on devices other than CUDA (#10610)

2025-09-19 02:38:02 +08:00
parent 1344ebc833
commit 6fd4816d9f
2 changed files with 2 additions and 2 deletions
--- 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():