Adding strong and weak scaling for OpenMP version (#320)

* Adding strong and weak scaling for OpenMP version * Add aug/openmp dry run on Dane
LLNL · Jul 24, 2024 · f1b6f0f · f1b6f0f
1 parent 7e27d20
commit f1b6f0f
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 8 deletions.
diff --git a/.github/workflows/run.yml b/.github/workflows/run.yml
@@ -100,6 +100,16 @@ jobs:
         run: |
           pip install -r ./requirements.txt
 
+      - name: Dry run amg2023/openmp on Dane with allocation modifier
+        run: |
+          ./bin/benchpark setup amg2023/openmp LLNL-Dane-DELL-sapphirerapids-OmniPath workspace/
+          . workspace/setup.sh
+          ramble \
+            --workspace-dir workspace/amg2023/openmp/LLNL-Dane-DELL-sapphirerapids-OmniPath/workspace \
+            --disable-progress-bar \
+            --disable-logger \
+            workspace setup --dry-run
+
       - name: Dry run amg2023/cuda on Sierra with allocation modifier
         run: |
           ./bin/benchpark setup amg2023/cuda LLNL-Sierra-IBM-power9-V100-Infiniband workspace/

diff --git a/experiments/kripke/openmp/ramble.yaml b/experiments/kripke/openmp/ramble.yaml
@@ -22,15 +22,15 @@ ramble:
     kripke:
       workloads:
         kripke:
+          env_vars:
           variables:
             experiment_setup: ''
-            n_nodes: '1'
-            n_ranks: ['1', '16']
-            n_threads_per_proc: ['32', '2']
+            n_ranks: '{npx}*{npy}*{npz}'
+            n_threads_per_proc: '1'
             omp_num_threads: '{n_threads_per_proc}'
             arch: 'OpenMP'
           experiments:
-            kripke_omp_{n_nodes}_{n_ranks}_{omp_num_threads}_{ngroups}_{gs}_{nquad}_{ds}_{lorder}_{nzx}_{nzy}_{nzz}_{npx}_{npy}_{npz}:
+            kripke_openmp_strong_{n_nodes}_{n_ranks}_{omp_num_threads}_{ngroups}_{gs}_{nquad}_{ds}_{lorder}_{nzx}_{nzy}_{nzz}_{npx}_{npy}_{npz}:
               variants:
                 package_manager: spack
               variables:
@@ -40,11 +40,26 @@ ramble:
                 ds: 128
                 lorder: 4
                 nzx: '64'
-                nzy: '32'
+                nzy: '64'
                 nzz: '32'
-                npx: ['1', '4']
-                npy: ['1', '2']
-                npz: ['1', '2']
+                npx: ['2', '2', '4', '4']
+                npy: ['2', '2', '2', '4']
+                npz: ['1', '2', '2', '2']
+            kripke_openmp_weak_{n_nodes}_{n_ranks}_{omp_num_threads}_{ngroups}_{gs}_{nquad}_{ds}_{lorder}_{nzx}_{nzy}_{nzz}_{npx}_{npy}_{npz}:
+              variants:
+                package_manager: spack
+              variables:
+                ngroups: 64
+                gs: 1
+                nquad: 128
+                ds: 128
+                lorder: 4
+                nzx: ['64', '64', '128', '128']
+                nzy: ['64', '64', '64', '128']
+                nzz: ['32', '64', '64', '64']
+                npx: ['2', '2', '4', '4']
+                npy: ['2', '2', '2', '4']
+                npz: ['1', '2', '2', '2']
   software:
     packages:
       kripke: