Skip to content

Commit

Permalink
Fix lint formatiing
Browse files Browse the repository at this point in the history
  • Loading branch information
Riyaz Haque committed Oct 6, 2024
1 parent 676b58b commit 41b7588
Showing 1 changed file with 15 additions and 5 deletions.
20 changes: 15 additions & 5 deletions var/exp_repo/experiments/amg2023/experiment.py
Original file line number Diff line number Diff line change
Expand Up @@ -99,18 +99,28 @@ def compute_applications_section(self):
variables[ny] = initial_n
variables[nz] = initial_n
zips_size = "size"
experiment_setup["zips"] = {f"{zips_size}" : [nx, ny, nz]}
experiment_setup["zips"] = {f"{zips_size}": [nx, ny, nz]}

if self.spec.satisfies("programming_model=openmp"):
experiment_setup["matrices"] = [{"size_nodes_threads": [f"{zips_size}", "n_nodes", "n_threads_per_proc"]}]
experiment_setup["matrices"] = [
{
"size_nodes_threads": [
f"{zips_size}",
"n_nodes",
"n_threads_per_proc",
]
}
]
elif self.spec.satisfies("programming_model=cuda") or self.spec.satisfies(
"programming_model=rocm"
):
experiment_setup["matrix"] = [f"{zips_size}"]
if self.spec.satisfies("programming_model=openmp"):
experiment_setup["exclude"] = {"where" : [
"{n_threads_per_proc} * {n_ranks} > {n_nodes} * {sys_cores_per_node}"
]}
experiment_setup["exclude"] = {
"where": [
"{n_threads_per_proc} * {n_ranks} > {n_nodes} * {sys_cores_per_node}"
]
}
else:
input_params = {}
if self.spec.satisfies("experiment=throughput"):
Expand Down

0 comments on commit 41b7588

Please sign in to comment.