From dde93ed1d41bb9fb9f745af4fb285074de7f8fb4 Mon Sep 17 00:00:00 2001
From: krishung5 <krish@nvidia.com>
Date: Fri, 15 Nov 2024 14:12:01 -0800
Subject: [PATCH] Address comment

---
 qa/L0_dyna_implicit_state/test.sh | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/qa/L0_dyna_implicit_state/test.sh b/qa/L0_dyna_implicit_state/test.sh
index 1e21853587..7ff3e31060 100755
--- a/qa/L0_dyna_implicit_state/test.sh
+++ b/qa/L0_dyna_implicit_state/test.sh
@@ -34,6 +34,9 @@ if [ -z "$REPO_VERSION" ]; then
     echo -e "\n***\n*** Test Failed\n***"
     exit 1
 fi
+if [ ! -z "$TEST_REPO_ARCH" ] && [[ "$REPO_VERSION" == *"$TEST_REPO_ARCH"* ]]; then
+    REPO_VERSION=${REPO_VERSION}_${TEST_REPO_ARCH}
+fi
 
 export ENSEMBLES=0
 BACKENDS=${BACKENDS:="onnx plan"}