Method submit

comfy/multigpu.py:62–63

(self, device: torch.device, fn, *args, **kwargs)

                result_q.put((None, e))

    def submit(self, device: torch.device, fn, *args, **kwargs):
        self._work_queues[device].put((fn, args, kwargs))

    def get_result(self, device: torch.device):
        return self._result_queues[device].get()