Multiple updates and refactorings (#231)

.github/workflows/publish.yml

@@ -41,9 +41,9 @@ jobs:
         # Using ubuntu-22.04 instead of 24.04 for more compatibility (glibc). Ideally we'd use the
         # manylinux docker image, but I haven't figured out how to install CUDA on manylinux.
         os: [ubuntu-22.04]
-        python-version: ["3.9", "3.10", "3.11", "3.12", "3.13"]
-        torch-version: ["2.5.1", "2.6.0", "2.7.1", "2.8.0"]
-        cuda-version: ["12.9.1", "13.0.0"]
+        python-version: ["3.8", "3.9", "3.10", "3.11", "3.12", "3.13"]
+        torch-version: ["2.4.0", "2.5.1", "2.6.0", "2.7.1", "2.8.0"]
+        cuda-version: ["12.9.1"]
         # We need separate wheels that either uses C++11 ABI (-D_GLIBCXX_USE_CXX11_ABI) or not.
         # Pytorch wheels currently don't use it, but nvcr images have Pytorch compiled with C++11 ABI.
         # Without this we get import error (undefined symbol: _ZN3c105ErrorC2ENS_14SourceLocationESs)
@@ -51,14 +51,9 @@ jobs:
         cxx11_abi: ["FALSE", "TRUE"]
         exclude:
           # see https://github.com/pytorch/pytorch/blob/main/RELEASE.md#release-compatibility-matrix
+          # Pytorch < 2.5 does not support Python 3.13
           - torch-version: "2.4.0"
             python-version: "3.13"
-        include:
-          - os: "ubuntu-22.04"
-            cxx11_abi: "FALSE"
-            cuda-version: "12.1.0"
-            python-version: "3.8"
-            torch-version: "2.1.0"
     uses: ./.github/workflows/_build.yml
     with:
       runs-on: ${{ matrix.os }}
@@ -69,35 +64,6 @@ jobs:
       release-version: ${{ needs.setup_release.outputs.release-version }}
       upload-to-release: true
       use-local-version: false
-  
-  check_for_ngc_images:
-    runs-on: ubuntu-latest
-    outputs:
-      images: ${{ steps.check_for_ngc_images.outputs.IMAGES }}
-    steps:
-      - name: Checkout repository
-        uses: actions/checkout@v4
-
-      - name: Check for NGC PyTorch images
-        id: check_for_ngc_images
-        run: |
-          bash ./.github/scripts/check_for_ngc_images.sh
-          echo "IMAGES=$(cat ngc_images.json| jq -cr)" | tee -a $GITHUB_OUTPUT
-
-  build_ngc_wheels:
-    name: Build Wheel for NGC PyTorch
-    needs: [setup_release, check_for_ngc_images]
-    strategy:
-      fail-fast: false
-      matrix:
-        os: [ubuntu-22.04]
-        container-image: ${{ fromJson(needs.check_for_ngc_images.outputs.images) }}
-    uses: ./.github/workflows/_build_in_container.yml
-    with:
-      runs-on: ${{ matrix.os }}
-      container-image: ${{ matrix.container-image }}
-      release-version: ${{ needs.setup_release.outputs.release-version }}
-      upload-to-release: true
 
   publish_package:
     name: Publish package