From 2a4c024ec6732ec1c34755d2ebc53aa16b74f7ac Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?oliver=20k=C3=B6nig?= Date: Thu, 29 Jan 2026 22:05:01 +0000 Subject: [PATCH 01/12] ci(fix): Pass release-version to NGC build MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: oliver könig --- .../workflows/attach-wheels-to-release.yml | 44 +++++++++---------- 1 file changed, 22 insertions(+), 22 deletions(-) diff --git a/.github/workflows/attach-wheels-to-release.yml b/.github/workflows/attach-wheels-to-release.yml index fb56622d77..12f890ba50 100644 --- a/.github/workflows/attach-wheels-to-release.yml +++ b/.github/workflows/attach-wheels-to-release.yml @@ -17,43 +17,43 @@ on: workflow_dispatch: inputs: runs-on: - description: 'The runner to use for the build' + description: "The runner to use for the build" required: true type: string default: ubuntu-22.04 release-version: - description: 'Release version' + description: "Release version" required: true - default: '0.1.0' + default: "0.1.0" python-version: - description: 'Python version' + description: "Python version" required: true - default: '3.12' + default: "3.12" torch-version: - description: 'Torch version' + description: "Torch version" required: true - default: '2.8.0' + default: "2.8.0" cuda-version: - description: 'CUDA version' + description: "CUDA version" required: true - default: '12.9.1' + default: "12.9.1" cudnn-version: - description: 'CUDNN version' + description: "CUDNN version" required: true - default: '9' + default: "9" cxx11_abi: - description: 'C++11 ABI' + description: "C++11 ABI" required: true type: choice - default: 'TRUE' + default: "TRUE" options: - - 'TRUE' - - 'FALSE' + - "TRUE" + - "FALSE" ngc-image: - description: 'NGC PyTorch image (will take precedence over the source build)' + description: "NGC PyTorch image (will take precedence over the source build)" required: false type: string - default: '' + default: "" jobs: pre-flight: runs-on: ubuntu-latest @@ -137,10 +137,10 @@ jobs: fail-fast: false matrix: ${{ fromJson(needs.pre-flight.outputs.build-wheel-matrix) }} steps: - - name: 'Checkout' + - name: "Checkout" uses: actions/checkout@v3 - - name: 'Build PyTorch Wheel' + - name: "Build PyTorch Wheel" uses: ./.github/actions/build-pytorch-wheel id: build-pytorch-wheel with: @@ -174,16 +174,16 @@ jobs: strategy: fail-fast: false matrix: - os: [ubuntu-22.04] container-image: ${{ fromJson(needs.pre-flight.outputs.ngc-images) }} steps: - - name: 'Checkout' + - name: "Checkout" uses: actions/checkout@v3 - - name: 'Build PyTorch Wheel' + - name: "Build PyTorch Wheel" uses: ./.github/actions/build-pytorch-wheel id: build-pytorch-wheel with: + release-version: ${{ matrix.release-version }} base-image: ${{ matrix.container-image }} - name: Upload Release Asset From c50f6c29acb57818881073abdc76729e909cac03 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?oliver=20k=C3=B6nig?= Date: Thu, 29 Jan 2026 22:06:58 +0000 Subject: [PATCH 02/12] fix MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: oliver könig --- .github/workflows/attach-wheels-to-release.yml | 2 ++ 1 file changed, 2 insertions(+) diff --git a/.github/workflows/attach-wheels-to-release.yml b/.github/workflows/attach-wheels-to-release.yml index 12f890ba50..7c524c6a4c 100644 --- a/.github/workflows/attach-wheels-to-release.yml +++ b/.github/workflows/attach-wheels-to-release.yml @@ -174,6 +174,8 @@ jobs: strategy: fail-fast: false matrix: + os: [ubuntu-22.04] + release-version: ${{ inputs.release-version }} container-image: ${{ fromJson(needs.pre-flight.outputs.ngc-images) }} steps: - name: "Checkout" From 036772b66c637a6e7ec50246da0eac1bc0bdca96 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?oliver=20k=C3=B6nig?= Date: Thu, 29 Jan 2026 22:07:47 +0000 Subject: [PATCH 03/12] fix MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: oliver könig --- .github/workflows/attach-wheels-to-release.yml | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/.github/workflows/attach-wheels-to-release.yml b/.github/workflows/attach-wheels-to-release.yml index 7c524c6a4c..dcd3b35a49 100644 --- a/.github/workflows/attach-wheels-to-release.yml +++ b/.github/workflows/attach-wheels-to-release.yml @@ -175,7 +175,6 @@ jobs: fail-fast: false matrix: os: [ubuntu-22.04] - release-version: ${{ inputs.release-version }} container-image: ${{ fromJson(needs.pre-flight.outputs.ngc-images) }} steps: - name: "Checkout" @@ -185,7 +184,7 @@ jobs: uses: ./.github/actions/build-pytorch-wheel id: build-pytorch-wheel with: - release-version: ${{ matrix.release-version }} + release-version: ${{ inputs.release-version }} base-image: ${{ matrix.container-image }} - name: Upload Release Asset From b16b612b75dd86a3c28a3b48fc021d173414db5c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?oliver=20k=C3=B6nig?= Date: Thu, 29 Jan 2026 22:11:26 +0000 Subject: [PATCH 04/12] fix MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: oliver könig --- .github/workflows/attach-wheels-to-release.yml | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/.github/workflows/attach-wheels-to-release.yml b/.github/workflows/attach-wheels-to-release.yml index dcd3b35a49..027c9349be 100644 --- a/.github/workflows/attach-wheels-to-release.yml +++ b/.github/workflows/attach-wheels-to-release.yml @@ -173,9 +173,7 @@ jobs: if: ${{ github.event_name == 'release' || inputs.ngc-image != '' }} strategy: fail-fast: false - matrix: - os: [ubuntu-22.04] - container-image: ${{ fromJson(needs.pre-flight.outputs.ngc-images) }} + matrix: ${{ fromJson(needs.pre-flight.outputs.build-wheel-matrix) }} steps: - name: "Checkout" uses: actions/checkout@v3 @@ -184,7 +182,7 @@ jobs: uses: ./.github/actions/build-pytorch-wheel id: build-pytorch-wheel with: - release-version: ${{ inputs.release-version }} + release-version: ${{ matrix.release-version }} base-image: ${{ matrix.container-image }} - name: Upload Release Asset From e3f71e42504987a21e7e795355baec9ca2b881dc Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?oliver=20k=C3=B6nig?= Date: Thu, 29 Jan 2026 22:17:06 +0000 Subject: [PATCH 05/12] fix MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: oliver könig --- .../workflows/attach-wheels-to-release.yml | 20 ++++++++++++++----- 1 file changed, 15 insertions(+), 5 deletions(-) diff --git a/.github/workflows/attach-wheels-to-release.yml b/.github/workflows/attach-wheels-to-release.yml index 027c9349be..9fa79d3b8a 100644 --- a/.github/workflows/attach-wheels-to-release.yml +++ b/.github/workflows/attach-wheels-to-release.yml @@ -60,7 +60,7 @@ jobs: outputs: build-wheel-matrix: ${{ steps.matrix.outputs.matrix }} release-assets-url: ${{ steps.release-assets-url.outputs.upload_url }} - ngc-images: ${{ steps.check_for_ngc_images.outputs.IMAGES }} + ngc-images-matrix: ${{ steps.check_for_ngc_images.outputs.matrix }} steps: - name: Checkout repository uses: actions/checkout@v4 @@ -123,11 +123,21 @@ jobs: run: | if [[ "$EVENT" == "release" ]]; then bash ./.github/scripts/check_for_ngc_images.sh - echo "IMAGES=$(cat ngc_images.json | jq -cr)" | tee -a $GITHUB_OUTPUT + MATRIX=$(echo '{ + "os": ["${{ inputs.runs-on }}"], + "release-version": ["${{ inputs.release-version }}"], + "image": "$(cat ngc_images.json | jq -cr)", + }' | jq -rc) else - echo 'IMAGES=["${{ inputs.ngc-image }}"]' | tee -a "$GITHUB_OUTPUT" + MATRIX=$(echo '{ + "os": ["${{ inputs.runs-on }}"], + "release-version": ["${{ inputs.release-version }}"], + "image": ["${{ inputs.ngc-image }}"], + }' | jq -rc) fi + echo "matrix=$MATRIX" | tee -a "$GITHUB_OUTPUT" + build_wheels: name: Build Wheel runs-on: ${{ matrix.os }} @@ -173,7 +183,7 @@ jobs: if: ${{ github.event_name == 'release' || inputs.ngc-image != '' }} strategy: fail-fast: false - matrix: ${{ fromJson(needs.pre-flight.outputs.build-wheel-matrix) }} + matrix: ${{ fromJson(needs.pre-flight.outputs.ngc-images-matrix) }} steps: - name: "Checkout" uses: actions/checkout@v3 @@ -182,7 +192,7 @@ jobs: uses: ./.github/actions/build-pytorch-wheel id: build-pytorch-wheel with: - release-version: ${{ matrix.release-version }} + release-version: ${{ inputs.release-version }} base-image: ${{ matrix.container-image }} - name: Upload Release Asset From d967e2d3ee24e6a15e1c5c2661208ff626bf4934 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?oliver=20k=C3=B6nig?= Date: Thu, 29 Jan 2026 22:18:35 +0000 Subject: [PATCH 06/12] fix MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: oliver könig --- .github/workflows/attach-wheels-to-release.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/attach-wheels-to-release.yml b/.github/workflows/attach-wheels-to-release.yml index 9fa79d3b8a..a98a81c586 100644 --- a/.github/workflows/attach-wheels-to-release.yml +++ b/.github/workflows/attach-wheels-to-release.yml @@ -126,13 +126,13 @@ jobs: MATRIX=$(echo '{ "os": ["${{ inputs.runs-on }}"], "release-version": ["${{ inputs.release-version }}"], - "image": "$(cat ngc_images.json | jq -cr)", + "image": "$(cat ngc_images.json | jq -cr)" }' | jq -rc) else MATRIX=$(echo '{ "os": ["${{ inputs.runs-on }}"], "release-version": ["${{ inputs.release-version }}"], - "image": ["${{ inputs.ngc-image }}"], + "image": ["${{ inputs.ngc-image }}"] }' | jq -rc) fi From 50f4a24486877ca4a55be5f17d82acf20276d034 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?oliver=20k=C3=B6nig?= Date: Thu, 29 Jan 2026 22:20:44 +0000 Subject: [PATCH 07/12] fix MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: oliver könig --- .github/workflows/attach-wheels-to-release.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/attach-wheels-to-release.yml b/.github/workflows/attach-wheels-to-release.yml index a98a81c586..a5f588000e 100644 --- a/.github/workflows/attach-wheels-to-release.yml +++ b/.github/workflows/attach-wheels-to-release.yml @@ -124,7 +124,7 @@ jobs: if [[ "$EVENT" == "release" ]]; then bash ./.github/scripts/check_for_ngc_images.sh MATRIX=$(echo '{ - "os": ["${{ inputs.runs-on }}"], + "os": ["ubuntu-22.04", "ubuntu-22.04-arm"], "release-version": ["${{ inputs.release-version }}"], "image": "$(cat ngc_images.json | jq -cr)" }' | jq -rc) From 75cfa4ca48ec5a7f838eca035b790c5a3ea5c8f4 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?oliver=20k=C3=B6nig?= Date: Thu, 29 Jan 2026 22:24:28 +0000 Subject: [PATCH 08/12] fix MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: oliver könig --- .github/workflows/attach-wheels-to-release.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/attach-wheels-to-release.yml b/.github/workflows/attach-wheels-to-release.yml index a5f588000e..42ecb44430 100644 --- a/.github/workflows/attach-wheels-to-release.yml +++ b/.github/workflows/attach-wheels-to-release.yml @@ -126,13 +126,13 @@ jobs: MATRIX=$(echo '{ "os": ["ubuntu-22.04", "ubuntu-22.04-arm"], "release-version": ["${{ inputs.release-version }}"], - "image": "$(cat ngc_images.json | jq -cr)" + "container-image": "$(cat ngc_images.json | jq -cr)" }' | jq -rc) else MATRIX=$(echo '{ "os": ["${{ inputs.runs-on }}"], "release-version": ["${{ inputs.release-version }}"], - "image": ["${{ inputs.ngc-image }}"] + "container-image": ["${{ inputs.ngc-image }}"] }' | jq -rc) fi From 7e6231208262ac3e7ad56376772425196541071a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?oliver=20k=C3=B6nig?= Date: Mon, 2 Feb 2026 23:55:11 +0000 Subject: [PATCH 09/12] permissions: write-all MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: oliver könig --- .github/workflows/attach-wheels-to-release.yml | 2 ++ 1 file changed, 2 insertions(+) diff --git a/.github/workflows/attach-wheels-to-release.yml b/.github/workflows/attach-wheels-to-release.yml index 42ecb44430..d5a50c2624 100644 --- a/.github/workflows/attach-wheels-to-release.yml +++ b/.github/workflows/attach-wheels-to-release.yml @@ -142,6 +142,7 @@ jobs: name: Build Wheel runs-on: ${{ matrix.os }} needs: pre-flight + permissions: write-all if: ${{ github.event_name == 'release' || inputs.ngc-image == '' }} strategy: fail-fast: false @@ -180,6 +181,7 @@ jobs: name: Build Wheels for NGC PyTorch images runs-on: ${{ matrix.os }} needs: pre-flight + permissions: write-all if: ${{ github.event_name == 'release' || inputs.ngc-image != '' }} strategy: fail-fast: false From 5fbed6cebe9a7f6ebb33a40d80aa3d264f37eb2f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?oliver=20k=C3=B6nig?= Date: Tue, 3 Feb 2026 01:04:02 +0100 Subject: [PATCH 10/12] Update .github/workflows/attach-wheels-to-release.yml MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com> Signed-off-by: oliver könig --- .github/workflows/attach-wheels-to-release.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/attach-wheels-to-release.yml b/.github/workflows/attach-wheels-to-release.yml index d5a50c2624..35c64e5874 100644 --- a/.github/workflows/attach-wheels-to-release.yml +++ b/.github/workflows/attach-wheels-to-release.yml @@ -125,7 +125,7 @@ jobs: bash ./.github/scripts/check_for_ngc_images.sh MATRIX=$(echo '{ "os": ["ubuntu-22.04", "ubuntu-22.04-arm"], - "release-version": ["${{ inputs.release-version }}"], + "release-version": ["${{ github.event.release.tag_name }}"], "container-image": "$(cat ngc_images.json | jq -cr)" }' | jq -rc) else From e7e8138fe0ac3a0199951b91ee84db08040c2351 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?oliver=20k=C3=B6nig?= Date: Tue, 3 Feb 2026 01:04:25 +0100 Subject: [PATCH 11/12] Update .github/workflows/attach-wheels-to-release.yml MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com> Signed-off-by: oliver könig --- .github/workflows/attach-wheels-to-release.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/attach-wheels-to-release.yml b/.github/workflows/attach-wheels-to-release.yml index 35c64e5874..9dcc3cde0c 100644 --- a/.github/workflows/attach-wheels-to-release.yml +++ b/.github/workflows/attach-wheels-to-release.yml @@ -126,7 +126,7 @@ jobs: MATRIX=$(echo '{ "os": ["ubuntu-22.04", "ubuntu-22.04-arm"], "release-version": ["${{ github.event.release.tag_name }}"], - "container-image": "$(cat ngc_images.json | jq -cr)" + "container-image": $(cat ngc_images.json | jq -cr) }' | jq -rc) else MATRIX=$(echo '{ From 04f305e90614bbe6ef9078fcfd0c10e868907475 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?oliver=20k=C3=B6nig?= Date: Tue, 3 Feb 2026 01:05:16 +0100 Subject: [PATCH 12/12] Update .github/workflows/attach-wheels-to-release.yml MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com> Signed-off-by: oliver könig --- .github/workflows/attach-wheels-to-release.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/attach-wheels-to-release.yml b/.github/workflows/attach-wheels-to-release.yml index 9dcc3cde0c..6283ab86ef 100644 --- a/.github/workflows/attach-wheels-to-release.yml +++ b/.github/workflows/attach-wheels-to-release.yml @@ -194,7 +194,7 @@ jobs: uses: ./.github/actions/build-pytorch-wheel id: build-pytorch-wheel with: - release-version: ${{ inputs.release-version }} + release-version: ${{ matrix.release-version }} base-image: ${{ matrix.container-image }} - name: Upload Release Asset