From a06e090d899768f8d373468a66c491d63d64ad67 Mon Sep 17 00:00:00 2001
From: kthui <18255193+kthui@users.noreply.github.com>
Date: Wed, 29 May 2024 10:18:21 -0700
Subject: [PATCH] Set BLS size to 256 and turn off restart

---
 qa/L0_backend_python/bls/test.sh | 2 +-
 qa/L0_backend_python/test.sh     | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/qa/L0_backend_python/bls/test.sh b/qa/L0_backend_python/bls/test.sh
index f4435eacaa..3dfd217d9d 100755
--- a/qa/L0_backend_python/bls/test.sh
+++ b/qa/L0_backend_python/bls/test.sh
@@ -100,7 +100,7 @@ if [[ ${TEST_WINDOWS} == 0 ]]; then
         echo "instance_group [ { kind: KIND_CPU} ]" >> models/libtorch_cpu/config.pbtxt
 
     # Test with different sizes of CUDA memory pool
-    for CUDA_MEMORY_POOL_SIZE_MB in 64 128 ; do
+    for CUDA_MEMORY_POOL_SIZE_MB in 64 256 ; do
         CUDA_MEMORY_POOL_SIZE_BYTES=$((CUDA_MEMORY_POOL_SIZE_MB * 1024 * 1024))
         SERVER_ARGS="--model-repository=${MODELDIR}/bls/models --backend-directory=${BACKEND_DIR} --log-verbose=1 --cuda-memory-pool-byte-size=0:${CUDA_MEMORY_POOL_SIZE_BYTES}"
         for TRIAL in non_decoupled decoupled ; do
diff --git a/qa/L0_backend_python/test.sh b/qa/L0_backend_python/test.sh
index bbaabbaf10..755457d952 100755
--- a/qa/L0_backend_python/test.sh
+++ b/qa/L0_backend_python/test.sh
@@ -448,7 +448,7 @@ SUBTESTS="lifecycle argument_validation logging custom_metrics"
 # [DLIS-6122] Disable model_control & request_rescheduling tests for Windows since they require load/unload
 # [DLIS-6123] Disable examples test for Windows since it requires updates to the example clients
 if [[ ${TEST_WINDOWS} == 0 ]]; then
-    SUBTESTS+=" restart model_control examples request_rescheduling"
+    SUBTESTS+=" model_control examples request_rescheduling"
 fi
 for TEST in ${SUBTESTS}; do
     # Run each subtest in a separate virtual environment to avoid conflicts