pytorch/torch/utils/viz
Zachary DeVito 719c493f0b MemoryViz: print stream 0 if other streams exist (#106483)
It is confusing to not print stream 0, but print other streams. It makes stream 0
allocations seem like they are missing a stream annotation. This change will print streams
for everything unless all the events are on stream 0, then it will just not print streams.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/106483
Approved by: https://github.com/albanD
ghstack dependencies: #106328, #106482
2023-08-03 00:42:13 +00:00
..
__init__.py Re-land _cycleviz.py: visualize reference cycles holding cuda memory (#104051) 2023-06-23 13:44:58 +00:00
_cycles.py [BE] f-stringify torch/ and scripts (#105538) 2023-07-21 19:35:24 +00:00
MemoryViz.js MemoryViz: print stream 0 if other streams exist (#106483) 2023-08-03 00:42:13 +00:00