From a043f7f2abb5d0a8297927f807361933ec0a5ac4 Mon Sep 17 00:00:00 2001
From: Yineng Zhang <me@zhyncs.com>
Date: Tue, 29 Apr 2025 17:51:18 -0700
Subject: [PATCH] chore: use torch 2.6 for sgl-kernel build (#5898)

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

diff --git a/sgl-kernel/build.sh b/sgl-kernel/build.sh
index 819283c51..65d4edc53 100755
--- a/sgl-kernel/build.sh
+++ b/sgl-kernel/build.sh
@@ -13,7 +13,7 @@ if [ ${CUDA_VERSION} = "12.8" ]; then
    TORCH_INSTALL="pip install --no-cache-dir --pre torch --index-url https://download.pytorch.org/whl/nightly/cu${CUDA_VERSION//.}"
 else
    DOCKER_IMAGE="pytorch/manylinux-builder:cuda${CUDA_VERSION}"
-   TORCH_INSTALL="pip install --no-cache-dir torch==2.5.1 --index-url https://download.pytorch.org/whl/cu${CUDA_VERSION//.}"
+   TORCH_INSTALL="pip install --no-cache-dir torch==2.6.0 --index-url https://download.pytorch.org/whl/cu${CUDA_VERSION//.}"
 fi
 
 docker run --rm \
diff --git a/sgl-kernel/pyproject.toml b/sgl-kernel/pyproject.toml
index cb5f388c2..916d46590 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.6.0",
   "wheel",
 ]
 build-backend = "scikit_build_core.build"
diff --git a/sgl-kernel/pyproject_rocm.toml b/sgl-kernel/pyproject_rocm.toml
index dd122531b..b72918b0e 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.6.0",
   "wheel",
 ]
 build-backend = "setuptools.build_meta"