CI: use uv

Author: akx

.github/scripts/build-cpu.sh

@@ -4,7 +4,7 @@ declare build_os
 
 set -xeuo pipefail
 
-pip install cmake==3.28.3
+uv tool install cmake==3.28.3
 
 if [ "${build_os:0:5}" == macos ] && [ "${build_arch}" == aarch64 ]; then
 	cmake -DCMAKE_OSX_ARCHITECTURES=arm64 -DCOMPUTE_BACKEND=cpu .

.github/scripts/build-cuda.sh

@@ -27,7 +27,7 @@ else
     [[ "${cuda_version}" == 13.*.* ]] && build_capability="75;80;86;89;90;100;120"
 fi
 
-[[ "${build_os}" = windows-* ]] && python3 -m pip install ninja
+uv tool install ninja cmake==3.28.3
 
 if [ "${build_os:0:6}" == ubuntu ]; then
     # We'll use Rocky Linux 8 in order to maintain manylinux 2.24 compatibility.
@@ -41,7 +41,6 @@ if [ "${build_os:0:6}" == ubuntu ]; then
         && cmake -DCOMPUTE_BACKEND=cuda -DCOMPUTE_CAPABILITY=\"${build_capability}\" . \
         && cmake --build . --config Release"
 else
-    pip install cmake==3.28.3
     cmake -G Ninja -DCOMPUTE_BACKEND=cuda -DCOMPUTE_CAPABILITY="${build_capability}" -DCMAKE_BUILD_TYPE=Release -S .
     cmake --build . --config Release
 fi

.github/scripts/build-rocm.sh

@@ -18,7 +18,7 @@ if [ "${build_os:0:6}" == ubuntu ]; then
     docker run --rm --platform "linux/$build_arch" -i \
         -w /src -v "$PWD:/src" "$image" sh -c \
         "apt-get update \
-      && pip install cmake==3.31.6 \
+      && uv tool install cmake==3.31.6 \
       && cmake -DCOMPUTE_BACKEND=hip -DCMAKE_BUILD_TYPE=MinSizeRel -DCMAKE_HIP_FLAGS=\"--offload-compress\" -DBNB_ROCM_ARCH=\"${bnb_rocm_arch}\" . \
       && cmake --build ."
 fi

.github/workflows/python-package.yml

@@ -140,6 +140,9 @@ jobs:
     runs-on: ${{ matrix.os }}
     steps:
       - uses: actions/checkout@v6
+      - uses: astral-sh/setup-uv@61cb8a9741eeb8a550a1b8544337180c0fc8476b
+        with:
+          python-version: '3.14'
       - name: Clean up disk space
         run: |
           echo "Disk space before cleanup:"
@@ -209,18 +212,17 @@ jobs:
           ls -lR output/
           cp output/${{ matrix.os }}/${{ matrix.arch }}/* bitsandbytes/
       - name: Set up Python ${{ matrix.python-version }}
-        uses: actions/setup-python@v6
+        uses: astral-sh/setup-uv@61cb8a9741eeb8a550a1b8544337180c0fc8476b # v7.2.0
         with:
           python-version: ${{ matrix.python-version }}
-          cache: pip
-      - run: pip install build wheel
-      - run: python -m build .
+          enable-cache: 'true'
+      - run: uv build
       - name: Determine and Set Platform Tag, then Tag Wheel
         shell: bash
         run: |
           PLATFORM_TAG=$(python .github/scripts/set_platform_tag.py "${{ matrix.arch }}")
           echo "PLATFORM_TAG=$PLATFORM_TAG"
-          wheel tags --remove --abi-tag=none --python-tag=py3 --platform-tag=$PLATFORM_TAG dist/bitsandbytes-*.whl
+          uvx wheel tags --remove --abi-tag=none --python-tag=py3 --platform-tag=$PLATFORM_TAG dist/bitsandbytes-*.whl
       - name: Upload build artifact
         uses: actions/upload-artifact@v6
         with:
@@ -378,12 +380,8 @@ jobs:
         with:
           name: bdist_wheel_${{ matrix.os }}_${{ matrix.arch }}
           path: wheels/
-      - name: Set up Python
-        uses: actions/setup-python@v6
-        with:
-          python-version: "3.12"
-      - run: pip install auditwheel
-      - run: python ./.github/scripts/auditwheel_show.py wheels/* | tee $GITHUB_STEP_SUMMARY
+      - uses: astral-sh/setup-uv@61cb8a9741eeb8a550a1b8544337180c0fc8476b # v7.2.0
+      - run: uv run --no-sync --with=auditwheel ./.github/scripts/auditwheel_show.py wheels/* | tee $GITHUB_STEP_SUMMARY
 
   publish-wheels:
     name: Publish wheels to PyPI

.github/workflows/test-runner.yml

@@ -193,10 +193,10 @@ jobs:
           path: bitsandbytes/
           merge-multiple: true
 
-      - name: Setup Python
-        uses: actions/setup-python@v6
+      - uses: astral-sh/setup-uv@61cb8a9741eeb8a550a1b8544337180c0fc8476b
         with:
           python-version: '3.10'
+          activate-environment: 'true'
 
       # Windows: Setup MSVC for torch.compile
       - name: Setup MSVC
@@ -205,21 +205,20 @@ jobs:
 
       - name: Install dependencies
         run: |
-          pip install torch==${{ inputs.torch_version }} --index-url ${{ inputs.pypi_index }}
-          pip install -e ".[test]" -v
-          pip install pytest-cov
+          uv pip install torch==${{ inputs.torch_version }} --index-url ${{ inputs.pypi_index }}
+          uv pip install pytest-cov -e ".[test]"
 
       # Windows: Downgrade NumPy for torch<2.4.1 compatibility
       # See: https://github.com/pytorch/pytorch/issues/131668
       - name: Downgrade NumPy
         if: inputs.platform == 'windows' && startsWith(inputs.torch_version, '2.3.')
-        run: pip install "numpy<2"
+        run: uv pip install "numpy<2"
 
       - name: Show installed packages
-        run: pip list
+        run: uv pip list
 
       - name: Show environment information
-        run: python -m torch.utils.collect_env
+        run: uv run -m torch.utils.collect_env
 
       - name: Run tests
-        run: pytest --durations=100
+        run: uv run pytest --durations=100