-
Notifications
You must be signed in to change notification settings - Fork 56
67 lines (56 loc) · 1.63 KB
/
_sandbox.yaml
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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
name: "~Sandbox"
on:
workflow_dispatch:
push:
permissions:
contents: read # to fetch code
actions: write # to cancel previous workflows
packages: write # to upload container
jobs:
# build-base:
# uses: ./.github/workflows/_build_base.yaml
# with:
# ARCHITECTURE: amd64
# secrets: inherit
# build-jax:
# needs: [build-base]
# uses: ./.github/workflows/_build_jax.yaml
# with:
# ARCHITECTURE: amd64
# BASE_IMAGE: ${{ needs.build-base.outputs.DOCKER_TAG }}
# secrets: inherit
# build-pax:
# needs: [build-jax]
# uses: ./.github/workflows/_build_pax.yaml
# with:
# ARCHITECTURE: amd64
# BASE_IMAGE: ${{ needs.build-jax.outputs.DOCKER_TAG_MEALKIT }}
# secrets: inherit
# build-t5x:
# needs: [build-jax]
# uses: ./.github/workflows/_build_t5x.yaml
# with:
# ARCHITECTURE: amd64
# BASE_IMAGE: ${{ needs.build-jax.outputs.DOCKER_TAG_MEALKIT }}
# secrets: inherit
# test-pax:
# needs: [build-pax]
# uses: ./.github/workflows/_test_pax.yaml
# with:
# PAX_IMAGE: ${{ needs.build-pax.outputs.DOCKER_TAG_FINAL }}
# secrets: inherit
# test-te:
# needs: [build-pax]
# uses: ./.github/workflows/_test_te.yaml
# with:
# TE_IMAGE: ${{ needs.build-pax.outputs.DOCKER_TAG_FINAL }}
# secrets: inherit
build-rosetta-pax:
uses: ./.github/workflows/_build_rosetta.yaml
# needs: [metadata, build-pax]
with:
ARCHITECTURE: amd64
BUILD_DATE: "2023-11-15"
BASE_IMAGE: ghcr.io/nvidia/jax-toolbox-internal:6896408819-upstream-pax-amd64-mealkit
BASE_LIBRARY: pax
secrets: inherit