Update publish.yml #2
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: Build wheels in a container | |
| on: | |
| workflow_dispatch: | |
| inputs: | |
| runs-on: | |
| description: "The runner to use for the build" | |
| required: true | |
| type: string | |
| default: ubuntu-22.04 | |
| container-image: | |
| description: "Container image" | |
| required: true | |
| type: string | |
| upload-to-release: | |
| description: "Upload wheel to this release" | |
| required: false | |
| type: boolean | |
| default: false | |
| release-version: | |
| description: "Upload wheel to this release" | |
| required: false | |
| type: string | |
| push: | |
| jobs: | |
| 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)" >> $GITHUB_OUTPUT | |
| build-wheels: | |
| needs: [check_for_ngc_images] | |
| uses: ./.github/workflows/_build_in_container.yml | |
| strategy: | |
| fail-fast: false | |
| matrix: | |
| container-image: ${{ fromJson(needs.check_for_ngc_images.outputs.images) }} | |
| with: | |
| runs-on: ${{ inputs.runs-on || 'ubuntu-22.04' }} | |
| container-image: ${{ matrix.container-image }} | |
| upload-to-release: ${{ inputs.upload-to-release || false }} | |
| release-version: ${{ inputs.release-version || 'v2.2.5' }} |