ci(torch): Build with full torchaudio v2.7.0, update NCCL base images
#180
Workflow file for this run
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: torch-nccl | |
| on: | |
| workflow_call: | |
| inputs: | |
| image-name: | |
| required: false | |
| type: string | |
| image-tag-suffix: | |
| required: false | |
| type: string | |
| workflow_dispatch: | |
| inputs: | |
| image-name: | |
| required: false | |
| description: "Custom name under which to publish the resulting container" | |
| type: string | |
| image-tag-suffix: | |
| required: false | |
| description: "Custom tag suffix listing library versions under which to publish the resulting container" | |
| type: string | |
| push: | |
| paths: | |
| - "torch/**" | |
| - ".github/configurations/torch-nccl.yml" | |
| - ".github/workflows/torch-nccl.yml" | |
| - ".github/workflows/torch.yml" | |
| - ".github/workflows/build.yml" | |
| jobs: | |
| get-config: | |
| name: Get torch:nccl Config | |
| uses: ./.github/workflows/read-configuration.yml | |
| with: | |
| path: ./.github/configurations/torch-nccl.yml | |
| build: | |
| name: Build torch:nccl | |
| needs: get-config | |
| strategy: | |
| matrix: ${{ fromJSON(needs.get-config.outputs.config) }} | |
| uses: ./.github/workflows/torch.yml | |
| secrets: inherit | |
| with: | |
| image-name: ${{ inputs.image-name }} | |
| tag: ${{ format('{0}-{1}', format('nccl-cuda{0}-{1}-nccl{2}', matrix.cuda, matrix.os, matrix.nccl), inputs.image-tag-suffix || format('torch{0}-vision{1}-audio{2}-abi{3}', matrix.torch, matrix.vision, matrix.audio, matrix.abi)) }} | |
| builder-base-image: ghcr.io/coreweave/nccl-tests:${{ matrix.cuda }}-devel-${{ matrix.os }}-nccl${{ matrix.nccl }}-${{ matrix.nccl-tests-hash }} | |
| base-image: ghcr.io/coreweave/nccl-tests:${{ matrix.cuda }}-devel-${{ matrix.os }}-nccl${{ matrix.nccl }}-${{ matrix.nccl-tests-hash }} | |
| torch-version: ${{ matrix.torch }} | |
| torchvision-version: ${{ matrix.vision }} | |
| torchaudio-version: ${{ matrix.audio }} | |
| additional-build-args: BUILD_CXX11_ABI=${{ matrix.abi }} | |
| cache-key: nccl-cuda${{ matrix.cuda }}-${{ matrix.os }} | |
| build-extras: true |