|
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+gite6cf3e9 ) ▼</a> |
| 245 | + <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6fefc52 ) ▼</a> |
246 | 246 | </div>
|
247 | 247 |
|
248 | 248 |
|
@@ -528,6 +528,7 @@ <h1>Source code for torch._jit_internal</h1><div class="highlight"><pre>
|
528 | 528 | <span class="kn">from</span> <span class="nn">torch._awaits</span> <span class="kn">import</span> <span class="n">_Await</span>
|
529 | 529 | <span class="kn">from</span> <span class="nn">torch._C</span> <span class="kn">import</span> <span class="n">_Await</span> <span class="k">as</span> <span class="n">CAwait</span><span class="p">,</span> <span class="n">Future</span> <span class="k">as</span> <span class="n">CFuture</span>
|
530 | 530 | <span class="kn">from</span> <span class="nn">torch._sources</span> <span class="kn">import</span> <span class="n">fake_range</span><span class="p">,</span> <span class="n">get_source_lines_and_file</span><span class="p">,</span> <span class="n">parse_def</span>
|
| 531 | +<span class="kn">from</span> <span class="nn">torch._utils_internal</span> <span class="kn">import</span> <span class="n">log_torchscript_usage</span> |
531 | 532 | <span class="kn">from</span> <span class="nn">torch.futures</span> <span class="kn">import</span> <span class="n">Future</span>
|
532 | 533 |
|
533 | 534 | <span class="n">IS_PY39_PLUS</span><span class="p">:</span> <span class="n">Final</span><span class="p">[</span><span class="nb">bool</span><span class="p">]</span> <span class="o">=</span> <span class="n">sys</span><span class="o">.</span><span class="n">version_info</span> <span class="o">>=</span> <span class="p">(</span><span class="mi">3</span><span class="p">,</span> <span class="mi">9</span><span class="p">)</span>
|
@@ -1064,6 +1065,7 @@ <h1>Source code for torch._jit_internal</h1><div class="highlight"><pre>
|
1064 | 1065 | <span class="sd"> # any compiled methods and wasn't decorated with `@torch.jit.export`</span>
|
1065 | 1066 | <span class="sd"> m = torch.jit.script(MyModule())</span>
|
1066 | 1067 | <span class="sd"> """</span>
|
| 1068 | + <span class="n">log_torchscript_usage</span><span class="p">(</span><span class="s2">"export"</span><span class="p">)</span> |
1067 | 1069 | <span class="n">fn</span><span class="o">.</span><span class="n">_torchscript_modifier</span> <span class="o">=</span> <span class="n">FunctionModifiers</span><span class="o">.</span><span class="n">EXPORT</span>
|
1068 | 1070 | <span class="k">return</span> <span class="n">fn</span></div>
|
1069 | 1071 |
|
@@ -1105,6 +1107,7 @@ <h1>Source code for torch._jit_internal</h1><div class="highlight"><pre>
|
1105 | 1107 | <span class="sd"> # exception raised</span>
|
1106 | 1108 | <span class="sd"> m(torch.rand(100))</span>
|
1107 | 1109 | <span class="sd"> """</span>
|
| 1110 | + <span class="n">log_torchscript_usage</span><span class="p">(</span><span class="s2">"unused"</span><span class="p">)</span> |
1108 | 1111 | <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">fn</span><span class="p">,</span> <span class="nb">property</span><span class="p">):</span>
|
1109 | 1112 | <span class="n">prop</span> <span class="o">=</span> <span class="n">fn</span>
|
1110 | 1113 | <span class="nb">setattr</span><span class="p">(</span> <span class="c1"># noqa: B010</span>
|
@@ -1192,6 +1195,7 @@ <h1>Source code for torch._jit_internal</h1><div class="highlight"><pre>
|
1192 | 1195 | <span class="sd"> import os</span>
|
1193 | 1196 | <span class="sd"> os.remove('m.pt')</span>
|
1194 | 1197 | <span class="sd"> """</span>
|
| 1198 | + <span class="n">log_torchscript_usage</span><span class="p">(</span><span class="s2">"ignore"</span><span class="p">)</span> |
1195 | 1199 |
|
1196 | 1200 | <span class="k">if</span> <span class="nb">callable</span><span class="p">(</span><span class="n">drop</span><span class="p">):</span>
|
1197 | 1201 | <span class="c1"># used without any args, so drop is actually a function</span>
|
|
0 commit comments