From 6ec6f24094174db0af05d9ffb5f2d839abd7d6f4 Mon Sep 17 00:00:00 2001
From: yzh119 <expye@outlook.com>
Date: Fri, 16 Feb 2024 23:51:55 +0000
Subject: [PATCH 1/2] fix version naming

---
 .github/workflows/release_wheel.yml | 2 +-
 scripts/run-ci-build-wheel.sh       | 4 +++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/release_wheel.yml b/.github/workflows/release_wheel.yml
index 8eaedf9a..cc9e9ce4 100644
--- a/.github/workflows/release_wheel.yml
+++ b/.github/workflows/release_wheel.yml
@@ -27,7 +27,7 @@ jobs:
       matrix:
         python: ["3.9", "3.10", "3.11"]
         cuda: ["11.8", "12.1"]
-        torch: ["2.1.0", "2.2.0"]
+        torch: ["2.1", "2.2"]
     runs-on: [self-hosted]
     steps:
       - uses: actions/checkout@v4
diff --git a/scripts/run-ci-build-wheel.sh b/scripts/run-ci-build-wheel.sh
index 88afc6fa..5507706e 100644
--- a/scripts/run-ci-build-wheel.sh
+++ b/scripts/run-ci-build-wheel.sh
@@ -24,6 +24,8 @@ mkdir -p $HOME
 export PATH="$HOME/.local/bin:$PATH"
 CUDA_MAJOR="${FLASHINFER_CI_CUDA_VERSION%.*}"
 CUDA_MINOR="${FLASHINFER_CI_CUDA_VERSION#*.}"
+TORCH_MAJOR="${FLASHINFER_CI_TORCH_VERSION%.*}"
+TORCH_MINOR="${FLASHINFER_CI_TORCH_VERSION#*.}"
 PYVER="${FLASHINFER_CI_PYTHON_VERSION//./}"
 export PATH="/opt/python/cp${PYVER}-cp${PYVER}/bin:$PATH"
 
@@ -40,7 +42,7 @@ echo "::endgroup::"
 
 echo "::group::Build wheel for FlashInfer"
 cd "$PROJECT_ROOT/python"
-MAX_JOBS=32 FLASHINFER_BUILD_VERSION="${FLASHINFER_BUILD_VERSION}+cu${CUDA_MAJOR}${CUDA_MINOR}+torch${FLASHINFER_CI_TORCH_VERSION}" python -m build --no-isolation
+MAX_JOBS=32 FLASHINFER_BUILD_VERSION="${FLASHINFER_BUILD_VERSION}+cu${CUDA_MAJOR}${CUDA_MINOR}+torch${TORCH_MAJOR}${TORCH_MINOR}" python -m build --no-isolation
 rm -f dist/*.tar.gz
 python -m build --no-isolation --sdist
 echo "::endgroup::"

From cd9934ce2a157120b7d3fe794767f1cebd378d93 Mon Sep 17 00:00:00 2001
From: yzh119 <expye@outlook.com>
Date: Sat, 17 Feb 2024 01:24:22 +0000
Subject: [PATCH 2/2] upd

---
 scripts/run-ci-build-wheel.sh | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/scripts/run-ci-build-wheel.sh b/scripts/run-ci-build-wheel.sh
index 5507706e..692431aa 100644
--- a/scripts/run-ci-build-wheel.sh
+++ b/scripts/run-ci-build-wheel.sh
@@ -24,8 +24,6 @@ mkdir -p $HOME
 export PATH="$HOME/.local/bin:$PATH"
 CUDA_MAJOR="${FLASHINFER_CI_CUDA_VERSION%.*}"
 CUDA_MINOR="${FLASHINFER_CI_CUDA_VERSION#*.}"
-TORCH_MAJOR="${FLASHINFER_CI_TORCH_VERSION%.*}"
-TORCH_MINOR="${FLASHINFER_CI_TORCH_VERSION#*.}"
 PYVER="${FLASHINFER_CI_PYTHON_VERSION//./}"
 export PATH="/opt/python/cp${PYVER}-cp${PYVER}/bin:$PATH"
 
@@ -42,7 +40,7 @@ echo "::endgroup::"
 
 echo "::group::Build wheel for FlashInfer"
 cd "$PROJECT_ROOT/python"
-MAX_JOBS=32 FLASHINFER_BUILD_VERSION="${FLASHINFER_BUILD_VERSION}+cu${CUDA_MAJOR}${CUDA_MINOR}+torch${TORCH_MAJOR}${TORCH_MINOR}" python -m build --no-isolation
+MAX_JOBS=32 FLASHINFER_BUILD_VERSION="${FLASHINFER_BUILD_VERSION}+cu${CUDA_MAJOR}${CUDA_MINOR}torch${FLASHINFER_CI_TORCH_VERSION}" python -m build --no-isolation
 rm -f dist/*.tar.gz
 python -m build --no-isolation --sdist
 echo "::endgroup::"