From 06b577b1e772d5bdeb1edd45936e15d64a706e4b Mon Sep 17 00:00:00 2001
From: "M. Chornyi" <99709299+mc-nv@users.noreply.github.com>
Date: Sat, 11 Jul 2026 07:49:00 -0700
Subject: [PATCH] fix: Enable OpenVINO model generation for SBSA

---
 qa/common/gen_qa_model_repository | 38 +++++++++++++++----------------
 1 file changed, 18 insertions(+), 20 deletions(-)

diff --git a/qa/common/gen_qa_model_repository b/qa/common/gen_qa_model_repository
index 7d4d02a535..6b752479d8 100755
--- a/qa/common/gen_qa_model_repository
+++ b/qa/common/gen_qa_model_repository
@@ -470,30 +470,28 @@ if [ "$TRITON_MODELS_USE_DOCKER" -eq 1 ] && which docker ; then
     log_message.status "docker container: copy script to container"
     docker cp . $DOCKER_VOLUME_CONTAINER:/mnt/$CI_JOB_ID/gen_srcdir
 
-    if [[ "aarch64" != $(uname -m) ]] ; then
 
-        log_message.status "docker run: $OPENVINOSCRIPT"
-        log_message.info "docker run $DOCKER_GPU_ARGS -v $DOCKER_VOLUME:/mnt $UBUNTU_IMAGE bash -e $TRITON_MDLS_SRC_DIR/$OPENVINOSCRIPT"
-        docker run \
-            --rm \
-            -e TRITON_GENSRCDIR=$TRITON_MDLS_SRC_DIR \
-            --label RUNNER_ID=$RUNNER_ID \
-            --label PROJECT_NAME=$PROJECT_NAME \
-            $DOCKER_GPU_ARGS \
-            -v $DOCKER_VOLUME:/mnt \
-            -t \
-            $UBUNTU_IMAGE \
-            bash -e $TRITON_MDLS_SRC_DIR/$OPENVINOSCRIPT
+    log_message.status "docker run: $OPENVINOSCRIPT"
+    log_message.info "docker run $DOCKER_GPU_ARGS -v $DOCKER_VOLUME:/mnt $UBUNTU_IMAGE bash -e $TRITON_MDLS_SRC_DIR/$OPENVINOSCRIPT"
+    docker run \
+        --rm \
+        -e TRITON_GENSRCDIR=$TRITON_MDLS_SRC_DIR \
+        --label RUNNER_ID=$RUNNER_ID \
+        --label PROJECT_NAME=$PROJECT_NAME \
+        $DOCKER_GPU_ARGS \
+        -v $DOCKER_VOLUME:/mnt \
+        -t \
+        $UBUNTU_IMAGE \
+        bash -e $TRITON_MDLS_SRC_DIR/$OPENVINOSCRIPT
 
-        exit_code=$?
+    exit_code=$?
 
-        if [ $exit_code -ne 0 ]; then
-            log_message.error "docker run: ${OPENVINOSCRIPT} failed"
-            exit 1
-        fi
+    if [ $exit_code -ne 0 ]; then
+        log_message.error "docker run: ${OPENVINOSCRIPT} failed"
+        exit 1
+    fi
 
-        rm $OPENVINOSCRIPT
-    fi # [[ "aarch64" != $(uname -m) ]]
+    rm $OPENVINOSCRIPT
 
     log_message.status "docker run: $ONNXSCRIPT"
     log_message.info "docker run $DOCKER_GPU_ARGS -v $DOCKER_VOLUME:/mnt $UBUNTU_IMAGE bash -e $TRITON_MDLS_SRC_DIR/$ONNXSCRIPT"