From 1e11d8ecff7b696021073d10e35bd5b545b925c8 Mon Sep 17 00:00:00 2001
From: Himani Anil Deshpande <himanidp@amazon.com>
Date: Tue, 16 Sep 2025 13:53:22 -0400
Subject: [PATCH 1/2] [Gb200][Prolog] Using generic function name

---
 .../test_gb200/test_gb200/91_nvidia_imex_prolog.sh            | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/integration-tests/tests/ultraserver/test_gb200/test_gb200/91_nvidia_imex_prolog.sh b/tests/integration-tests/tests/ultraserver/test_gb200/test_gb200/91_nvidia_imex_prolog.sh
index dc2540217d..5cd11dad2e 100644
--- a/tests/integration-tests/tests/ultraserver/test_gb200/test_gb200/91_nvidia_imex_prolog.sh
+++ b/tests/integration-tests/tests/ultraserver/test_gb200/test_gb200/91_nvidia_imex_prolog.sh
@@ -34,7 +34,7 @@ function get_instance_type() {
   curl -s -H "X-aws-ec2-metadata-token: ${token}" http://169.254.169.254/latest/meta-data/instance-type
 }
 
-function return_unless_gb200_with_imex() {
+function return_if_unsupported_instance_type() {
   local instance_type=$(get_instance_type)
 
   if [[ ! ${instance_type} =~ ${ALLOWED_INSTANCE_TYPES} ]]; then
@@ -163,7 +163,7 @@ function create_default_imex_channel() {
 {
   info "PROLOG Start JobId=${SLURM_JOB_ID}: $0"
 
-  return_unless_gb200_with_imex
+  return_if_unsupported_instance_type
 
   create_default_imex_channel
 

From dd429f2e6f976d4857b2da2a995bb317c6018f50 Mon Sep 17 00:00:00 2001
From: Himani Anil Deshpande <himanidp@amazon.com>
Date: Tue, 16 Sep 2025 09:55:21 -0400
Subject: [PATCH 2/2] [Gb200][Prolog] Re-direct logs to syslogs

---
 .../ultraserver/test_gb200/test_gb200/91_nvidia_imex_prolog.sh  | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/integration-tests/tests/ultraserver/test_gb200/test_gb200/91_nvidia_imex_prolog.sh b/tests/integration-tests/tests/ultraserver/test_gb200/test_gb200/91_nvidia_imex_prolog.sh
index 5cd11dad2e..f90f2eca4c 100644
--- a/tests/integration-tests/tests/ultraserver/test_gb200/test_gb200/91_nvidia_imex_prolog.sh
+++ b/tests/integration-tests/tests/ultraserver/test_gb200/test_gb200/91_nvidia_imex_prolog.sh
@@ -188,4 +188,4 @@ function create_default_imex_channel() {
 
   prolog_end
 
-} >> "${LOG_FILE_PATH}" 2>&1
+} 2>&1 | tee -a "${LOG_FILE_PATH}" | logger -t "91_nvidia_imex_prolog"