Skip to content

Commit

Permalink
Update doc from commit ad14582
Browse files Browse the repository at this point in the history
  • Loading branch information
torchxlabot2 committed Dec 19, 2023
1 parent 863c7a1 commit 7305298
Show file tree
Hide file tree
Showing 13 changed files with 19 additions and 16 deletions.
2 changes: 1 addition & 1 deletion master/_modules/index.html
Original file line number Diff line number Diff line change
Expand Up @@ -225,7 +225,7 @@


<div class="version">
master (2.2.0+git6f3f38f )
master (2.2.0+gitad14582 )
</div>


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


<div class="version">
master (2.2.0+git6f3f38f )
master (2.2.0+gitad14582 )
</div>


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


<div class="version">
master (2.2.0+git6f3f38f )
master (2.2.0+gitad14582 )
</div>


Expand Down Expand Up @@ -411,7 +411,7 @@ <h1>Source code for torch_xla.core.xla_model</h1><div class="highlight"><pre>
<span class="w"> </span><span class="sd">&quot;&quot;&quot;Returns a list of supported devices of a given kind.</span>

<span class="sd"> Args:</span>
<span class="sd"> devkind (string..., optional): If specified, one of `TPU`, `GPU`, `XPU`, </span>
<span class="sd"> devkind (string..., optional): If specified, one of `TPU`, `GPU`, `XPU`,</span>
<span class="sd"> `NEURON` or `CPU` (the &#39;GPU&#39; XLA device is currently not implemented).</span>
<span class="sd"> max_devices (int, optional): The maximum number of devices to be returned of</span>
<span class="sd"> that kind.</span>
Expand Down Expand Up @@ -521,7 +521,7 @@ <h1>Source code for torch_xla.core.xla_model</h1><div class="highlight"><pre>
<span class="sd"> n (int, optional): The specific instance (ordinal) to be returned. If</span>
<span class="sd"> specified, the specific XLA device instance will be returned. Otherwise</span>
<span class="sd"> the first device of `devkind` will be returned.</span>
<span class="sd"> devkind (string..., optional): If specified, one of `TPU`, `CUDA`, `XPU` </span>
<span class="sd"> devkind (string..., optional): If specified, one of `TPU`, `CUDA`, `XPU`</span>
<span class="sd"> `NEURON`, `ROCM` or `CPU`.</span>

<span class="sd"> Returns:</span>
Expand All @@ -545,7 +545,10 @@ <h1>Source code for torch_xla.core.xla_model</h1><div class="highlight"><pre>
<span class="k">return</span> <span class="n">_DEVICES</span><span class="o">.</span><span class="n">value</span><span class="p">[</span><span class="nb">int</span><span class="p">(</span><span class="n">m</span><span class="o">.</span><span class="n">group</span><span class="p">(</span><span class="mi">1</span><span class="p">))]</span>


<span class="k">def</span> <span class="nf">xla_real_devices</span><span class="p">(</span><span class="n">devices</span><span class="p">):</span>
<span class="k">def</span> <span class="nf">xla_real_devices</span><span class="p">(</span><span class="n">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="k">if</span> <span class="ow">not</span> <span class="n">devices</span><span class="p">:</span>
<span class="n">devices</span> <span class="o">=</span> <span class="n">get_xla_supported_devices</span><span class="p">()</span>

<span class="k">return</span> <span class="p">[</span><span class="n">_xla_real_device</span><span class="p">(</span><span class="n">device</span><span class="p">)</span> <span class="k">for</span> <span class="n">device</span> <span class="ow">in</span> <span class="n">devices</span><span class="p">]</span>


Expand All @@ -557,7 +560,7 @@ <h1>Source code for torch_xla.core.xla_model</h1><div class="highlight"><pre>
<span class="sd"> real device.</span>

<span class="sd"> Returns:</span>
<span class="sd"> A string representation of the hardware type (`CPU`, `TPU`, `XPU`, `NEURON`, `GPU`, `CUDA`, `ROCM`) </span>
<span class="sd"> A string representation of the hardware type (`CPU`, `TPU`, `XPU`, `NEURON`, `GPU`, `CUDA`, `ROCM`)</span>
<span class="sd"> of the given device.</span>
<span class="sd"> &quot;&quot;&quot;</span>
<span class="n">real_device</span> <span class="o">=</span> <span class="n">_xla_real_device</span><span class="p">(</span><span class="n">device</span><span class="p">)</span>
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 @@ -225,7 +225,7 @@


<div class="version">
master (2.2.0+git6f3f38f )
master (2.2.0+gitad14582 )
</div>


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


<div class="version">
master (2.2.0+git6f3f38f )
master (2.2.0+gitad14582 )
</div>


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


<div class="version">
master (2.2.0+git6f3f38f )
master (2.2.0+gitad14582 )
</div>


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


<div class="version">
master (2.2.0+git6f3f38f )
master (2.2.0+gitad14582 )
</div>


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


<div class="version">
master (2.2.0+git6f3f38f )
master (2.2.0+gitad14582 )
</div>


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


<div class="version">
master (2.2.0+git6f3f38f )
master (2.2.0+gitad14582 )
</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 @@ -225,7 +225,7 @@


<div class="version">
master (2.2.0+git6f3f38f )
master (2.2.0+gitad14582 )
</div>


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


<div class="version">
master (2.2.0+git6f3f38f )
master (2.2.0+gitad14582 )
</div>


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


<div class="version">
master (2.2.0+git6f3f38f )
master (2.2.0+gitad14582 )
</div>


Expand Down

0 comments on commit 7305298

Please sign in to comment.