Explorar o código

test shmemory

justheuristic %!s(int64=4) %!d(string=hai) anos
pai
achega
66274a2906
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      hivemind/moe/server/task_pool.py

+ 1 - 1
hivemind/moe/server/task_pool.py

@@ -100,7 +100,7 @@ class TaskPool(TaskPoolBase):
 
     def submit_task(self, *args: torch.Tensor) -> Future:
         """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:
             exc = ValueError(f"Task size greater than max_batch_size ({self.max_batch_size}), it can't be processed")
             task.future.set_exception(exc)