(self, region_list: List[Region], memory_budget: float = -1.0, error_factor: float = 0.95)
| 51 | """ |
| 52 | |
| 53 | def __init__(self, region_list: List[Region], memory_budget: float = -1.0, error_factor: float = 0.95) -> None: |
| 54 | self.region_list = region_list |
| 55 | |
| 56 | self.error_factor: float = error_factor |
| 57 | if memory_budget > 0: |
| 58 | self.memory_budget = memory_budget * self.error_factor |
| 59 | else: |
| 60 | self.memory_budget = ( |
| 61 | torch.cuda.get_device_properties(get_accelerator().get_current_device()).total_memory |
| 62 | * self.error_factor |
| 63 | ) |
| 64 | |
| 65 | self.link_to_bandwidth: Dict[str, Dict[float, float]] = self._profile_bandwidth() |
| 66 | self.comp_power: float = self._extract_computing_power() |
| 67 | |
| 68 | @abstractmethod |
| 69 | def _call_solver(self): |
no test coverage detected