|
242 | 242 | <div class="pytorch-left-menu-search">
|
243 | 243 |
|
244 | 244 | <div class="version">
|
245 |
| - <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+giteda279c ) ▼</a> |
| 245 | + <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+gitc81c9ba ) ▼</a> |
246 | 246 | </div>
|
247 | 247 |
|
248 | 248 |
|
@@ -494,6 +494,9 @@ <h1>Source code for torch._logging._internal</h1><div class="highlight"><pre>
|
494 | 494 | <span class="kn">from</span> <span class="nn">typing</span> <span class="kn">import</span> <span class="n">Any</span><span class="p">,</span> <span class="n">Callable</span><span class="p">,</span> <span class="n">Dict</span><span class="p">,</span> <span class="n">List</span><span class="p">,</span> <span class="n">Optional</span><span class="p">,</span> <span class="n">Set</span><span class="p">,</span> <span class="n">Tuple</span><span class="p">,</span> <span class="n">Union</span>
|
495 | 495 | <span class="kn">from</span> <span class="nn">weakref</span> <span class="kn">import</span> <span class="n">WeakSet</span>
|
496 | 496 |
|
| 497 | +<span class="kn">import</span> <span class="nn">torch._logging.structured</span> |
| 498 | +<span class="kn">from</span> <span class="nn">torch.utils._traceback</span> <span class="kn">import</span> <span class="n">CapturedTraceback</span> |
| 499 | + |
497 | 500 | <span class="n">log</span> <span class="o">=</span> <span class="n">logging</span><span class="o">.</span><span class="n">getLogger</span><span class="p">(</span><span class="vm">__name__</span><span class="p">)</span>
|
498 | 501 |
|
499 | 502 | <span class="c1"># This is a synthetic logger which doesn't correspond to an actual logger,</span>
|
@@ -1545,6 +1548,12 @@ <h1>Source code for torch._logging._internal</h1><div class="highlight"><pre>
|
1545 | 1548 | <span class="n">record</span><span class="p">[</span><span class="s2">"frame_id"</span><span class="p">]</span> <span class="o">=</span> <span class="n">trace_id</span><span class="o">.</span><span class="n">compile_id</span><span class="o">.</span><span class="n">frame_id</span>
|
1546 | 1549 | <span class="n">record</span><span class="p">[</span><span class="s2">"frame_compile_id"</span><span class="p">]</span> <span class="o">=</span> <span class="n">trace_id</span><span class="o">.</span><span class="n">compile_id</span><span class="o">.</span><span class="n">frame_compile_id</span>
|
1547 | 1550 | <span class="n">record</span><span class="p">[</span><span class="s2">"attempt"</span><span class="p">]</span> <span class="o">=</span> <span class="n">trace_id</span><span class="o">.</span><span class="n">attempt</span>
|
| 1551 | + <span class="k">else</span><span class="p">:</span> |
| 1552 | + <span class="c1"># Record the stack of the log call to better diagnose why we</span> |
| 1553 | + <span class="c1"># don't have a frame id for it</span> |
| 1554 | + <span class="n">record</span><span class="p">[</span><span class="s2">"stack"</span><span class="p">]</span> <span class="o">=</span> <span class="n">torch</span><span class="o">.</span><span class="n">_logging</span><span class="o">.</span><span class="n">structured</span><span class="o">.</span><span class="n">from_traceback</span><span class="p">(</span> |
| 1555 | + <span class="n">CapturedTraceback</span><span class="o">.</span><span class="n">extract</span><span class="p">(</span><span class="n">skip</span><span class="o">=</span><span class="mi">1</span><span class="p">)</span><span class="o">.</span><span class="n">summary</span><span class="p">()</span> |
| 1556 | + <span class="p">)</span> |
1548 | 1557 | <span class="n">payload</span> <span class="o">=</span> <span class="n">payload_fn</span><span class="p">()</span>
|
1549 | 1558 | <span class="k">if</span> <span class="n">payload</span> <span class="ow">is</span> <span class="ow">not</span> <span class="kc">None</span><span class="p">:</span>
|
1550 | 1559 | <span class="k">if</span> <span class="ow">not</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">payload</span><span class="p">,</span> <span class="nb">str</span><span class="p">):</span>
|
|
0 commit comments