Generate Python docs from pytorch/pytorch@fb696bf

main/_modules/index.html

@@ -280,7 +280,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+git75fa100 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+gitfb696bf ) &#x25BC</a>
     </div>
 
 

main/_modules/torch.html

@@ -280,7 +280,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+git75fa100 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+gitfb696bf ) &#x25BC</a>
     </div>
 
 
@@ -3087,6 +3087,53 @@ <h1>Source code for torch</h1><div class="highlight"><pre>
 <span class="kn">from</span> <span class="nn">torch</span> <span class="kn">import</span> <span class="n">_logging</span>
 
 <span class="n">_logging</span><span class="o">.</span><span class="n">_init_logs</span><span class="p">()</span>
+
+
+<span class="k">def</span> <span class="nf">_import_device_backends</span><span class="p">():</span>
+<span class="w">    </span><span class="sd">&quot;&quot;&quot;</span>
+<span class="sd">    Leverage the Python plugin mechanism to load out-of-the-tree device extensions.</span>
+<span class="sd">    See this RFC: https://github.com/pytorch/pytorch/issues/122468</span>
+<span class="sd">    &quot;&quot;&quot;</span>
+    <span class="kn">from</span> <span class="nn">importlib.metadata</span> <span class="kn">import</span> <span class="n">entry_points</span>
+
+    <span class="n">group_name</span> <span class="o">=</span> <span class="s2">&quot;torch.backends&quot;</span>
+    <span class="k">if</span> <span class="n">sys</span><span class="o">.</span><span class="n">version_info</span> <span class="o">&lt;</span> <span class="p">(</span><span class="mi">3</span><span class="p">,</span> <span class="mi">10</span><span class="p">):</span>
+        <span class="n">backend_extensions</span> <span class="o">=</span> <span class="n">entry_points</span><span class="p">()</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="n">group_name</span><span class="p">,</span> <span class="p">())</span>
+    <span class="k">else</span><span class="p">:</span>
+        <span class="n">backend_extensions</span> <span class="o">=</span> <span class="n">entry_points</span><span class="p">(</span><span class="n">group</span><span class="o">=</span><span class="n">group_name</span><span class="p">)</span>
+
+    <span class="k">for</span> <span class="n">backend_extension</span> <span class="ow">in</span> <span class="n">backend_extensions</span><span class="p">:</span>
+        <span class="k">try</span><span class="p">:</span>
+            <span class="c1"># Load the extension</span>
+            <span class="n">entrypoint</span> <span class="o">=</span> <span class="n">backend_extension</span><span class="o">.</span><span class="n">load</span><span class="p">()</span>
+            <span class="c1"># Call the entrypoint</span>
+            <span class="n">entrypoint</span><span class="p">()</span>
+        <span class="k">except</span> <span class="ne">Exception</span> <span class="k">as</span> <span class="n">err</span><span class="p">:</span>
+            <span class="k">raise</span> <span class="ne">RuntimeError</span><span class="p">(</span>
+                <span class="sa">f</span><span class="s2">&quot;Failed to load the backend extension: </span><span class="si">{</span><span class="n">backend_extension</span><span class="o">.</span><span class="n">name</span><span class="si">}</span><span class="s2">. &quot;</span>
+                <span class="sa">f</span><span class="s2">&quot;You can disable extension auto-loading with TORCH_DEVICE_BACKEND_AUTOLOAD=0.&quot;</span>
+            <span class="p">)</span> <span class="kn">from</span> <span class="nn">err</span>
+
+
+<span class="k">def</span> <span class="nf">_is_device_backend_autoload_enabled</span><span class="p">()</span> <span class="o">-&gt;</span> <span class="n">builtins</span><span class="o">.</span><span class="n">bool</span><span class="p">:</span>
+<span class="w">    </span><span class="sd">&quot;&quot;&quot;</span>
+<span class="sd">    Whether autoloading out-of-the-tree device extensions is enabled.</span>
+<span class="sd">    The switch depends on the value of the environment variable</span>
+<span class="sd">    `TORCH_DEVICE_BACKEND_AUTOLOAD`.</span>
+
+<span class="sd">    Returns:</span>
+<span class="sd">        bool: Whether to enable autoloading the extensions. Enabled by default.</span>
+
+<span class="sd">    Examples:</span>
+<span class="sd">        &gt;&gt;&gt; torch._is_device_backend_autoload_enabled()</span>
+<span class="sd">        True</span>
+<span class="sd">    &quot;&quot;&quot;</span>
+    <span class="c1"># enabled by default</span>
+    <span class="k">return</span> <span class="n">os</span><span class="o">.</span><span class="n">getenv</span><span class="p">(</span><span class="s2">&quot;TORCH_DEVICE_BACKEND_AUTOLOAD&quot;</span><span class="p">,</span> <span class="s2">&quot;1&quot;</span><span class="p">)</span> <span class="o">==</span> <span class="s2">&quot;1&quot;</span>
+
+
+<span class="k">if</span> <span class="n">_is_device_backend_autoload_enabled</span><span class="p">():</span>
+    <span class="n">_import_device_backends</span><span class="p">()</span>
 </pre></div>
 
              </article>

