Support binding agent to core

2023-09-02 19:55:35 +02:00
parent ed5081f1b8
commit 695046a123
2 changed files with 7 additions and 0 deletions
@@ -294,6 +294,9 @@ class Slate():

        Runner = self.runners[runnerName]

+        if orig_config.consume('scheduler.bind_agent_to_core', False):
+            os.sched_setaffinity(0, [p_ind % os.cpu_count()])
+
        for r in rep_ids:
            self.run_id = r
            config = copy.deepcopy(orig_config)
@@ -324,6 +327,9 @@ class Slate():

        Runner = self.runners[runnerName]

+        if orig_config.consume('scheduler.bind_agent_to_core', False):
+            os.sched_setaffinity(0, [p_ind % os.cpu_count()])
+
        with wandb.init(
            project=project,
            reinit=self.consume(wandbC, 'reinit', DEFAULT_REINIT),