chore: relax the torch version restriction for sgl-kernel compilation (#5288)

55986343 · Yineng Zhang · GitHub · b75275b6 · 55986343 · 55986343
Unverified Commit 55986343 authored Apr 11, 2025 by Yineng Zhang Committed by GitHub Apr 11, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

sgl-kernel/pyproject.toml sgl-kernel/pyproject.toml +1 -1

sgl-kernel/pyproject_rocm.toml sgl-kernel/pyproject_rocm.toml +1 -1

No files found.
--- a/sgl-kernel/pyproject.toml
+++ b/sgl-kernel/pyproject.toml
 [build-system]
 requires = [
  "scikit-build-core>=0.10",
-  "torch==2.5.1",
+  "torch>=2.5.1",
  "wheel",
 ]
 build-backend = "scikit_build_core.build"

--- a/sgl-kernel/pyproject_rocm.toml
+++ b/sgl-kernel/pyproject_rocm.toml
@@ -2,7 +2,7 @@
 requires = [
  "setuptools>=75.0",
  "scikit-build-core>=0.10",
-  "torch==2.5.1",
+  "torch>=2.5.1",
  "wheel",
 ]
 build-backend = "setuptools.build_meta"