(self, name="default")
| 70 | self.agg = {} |
| 71 | |
| 72 | def tick(self, name="default"): |
| 73 | if name not in self.starts: |
| 74 | self.starts[name] = torch.cuda.Event(enable_timing=True) |
| 75 | self.ends[name] = torch.cuda.Event(enable_timing=True) |
| 76 | self.starts[name].record() |
| 77 | else: |
| 78 | ms = self.tock(name, evict=True, print_ms=False) |
| 79 | |
| 80 | def tock(self, name="default", evict=True, print_ms=True): |
| 81 | if name in self.ends: |