|
@@ -100,7 +100,7 @@ class TaskPool(TaskPoolBase):
|
|
|
|
|
|
def submit_task(self, *args: torch.Tensor) -> Future:
|
|
def submit_task(self, *args: torch.Tensor) -> Future:
|
|
"""Add task to this pool's queue, return Future for its output"""
|
|
"""Add task to this pool's queue, return Future for its output"""
|
|
- task = Task(MPFuture(synchronize=False), args)
|
|
|
|
|
|
+ task = Task(MPFuture(), args)
|
|
if self.get_task_size(task) > self.max_batch_size:
|
|
if self.get_task_size(task) > self.max_batch_size:
|
|
exc = ValueError(f"Task size greater than max_batch_size ({self.max_batch_size}), it can't be processed")
|
|
exc = ValueError(f"Task size greater than max_batch_size ({self.max_batch_size}), it can't be processed")
|
|
task.future.set_exception(exc)
|
|
task.future.set_exception(exc)
|