Fixed some issues and added abstraction variable

Author: prathi-wind

toolchain/templates/bridges2.mako

@@ -15,7 +15,7 @@
 % if account:
 #SBATCH --account="${account}"
 % endif
-% if gpu != gpuConfigOptions.NONE.value:
+% if is_gpu_active:
 #SBATCH --gpu-bind=verbose,closest
 #SBATCH --gres=gpu:v100-16:${tasks_per_node}
 % endif
@@ -32,7 +32,7 @@ ${helpers.template_prologue()}
 
 ok ":) Loading modules:\n"
 cd "${MFC_ROOT_DIR}"
-. ./mfc.sh load -c b -m ${'g' if gpu else 'c'}
+. ./mfc.sh load -c b -m ${'g' if is_gpu_active else 'c'}
 cd - > /dev/null
 echo
 

toolchain/templates/carpenter-cray.mako

@@ -27,7 +27,7 @@ ${helpers.template_prologue()}
 
 ok ":) Loading modules:\n"
 cd "${MFC_ROOT_DIR}"
-. ./mfc.sh load -c cc -m ${'g' if gpu else 'c'}
+. ./mfc.sh load -c cc -m ${'g' if is_gpu_active else 'c'}
 cd - > /dev/null
 echo
 

toolchain/templates/carpenter.mako

@@ -27,7 +27,7 @@ ${helpers.template_prologue()}
 
 ok ":) Loading modules:\n"
 cd "${MFC_ROOT_DIR}"
-. ./mfc.sh load -c c -m ${'g' if gpu else 'c'}
+. ./mfc.sh load -c c -m ${'g' if is_gpu_active else 'c'}
 cd - > /dev/null
 echo
 

toolchain/templates/default.mako

@@ -50,7 +50,7 @@ if engine == 'batch':
                 (set -x; ${profiler}   \
                     jsrun --nrs          ${tasks_per_node*nodes} \
                         --cpu_per_rs   1                       \
-                        --gpu_per_rs   ${1 if gpu else 0}      \
+                        --gpu_per_rs   ${1 if is_gpu_active else 0}      \
                         --tasks_per_rs 1                       \
                         "${target.get_install_binpath(case)}")
             elif [ "$binary" == "srun" ]; then

toolchain/templates/delta.mako

@@ -15,7 +15,7 @@
 % if account:
 #SBATCH --account="${account}"
 % endif
-% if gpu != gpuConfigOptions.NONE.value:
+% if is_gpu_active:
 #SBATCH --gpus-per-node=${tasks_per_node}
 #SBATCH --mem=208G
 #SBATCH --gpu-bind=closest
@@ -33,7 +33,7 @@ ${helpers.template_prologue()}
 
 ok ":) Loading modules:\n"
 cd "${MFC_ROOT_DIR}"
-. ./mfc.sh load -c d -m ${'g' if gpu else 'c'}
+. ./mfc.sh load -c d -m ${'g' if is_gpu_active else 'c'}
 cd - > /dev/null
 echo
 

toolchain/templates/deltaai.mako

@@ -15,7 +15,7 @@
 % if account:
 #SBATCH --account="${account}"
 % endif
-% if gpu != gpuConfigOptions.NONE.value:
+% if is_gpu_active:
 #SBATCH --gpus-per-node=${tasks_per_node}
 #SBATCH --mem=208G
 #SBATCH --gpu-bind=closest
@@ -33,7 +33,7 @@ ${helpers.template_prologue()}
 
 ok ":) Loading modules:\n"
 cd "${MFC_ROOT_DIR}"
-. ./mfc.sh load -c dai -m ${'g' if gpu else 'c'}
+. ./mfc.sh load -c dai -m ${'g' if is_gpu_active else 'c'}
 cd - > /dev/null
 echo
 

toolchain/templates/frontier.mako

@@ -11,7 +11,7 @@
 #SBATCH --time=${walltime}
 #SBATCH --cpus-per-task=7
 #SBATCH -C nvme
-% if gpu != gpuConfigOptions.NONE.value:
+% if is_gpu_active:
 #SBATCH --gpus-per-task=1
 #SBATCH --gpu-bind=closest
 % endif
@@ -35,12 +35,12 @@ ${helpers.template_prologue()}
 ok ":) Loading modules:\n"
 cd "${MFC_ROOT_DIR}"
 % if engine == 'batch':
-. ./mfc.sh load -c f -m ${'g' if gpu else 'c'}
+. ./mfc.sh load -c f -m ${'g' if is_gpu_active else 'c'}
 % endif
 cd - > /dev/null
 echo
 
-% if gpu != gpuConfigOptions.NONE.value:
+% if is_gpu_active:
     export MPICH_GPU_SUPPORT_ENABLED=1
 % else:
     export MPICH_GPU_SUPPORT_ENABLED=0
@@ -67,7 +67,7 @@ ulimit -s unlimited
         % if engine == 'interactive':
                 --unbuffered --nodes ${nodes} --ntasks-per-node ${tasks_per_node} \
                 --cpus-per-task 7                                    \
-            % if gpu != gpuConfigOptions.NONE.value:
+            % if is_gpu_active:
                 --gpus-per-task 1 --gpu-bind closest                 \
             % endif
             ${profiler} "${target.get_install_binpath(case)}")

toolchain/templates/hipergator.mako

@@ -10,7 +10,7 @@
 #SBATCH --output="${name}.out"
 #SBATCH --time=${walltime}
 #SBATCH --cpus-per-task=7
-% if gpu != gpuConfigOptions.NONE.value:
+% if is_gpu_active:
 #SBATCH --gpus-per-task=1
 #SBATCH --gpu-bind=closest
 % endif
@@ -36,7 +36,7 @@ ${helpers.template_prologue()}
 ok ":) Loading modules:\n"
 cd "${MFC_ROOT_DIR}"
 % if engine == 'batch':
-. ./mfc.sh load -c h -m ${'g' if gpu else 'c'}
+. ./mfc.sh load -c h -m ${'g' if is_gpu_active else 'c'}
 % endif
 cd - > /dev/null
 echo

toolchain/templates/include/helpers.mako

@@ -1,5 +1,6 @@
 <%! import os %>
 
+<% is_gpu_active = (gpu != gpuConfigOptions.NONE.value) %>
 <%def name="template_prologue()">
 % if os.name != 'nt':
     #>

toolchain/templates/nautilus.mako

@@ -15,7 +15,7 @@
 % if account:
 #SBATCH --account="${account}"
 % endif
-% if gpu != gpuConfigOptions.NONE.value:
+% if is_gpu_active:
 #SBATCH --gpu-bind=verbose,closest
 #SBATCH --gres=gpu:v100-16:${tasks_per_node}
 % endif
@@ -32,7 +32,7 @@ ${helpers.template_prologue()}
 
 ok ":) Loading modules:\n"
 cd "${MFC_ROOT_DIR}"
-. ./mfc.sh load -c n -m ${'g' if gpu else 'c'}
+. ./mfc.sh load -c n -m ${'g' if is_gpu_active else 'c'}
 cd - > /dev/null
 echo