openmp implementation for the HPL benchmark (#69)

* openmp implementation for the HPL benchmark * Changes for ats4 hpl implementation --------- Co-authored-by: pearce8 <[email protected]>
LLNL · Dec 14, 2023 · 21fc295 · 21fc295
1 parent 73a0b8e
commit 21fc295
Show file tree

Hide file tree

Showing 4 changed files with 65 additions and 0 deletions.
diff --git a/configs/ats4/spack.yaml b/configs/ats4/spack.yaml
@@ -10,6 +10,8 @@ spack:
       spack_spec: [email protected]
     blas-rocm:
       spack_spec: [email protected]
+    blas:
+      spack_spec: [email protected]
     lapack:
       spack_spec: [email protected]
     mpi-rocm-gtl:

diff --git a/configs/cts1/spack.yaml b/configs/cts1/spack.yaml
@@ -8,6 +8,8 @@ spack:
       spack_spec: [email protected]
     compiler-intel:
       spack_spec: [email protected]
+    blas:
+      spack_spec: [email protected]
     lapack:
       spack_spec: [email protected]
     mpi-gcc:

diff --git a/experiments/hpl/openmp/execute_experiment.tpl b/experiments/hpl/openmp/execute_experiment.tpl
@@ -0,0 +1,12 @@
+#!/bin/bash
+{batch_nodes}
+{batch_ranks}
+{batch_timeout}
+
+cd {experiment_run_dir}
+
+{spack_setup}
+
+{experiment_setup}
+
+{command}
diff --git a/experiments/hpl/openmp/ramble.yaml b/experiments/hpl/openmp/ramble.yaml
@@ -0,0 +1,49 @@
+ramble:
+  include:
+  - ./configs/spack.yaml
+  - ./configs/variables.yaml
+
+  config:
+    deprecated: true
+    spack_flags:
+      install: '--add --keep-stage'
+      concretize: '-U -f'
+
+  applications:
+    hpl:
+      workloads:
+        standard:
+          env_vars:
+            set:
+              OMP_NUM_THREADS: '{omp_num_threads}'
+          variables:
+            n_ranks: '{processes_per_node} * {n_nodes}'
+            N-Grids: 1
+            Ps: 2
+            Qs: 4
+            N-Ns: 1
+            Ns: 10000
+            N-NBs: 1
+            NBs: 128
+            experiment_setup: ''
+            processes_per_node: '8'
+            n_nodes: '1'
+            omp_num_threads: ['2', '4', '8']
+          experiments:
+            hpl_omp_problem1_{n_nodes}_{n_ranks}_{omp_num_threads}_{Ps}_{Qs}_{Ns}_{NBs}:
+              variables:
+                env_name: hpl-omp
+              matrix:
+                - omp_num_threads
+  spack:
+    concretized: true
+    packages:
+      hpl-omp:
+        spack_spec: [email protected] +openmp
+        compiler: default-compiler
+    environments:
+      hpl-omp:
+        packages:
+        - blas
+        - default-mpi
+        - hpl-omp