From 6b30ccc60cd5ec6c4c57fb19e0a68c32464c0224 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 11 Feb 2023 16:14:02 -0600 Subject: [PATCH] log more about selected device --- api/onnx_web/device_pool.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api/onnx_web/device_pool.py b/api/onnx_web/device_pool.py index 0f22bdb6..c7cd6395 100644 --- a/api/onnx_web/device_pool.py +++ b/api/onnx_web/device_pool.py @@ -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)