Mixed Precision Gmres

TODO_CB_GMRES.txt

@@ -0,0 +1,10 @@
+TODO CB-GMRES for Paper review
+
+1. change stopping criterion: 1e-12
+2. Use Cusparse ILU preconditioner
+3. Always run GmresMixed<float, float> in addition to the other benchmarks
+4. Run both Classical and Modified Grahm-Schmidt and compare them
+   (Add regular Ginkgo Gmres<double> to the benchmark and compare MGS with CGS)
+5. Run everything with Restart = 300
+6. Remove matrices from same group (e.g. af_*) from the plots
+7. Run MGS vs CGS without preconditioner

benchmark/run_all_benchmarks.sh

@@ -11,6 +11,11 @@ if [ ! "${DRY_RUN}" ]; then
     DRY_RUN="false"
 fi
 
+if [ ! "${KEEP_MTX_FILES}" ]; then
+    echo "KEEP_MTX_FILES    environment variable not set - assuming \"false\"" 1>&2
+    KEEP_MTX_FILES="false"
+fi
+
 if [ ! "${EXECUTOR}" ]; then
     echo "EXECUTOR    environment variable not set - assuming \"cuda\"" 1>&2
     EXECUTOR="cuda"
@@ -50,6 +55,11 @@ if [ ! "${SOLVERS_MAX_ITERATIONS}" ]; then
     SOLVERS_MAX_ITERATIONS=10000
 fi
 
+if [ ! "${SOLVERS_KRYLOV_DIM}" ]; then
+    echo "SOLVERS_KRYLOV_DIM    environment variable not set - assuming \"100\"" 1>&2
+    SOLVERS_KRYLOV_DIM=100
+fi
+
 if [ ! "${SYSTEM_NAME}" ]; then
     echo "SYSTEM_MANE environment variable not set - assuming \"unknown\"" 1>&2
     SYSTEM_NAME="unknown"
@@ -60,6 +70,11 @@ if [ ! "${DEVICE_ID}" ]; then
     DEVICE_ID="0"
 fi
 
+if [ ! "${SOLVERS_NUM_RHS}" ]; then
+    echo "SOLVERS_NUM_RHS environment variable not set - assuming \"1\"" 1>&2
+    SOLVERS_NUM_RHS="1"
+fi
+
 # Control whether to run detailed benchmarks or not.
 # Default setting is detailed=false. To activate, set DETAILED=1.
 if  [ ! "${DETAILED}" ] || [ "${DETAILED}" -eq 0 ]; then
@@ -68,6 +83,22 @@ else
     DETAILED_STR="--detailed=true"
 fi
 
+# Control whether to print additional preconditioner information for solver
+# benchmarks. Requires `DETAILED` to be set to have an effect.
+if  [ ! "${SOLVERS_PRINT_PRECOND}" ] || [ "${SOLVERS_PRINT_PRECOND}" -ne 0 ]; then
+    PRINT_PRECOND_STR="--print_preconditioner_information=true"
+else
+    PRINT_PRECOND_STR="--print_preconditioner_information=false"
+fi
+
+# Control whether to randomize the right hand side.
+# Default setting is randomize=true. To deactivate, set SOLVERS_RANDOMIZE_RHS=0.
+if  [ ! "${SOLVERS_RANDOMIZE_RHS}" ] || [ "${SOLVERS_RANDOMIZE_RHS}" -ne 0 ]; then
+    SOLVERS_RND_STR="--randomize_rhs=true"
+else
+    SOLVERS_RND_STR="--randomize_rhs=false"
+fi
+
 # This allows using a matrix list file for benchmarking.
 # The file should contains a suitesparse matrix on each line.
 # The allowed formats to target suitesparse matrix is:
@@ -166,8 +197,11 @@ run_solver_benchmarks() {
     ./solver/solver --backup="$1.bkp" --double_buffer="$1.bkp2" \
                     --executor="${EXECUTOR}" --solvers="${SOLVERS}" \
                     --preconditioners="${PRECONDS}" \
-                    --max_iters=${SOLVERS_MAX_ITERATIONS} --rel_res_goal=${SOLVERS_PRECISION} \
-                    ${DETAILED_STR} --device_id="${DEVICE_ID}" \
+                    --max_iters="${SOLVERS_MAX_ITERATIONS}" \
+                    --rel_res_goal="${SOLVERS_PRECISION}" \
+                    --krylov_dim="${SOLVERS_KRYLOV_DIM}" "${SOLVERS_RND_STR}" \
+                    --nrhs="${SOLVERS_NUM_RHS}" "${DETAILED_STR}" \
+                    --device_id="${DEVICE_ID}" "${PRINT_PRECOND_STR}" \
                     <"$1.imd" 2>&1 >"$1"
     keep_latest "$1" "$1.bkp" "$1.bkp2" "$1.imd"
 }
@@ -380,7 +414,8 @@ for bsize in ${BLOCK_SIZES}; do
         run_preconditioner_benchmarks "${RESULT_FILE}"
 
         echo -e "${PREFIX}Cleaning up problem ${GROUP}/${NAME}" 1>&2
-        [ "${DRY_RUN}" != "true" ] && rm -r "/tmp/${GROUP}/${NAME}.mtx"
+        [ "${DRY_RUN}" != "true" ] && [ "${KEEP_MTX_FILES}" != "false" ] && \
+            rm -r "/tmp/${GROUP}/${NAME}.mtx"
     done
     if [ "${ID}" -ge "${LOOP_END}" ]; then
         break