From f6a6c997bb30782a1d543e7544b161f52a4dd558 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 4 Feb 2023 15:56:32 -0600 Subject: [PATCH] lint(api): log which device is assigned to which job --- api/onnx_web/device_pool.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/api/onnx_web/device_pool.py b/api/onnx_web/device_pool.py index fdeda89e..47fb8383 100644 --- a/api/onnx_web/device_pool.py +++ b/api/onnx_web/device_pool.py @@ -142,7 +142,10 @@ class DevicePoolExecutor: self.jobs[:] = [job for job in self.jobs if job.future.done()] def submit(self, key: str, fn: Callable[..., None], /, *args, **kwargs) -> None: - context = JobContext(key, self.devices, device_index=self.get_next_device()) + device = self.get_next_device() + logger.info('assigning job %s to device %s', key, device) + + context = JobContext(key, self.devices, device_index=device) future = self.pool.submit(fn, context, *args, **kwargs) job = Job(key, future, context) self.jobs.append(job)