|
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+git6f13d68 ) ▼</a> |
| 238 | + <a href='https://pytorch.org/docs/versions.html'>main (2.1.0a0+git48346a4 ) ▼</a> |
239 | 239 | </div> |
240 | 240 |
|
241 | 241 |
|
|
309 | 309 | <li class="toctree-l1"><a class="reference internal" href="../compile/custom-backends.html">Custom Backends</a></li> |
310 | 310 | <li class="toctree-l1"><a class="reference internal" href="../compile/deep-dive.html">TorchDynamo Deeper Dive</a></li> |
311 | 311 | <li class="toctree-l1"><a class="reference internal" href="../compile/performance-dashboard.html">PyTorch 2.0 Performance Dashboard</a></li> |
| 312 | +<li class="toctree-l1"><a class="reference internal" href="../compile/torchfunc-and-torchcompile.html">torch.func interaction with torch.compile</a></li> |
312 | 313 | <li class="toctree-l1"><a class="reference internal" href="../ir.html">IRs</a></li> |
| 314 | +<li class="toctree-l1"><a class="reference internal" href="../compile/dynamic-shapes.html">Dynamic shapes</a></li> |
| 315 | +<li class="toctree-l1"><a class="reference internal" href="../compile/fake-tensor.html">Fake tensor</a></li> |
| 316 | +<li class="toctree-l1"><a class="reference internal" href="../logging.html">torch._logging</a></li> |
| 317 | +<li class="toctree-l1"><a class="reference internal" href="../compile/transformations.html">Writing Graph Transformations on ATen IR</a></li> |
313 | 318 | </ul> |
314 | 319 | <p class="caption" role="heading"><span class="caption-text">Language Bindings</span></p> |
315 | 320 | <ul> |
@@ -1964,6 +1969,10 @@ <h1>Source code for torch</h1><div class="highlight"><pre> |
1964 | 1969 | <span class="n">options</span> <span class="ow">or</span> <span class="p">()</span> |
1965 | 1970 | <span class="p">),</span> <span class="s2">"triton.cudagraphs does not support dynamic shapes. Please set dynamic=False or triton.cudagraphs=False"</span> |
1966 | 1971 |
|
| 1972 | + <span class="c1"># FIXME: CUPTI Lazy Re-init and CUDA Graph crashes with CUDA 11.</span> |
| 1973 | + <span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">config</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">"triton.cudagraphs"</span><span class="p">,</span> <span class="kc">False</span><span class="p">):</span> |
| 1974 | + <span class="n">os</span><span class="o">.</span><span class="n">environ</span><span class="p">[</span><span class="s2">"DISABLE_CUPTI_LAZY_REINIT"</span><span class="p">]</span> <span class="o">=</span> <span class="s2">"1"</span> |
| 1975 | + |
1967 | 1976 | <span class="k">def</span> <span class="fm">__eq__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">other</span><span class="p">):</span> |
1968 | 1977 | <span class="k">return</span> <span class="p">(</span><span class="nb">isinstance</span><span class="p">(</span><span class="n">other</span><span class="p">,</span> <span class="n">_TorchCompileInductorWrapper</span><span class="p">)</span> <span class="ow">and</span> |
1969 | 1978 | <span class="bp">self</span><span class="o">.</span><span class="n">config</span> <span class="o">==</span> <span class="n">other</span><span class="o">.</span><span class="n">config</span> <span class="ow">and</span> |
@@ -2041,7 +2050,7 @@ <h1>Source code for torch</h1><div class="highlight"><pre> |
2041 | 2050 | <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> |
2042 | 2051 |
|
2043 | 2052 |
|
2044 | | -<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> |
| 2053 | +<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> |
2045 | 2054 | <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> |
2046 | 2055 | <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> |
2047 | 2056 | <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