Fix --cpuset

parttimenerd · Jul 21, 2020 · c972086 · c972086
1 parent d1a3ab5
commit c972086
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/temci/run/cpuset.py b/temci/run/cpuset.py
@@ -105,6 +105,8 @@ def move_process_to_set(self, pid: int, set_id: int):
 
     def get_sub_set(self, set_id: int) -> str:
         """ Gets the name of the benchmarking cpu set with the given id / number (starting at zero). """
+        if self.parallel == 0:
+            return CONTROLLER_SUB_BENCH_SET
         if self.active:
             typecheck(set_id, Int(range=range(0, self.parallel_number)))
         return SUB_BENCH_SET.format(set_id)

diff --git a/temci/run/run_worker_pool.py b/temci/run/run_worker_pool.py
@@ -201,7 +201,7 @@ def submit(self, block: RunProgramBlock, id: int, runs: int):
         typecheck(runs, NaturalNumber())
         typecheck(id, NaturalNumber())
         block.is_enqueued = True
-        self.result_queue.put((block, self.run_driver.benchmark(block, runs, timeout=self.next_block_timeout()), id))
+        self.result_queue.put((block, self.run_driver.benchmark(block, runs, self.cpuset, timeout=self.next_block_timeout()), id))
         block.is_enqueued = False
 
     def results(self, expected_num: int) -> ResultGenerator: