Skip to content

Commit

Permalink
lint
Browse files Browse the repository at this point in the history
  • Loading branch information
august-knox committed Jan 9, 2025
1 parent 360927f commit e7370d2
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions experiments/laghos/experiment.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@
from benchpark.cuda import CudaExperiment
from benchpark.rocm import ROCmExperiment


class Laghos(
Experiment,
StrongScaling,
Expand All @@ -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"):
Expand Down

0 comments on commit e7370d2

Please sign in to comment.