(self)
| 91 | # manages threads internally. |
| 92 | |
| 93 | def setup_engine(self): |
| 94 | # Force all the ranks to wait here, and start creating the executor simultaneously. |
| 95 | # Only call barrier if we have multiple ranks to avoid hanging in single-process tests |
| 96 | if mpi_comm().Get_size() > 1: |
| 97 | mpi_comm().barrier() |
| 98 | |
| 99 | super().setup_engine() |
| 100 | |
| 101 | def shutdown(self): |
| 102 | logger_debug(f"[worker] RpcWorker #{mpi_rank()} is shutting down", |