fix syntax

Author: yzh119

.github/workflows/release-flashinfer-aot-modules-wheel.yml

@@ -79,13 +79,17 @@ jobs:
             - name: Build wheel in container
               env:
                   DOCKER_IMAGE: ${{ matrix.arch == 'aarch64' && format('pytorch/manylinux2_28_aarch64-builder:cuda{0}', matrix.cuda) || format('pytorch/manylinux2_28-builder:cuda{0}', matrix.cuda) }}
+                  CUDA_MAJOR: ${{ matrix.cuda.split('.')[0] }}
+                  CUDA_MINOR: ${{ matrix.cuda.split('.')[1] }}
               run: |
                   chown -R $CI_UID:$CI_GID ${{ github.workspace }}
 
                   # Run the build script inside the container with proper mounts
                   docker run --rm \
                     -v ${{ github.workspace }}:/workspace \
                     -e CUDA_VERSION="${{ matrix.cuda }}" \
+                    -e CUDA_MAJOR="${{ matrix.cuda.split('.')[0] }}" \
+                    -e CUDA_MINOR="${{ matrix.cuda.split('.')[1] }}" \
                     -e ARCH="${{ matrix.arch }}" \
                     -e FLASHINFER_CUDA_ARCH_LIST="${FLASHINFER_CUDA_ARCH_LIST}" \
                     --user $CI_UID:$CI_GID \

scripts/build_flashinfer_aot_modules_whl.sh

@@ -10,7 +10,9 @@ echo "=========================================="
 
 # Display build environment info
 echo "CUDA Version: ${CUDA_VERSION}"
-echo "Architecture: ${ARCH}"
+echo "CPU Architecture: ${ARCH}"
+echo "CUDA Major: ${CUDA_MAJOR}"
+echo "CUDA Minor: ${CUDA_MINOR}"
 echo "CUDA Architectures: ${FLASHINFER_CUDA_ARCH_LIST}"
 echo "Working directory: $(pwd)"
 echo ""
@@ -32,6 +34,15 @@ echo ""
 echo "Built wheels:"
 ls -lh dist/
 
+echo "::group::Install PyTorch"
+pip install torch==2.8 --index-url "https://download.pytorch.org/whl/cu${CUDA_MAJOR}${CUDA_MINOR}"
+echo "::endgroup::"
+
+echo "::group::Install build system"
+pip install ninja numpy
+pip install --upgrade setuptools packaging wheel build
+echo "::endgroup::"
+
 # Copy wheels to output directory if specified
 if [ -n "${OUTPUT_DIR}" ]; then
     echo ""