main/_modules/torch/__config__.html

@@ -280,7 +280,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+git75fa100 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+gitfb696bf ) &#x25BC</a>
     </div>
 
 

main/_modules/torch/__future__.html

@@ -280,7 +280,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+git75fa100 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+gitfb696bf ) &#x25BC</a>
     </div>
 
 

main/_modules/torch/_jit_internal.html

@@ -280,7 +280,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+git75fa100 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+gitfb696bf ) &#x25BC</a>
     </div>
 
 
@@ -2045,6 +2045,15 @@ <h1>Source code for torch._jit_internal</h1><div class="highlight"><pre>
     <span class="k">return</span> <span class="n">tensors</span>
 
 
+<span class="k">def</span> <span class="nf">_get_model_id</span><span class="p">(</span><span class="n">obj</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="n">Optional</span><span class="p">[</span><span class="nb">str</span><span class="p">]:</span>
+    <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">obj</span><span class="p">,</span> <span class="n">torch</span><span class="o">.</span><span class="n">jit</span><span class="o">.</span><span class="n">ScriptModule</span><span class="p">):</span>
+        <span class="k">return</span> <span class="nb">str</span><span class="p">(</span><span class="n">obj</span><span class="o">.</span><span class="n">_c</span><span class="o">.</span><span class="n">_type</span><span class="p">())</span>
+    <span class="k">elif</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">obj</span><span class="p">,</span> <span class="n">torch</span><span class="o">.</span><span class="n">jit</span><span class="o">.</span><span class="n">ScriptFunction</span><span class="p">):</span>
+        <span class="k">return</span> <span class="n">obj</span><span class="o">.</span><span class="n">qualified_name</span>
+    <span class="k">else</span><span class="p">:</span>
+        <span class="k">return</span> <span class="kc">None</span>
+
+
 <span class="c1"># In Python-3.11+ typed enums (i.e. IntEnum for example) retain number of base class methods in subclass</span>
 <span class="c1"># that were previously dropped. To preserve the behavior, explicitly drop them there</span>
 

main/_modules/torch/_lobpcg.html

@@ -280,7 +280,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+git75fa100 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+gitfb696bf ) &#x25BC</a>
     </div>
 
 

main/_modules/torch/_logging/_internal.html

@@ -280,7 +280,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+git75fa100 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+gitfb696bf ) &#x25BC</a>
     </div>
 
 

main/_modules/torch/_lowrank.html

@@ -280,7 +280,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+git75fa100 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+gitfb696bf ) &#x25BC</a>
     </div>
 
 

main/_modules/torch/_tensor.html

@@ -280,7 +280,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+git75fa100 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+gitfb696bf ) &#x25BC</a>
     </div>
 
 
