|
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 (1.14.0a0+git7f4b4d2 ) ▼</a> |
| 238 | + <a href='https://pytorch.org/docs/versions.html'>master (1.14.0a0+gita00bd6f ) ▼</a> |
239 | 239 | </div> |
240 | 240 |
|
241 | 241 |
|
|
323 | 323 | <li class="toctree-l1"><a class="reference internal" href="../../distributed.optim.html">torch.distributed.optim</a></li> |
324 | 324 | <li class="toctree-l1"><a class="reference internal" href="../../distributed.checkpoint.html">torch.distributed.checkpoint</a></li> |
325 | 325 | <li class="toctree-l1"><a class="reference internal" href="../../distributions.html">torch.distributions</a></li> |
| 326 | +<li class="toctree-l1"><a class="reference internal" href="../../_dynamo.html">torch._dynamo</a></li> |
326 | 327 | <li class="toctree-l1"><a class="reference internal" href="../../fft.html">torch.fft</a></li> |
327 | 328 | <li class="toctree-l1"><a class="reference internal" href="../../futures.html">torch.futures</a></li> |
328 | 329 | <li class="toctree-l1"><a class="reference internal" href="../../fx.html">torch.fx</a></li> |
@@ -547,7 +548,8 @@ <h1>Source code for torch._tensor</h1><div class="highlight"><pre> |
547 | 548 | <span class="c1"># Update the test in test_serialization if you remove 'meta' from here</span> |
548 | 549 | <span class="k">if</span> <span class="p">(</span> |
549 | 550 | <span class="bp">self</span><span class="o">.</span><span class="n">is_sparse</span> |
550 | | - <span class="ow">or</span> <span class="bp">self</span><span class="o">.</span><span class="n">device</span><span class="o">.</span><span class="n">type</span> <span class="ow">in</span> <span class="p">[</span><span class="s2">"lazy"</span><span class="p">,</span> <span class="s2">"xla"</span><span class="p">,</span> <span class="s2">"mps"</span><span class="p">,</span> <span class="s2">"ort"</span><span class="p">,</span> <span class="s2">"meta"</span><span class="p">,</span> <span class="s2">"hpu"</span><span class="p">]</span> |
| 551 | + <span class="ow">or</span> <span class="bp">self</span><span class="o">.</span><span class="n">device</span><span class="o">.</span><span class="n">type</span> |
| 552 | + <span class="ow">in</span> <span class="p">[</span><span class="s2">"lazy"</span><span class="p">,</span> <span class="s2">"xla"</span><span class="p">,</span> <span class="s2">"mps"</span><span class="p">,</span> <span class="s2">"ort"</span><span class="p">,</span> <span class="s2">"meta"</span><span class="p">,</span> <span class="s2">"hpu"</span><span class="p">,</span> <span class="s2">"ipu"</span><span class="p">]</span> |
551 | 553 | <span class="ow">or</span> <span class="p">(</span> |
552 | 554 | <span class="ow">not</span> <span class="n">torch</span><span class="o">.</span><span class="n">_C</span><span class="o">.</span><span class="n">_has_storage</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> |
553 | 555 | <span class="ow">and</span> <span class="bp">self</span><span class="o">.</span><span class="n">device</span><span class="o">.</span><span class="n">type</span> <span class="o">==</span> <span class="s2">"privateuseone"</span> |
@@ -781,22 +783,32 @@ <h1>Source code for torch._tensor</h1><div class="highlight"><pre> |
781 | 783 | <span class="s2">"sparse tensor __reduce_ex__ for layout `</span><span class="si">%s</span><span class="s2">`"</span> <span class="o">%</span> <span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">layout</span><span class="p">)</span> |
782 | 784 | <span class="p">)</span> |
783 | 785 | <span class="k">return</span> <span class="p">(</span><span class="n">torch</span><span class="o">.</span><span class="n">_utils</span><span class="o">.</span><span class="n">_rebuild_sparse_tensor</span><span class="p">,</span> <span class="n">args_sparse</span><span class="p">)</span> |
784 | | - <span class="k">elif</span> <span class="bp">self</span><span class="o">.</span><span class="n">is_sparse_csr</span><span class="p">:</span> |
785 | | - <span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">layout</span> <span class="o">==</span> <span class="n">torch</span><span class="o">.</span><span class="n">sparse_csr</span><span class="p">:</span> |
786 | | - <span class="n">args_sparse_csr</span> <span class="o">=</span> <span class="p">(</span> |
787 | | - <span class="bp">self</span><span class="o">.</span><span class="n">layout</span><span class="p">,</span> |
788 | | - <span class="p">(</span> |
789 | | - <span class="bp">self</span><span class="o">.</span><span class="n">crow_indices</span><span class="p">(),</span> |
790 | | - <span class="bp">self</span><span class="o">.</span><span class="n">col_indices</span><span class="p">(),</span> |
791 | | - <span class="bp">self</span><span class="o">.</span><span class="n">values</span><span class="p">(),</span> |
792 | | - <span class="bp">self</span><span class="o">.</span><span class="n">size</span><span class="p">(),</span> |
793 | | - <span class="p">),</span> |
| 786 | + <span class="k">elif</span> <span class="bp">self</span><span class="o">.</span><span class="n">layout</span> <span class="ow">in</span> <span class="p">{</span> |
| 787 | + <span class="n">torch</span><span class="o">.</span><span class="n">sparse_csr</span><span class="p">,</span> |
| 788 | + <span class="n">torch</span><span class="o">.</span><span class="n">sparse_csc</span><span class="p">,</span> |
| 789 | + <span class="n">torch</span><span class="o">.</span><span class="n">sparse_bsr</span><span class="p">,</span> |
| 790 | + <span class="n">torch</span><span class="o">.</span><span class="n">sparse_bsc</span><span class="p">,</span> |
| 791 | + <span class="p">}:</span> |
| 792 | + <span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">layout</span> <span class="ow">in</span> <span class="p">{</span><span class="n">torch</span><span class="o">.</span><span class="n">sparse_csr</span><span class="p">,</span> <span class="n">torch</span><span class="o">.</span><span class="n">sparse_bsr</span><span class="p">}:</span> |
| 793 | + <span class="n">compressed_indices</span><span class="p">,</span> <span class="n">plain_indices</span> <span class="o">=</span> <span class="p">(</span> |
| 794 | + <span class="bp">self</span><span class="o">.</span><span class="n">crow_indices</span><span class="p">(),</span> |
| 795 | + <span class="bp">self</span><span class="o">.</span><span class="n">col_indices</span><span class="p">(),</span> |
794 | 796 | <span class="p">)</span> |
795 | 797 | <span class="k">else</span><span class="p">:</span> |
796 | | - <span class="k">raise</span> <span class="ne">NotImplementedError</span><span class="p">(</span> |
797 | | - <span class="s2">"sparse csr tensor __reduce_ex__ for layout `</span><span class="si">%s</span><span class="s2">`"</span> <span class="o">%</span> <span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">layout</span><span class="p">)</span> |
| 798 | + <span class="n">compressed_indices</span><span class="p">,</span> <span class="n">plain_indices</span> <span class="o">=</span> <span class="p">(</span> |
| 799 | + <span class="bp">self</span><span class="o">.</span><span class="n">ccol_indices</span><span class="p">(),</span> |
| 800 | + <span class="bp">self</span><span class="o">.</span><span class="n">row_indices</span><span class="p">(),</span> |
798 | 801 | <span class="p">)</span> |
799 | | - <span class="k">return</span> <span class="p">(</span><span class="n">torch</span><span class="o">.</span><span class="n">_utils</span><span class="o">.</span><span class="n">_rebuild_sparse_csr_tensor</span><span class="p">,</span> <span class="n">args_sparse_csr</span><span class="p">)</span> |
| 802 | + <span class="n">args_sparse_compressed</span> <span class="o">=</span> <span class="p">(</span> |
| 803 | + <span class="bp">self</span><span class="o">.</span><span class="n">layout</span><span class="p">,</span> |
| 804 | + <span class="p">(</span> |
| 805 | + <span class="n">compressed_indices</span><span class="p">,</span> |
| 806 | + <span class="n">plain_indices</span><span class="p">,</span> |
| 807 | + <span class="bp">self</span><span class="o">.</span><span class="n">values</span><span class="p">(),</span> |
| 808 | + <span class="bp">self</span><span class="o">.</span><span class="n">size</span><span class="p">(),</span> |
| 809 | + <span class="p">),</span> |
| 810 | + <span class="p">)</span> |
| 811 | + <span class="k">return</span> <span class="p">(</span><span class="n">torch</span><span class="o">.</span><span class="n">_utils</span><span class="o">.</span><span class="n">_rebuild_sparse_tensor</span><span class="p">,</span> <span class="n">args_sparse_compressed</span><span class="p">)</span> |
800 | 812 | <span class="k">elif</span> <span class="p">(</span> |
801 | 813 | <span class="bp">self</span><span class="o">.</span><span class="n">data_ptr</span><span class="p">()</span> <span class="o">==</span> <span class="mi">0</span> |
802 | 814 | <span class="ow">and</span> <span class="nb">type</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="ow">is</span> <span class="ow">not</span> <span class="n">torch</span><span class="o">.</span><span class="n">Tensor</span> |
|
0 commit comments