From 2e6810a89ee99f8a7a57b06f8bd9c847ac3205c7 Mon Sep 17 00:00:00 2001 From: Kajal Jain Date: Mon, 18 May 2026 18:29:28 -0500 Subject: [PATCH] ci: build container on Azure for main builds Add a parallel build-container-azure job that builds and pushes the container image to the Azure registry (nemoci.azurecr.io) on every main branch build. This runs alongside the existing GCP build since the architectures differ (GCP=ARM64, Azure=x86_64) and images cannot be copied across. External contributor PR builds use Azure runners with inline cache from nemoci.azurecr.io/rl:main. Without this job, the Azure :main tag goes stale whenever the Dockerfile changes, causing cache misses that can fail with nvcr.io rate-limit 401 errors. Co-Authored-By: Claude Opus 4.6 Signed-off-by: Kajal Jain --- .github/workflows/cicd-main.yml | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/.github/workflows/cicd-main.yml b/.github/workflows/cicd-main.yml index e1c802b5bf..b9e08d9a81 100644 --- a/.github/workflows/cicd-main.yml +++ b/.github/workflows/cicd-main.yml @@ -335,6 +335,27 @@ jobs: docker rmi "${SRC}" "${DST}" 2>/dev/null || true rm -rf /tmp/uv-cache /tmp/Dockerfile.uv-cache + build-container-azure: + if: >- + ${{ + github.ref == 'refs/heads/main' && + needs.pre-flight.outputs.test_level != 'none' && + needs.pre-flight.outputs.image_tag == '' + }} + needs: [pre-flight, org-member-pre-flight] + uses: NVIDIA-NeMo/FW-CI-templates/.github/workflows/_build_container.yml@v0.78.0 + with: + build-ref: ${{ needs.pre-flight.outputs.test_sha }} + image-name: ${{ vars.CI_CONTAINER_NAME }} + dockerfile: docker/Dockerfile + runner: ${{ format('{0}-gpu-x2', vars.NON_NVIDIA_RUNNER_PREFIX) }} + image-label: ${{ vars.CI_CONTAINER_NAME }} + target: release + registry: ${{ vars.NON_NVIDIA_CONTAINER_REGISTRY }} + build-args: | + MAX_JOBS=4 + NEMO_RL_COMMIT=${{ needs.pre-flight.outputs.test_sha }} + cicd-doc-tests: strategy: fail-fast: false