Adding --apex:kokkos-counters flag

UO-OACISS · Sep 11, 2024 · 85b694d · 85b694d
1 parent a99f8ba
commit 85b694d
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/scripts/apex_exec b/src/scripts/apex_exec
@@ -62,6 +62,7 @@ where APEX options are zero or more of:
     --apex:kokkos                 enable Kokkos support
     --apex:kokkos-tuning          enable Kokkos runtime autotuning support
     --apex:kokkos-fence           enable Kokkos fences for async kernels
+    --apex:kokkos-counters        enable Kokkos counters for allocations and data transfers
     --apex:raja                   enable RAJA support
     --apex:pthread                enable pthread wrapper support (default: off)
     --apex:track-pthread          track pthread lifetime (forces --apex:pthread on)
@@ -106,6 +107,7 @@ openacc=no
 kokkos=no
 kokkos_tuning=no
 kokkos_fence=no
+kokkos_counters=no
 raja=no
 otf2=no
 gtrace=no
@@ -175,6 +177,11 @@ while (( "$#" )); do
       kokkos_fence=yes
       shift
       ;;
+    --apex:kokkos-counters)
+      kokkos=yes
+      kokkos_counters=yes
+      shift
+      ;;
     --apex:kokkos_fence|--apex:kokkos-fence)
       kokkos=yes
       kokkos_fence=yes
@@ -749,6 +756,9 @@ fi
 if [ ${kokkos_tuning} = yes ] ; then
     export APEX_KOKKOS_TUNING=1
 fi
+if [ ${kokkos_counters} = yes ] ; then
+    export APEX_KOKKOS_COUNTERS=1
+fi
 if [ ${raja} = yes ] ; then
     export RAJA_PLUGINS=${BASEDIR}/${LIBDIR}/${APEX_LIBRARY_NAME}${SHLIBX}
 fi