-
Notifications
You must be signed in to change notification settings - Fork 4
53 lines (50 loc) · 1.98 KB
/
torch-nccl.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
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.image.cuda, matrix.image.os, matrix.image.nccl), inputs.image-tag-suffix || format('torch{0}-vision{1}-audio{2}', matrix.torch, matrix.vision, matrix.audio)) }}
builder-base-image: ghcr.io/coreweave/nccl-tests:${{ matrix.image.cuda }}-${{ matrix.image.cudnn }}-devel-${{ matrix.image.os }}-nccl${{ matrix.image.nccl }}-${{ matrix.image.nccl-tests-hash }}
base-image: ghcr.io/coreweave/nccl-tests:${{ matrix.image.cuda }}-${{ matrix.image.cudnn }}-devel-${{ matrix.image.os }}-nccl${{ matrix.image.nccl }}-${{ matrix.image.nccl-tests-hash }}
torch-version: ${{ matrix.torch }}
torchvision-version: ${{ matrix.vision }}
torchaudio-version: ${{ matrix.audio }}
cache-key: nccl-cuda${{ matrix.image.cuda }}-${{ matrix.image.os }}
build-extras: true