Skip to content

Commit

Permalink
Update doc from commit 38ed80e
Browse files Browse the repository at this point in the history
  • Loading branch information
@torchxlabot2
torchxlabot2 committed Dec 20, 2024
1 parent 8932aeb commit 89c62ea
Show file tree
Hide file tree
Showing 45 changed files with 60 additions and 45 deletions.
2 changes: 1 addition & 1 deletion release/r2.6/_modules/index.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -265,7 +265,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
19 changes: 17 additions & 2 deletions release/r2.6/_modules/torch_xla/core/xla_model.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -265,7 +265,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down Expand Up @@ -617,6 +617,21 @@ <h1>Source code for torch_xla.core.xla_model</h1><div class="highlight"><pre>
<span class="k">return</span> <span class="n">real_device</span><span class="o">.</span><span class="n">split</span><span class="p">(</span><span class="s1">&#39;:&#39;</span><span class="p">)[</span><span class="mi">0</span><span class="p">]</span></div>


<span class="k">def</span> <span class="nf">xla_device_kind</span><span class="p">(</span><span class="n">device</span><span class="p">:</span> <span class="n">Optional</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">torch</span><span class="o">.</span><span class="n">device</span><span class="p">]]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">str</span><span class="p">:</span>
<span class="w"> </span><span class="sd">&quot;&quot;&quot;Returns vendor-dependent string that uniquely identifies the kind of</span>
<span class="sd"> device.</span>

<span class="sd"> Args:</span>
<span class="sd"> device (string or torch.device): The xla device</span>

<span class="sd"> Returns:</span>
<span class="sd"> A vendor-dependent device kind string.</span>
<span class="sd"> &quot;&quot;&quot;</span>
<span class="k">if</span> <span class="n">device</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
<span class="n">device</span> <span class="o">=</span> <span class="n">torch_xla</span><span class="o">.</span><span class="n">_XLAC</span><span class="o">.</span><span class="n">_xla_get_default_device</span><span class="p">()</span>
<span class="k">return</span> <span class="n">torch_xla</span><span class="o">.</span><span class="n">_XLAC</span><span class="o">.</span><span class="n">_xla_device_kind</span><span class="p">(</span><span class="nb">str</span><span class="p">(</span><span class="n">device</span><span class="p">))</span>


<span class="k">def</span> <span class="nf">xla_replication_devices</span><span class="p">(</span>
<span class="n">local_devices</span><span class="p">:</span> <span class="n">Optional</span><span class="p">[</span><span class="n">List</span><span class="p">[</span><span class="n">torch</span><span class="o">.</span><span class="n">device</span><span class="p">]]</span> <span class="o">=</span> <span class="kc">None</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="n">List</span><span class="p">[</span><span class="nb">str</span><span class="p">]:</span>
<span class="n">real_devices</span> <span class="o">=</span> <span class="n">xla_real_devices</span><span class="p">(</span><span class="n">local_devices</span><span class="p">)</span>
Expand Down Expand Up @@ -1691,7 +1706,7 @@ <h1>Source code for torch_xla.core.xla_model</h1><div class="highlight"><pre>

<span class="k">def</span> <span class="nf">convert_fn</span><span class="p">(</span><span class="n">tensors</span><span class="p">):</span>
<span class="n">torch_xla</span><span class="o">.</span><span class="n">_XLAC</span><span class="o">.</span><span class="n">_xla_sync_multi</span><span class="p">(</span>
<span class="n">tensors</span><span class="p">,</span> <span class="n">devices</span><span class="o">=</span><span class="p">[],</span> <span class="n">wait</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span> <span class="n">sync_xla_data</span><span class="o">=</span><span class="kc">True</span><span class="p">)</span>
<span class="n">tensors</span><span class="p">,</span> <span class="n">devices</span><span class="o">=</span><span class="p">[],</span> <span class="n">wait</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span> <span class="n">sync_xla_data</span><span class="o">=</span><span class="kc">False</span><span class="p">)</span>
<span class="k">if</span> <span class="ow">not</span> <span class="n">convert</span><span class="p">:</span>
<span class="k">return</span> <span class="n">tensors</span>
<span class="k">return</span> <span class="n">torch_xla</span><span class="o">.</span><span class="n">_XLAC</span><span class="o">.</span><span class="n">_xla_get_cpu_tensors</span><span class="p">(</span><span class="n">tensors</span><span class="p">)</span>
Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/_modules/torch_xla/debug/metrics.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -265,7 +265,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/_modules/torch_xla/distributed/parallel_loader.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -265,7 +265,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/_modules/torch_xla/distributed/spmd/xla_sharding.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -265,7 +265,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/_modules/torch_xla/distributed/xla_multiprocessing.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -265,7 +265,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/_modules/torch_xla/experimental/eager.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -265,7 +265,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/_modules/torch_xla/runtime.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -265,7 +265,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/_modules/torch_xla/torch_xla.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -265,7 +265,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/accelerators/gpu.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -267,7 +267,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/accelerators/tpu.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -267,7 +267,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/contribute/bazel.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -266,7 +266,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/contribute/codegen_migration.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -267,7 +267,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/contribute/configure-environment.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -267,7 +267,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/contribute/op_lowering.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -267,7 +267,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/contribute/plugins.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -267,7 +267,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/features/distop.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -265,7 +265,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/features/pallas.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -267,7 +267,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/features/stablehlo.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -267,7 +267,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/features/triton.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -267,7 +267,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/genindex.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -265,7 +265,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/index.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -266,7 +266,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/learn/api-guide.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -267,7 +267,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/learn/dynamic_shape.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -267,7 +267,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/learn/eager.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -267,7 +267,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/learn/pjrt.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -267,7 +267,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/learn/pytorch-on-xla-devices.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -267,7 +267,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/learn/troubleshoot.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -267,7 +267,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/learn/xla-overview.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -267,7 +267,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/notes/source_of_recompilation.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -265,7 +265,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
Binary file modified release/r2.6/objects.inv
View file
Binary file not shown.
2 changes: 1 addition & 1 deletion release/r2.6/perf/amp.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -267,7 +267,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/perf/ddp.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -267,7 +267,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/perf/dynamo.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -267,7 +267,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/perf/fori_loop.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -267,7 +267,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/perf/fsdp.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -267,7 +267,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/perf/fsdpv2.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -267,7 +267,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/perf/quantized_ops.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -267,7 +267,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/perf/recompilation.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -267,7 +267,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/perf/spmd_advanced.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -267,7 +267,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/perf/spmd_basic.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -267,7 +267,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
2 changes: 1 addition & 1 deletion release/r2.6/perf/spmd_distributed_checkpoint.html
View file
Original file line number Diff line number Diff line change
Expand Up @@ -267,7 +267,7 @@


<div class="version">
master (2.6.0+git32e3c60 )
master (2.6.0+git38ed80e )
</div>


Expand Down
Loading

0 comments on commit 89c62ea

Please sign in to comment.