Adding allocation modifier to Kripke experiments (#269)

* Adding allocation modifier to Kripke experiments * Add dryruns for all 3 versions of Kripke * Minor * Trying to fix the openmp version * Trying to fix the openmp version * Trying to fix the openmp version * Trying to fix the openmp version
LLNL · Jun 7, 2024 · bb1aca9 · bb1aca9
1 parent 5113774
commit bb1aca9
Show file tree

Hide file tree

Showing 4 changed files with 34 additions and 10 deletions.
diff --git a/.github/workflows/run.yml b/.github/workflows/run.yml
@@ -120,7 +120,27 @@ jobs:
             --disable-logger \
             workspace setup --dry-run
 
-      - name: Dry run kripke/rocm on Tioga
+      - name: Dry run kripke/openmp on nosite-x86_64 with allocation modifier
+        run: |
+          ./bin/benchpark setup kripke/openmp nosite-x86_64 workspace/
+          . workspace/setup.sh
+          ramble \
+            --workspace-dir workspace/kripke/openmp/nosite-x86_64/workspace \
+            --disable-progress-bar \
+            --disable-logger \
+            workspace setup --dry-run
+
+      - name: Dry run kripke/cuda on Sierra with allocation modifier
+        run: |
+          ./bin/benchpark setup kripke/cuda LLNL-Sierra-IBM-power9-V100-Infiniband workspace/
+          . workspace/setup.sh
+          ramble \
+            --workspace-dir workspace/kripke/cuda/LLNL-Sierra-IBM-power9-V100-Infiniband/workspace \
+            --disable-progress-bar \
+            --disable-logger \
+            workspace setup --dry-run
+
+      - name: Dry run kripke/rocm on Tioga with allocation modifier
         run: |
           ./bin/benchpark setup kripke/rocm LLNL-Tioga-HPECray-zen3-MI250X-Slingshot workspace/
           . workspace/setup.sh

diff --git a/experiments/kripke/cuda/ramble.yaml b/experiments/kripke/cuda/ramble.yaml
@@ -15,16 +15,17 @@ ramble:
       install: '--add --keep-stage'
       concretize: '-U -f'
 
+  modifiers:
+  - name: allocation
+
   applications:
     kripke:
       workloads:
         kripke:
           env_vars:
           variables:
-            n_ranks: '{processes_per_node} * {n_nodes}'
             experiment_setup: ''
-            processes_per_node: '2'
-            n_nodes: '2'
+            n_gpus: '4'
             arch: 'CUDA'
           experiments:
             kripke_cuda_{n_nodes}_{ngroups}_{gs}_{nquad}_{ds}_{lorder}_{nzx}_{nzy}_{nzz}_{npx}_{npy}_{npz}:

diff --git a/experiments/kripke/openmp/ramble.yaml b/experiments/kripke/openmp/ramble.yaml
@@ -15,6 +15,9 @@ ramble:
       install: '--add --keep-stage'
       concretize: '-U -f'
 
+  modifiers:
+  - name: allocation
+
   applications:
     kripke:
       workloads:
@@ -23,11 +26,10 @@ ramble:
             set:
               OMP_NUM_THREADS: '{omp_num_threads}'
           variables:
-            n_ranks: '{processes_per_node} * {n_nodes}'
             experiment_setup: ''
-            processes_per_node: ['1', '16']
             n_nodes: '1'
-            omp_num_threads: ['32', '2']
+            n_threads_per_proc: ['32', '2']
+            omp_num_threads: n_threads_per_proc
             arch: 'OpenMP'
           experiments:
             kripke_omp_{n_nodes}_{omp_num_threads}_{ngroups}_{gs}_{nquad}_{ds}_{lorder}_{nzx}_{nzy}_{nzz}_{npx}_{npy}_{npz}:

diff --git a/experiments/kripke/rocm/ramble.yaml b/experiments/kripke/rocm/ramble.yaml
@@ -15,16 +15,17 @@ ramble:
       install: '--add --keep-stage'
       concretize: '-U -f'
 
+  modifiers:
+  - name: allocation
+
   applications:
     kripke:
       workloads:
         kripke:
           env_vars:
           variables:
-            n_ranks: '{processes_per_node} * {n_nodes}'
             experiment_setup: ''
-            processes_per_node: ['8']
-            n_nodes: '1'
+            n_gpus: '8'
             arch: 'HIP'
           experiments:
             kripke_rocm_{n_nodes}_{ngroups}_{gs}_{nquad}_{ds}_{lorder}_{nzx}_{nzy}_{nzz}_{npx}_{npy}_{npz}: