add check for existance of output dir for memtest

Author: PrometheusPi

bin/cuda_memtest.sh

@@ -39,8 +39,13 @@ output=`cuda_memtest --disable_all --device $host_rank $enable_gpu_tests --num_p
 
 if [ $? -ne 0 ] ; then
    host_name=`hostname`
-   echo -e "$output" > $old_path/cuda_memtest_"$host_name"_"$host_rank".err
-   echo cuda_memtest crash: see file $old_path/cuda_memtest_"$host_name"_"$host_rank".err >&2
-   exit 1
+   if [ ! -d "$old_path" ]; then
+      echo "Error: $0 did not find directory: $old_path (on host: $host_name with rank: $host_rank)" >&2
+      exit 2
+   else
+      echo -e "$output" > $old_path/cuda_memtest_"$host_name"_"$host_rank".err
+      echo cuda_memtest crash: see file $old_path/cuda_memtest_"$host_name"_"$host_rank".err >&2
+      exit 1
+   fi
 fi
 exit 0