(self, device: torch.device | None = None, warmup: int = 3, repeat: int = 10)
| 162 | """ |
| 163 | |
| 164 | def __init__(self, device: torch.device | None = None, warmup: int = 3, repeat: int = 10): |
| 165 | self.device = device if device is not None else device_ctx.device |
| 166 | self.warmup = warmup |
| 167 | self.repeat = repeat |
| 168 | self.gpu_info = GPUProfiler.get_target_info( |
| 169 | device_index=self.device.index if self.device.type == "cuda" else -1 |
| 170 | ) |
| 171 | self._metrics: list[BenchmarkMetrics] = [] |
| 172 | |
| 173 | def _sync(self) -> None: |
| 174 | if self.device.type == "cuda": |
nothing calls this directly
no test coverage detected