(self)
| 42 | |
| 43 | class ProfilerConfig: |
| 44 | def __init__(self): |
| 45 | self.enabled = False |
| 46 | self.activities = [ |
| 47 | torch.profiler.ProfilerActivity.CPU, |
| 48 | torch.profiler.ProfilerActivity.CUDA, |
| 49 | ] |
| 50 | |
| 51 | # 基础配置 |
| 52 | self.wait = 4900 # 开始记录前等待的步数 |
| 53 | self.warmup = 50 # 预热步数 |
| 54 | self.active = 30_000 # 实际分析的步数 |
| 55 | # self.repeat = 2 # 重复次数 |
| 56 | # self.skip_first = 10 # 跳过前N步（可选） |
| 57 | |
| 58 | # 创建schedule |
| 59 | self.schedule = self._create_schedule() |
| 60 | |
| 61 | # 其他profiler设置 |
| 62 | self.on_trace_ready = torch.profiler.tensorboard_trace_handler('./log/profiler') |
| 63 | self.record_shapes = True |
| 64 | self.profile_memory = True |
| 65 | self.with_stack = True |
| 66 | |
| 67 | def _create_schedule(self): |
| 68 |
nothing calls this directly
no test coverage detected