From b22f2bf0509ca6e3f16c9cb1966796365547673d Mon Sep 17 00:00:00 2001
From: zhyncs <me@zhyncs.com>
Date: Fri, 14 Feb 2025 06:32:13 +0800
Subject: [PATCH] set pip path

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

diff --git a/scripts/run-ci-build-wheel.sh b/scripts/run-ci-build-wheel.sh
index 53d1925d0..963b0f2c4 100644
--- a/scripts/run-ci-build-wheel.sh
+++ b/scripts/run-ci-build-wheel.sh
@@ -12,6 +12,7 @@ assert_env() {
 
 assert_env FLASHINFER_CI_CACHE
 assert_env FLASHINFER_CI_CUDA_VERSION
+assert_env FLASHINFER_CI_PYTHON_VERSION
 assert_env FLASHINFER_CI_TORCH_VERSION
 assert_env TORCH_CUDA_ARCH_LIST
 PROJECT_ROOT="$(cd "$(dirname "$0")/.." && pwd)"
@@ -25,6 +26,8 @@ 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"
 
 FLASHINFER_LOCAL_VERSION="cu${CUDA_MAJOR}${CUDA_MINOR}torch${FLASHINFER_CI_TORCH_VERSION}"
 if [ -n "${FLASHINFER_GIT_SHA}" ]; then