From e7370d2e03df00b3963ff47dfe8183aef9155906 Mon Sep 17 00:00:00 2001 From: august-knox Date: Thu, 9 Jan 2025 12:41:27 -0800 Subject: [PATCH] lint --- experiments/laghos/experiment.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/experiments/laghos/experiment.py b/experiments/laghos/experiment.py index 3c1293be..aeb062c4 100644 --- a/experiments/laghos/experiment.py +++ b/experiments/laghos/experiment.py @@ -11,6 +11,7 @@ from benchpark.cuda import CudaExperiment from benchpark.rocm import ROCmExperiment + class Laghos( Experiment, StrongScaling, @@ -35,12 +36,12 @@ def compute_applications_section(self): # Number of initial nodes n_resources = {"n_nodes": 1} - device= "n_ranks" + device = "n_ranks" if self.spec.satisfies("+cuda"): self.add_experiment_variable("device", "cuda", True) elif self.spec.satisfies("+rocm"): self.add_experiment_variable("device", "hip", True) - + if self.spec.satisfies("+cuda") or self.spec.satisfies("+rocm"): device = "n_gpus" if self.spec.satisfies("+single_node"):