log more about selected device
This commit is contained in:
parent
38c8a8a413
commit
6b30ccc60c
|
@ -187,7 +187,7 @@ class DevicePoolExecutor:
|
|||
|
||||
def submit(self, key: str, fn: Callable[..., None], /, *args, needs_device: Optional[DeviceParams] = None, **kwargs) -> None:
|
||||
device = self.get_next_device(needs_device=needs_device)
|
||||
logger.info("assigning job %s to device %s", key, device)
|
||||
logger.info("assigning job %s to device %s: %s", key, device, self.devices[device])
|
||||
|
||||
context = JobContext(key, self.devices, device_index=device)
|
||||
future = self.pool.submit(fn, context, *args, **kwargs)
|
||||
|
|
Loading…
Reference in New Issue