Update container ver and allow running local ci for a specific arch

nv-morpheus · Dec 20, 2024 · 1cdaaf8 · 1cdaaf8
1 parent bab4a65
commit 1cdaaf8
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/ci/scripts/run_ci_local.sh b/ci/scripts/run_ci_local.sh
@@ -46,6 +46,7 @@ function git_ssh_to_https()
     echo $url | sed -e 's|^git@github\.com:|https://github.com/|'
 }
 
+CI_ARCH=${CI_ARCH:-$(dpkg --print-architecture)}
 MRC_ROOT=${MRC_ROOT:-$(git rev-parse --show-toplevel)}
 
 GIT_URL=$(git remote get-url origin)
@@ -58,7 +59,7 @@ GIT_BRANCH=$(git branch --show-current)
 GIT_COMMIT=$(git log -n 1 --pretty=format:%H)
 
 BASE_LOCAL_CI_TMP=${BASE_LOCAL_CI_TMP:-${MRC_ROOT}/.tmp/local_ci_tmp}
-CONTAINER_VER=${CONTAINER_VER:-241002}
+CONTAINER_VER=${CONTAINER_VER:-241219}
 CUDA_VER=${CUDA_VER:-12.5}
 DOCKER_EXTRA_ARGS=${DOCKER_EXTRA_ARGS:-""}
 
@@ -99,7 +100,7 @@ for STAGE in "${STAGES[@]}"; do
     cp ${MRC_ROOT}/ci/scripts/bootstrap_local_ci.sh ${LOCAL_CI_TMP}
 
 
-    DOCKER_RUN_ARGS="--rm -ti --net=host -v "${LOCAL_CI_TMP}":/ci_tmp ${ENV_LIST} --env STAGE=${STAGE}"
+    DOCKER_RUN_ARGS="--rm -ti --net=host --platform=linux/${CI_ARCH} -v "${LOCAL_CI_TMP}":/ci_tmp ${ENV_LIST} --env STAGE=${STAGE}"
     if [[ "${STAGE}" =~ "test" || "${STAGE}" =~ "codecov" || "${USE_GPU}" == "1" ]]; then
         CONTAINER="${TEST_CONTAINER}"
         DOCKER_RUN_ARGS="${DOCKER_RUN_ARGS} --runtime=nvidia --gpus all --cap-add=sys_nice --cap-add=sys_ptrace"