apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

log more about selected device

This commit is contained in:
Sean Sube 2023-02-11 16:14:02 -06:00
parent 38c8a8a413
commit 6b30ccc60c
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/device_pool.py
View File

@ -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)