Add PLATFORMS to 12.1 workflow

NVIDIA · Oct 10, 2023 · 6eaf361 · 6eaf361
1 parent eb14bed
commit 6eaf361
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/.github/workflows/pax-cuda-121.yaml b/.github/workflows/pax-cuda-121.yaml
@@ -65,6 +65,7 @@ jobs:
       BUILD_DATE: ${{ needs.metadata.outputs.BUILD_DATE }}
       BASE_IMAGE: ${{ needs.build-pax.outputs.DOCKER_TAGS }}
       BASE_LIBRARY: pax
+      PLATFORMS: '["amd64"]'
     secrets: inherit
 
   build-t5x:
@@ -82,6 +83,7 @@ jobs:
       BUILD_DATE: ${{ needs.metadata.outputs.BUILD_DATE }}
       BASE_IMAGE: ${{ needs.build-pax.outputs.DOCKER_TAGS }}
       BASE_LIBRARY: t5x
+      PLATFORMS: '["amd64"]'
     secrets: inherit
 
   build-summary: