diff --git a/projects/rocshmem/scripts/functional_tests/driver.sh b/projects/rocshmem/scripts/functional_tests/driver.sh index 865b2fb78f..ff200c76af 100755 --- a/projects/rocshmem/scripts/functional_tests/driver.sh +++ b/projects/rocshmem/scripts/functional_tests/driver.sh @@ -133,6 +133,7 @@ ExecTest() { then NUM_GPUS=${NUM_GPUS:-$(rocm-smi --showserial | grep GPU | wc -l)} fi + NUM_GPUS=${NUM_GPUS:-0} NUM_GPUS=$(($NUM_GPUS > 0? $NUM_GPUS: 8)) TEST_NUM=${TEST_NUMBERS[$TEST_NAME]}