Fix task scaling; test WorkersAndBlocks with process-based tasks.

pp-mo · pp-mo · commit 84ff4ad5d6cf · 2024-05-16T15:12:44.000+01:00
diff --git a/benchmarks/benchmarks/memtrace_evaluation/__init__.py b/benchmarks/benchmarks/memtrace_evaluation/__init__.py
@@ -29,11 +29,11 @@ def _setup(self, **kwargs):
         nblocks = params["nblocks"]
         nworkers = params["nworkers"]
 
-        nyfull = ysize // nblocks
+        ny_task = ysize // nblocks
         use_processes = {"threads": False, "processes": True}[runtype]
         self.task = SampleParallelTask(
             n_blocks=nblocks,
-            outerdim=nyfull // nblocks,
+            outerdim=ny_task,
             innerdim=nx,
             n_workers=nworkers,
             use_process_workers=use_processes,
@@ -113,6 +113,12 @@ def track_addmem_calc(self, nblocks, nworkers):
         return self.run_addedmem_calc()
 
 
+class MemcheckBlocksAndWorkers_processes(MemcheckBlocksAndWorkers):
+    def setup(self, nblocks, nworkers):
+        self.default_params["runtype"] = "processes"
+        super().setup(nblocks, nworkers)
+
+
 class MemcheckBlocksAndWorkers_Rss(MemcheckBlocksAndWorkers):
     def setup(self, nblocks, nworkers):
         self.default_params["measure"] = "rss"
