Browse Source

default to fifo

justheuristic 3 years ago
parent
commit
b7ed72c4d6
2 changed files with 3 additions and 2 deletions
  1. 2 1
      src/server/backend.py
  2. 1 1
      src/server/task_prioritizer.py

+ 2 - 1
src/server/backend.py

@@ -1,11 +1,12 @@
 """Code for serving bloom blocks via hivemind-server"""
 """Code for serving bloom blocks via hivemind-server"""
+import ctypes
 import multiprocessing as mp
 import multiprocessing as mp
 import os
 import os
 import threading
 import threading
 from concurrent.futures import Future
 from concurrent.futures import Future
 from dataclasses import dataclass, field
 from dataclasses import dataclass, field
 from queue import Empty, PriorityQueue
 from queue import Empty, PriorityQueue
-from typing import Any, Dict, Optional, Sequence, Tuple
+from typing import Any, Dict, Optional, Sequence, Tuple, List
 
 
 import torch
 import torch
 from hivemind import BatchTensorDescriptor, use_hivemind_log_handler
 from hivemind import BatchTensorDescriptor, use_hivemind_log_handler

+ 1 - 1
src/server/task_prioritizer.py

@@ -16,5 +16,5 @@ class TaskPrioritizerBase(ABC):
 class DummyTaskPrioritizer(TaskPrioritizerBase):
 class DummyTaskPrioritizer(TaskPrioritizerBase):
     """Simple implementation of DustBroker which counts amount of dust per task size"""
     """Simple implementation of DustBroker which counts amount of dust per task size"""
 
 
-    def __call__(self, *input: torch.Tensor, points: float = 0.0, **kwargs) -> float:
+    def prioritize(self, *input: torch.Tensor, points: float = 0.0, **kwargs) -> float:
         return 0.0
         return 0.0