|
192 | 192 | <div class="pytorch-left-menu-search">
|
193 | 193 |
|
194 | 194 | <div class="version">
|
195 |
| - <a href='https://pytorch.org/docs/versions.html'>master (1.10.0a0+git3b69ec8 ) ▼</a> |
| 195 | + <a href='https://pytorch.org/docs/versions.html'>master (1.10.0a0+gitabc594d ) ▼</a> |
196 | 196 | </div>
|
197 | 197 |
|
198 | 198 |
|
@@ -411,6 +411,7 @@ <h1>Source code for torch.distributed.elastic.events</h1><div class="highlight">
|
411 | 411 |
|
412 | 412 | <span class="sd">"""</span>
|
413 | 413 |
|
| 414 | +<span class="kn">import</span> <span class="nn">os</span> |
414 | 415 | <span class="kn">import</span> <span class="nn">logging</span>
|
415 | 416 |
|
416 | 417 | <span class="kn">from</span> <span class="nn">torch.distributed.elastic.events.handlers</span> <span class="kn">import</span> <span class="n">get_logging_handler</span>
|
@@ -438,14 +439,14 @@ <h1>Source code for torch.distributed.elastic.events</h1><div class="highlight">
|
438 | 439 | <span class="k">return</span> <span class="n">_events_logger</span>
|
439 | 440 | <span class="n">logging_handler</span> <span class="o">=</span> <span class="n">get_logging_handler</span><span class="p">(</span><span class="n">destination</span><span class="p">)</span>
|
440 | 441 | <span class="n">_events_logger</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="sa">f</span><span class="s2">"torchelastic-events-</span><span class="si">{</span><span class="n">destination</span><span class="si">}</span><span class="s2">"</span><span class="p">)</span>
|
441 |
| - <span class="n">_events_logger</span><span class="o">.</span><span class="n">setLevel</span><span class="p">(</span><span class="n">logging</span><span class="o">.</span><span class="n">DEBUG</span><span class="p">)</span> |
| 442 | + <span class="n">_events_logger</span><span class="o">.</span><span class="n">setLevel</span><span class="p">(</span><span class="n">os</span><span class="o">.</span><span class="n">environ</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"LOGLEVEL"</span><span class="p">,</span> <span class="s2">"INFO"</span><span class="p">))</span> |
442 | 443 | <span class="c1"># Do not propagate message to the root logger</span>
|
443 | 444 | <span class="n">_events_logger</span><span class="o">.</span><span class="n">propagate</span> <span class="o">=</span> <span class="kc">False</span>
|
444 | 445 | <span class="n">_events_logger</span><span class="o">.</span><span class="n">addHandler</span><span class="p">(</span><span class="n">logging_handler</span><span class="p">)</span>
|
445 | 446 | <span class="k">return</span> <span class="n">_events_logger</span>
|
446 | 447 |
|
447 | 448 |
|
448 |
| -<div class="viewcode-block" id="record"><a class="viewcode-back" href="../../../../elastic/events.html#torch.distributed.elastic.events.record">[docs]</a><span class="k">def</span> <span class="nf">record</span><span class="p">(</span><span class="n">event</span><span class="p">:</span> <span class="n">Event</span><span class="p">,</span> <span class="n">destination</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="s2">"console"</span><span class="p">)</span> <span class="o">-></span> <span class="kc">None</span><span class="p">:</span> |
| 449 | +<div class="viewcode-block" id="record"><a class="viewcode-back" href="../../../../elastic/events.html#torch.distributed.elastic.events.record">[docs]</a><span class="k">def</span> <span class="nf">record</span><span class="p">(</span><span class="n">event</span><span class="p">:</span> <span class="n">Event</span><span class="p">,</span> <span class="n">destination</span><span class="p">:</span> <span class="nb">str</span> <span class="o">=</span> <span class="s2">"null"</span><span class="p">)</span> <span class="o">-></span> <span class="kc">None</span><span class="p">:</span> |
449 | 450 | <span class="n">_get_or_create_logger</span><span class="p">(</span><span class="n">destination</span><span class="p">)</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="n">event</span><span class="o">.</span><span class="n">serialize</span><span class="p">())</span></div>
|
450 | 451 | </pre></div>
|
451 | 452 |
|
|
0 commit comments