Skip to content

Commit 2efa294

Browse files
committed
Set custom image pull secret in maxtext GKE workflow
1 parent 293f0dc commit 2efa294

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

.github/workflows/_test_maxtext_gke_xpk.yaml

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,11 +24,21 @@ jobs:
2424
NUM_NODES: 2
2525

2626
steps:
27+
- name: Login to nvcr.io Container Registry
28+
uses: docker/login-action@v3
29+
with:
30+
registry: nvcr.io
31+
username: '$oauthtoken'
32+
password: ${{ secrets.NVCR_TOKEN }}
33+
- name: K8s GHCR store and delete token
34+
id: store-token
35+
uses: ./.github/actions/store-delete-k8s-ghcr
2736
- name: Run XPK workload on cluster
2837
uses: ./.github/actions/gke-xpk
2938
with:
3039
IMAGE: ${{ env.MAXTEXT_IMAGE }}
3140
WORKLOAD_NAME_PREFIX: ${{ env.WORKLOAD_NAME_PREFIX }}
41+
IMAGE_PULL_SECRET_NAME: ${{ steps.store-token.outputs.token-name }}
3242
COMMAND: |
3343
export NCCL_NET_PLUGIN=/opt/hpcx/nccl_rdma_sharp_plugin/lib/libnccl-net.so;
3444
export NCCL_TUNER_PLUGIN=none;

0 commit comments

Comments
 (0)