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:
|
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)
|
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)
|
context = JobContext(key, self.devices, device_index=device)
|
||||||
future = self.pool.submit(fn, context, *args, **kwargs)
|
future = self.pool.submit(fn, context, *args, **kwargs)
|
||||||
|
|
Loading…
Reference in New Issue