From 5598634326413b7898910c26974bd05b92c7f519 Mon Sep 17 00:00:00 2001
From: Yineng Zhang <me@zhyncs.com>
Date: Fri, 11 Apr 2025 02:05:53 -0700
Subject: [PATCH] chore: relax the torch version restriction for sgl-kernel
 compilation (#5288)

---
 sgl-kernel/pyproject.toml      | 2 +-
 sgl-kernel/pyproject_rocm.toml | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/sgl-kernel/pyproject.toml b/sgl-kernel/pyproject.toml
index d07fda861..278af1e46 100644
--- 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"
diff --git a/sgl-kernel/pyproject_rocm.toml b/sgl-kernel/pyproject_rocm.toml
index bab23cfdb..c13f3f026 100644
--- 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"