IFU-master-2025-06-03

.github/scripts/build-rocm.sh

@@ -0,0 +1,21 @@
+#!/bin/bash
+declare build_arch
+declare build_os
+declare rocm_version
+
+set -xeuo pipefail
+bnb_rocm_arch="gfx90a;gfx942;gfx1100"
+if [ "${build_os:0:6}" == ubuntu ]; then
+	image=rocm/dev-ubuntu-22.04:${rocm_version}-complete
+	echo "Using image $image"
+	docker run --rm --platform "linux/$build_arch" -i \
+		-w /src -v "$PWD:/src" "$image" sh -c \
+		"apt-get update \
+      && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends cmake \
+      && cmake -DCOMPUTE_BACKEND=hip -DBNB_ROCM_ARCH=\"${bnb_rocm_arch}\" . \
+      && cmake --build ."
+fi
+
+output_dir="output/${build_os}/${build_arch}"
+mkdir -p "${output_dir}"
+(shopt -s nullglob && cp bitsandbytes/*.{so,dylib,dll} "${output_dir}")

.github/workflows/python-package.yml

@@ -102,10 +102,55 @@ jobs:
           path: output/*
           retention-days: 7
 
+  build-shared-libs-rocm:
+    strategy:
+      matrix:
+        os: [ubuntu-22.04]
+        arch: [x86_64]
+        rocm_version:
+          ["6.1.2", "6.2.4", "6.3.2"]
+    runs-on: ${{ matrix.os }}
+    steps:
+      - uses: actions/checkout@v4
+      - name: Set up Docker multiarch
+        uses: docker/setup-qemu-action@v3
+      - name: Clean up disk space
+        run: |
+          sudo rm -rf \
+              /usr/share/dotnet \
+              /opt/ghc \
+              "/usr/local/share/boost" \
+              "$AGENT_TOOLSDIRECTORY" \
+              /opt/hostedtoolcache \
+              /opt/google/chrome \
+              /opt/microsoft/msedge \
+              /opt/microsoft/powershell \
+              /opt/pipx \
+              /usr/lib/mono \
+              /usr/local/julia* \
+              /usr/local/lib/android \
+              /usr/local/lib/node_modules \
+              /usr/local/share/chromium \
+              /usr/local/share/powershell \
+              /usr/share/swift
+      - name: Build C++
+        run: bash .github/scripts/build-rocm.sh
+        env:
+          build_os: ${{ matrix.os }}
+          build_arch: ${{ matrix.arch }}
+          rocm_version: ${{ matrix.rocm_version }}
+      - name: Upload build artifact
+        uses: actions/upload-artifact@v4
+        with:
+          name: shared_library_rocm_${{ matrix.os }}_${{ matrix.arch }}_${{ matrix.rocm_version }}
+          path: output/*
+          retention-days: 7
+
   build-wheels:
     needs:
       - build-shared-libs
       - build-shared-libs-cuda
+      - build-shared-libs-rocm
     strategy:
       matrix:
         os: [ubuntu-22.04, ubuntu-22.04-arm, windows-latest, macos-latest]
@@ -171,10 +216,17 @@ jobs:
           path: tmp/
           pattern: "bdist_wheel_*"
           merge-multiple: true
-  
+
       - name: Inspect tmp directory after downloading artifacts
-        run: ls -alFR tmp/
-
+        run: |
+          ls -alFR tmp/
+          WHEEL_COUNT=$(find tmp/ -type f -name "*.whl" | wc -l)
+          echo "Found $WHEEL_COUNT wheel files"
+          if [ "$WHEEL_COUNT" -eq 0 ]; then
+            echo "::error::No wheel files found in tmp directory! Cannot proceed with release."
+            exit 1
+          fi
+
       - name: Move and rename wheel files with pattern replacement
         run: |
           mkdir -p wheels/
@@ -199,37 +251,67 @@ jobs:
 
       - name: Inspect wheels directory after renaming files
         run: ls -alFR wheels/
-
+
+      - uses: actions/checkout@v4
+        with:
+          path: repo
       - name: Delete old pre-release (if exists)
         run: |
-          gh release delete continuous-release_main --cleanup-tag -y || true
+          cd repo && gh release delete continuous-release_main --cleanup-tag -y
+        env:
+          GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+
+      - name: Ensure tag exists
+        run: |
+          cd repo
+          git tag -f continuous-release_main
+          git push -f origin continuous-release_main
         env:
           GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
 
       - name: Generate pip install commands for release body
         run: |
           cat > body.md << 'ENDOFMARKDOWN'
-          ## Latest `main` Wheel Pre-release
+          ## Latest `main` pre-release wheel
 
           This pre-release contains the latest development wheels for all supported platforms, rebuilt automatically on every commit to the `main` branch.
 
-          **How to install:**  
+          **How to install:**
           Pick the correct command for your platform and run it in your terminal:
 
           ENDOFMARKDOWN
 
           for whl in wheels/*.whl; do
             fname=$(basename "$whl")
             url="https://github.com/bitsandbytes-foundation/bitsandbytes/releases/download/continuous-release_main/$fname"
+
+            if [[ "$fname" == *"manylinux_2_24_x86_64"* ]]; then
+              echo "### Linux (x86_64)" >> body.md
+            elif [[ "$fname" == *"manylinux_2_24_aarch64"* ]]; then
+              echo "### Linux (ARM/aarch64)" >> body.md
+            elif [[ "$fname" == *"win_amd64"* ]]; then
+              echo "### Windows (x86_64)" >> body.md
+            else
+              echo "### Other platform" >> body.md
+            fi
+
             echo "\`\`\`sh" >> body.md
-            echo "pip install $url" >> body.md
+            echo "pip install --force-reinstall $url" >> body.md
             echo "\`\`\`" >> body.md
             echo "" >> body.md
           done
 
           cat >> body.md << 'ENDOFMARKDOWN'
-          > **Note:**  
+          > **Note:**
           > These wheels are updated automatically with every commit to `main` and become available as soon as the [python-package.yml](.github/workflows/python-package.yml) workflow finishes.
+
+          The version number is replaced with 1.33.7-preview in order to keep the link stable, this however does not affect the installed version at all:
+          ```
+          > pip install https://.../bitsandbytes-1.33.7-preview-py3-none-manylinux_2_24_x86_64.whl
+          Collecting bitsandbytes==1.33.7rc0
+          ...
+          Successfully installed bitsandbytes-0.46.0.dev0
+          ```
           ENDOFMARKDOWN
 
           # for debugging:
@@ -245,7 +327,6 @@ jobs:
           tag_name: continuous-release_main
           make_latest: false
           draft: false
-          target_commitish: ${{ github.sha }}
 
   audit-wheels:
     needs: build-wheels