mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
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 |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| _cycles.py | ||
| MemoryViz.js | ||