mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-05 02:51:59 +00:00
fix conflicts to beautify the code
This commit is contained in:
@@ -20,7 +20,7 @@ class PlacementPolicy(ABC):
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
gemini_manager: "GeminiManager",
|
||||
gemini_manager: "GeminiManager", # TODO @botbw: solve circular import
|
||||
chunk_manager: ChunkManager,
|
||||
mem_stats_collector: Optional[ChunkMemStatsCollector] = None,
|
||||
max_prefetch: int = 0,
|
||||
@@ -41,9 +41,8 @@ class PlacementPolicy(ABC):
|
||||
) -> None:
|
||||
raise NotImplementedError
|
||||
|
||||
@abstractmethod
|
||||
def get_prefetch_chunks(self) -> List[Chunk]:
|
||||
raise NotImplementedError
|
||||
return [] # no prefetch by default
|
||||
|
||||
|
||||
class StaticPlacementPolicy(PlacementPolicy):
|
||||
|
Reference in New Issue
Block a user