(self, show_dataflow=True, show_memory=True, use_real_thread_id=True)
| 831 | self._allocator_maximums = alloc_maxes |
| 832 | |
| 833 | def analyze_step_stats(self, show_dataflow=True, show_memory=True, use_real_thread_id=True): |
| 834 | self._allocate_pids() |
| 835 | self._assign_lanes(use_real_thread_id) |
| 836 | self._analyze_tensors(show_memory) |
| 837 | self._show_compute(show_dataflow) |
| 838 | if show_memory: |
| 839 | self._show_memory_counters() |
| 840 | return StepStatsAnalysis( |
| 841 | chrome_trace=self._chrome_trace, |
| 842 | allocator_maximums=self._allocator_maximums) |
| 843 | |
| 844 | def generate_chrome_trace_format(self, show_dataflow=True, show_memory=False, use_real_thread_id=True): |
| 845 | """Produces a trace in Chrome Trace Format. |
no test coverage detected