MCPcopy
hub / github.com/hpcaitech/ColossalAI / __init__

Method __init__

colossalai/auto_parallel/offload/solver.py:300–308  ·  view source on GitHub ↗
(self, region_list: List[Region], memory_budget: float = -1.0, search_window_size: int = 3)

Source from the content-addressed store, hash-verified

298
299class AsynGreedySolver(Solver):
300 def __init__(self, region_list: List[Region], memory_budget: float = -1.0, search_window_size: int = 3):
301 super().__init__(region_list, memory_budget)
302
303 self.search_window_size = search_window_size
304 # Records the prefetch execution location of the offloaded region
305 self.region_to_region_map = {}
306 self.best_ts: AsynTrainingSimulator = None
307
308 self._init_state()
309
310 def _init_state(self):
311 """

Callers

nothing calls this directly

Calls 2

_init_stateMethod · 0.95
__init__Method · 0.45

Tested by

no test coverage detected