浏览代码

default to fifo

justheuristic 3 年之前
父节点
当前提交
b7ed72c4d6
共有 2 个文件被更改,包括 3 次插入2 次删除
  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