diff --git a/hwbench/bench/benchmark.py b/hwbench/bench/benchmark.py index fb7d27a..2978465 100644 --- a/hwbench/bench/benchmark.py +++ b/hwbench/bench/benchmark.py @@ -19,6 +19,7 @@ def __init__( self.job_number = job_number self.enginemodule = enginemodule self.parameters = parameters + self.parameters.benchmark = self def get_enginemodule(self) -> EngineModuleBase: return self.enginemodule diff --git a/hwbench/bench/parameters.py b/hwbench/bench/parameters.py index 71002fd..22e8870 100644 --- a/hwbench/bench/parameters.py +++ b/hwbench/bench/parameters.py @@ -35,6 +35,13 @@ def __init__( self.monitoring = monitoring self.skip_method = skip_method self.sync_start = sync_start + self.benchmark = None + + def get_benchmark(self): + return self.benchmark + + def set_benchmark(self, benchmark): + self.benchmark = benchmark def get_pinned_cpu(self): if self.pinned_cpu == "none":