@@ -1684,14 +1684,19 @@ <h1>Source code for torch._tensor</h1><div class="highlight"><pre>
         <span class="n">typestr</span> <span class="o">=</span> <span class="p">{</span>
             <span class="n">torch</span><span class="o">.</span><span class="n">complex64</span><span class="p">:</span> <span class="s2">&quot;&lt;c8&quot;</span><span class="p">,</span>
             <span class="n">torch</span><span class="o">.</span><span class="n">complex128</span><span class="p">:</span> <span class="s2">&quot;&lt;c16&quot;</span><span class="p">,</span>
+            <span class="n">torch</span><span class="o">.</span><span class="n">bfloat16</span><span class="p">:</span> <span class="s2">&quot;&lt;f2&quot;</span><span class="p">,</span>
             <span class="n">torch</span><span class="o">.</span><span class="n">float16</span><span class="p">:</span> <span class="s2">&quot;&lt;f2&quot;</span><span class="p">,</span>
             <span class="n">torch</span><span class="o">.</span><span class="n">float32</span><span class="p">:</span> <span class="s2">&quot;&lt;f4&quot;</span><span class="p">,</span>
             <span class="n">torch</span><span class="o">.</span><span class="n">float64</span><span class="p">:</span> <span class="s2">&quot;&lt;f8&quot;</span><span class="p">,</span>
             <span class="n">torch</span><span class="o">.</span><span class="n">uint8</span><span class="p">:</span> <span class="s2">&quot;|u1&quot;</span><span class="p">,</span>
             <span class="n">torch</span><span class="o">.</span><span class="n">int8</span><span class="p">:</span> <span class="s2">&quot;|i1&quot;</span><span class="p">,</span>
+            <span class="n">torch</span><span class="o">.</span><span class="n">uint16</span><span class="p">:</span> <span class="s2">&quot;&lt;u2&quot;</span><span class="p">,</span>
             <span class="n">torch</span><span class="o">.</span><span class="n">int16</span><span class="p">:</span> <span class="s2">&quot;&lt;i2&quot;</span><span class="p">,</span>
+            <span class="n">torch</span><span class="o">.</span><span class="n">uint32</span><span class="p">:</span> <span class="s2">&quot;&lt;u4&quot;</span><span class="p">,</span>
             <span class="n">torch</span><span class="o">.</span><span class="n">int32</span><span class="p">:</span> <span class="s2">&quot;&lt;i4&quot;</span><span class="p">,</span>
+            <span class="n">torch</span><span class="o">.</span><span class="n">uint64</span><span class="p">:</span> <span class="s2">&quot;&lt;u8&quot;</span><span class="p">,</span>
             <span class="n">torch</span><span class="o">.</span><span class="n">int64</span><span class="p">:</span> <span class="s2">&quot;&lt;i8&quot;</span><span class="p">,</span>
+            <span class="n">torch</span><span class="o">.</span><span class="n">bool</span><span class="p">:</span> <span class="s2">&quot;|b1&quot;</span><span class="p">,</span>
         <span class="p">}[</span><span class="bp">self</span><span class="o">.</span><span class="n">dtype</span><span class="p">]</span>
 
         <span class="n">itemsize</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">element_size</span><span class="p">()</span>

main/_modules/torch/_tensor_str.html

@@ -280,7 +280,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+git75fa100 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+gitfb696bf ) &#x25BC</a>
     </div>
 
 

main/_modules/torch/amp/autocast_mode.html

@@ -280,7 +280,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+git75fa100 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+gitfb696bf ) &#x25BC</a>
     </div>
 
 

main/_modules/torch/ao/nn/intrinsic/modules/fused.html

@@ -280,7 +280,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+git75fa100 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+gitfb696bf ) &#x25BC</a>
     </div>
 
 

main/_modules/torch/ao/nn/intrinsic/qat/modules/conv_fused.html

