[Enhancement] Update CUDA toolkit configurations and streamline environment setup for testing

Author: SigureMo

.github/workflows/dist-paddle.yml

@@ -90,8 +90,8 @@ jobs:
     strategy:
       matrix:
         target:
-          - { runner: ubuntu-latest, toolkit: "CUDA-12.8" }
-          - { runner: ubuntu-24.04-arm, toolkit: "CUDA-12.8" }
+          - { runner: ubuntu-latest, toolkit: "CUDA-12.8", test_backends: "cu118 cu130" }
+          - { runner: ubuntu-24.04-arm, toolkit: "CUDA-12.8", test_backends: "cu126 cu130" }
           - { runner: macos-latest, toolkit: "Metal" }
         python-version:
           # Wheels are built with Python 3.8 Limited API, they should work with all Python >= 3.8.
@@ -138,23 +138,10 @@ jobs:
             CUDA_VERSION="${CUDA_VERSION##*-}"
             CUDA_VERSION_MAJMIN="$(echo ${CUDA_VERSION} | cut -d '.' -f-2)"
             CUDA_VERSION_MAJMIN_NODOT="${CUDA_VERSION_MAJMIN//./}"
-            echo "CUDA_VERSION=${CUDA_VERSION}" | tee -a "${GITHUB_ENV}"
-            if [[ "${{ matrix.target.toolkit }}" == "Nightly-"* ]]; then
-              # Use torch nightly builds
-              export UV_INDEX="https://download.pytorch.org/whl/nightly/cu${CUDA_VERSION_MAJMIN_NODOT}"
-            else
-              export UV_INDEX="https://download.pytorch.org/whl/cu${CUDA_VERSION_MAJMIN_NODOT}"
-              echo "UV_TORCH_BACKEND=cu${CUDA_VERSION_MAJMIN_NODOT}" | tee -a "${GITHUB_ENV}"
-            fi
-            echo "UV_INDEX=${UV_INDEX}" | tee -a "${GITHUB_ENV}"
+            echo "UV_TORCH_BACKEND=cu${CUDA_VERSION_MAJMIN_NODOT}" | tee -a "${GITHUB_ENV}"
           fi
           if [[ "${{ env.IS_RELEASE }}" == "true" ]]; then
-            if [[ "${{ matrix.target.toolkit }}" == "Nightly-"* ]]; then
-              # Avoid using same file name for different toolkit.
-              echo "NO_GIT_VERSION=ON" | tee -a "${GITHUB_ENV}"
-            else
-              echo "NO_VERSION_LABEL=ON" | tee -a "${GITHUB_ENV}"
-            fi
+            echo "NO_VERSION_LABEL=ON" | tee -a "${GITHUB_ENV}"
           fi
           if [[ "${{ runner.os }}" == "Linux" ]]; then
             HOST_CCACHE_DIR="$(ccache --get-config cache_dir)"
@@ -172,7 +159,7 @@ jobs:
         id: setup-uv
         uses: astral-sh/setup-uv@v7
         with:
-          python-version: "3.12"
+          python-version: ${{ matrix.python-version }}
           activate-environment: true
 
       - name: Test built wheels
@@ -183,10 +170,6 @@ jobs:
               set -e
               uv venv --python=3.12 test-venv
               source test-venv/bin/activate
-              uv pip install --upgrade pip setuptools wheel
-              if [[ "${UV_INDEX}" == *"/nightly/"* ]]; then
-                uv pip install --prerelease=allow -v torch
-              fi
               uv pip install -v "${WHEEL}"
               (
                 set -e