Skip to content

Commit

Permalink
experiment.py to run genesis on fugaku (#544)
Browse files Browse the repository at this point in the history
  • Loading branch information
SBA0486 authored Jan 14, 2025
1 parent 294762a commit 9562a6a
Showing 1 changed file with 8 additions and 3 deletions.
11 changes: 8 additions & 3 deletions experiments/genesis/experiment.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,10 +40,10 @@ def compute_applications_section(self):
self.add_experiment_variable("maxiter_inner", "50")

if self.spec.satisfies("+openmp"):
self.add_experiment_variable("n_nodes", ["1"], True)
self.add_experiment_variable("processes_per_node", ["8"])
self.add_experiment_variable("n_nodes", ["2"], True)
self.add_experiment_variable("processes_per_node", ["4"])
self.add_experiment_variable("n_ranks", "{processes_per_node} * {n_nodes}")
self.add_experiment_variable("omp_num_threads", ["48"])
self.add_experiment_variable("omp_num_threads", ["12"])
self.add_experiment_variable("arch", "OpenMP")

def compute_spack_section(self):
Expand All @@ -53,10 +53,15 @@ def compute_spack_section(self):
system_specs = {}
system_specs["compiler"] = "default-compiler"
system_specs["mpi"] = "default-mpi"
system_specs["lapack"] = "lapack"

# if package_spec left empty spack will use external
self.add_spack_spec(system_specs["mpi"])
self.add_spack_spec(system_specs["lapack"])

self.add_spack_spec(
self.name, [f"genesis@{app_version} +mpi", system_specs["compiler"]]
)
self.add_spack_spec(
system_specs["lapack"], [system_specs["lapack"], system_specs["compiler"]]
)

0 comments on commit 9562a6a

Please sign in to comment.