|
235 | 235 | <div class="pytorch-left-menu-search">
|
236 | 236 |
|
237 | 237 | <div class="version">
|
238 |
| - <a href='https://pytorch.org/docs/versions.html'>master (2.1.0a0+git7e0c26d ) ▼</a> |
| 238 | + <a href='https://pytorch.org/docs/versions.html'>master (2.1.0a0+gitdda9523 ) ▼</a> |
239 | 239 | </div>
|
240 | 240 |
|
241 | 241 |
|
@@ -839,7 +839,7 @@ <h1>Source code for torch._dynamo.eval_frame</h1><div class="highlight"><pre>
|
839 | 839 | <span class="ow">or</span> <span class="n">skipfiles</span><span class="o">.</span><span class="n">check</span><span class="p">(</span><span class="n">frame</span><span class="o">.</span><span class="n">f_code</span><span class="o">.</span><span class="n">co_filename</span><span class="p">)</span>
|
840 | 840 | <span class="ow">or</span> <span class="n">config</span><span class="o">.</span><span class="n">disable</span>
|
841 | 841 | <span class="p">):</span>
|
842 |
| - <span class="n">log</span><span class="o">.</span><span class="n">debug</span><span class="p">(</span><span class="sa">f</span><span class="s2">"skipping </span><span class="si">{</span><span class="n">frame</span><span class="o">.</span><span class="n">f_code</span><span class="o">.</span><span class="n">co_name</span><span class="si">}</span><span class="s2"> </span><span class="si">{</span><span class="n">frame</span><span class="o">.</span><span class="n">f_code</span><span class="o">.</span><span class="n">co_filename</span><span class="si">}</span><span class="s2">"</span><span class="p">)</span> |
| 842 | + <span class="n">log</span><span class="o">.</span><span class="n">debug</span><span class="p">(</span><span class="s2">"skipping </span><span class="si">%s</span><span class="s2"> </span><span class="si">%s</span><span class="s2">"</span><span class="p">,</span> <span class="n">frame</span><span class="o">.</span><span class="n">f_code</span><span class="o">.</span><span class="n">co_name</span><span class="p">,</span> <span class="n">frame</span><span class="o">.</span><span class="n">f_code</span><span class="o">.</span><span class="n">co_filename</span><span class="p">)</span> |
843 | 843 | <span class="k">return</span> <span class="kc">None</span>
|
844 | 844 | <span class="k">if</span> <span class="n">frame</span><span class="o">.</span><span class="n">f_code</span><span class="o">.</span><span class="n">co_filename</span> <span class="o">==</span> <span class="s2">"<string>"</span> <span class="ow">and</span> <span class="n">frame</span><span class="o">.</span><span class="n">f_code</span><span class="o">.</span><span class="n">co_name</span> <span class="o">==</span> <span class="s2">"__new__"</span><span class="p">:</span>
|
845 | 845 | <span class="c1"># nametuple constructor</span>
|
@@ -901,8 +901,13 @@ <h1>Source code for torch._dynamo.eval_frame</h1><div class="highlight"><pre>
|
901 | 901 | <span class="k">def</span> <span class="nf">check_if_dynamo_supported</span><span class="p">():</span>
|
902 | 902 | <span class="k">if</span> <span class="n">sys</span><span class="o">.</span><span class="n">platform</span> <span class="o">==</span> <span class="s2">"win32"</span><span class="p">:</span>
|
903 | 903 | <span class="k">raise</span> <span class="ne">RuntimeError</span><span class="p">(</span><span class="s2">"Windows not yet supported for torch.compile"</span><span class="p">)</span>
|
904 |
| - <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> <span class="mi">11</span><span class="p">):</span> |
905 |
| - <span class="k">raise</span> <span class="ne">RuntimeError</span><span class="p">(</span><span class="s2">"Python 3.11+ not yet supported for torch.compile"</span><span class="p">)</span> |
| 904 | + <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> <span class="mi">12</span><span class="p">):</span> |
| 905 | + <span class="k">raise</span> <span class="ne">RuntimeError</span><span class="p">(</span><span class="s2">"Python 3.12+ not yet supported for torch.compile"</span><span class="p">)</span> |
| 906 | + <span class="k">elif</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">11</span><span class="p">):</span> |
| 907 | + <span class="n">warnings</span><span class="o">.</span><span class="n">warn</span><span class="p">(</span> |
| 908 | + <span class="s2">"torch.compile support of Python 3.11 is experimental. "</span> |
| 909 | + <span class="s2">"Program may generate incorrect results or segfault."</span> |
| 910 | + <span class="p">)</span> |
906 | 911 |
|
907 | 912 |
|
908 | 913 | <span class="k">def</span> <span class="nf">is_dynamo_supported</span><span class="p">():</span>
|
|
0 commit comments