Skip to content

Commit

Permalink
Update doc from commit dc20b2d
Browse files Browse the repository at this point in the history
  • Loading branch information
torchxlabot2 committed Oct 30, 2024
1 parent 5f3cba2 commit c8545d4
Show file tree
Hide file tree
Showing 45 changed files with 48 additions and 47 deletions.
2 changes: 1 addition & 1 deletion master/_modules/index.html
Original file line number Diff line number Diff line change
Expand Up @@ -265,7 +265,7 @@


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


Expand Down
9 changes: 5 additions & 4 deletions master/_modules/torch_xla/core/xla_model.html
Original file line number Diff line number Diff line change
Expand Up @@ -265,7 +265,7 @@


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


Expand Down Expand Up @@ -1722,9 +1722,10 @@ <h1>Source code for torch_xla.core.xla_model</h1><div class="highlight"><pre>
<span class="k">if</span> <span class="n">sharding</span> <span class="ow">and</span> <span class="n">tensor</span><span class="o">.</span><span class="n">dim</span><span class="p">()</span> <span class="o">&gt;</span> <span class="mi">0</span> <span class="ow">and</span> <span class="p">(</span><span class="n">tensor</span><span class="o">.</span><span class="n">size</span><span class="p">()[</span><span class="mi">0</span><span class="p">]</span> <span class="o">%</span>
<span class="n">local_runtime_device_count</span><span class="p">)</span> <span class="o">!=</span> <span class="mi">0</span><span class="p">:</span>
<span class="k">raise</span> <span class="ne">RuntimeError</span><span class="p">(</span>
<span class="s2">&quot;When minibatch is configured, the per-host batch size must be divisible &quot;</span>
<span class="o">+</span> <span class="s2">&quot;by local runtime device count. Per host input data shape &quot;</span> <span class="o">+</span>
<span class="sa">f</span><span class="s2">&quot;= </span><span class="si">{</span><span class="n">tensor</span><span class="o">.</span><span class="n">size</span><span class="p">()</span><span class="si">}</span><span class="s2">, local_runtime_device_count = </span><span class="si">{</span><span class="n">local_runtime_device_count</span><span class="si">}</span><span class="s2">&quot;</span>
<span class="s2">&quot;When minibatch is configured, batch dimension of the tensor &quot;</span> <span class="o">+</span>
<span class="s2">&quot;must be divisible by local runtime device count.input data shape &quot;</span>
<span class="o">+</span>
<span class="sa">f</span><span class="s2">&quot;=</span><span class="si">{</span><span class="n">tensor</span><span class="o">.</span><span class="n">size</span><span class="p">()</span><span class="si">}</span><span class="s2">, local_runtime_device_count = </span><span class="si">{</span><span class="n">local_runtime_device_count</span><span class="si">}</span><span class="s2">&quot;</span>
<span class="p">)</span>

<span class="n">xtensors</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_tensors_from_aten</span><span class="p">(</span><span class="n">tensors</span><span class="p">,</span> <span class="n">devices</span><span class="p">,</span>
Expand Down
2 changes: 1 addition & 1 deletion master/_modules/torch_xla/debug/metrics.html
Original file line number Diff line number Diff line change
Expand Up @@ -265,7 +265,7 @@


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -265,7 +265,7 @@


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -265,7 +265,7 @@


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


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


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


Expand Down
2 changes: 1 addition & 1 deletion master/perf/spmd_gpu.html
Original file line number Diff line number Diff line change
Expand Up @@ -267,7 +267,7 @@


<div class="version">
master (2.6.0+git3efe1eb )
master (2.6.0+gitdc20b2d )
</div>


Expand Down
Loading

0 comments on commit c8545d4

Please sign in to comment.