@@ -280,7 +280,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+git75fa100 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+gitfb696bf ) &#x25BC</a>
     </div>
 
 

main/_modules/torch/ao/nn/intrinsic/qat/modules/linear_relu.html

@@ -280,7 +280,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+git75fa100 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+gitfb696bf ) &#x25BC</a>
     </div>
 
 

main/_modules/torch/ao/nn/intrinsic/quantized/dynamic/modules/linear_relu.html

@@ -280,7 +280,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+git75fa100 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+gitfb696bf ) &#x25BC</a>
     </div>
 
 

main/_modules/torch/ao/nn/intrinsic/quantized/modules/bn_relu.html

@@ -280,7 +280,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+git75fa100 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+gitfb696bf ) &#x25BC</a>
     </div>
 
 

main/_modules/torch/ao/nn/intrinsic/quantized/modules/conv_relu.html

@@ -280,7 +280,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+git75fa100 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+gitfb696bf ) &#x25BC</a>
     </div>
 
 

main/_modules/torch/ao/nn/intrinsic/quantized/modules/linear_relu.html

@@ -280,7 +280,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+git75fa100 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+gitfb696bf ) &#x25BC</a>
     </div>
 
 

main/_modules/torch/ao/nn/qat/dynamic/modules/linear.html

@@ -280,7 +280,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+git75fa100 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+gitfb696bf ) &#x25BC</a>
     </div>
 
 

main/_modules/torch/ao/nn/qat/modules/conv.html

@@ -280,7 +280,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+git75fa100 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+gitfb696bf ) &#x25BC</a>
     </div>
 
 

main/_modules/torch/ao/nn/qat/modules/linear.html

@@ -280,7 +280,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+git75fa100 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+gitfb696bf ) &#x25BC</a>
     </div>
 
 

main/_modules/torch/ao/nn/quantizable/modules/activation.html

@@ -280,7 +280,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+git75fa100 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+gitfb696bf ) &#x25BC</a>
     </div>
 
 

main/_modules/torch/ao/nn/quantizable/modules/rnn.html

@@ -280,7 +280,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+git75fa100 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+gitfb696bf ) &#x25BC</a>
     </div>
 
 

main/_modules/torch/ao/nn/quantized/dynamic/modules/linear.html

@@ -280,7 +280,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+git75fa100 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+gitfb696bf ) &#x25BC</a>
     </div>
 
 

main/_modules/torch/ao/nn/quantized/dynamic/modules/rnn.html

@@ -280,7 +280,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+git75fa100 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+gitfb696bf ) &#x25BC</a>
     </div>
 
 

main/_modules/torch/ao/nn/quantized/functional.html

@@ -280,7 +280,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+git75fa100 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+gitfb696bf ) &#x25BC</a>
     </div>
 
 

main/_modules/torch/ao/nn/quantized/modules/activation.html

@@ -280,7 +280,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+git75fa100 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+gitfb696bf ) &#x25BC</a>
     </div>
 
 

main/_modules/torch/ao/nn/quantized/modules/batchnorm.html

@@ -280,7 +280,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+git75fa100 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+gitfb696bf ) &#x25BC</a>
     </div>
 
 

main/_modules/torch/ao/nn/quantized/modules/conv.html

@@ -280,7 +280,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+git75fa100 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+gitfb696bf ) &#x25BC</a>
     </div>
 
 

main/_modules/torch/ao/nn/quantized/modules/embedding_ops.html

@@ -280,7 +280,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+git75fa100 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+gitfb696bf ) &#x25BC</a>
     </div>
 
 

main/_modules/torch/ao/nn/quantized/modules/functional_modules.html

@@ -280,7 +280,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+git75fa100 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+gitfb696bf ) &#x25BC</a>
     </div>
 
 

main/_modules/torch/ao/nn/quantized/modules/linear.html

@@ -280,7 +280,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+git75fa100 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.5.0a0+gitfb696bf ) &#x25BC</a>
     </div>