Allocate fake process ids for each device in the StepStats.
(self)
| 639 | return "/host:CPU Launch" in device_name |
| 640 | |
| 641 | def _allocate_pids(self): |
| 642 | """Allocate fake process ids for each device in the StepStats.""" |
| 643 | self._allocators_pid = self._alloc_pid() |
| 644 | self._chrome_trace.emit_pid('Allocators', self._allocators_pid) |
| 645 | |
| 646 | # Add processes in the Chrome trace to show compute and data activity. |
| 647 | for dev_stats in self._step_stats.dev_stats: |
| 648 | device_pid = self._alloc_pid() |
| 649 | self._device_pids[dev_stats.device] = device_pid |
| 650 | tensors_pid = self._alloc_pid() |
| 651 | self._tensor_pids[dev_stats.device] = tensors_pid |
| 652 | self._chrome_trace.emit_pid(dev_stats.device + ' Compute', device_pid) |
| 653 | self._chrome_trace.emit_pid(dev_stats.device + ' Tensors', tensors_pid) |
| 654 | |
| 655 | def _analyze_tensors(self, show_memory): |
| 656 | """Analyze tensor references to track dataflow.""" |
no test coverage detected