github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix conflicts to beautify the code

pull/5733/head
genghaozhe 2024-05-20 04:09:51 +00:00
parent d22bf30ca6 f5a5287f87
commit 7416e4943b
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
colossalai/zero/gemini/placement_policy.py
View File

@ -20,7 +20,7 @@ class PlacementPolicy(ABC):
def __init__( def __init__(
self, self,
gemini_manager: "GeminiManager", gemini_manager: "GeminiManager", # TODO @botbw: solve circular import
chunk_manager: ChunkManager, chunk_manager: ChunkManager,
mem_stats_collector: Optional[ChunkMemStatsCollector] = None, mem_stats_collector: Optional[ChunkMemStatsCollector] = None,
max_prefetch: int = 0, max_prefetch: int = 0,
@ -41,9 +41,8 @@ class PlacementPolicy(ABC):
) -> None: ) -> None:
raise NotImplementedError raise NotImplementedError
@abstractmethod
def get_prefetch_chunks(self) -> List[Chunk]: def get_prefetch_chunks(self) -> List[Chunk]:
raise NotImplementedError return [] # no prefetch by default
class StaticPlacementPolicy(PlacementPolicy): class StaticPlacementPolicy(PlacementPolicy):