(self)
| 79 | |
| 80 | class ProfilerConfig: |
| 81 | def __init__(self): |
| 82 | self.enabled = False |
| 83 | self.activities = [ |
| 84 | torch.profiler.ProfilerActivity.CPU, |
| 85 | torch.profiler.ProfilerActivity.CUDA, |
| 86 | ] |
| 87 | |
| 88 | self.wait = 1 |
| 89 | self.warmup = 2 |
| 90 | self.active = 30_000 |
| 91 | |
| 92 | self.schedule = self._create_schedule() |
| 93 | |
| 94 | self.on_trace_ready = torch.profiler.tensorboard_trace_handler('./log/profiler') |
| 95 | self.record_shapes = True |
| 96 | self.profile_memory = True |
| 97 | self.with_stack = True |
| 98 | |
| 99 | def _create_schedule(self): |
| 100 | return torch.profiler.schedule( |
nothing calls this directly
no test coverage detected