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

2025-04-11 02:05:53 -07:00
parent b75275b6f2
commit 5598634326
2 changed files with 2 additions and 2 deletions
--- a/sgl-kernel/pyproject.toml
+++ b/sgl-kernel/pyproject.toml
@@ -1,7 +1,7 @@
 [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"