|
235 | 235 | <div class="pytorch-left-menu-search">
|
236 | 236 |
|
237 | 237 | <div class="version">
|
238 |
| - <a href='https://pytorch.org/docs/versions.html'>main (2.1.0a0+git788ff06 ) ▼</a> |
| 238 | + <a href='https://pytorch.org/docs/versions.html'>main (2.1.0a0+gitcde597e ) ▼</a> |
239 | 239 | </div>
|
240 | 240 |
|
241 | 241 |
|
@@ -487,11 +487,18 @@ <h1>Source code for torch</h1><div class="highlight"><pre>
|
487 | 487 | <span class="k">if</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>
|
488 | 488 | <span class="k">raise</span> <span class="ne">Exception</span><span class="p">(</span><span class="s2">"Python 2 has reached end-of-life and is no longer supported by PyTorch."</span><span class="p">)</span>
|
489 | 489 |
|
| 490 | +<span class="c1"># multipy/deploy is setting this import before importing torch, this is the most</span> |
| 491 | +<span class="c1"># reliable way we have to detect if we're running within deploy.</span> |
| 492 | +<span class="c1"># https://github.com/pytorch/multipy/blob/d60f34ad38c371e441fe7ffdb77a3c3dda5a5d19/multipy/runtime/interpreter/interpreter_impl.cpp#L134-L137</span> |
| 493 | +<span class="k">def</span> <span class="nf">_running_with_deploy</span><span class="p">():</span> |
| 494 | + <span class="k">return</span> <span class="n">sys</span><span class="o">.</span><span class="n">modules</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"torch._meta_registrations"</span><span class="p">,</span> <span class="kc">None</span><span class="p">)</span> <span class="ow">is</span> <span class="nb">object</span> |
| 495 | + |
490 | 496 | <span class="kn">from</span> <span class="nn">._utils</span> <span class="kn">import</span> <span class="n">_import_dotted_name</span><span class="p">,</span> <span class="n">classproperty</span>
|
491 | 497 | <span class="kn">from</span> <span class="nn">._utils_internal</span> <span class="kn">import</span> <span class="n">get_file_path</span><span class="p">,</span> <span class="n">prepare_multiprocessing_environment</span><span class="p">,</span> \
|
492 | 498 | <span class="n">USE_RTLD_GLOBAL_WITH_LIBTORCH</span><span class="p">,</span> <span class="n">USE_GLOBAL_DEPS</span>
|
| 499 | + |
493 | 500 | <span class="c1"># TODO(torch_deploy) figure out how to freeze version.py in fbcode build</span>
|
494 |
| -<span class="k">if</span> <span class="n">sys</span><span class="o">.</span><span class="n">executable</span> <span class="o">==</span> <span class="s1">'torch_deploy'</span><span class="p">:</span> |
| 501 | +<span class="k">if</span> <span class="n">_running_with_deploy</span><span class="p">():</span> |
495 | 502 | <span class="n">__version__</span> <span class="o">=</span> <span class="s2">"torch-deploy-1.8"</span>
|
496 | 503 | <span class="k">else</span><span class="p">:</span>
|
497 | 504 | <span class="kn">from</span> <span class="nn">.torch_version</span> <span class="kn">import</span> <span class="n">__version__</span> <span class="k">as</span> <span class="n">__version__</span>
|
@@ -625,7 +632,7 @@ <h1>Source code for torch</h1><div class="highlight"><pre>
|
625 | 632 |
|
626 | 633 | <span class="c1"># See Note [Global dependencies]</span>
|
627 | 634 | <span class="k">def</span> <span class="nf">_load_global_deps</span><span class="p">():</span>
|
628 |
| - <span class="k">if</span> <span class="n">sys</span><span class="o">.</span><span class="n">executable</span> <span class="o">==</span> <span class="s1">'torch_deploy'</span> <span class="ow">or</span> <span class="n">platform</span><span class="o">.</span><span class="n">system</span><span class="p">()</span> <span class="o">==</span> <span class="s1">'Windows'</span><span class="p">:</span> |
| 635 | + <span class="k">if</span> <span class="n">_running_with_deploy</span><span class="p">()</span> <span class="ow">or</span> <span class="n">platform</span><span class="o">.</span><span class="n">system</span><span class="p">()</span> <span class="o">==</span> <span class="s1">'Windows'</span><span class="p">:</span> |
629 | 636 | <span class="k">return</span>
|
630 | 637 |
|
631 | 638 | <span class="n">lib_name</span> <span class="o">=</span> <span class="s1">'libtorch_global_deps'</span> <span class="o">+</span> <span class="p">(</span><span class="s1">'.dylib'</span> <span class="k">if</span> <span class="n">platform</span><span class="o">.</span><span class="n">system</span><span class="p">()</span> <span class="o">==</span> <span class="s1">'Darwin'</span> <span class="k">else</span> <span class="s1">'.so'</span><span class="p">)</span>
|
@@ -659,7 +666,7 @@ <h1>Source code for torch</h1><div class="highlight"><pre>
|
659 | 666 |
|
660 | 667 |
|
661 | 668 | <span class="k">if</span> <span class="p">(</span><span class="n">USE_RTLD_GLOBAL_WITH_LIBTORCH</span> <span class="ow">or</span> <span class="n">os</span><span class="o">.</span><span class="n">getenv</span><span class="p">(</span><span class="s1">'TORCH_USE_RTLD_GLOBAL'</span><span class="p">))</span> <span class="ow">and</span> \
|
662 |
| - <span class="p">(</span><span class="n">sys</span><span class="o">.</span><span class="n">executable</span> <span class="o">==</span> <span class="s2">"torch_deploy"</span> <span class="ow">or</span> <span class="n">platform</span><span class="o">.</span><span class="n">system</span><span class="p">()</span> <span class="o">!=</span> <span class="s1">'Windows'</span><span class="p">):</span> |
| 669 | + <span class="p">(</span><span class="n">_running_with_deploy</span><span class="p">()</span> <span class="ow">or</span> <span class="n">platform</span><span class="o">.</span><span class="n">system</span><span class="p">()</span> <span class="o">!=</span> <span class="s1">'Windows'</span><span class="p">):</span> |
663 | 670 | <span class="c1"># Do it the hard way. You might want to load libtorch with RTLD_GLOBAL in a</span>
|
664 | 671 | <span class="c1"># few circumstances:</span>
|
665 | 672 | <span class="c1">#</span>
|
@@ -1745,7 +1752,7 @@ <h1>Source code for torch</h1><div class="highlight"><pre>
|
1745 | 1752 | <span class="c1">################################################################################</span>
|
1746 | 1753 |
|
1747 | 1754 | <span class="k">def</span> <span class="nf">manager_path</span><span class="p">():</span>
|
1748 |
| - <span class="k">if</span> <span class="n">sys</span><span class="o">.</span><span class="n">executable</span> <span class="o">==</span> <span class="s1">'torch_deploy'</span> <span class="ow">or</span> <span class="n">platform</span><span class="o">.</span><span class="n">system</span><span class="p">()</span> <span class="o">==</span> <span class="s1">'Windows'</span><span class="p">:</span> |
| 1755 | + <span class="k">if</span> <span class="n">_running_with_deploy</span><span class="p">()</span> <span class="ow">or</span> <span class="n">platform</span><span class="o">.</span><span class="n">system</span><span class="p">()</span> <span class="o">==</span> <span class="s1">'Windows'</span><span class="p">:</span> |
1749 | 1756 | <span class="k">return</span> <span class="sa">b</span><span class="s2">""</span>
|
1750 | 1757 | <span class="n">path</span> <span class="o">=</span> <span class="n">get_file_path</span><span class="p">(</span><span class="s1">'torch'</span><span class="p">,</span> <span class="s1">'bin'</span><span class="p">,</span> <span class="s1">'torch_shm_manager'</span><span class="p">)</span>
|
1751 | 1758 | <span class="n">prepare_multiprocessing_environment</span><span class="p">(</span><span class="n">get_file_path</span><span class="p">(</span><span class="s1">'torch'</span><span class="p">))</span>
|
@@ -2034,7 +2041,7 @@ <h1>Source code for torch</h1><div class="highlight"><pre>
|
2034 | 2041 | <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">compiler_fn</span><span class="p">(</span><span class="n">model_</span><span class="p">,</span> <span class="n">inputs_</span><span class="p">,</span> <span class="o">**</span><span class="bp">self</span><span class="o">.</span><span class="n">kwargs</span><span class="p">)</span>
|
2035 | 2042 |
|
2036 | 2043 |
|
2037 |
| -<div class="viewcode-block" id="compile"><a class="viewcode-back" href="../generated/torch.compile.html#torch.compile">[docs]</a><span class="k">def</span> <span class="nf">compile</span><span class="p">(</span><span class="n">model</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">Callable</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span> <span class="o">*</span><span class="p">,</span> |
| 2044 | +<div class="viewcode-block" id="compile"><a class="viewcode-back" href="../compile/generated/torch.compile.html#torch.compile">[docs]</a><span class="k">def</span> <span class="nf">compile</span><span class="p">(</span><span class="n">model</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">Callable</span><span class="p">]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">,</span> <span class="o">*</span><span class="p">,</span> |
2038 | 2045 | <span class="n">fullgraph</span><span class="p">:</span> <span class="n">builtins</span><span class="o">.</span><span class="n">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">,</span>
|
2039 | 2046 | <span class="n">dynamic</span><span class="p">:</span> <span class="n">builtins</span><span class="o">.</span><span class="n">bool</span> <span class="o">=</span> <span class="kc">False</span><span class="p">,</span>
|
2040 | 2047 | <span class="n">backend</span><span class="p">:</span> <span class="n">Union</span><span class="p">[</span><span class="nb">str</span><span class="p">,</span> <span class="n">Callable</span><span class="p">]</span> <span class="o">=</span> <span class="s2">"inductor"</span><span class="p">,</span>
|
|
0 commit comments