Skip to content

Commit 89a037b

Browse files
committed
Now log Dask worker plugin registration for timing information
1 parent 56c1719 commit 89a037b

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

ipsframework/services.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3310,13 +3310,17 @@ def _make_worker_args(num_workers: int, num_threads: int, use_shifter: bool, shi
33103310
if dask_worker_plugin is not None:
33113311
# TODO But what if there is more than one worker plugin?
33123312
# TODO And what about scheduler plugins?
3313+
self.debug('Register user provided plugin')
33133314
self.dask_client.register_plugin(dask_worker_plugin)
3315+
self.debug('Done registering user provided pluging')
33143316

33153317
# Regardless of any other worker plugins, we need this plugin to setup
33163318
# the DVM for the workers so that OpenMPI can work properly.
3319+
self.debug('Registered DVMPlugin')
33173320
self.dask_client.register_plugin(DVMPlugin(logger=services.logger,
33183321
oversubscribe=oversubscribe,
33193322
hwthreads=hwthreads))
3323+
self.debug('Registered DVMPlugin')
33203324

33213325
try:
33223326
# FIXME this is deprecated, but be mindful of blithely deleting

0 commit comments

Comments
 (0)