Generate Python docs from pytorch/pytorch@9629835

pytorch · Sep 25, 2024 · 663970b · 663970b
1 parent 4e70f48
commit 663970b
Show file tree

Hide file tree

Showing 41 changed files with 49 additions and 40 deletions.
diff --git a/nightly/_modules/index.html b/nightly/_modules/index.html
@@ -265,7 +265,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/_modules/torch/_functorch/aot_autograd.html b/nightly/_modules/torch/_functorch/aot_autograd.html
@@ -265,7 +265,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 
@@ -408,6 +408,7 @@ <h1>Source code for torch._functorch.aot_autograd</h1><div class="highlight"><pr
 <span class="kn">from</span> <span class="nn">torch.fx.experimental.proxy_tensor</span> <span class="kn">import</span> <span class="n">make_fx</span>
 <span class="kn">from</span> <span class="nn">torch.fx.experimental.symbolic_shapes</span> <span class="kn">import</span> <span class="n">ShapeEnv</span>
 <span class="kn">from</span> <span class="nn">torch.utils._python_dispatch</span> <span class="kn">import</span> <span class="n">is_traceable_wrapper_subclass</span>
+<span class="kn">from</span> <span class="nn">torch.utils.weak</span> <span class="kn">import</span> <span class="n">TensorWeakRef</span>
 
 
 <span class="n">static_inputs_log</span> <span class="o">=</span> <span class="n">torch</span><span class="o">.</span><span class="n">_logging</span><span class="o">.</span><span class="n">getArtifactLogger</span><span class="p">(</span>
@@ -485,6 +486,7 @@ <h1>Source code for torch._functorch.aot_autograd</h1><div class="highlight"><pr
     <span class="n">create_metadata_for_subclass</span><span class="p">,</span>
     <span class="n">requires_subclass_dispatch</span><span class="p">,</span>
     <span class="n">unwrap_tensor_subclasses</span><span class="p">,</span>
+    <span class="n">unwrap_tensor_subclasses_with_indices_to_original</span><span class="p">,</span>
     <span class="n">wrap_tensor_subclasses</span><span class="p">,</span>
     <span class="n">wrap_tensor_subclasses_maybe_joint</span><span class="p">,</span>
 <span class="p">)</span>
@@ -1364,6 +1366,13 @@ <h1>Source code for torch._functorch.aot_autograd</h1><div class="highlight"><pr
 
     <span class="k">if</span> <span class="n">tracing_context</span> <span class="o">:=</span> <span class="n">torch</span><span class="o">.</span><span class="n">_guards</span><span class="o">.</span><span class="n">TracingContext</span><span class="o">.</span><span class="n">try_get</span><span class="p">():</span>
         <span class="n">tracing_context</span><span class="o">.</span><span class="n">params_flat</span> <span class="o">=</span> <span class="n">params_flat</span>
+        <span class="p">(</span>
+            <span class="n">params_flat_unwrap_subclasses</span><span class="p">,</span>
+            <span class="n">tracing_context</span><span class="o">.</span><span class="n">params_unwrapped_to_flat_index</span><span class="p">,</span>
+        <span class="p">)</span> <span class="o">=</span> <span class="n">unwrap_tensor_subclasses_with_indices_to_original</span><span class="p">(</span><span class="n">params_flat</span><span class="p">)</span>
+        <span class="n">tracing_context</span><span class="o">.</span><span class="n">params_flat_unwrap_subclasses</span> <span class="o">=</span> <span class="p">[</span>
+            <span class="n">TensorWeakRef</span><span class="p">(</span><span class="n">p</span><span class="p">)</span> <span class="k">for</span> <span class="n">p</span> <span class="ow">in</span> <span class="n">params_flat_unwrap_subclasses</span>
+        <span class="p">]</span>
 
     <span class="n">aot_autograd_arg_pos_to_source</span> <span class="o">=</span> <span class="kc">None</span>
     <span class="c1"># Then, the params 1:1 mapped sources, if relevant.</span>

diff --git a/nightly/_modules/torch/_functorch/compilers.html b/nightly/_modules/torch/_functorch/compilers.html
@@ -265,7 +265,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/_modules/torch/_functorch/deprecated.html b/nightly/_modules/torch/_functorch/deprecated.html
@@ -265,7 +265,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/_modules/torch/_functorch/partitioners.html b/nightly/_modules/torch/_functorch/partitioners.html
@@ -265,7 +265,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/aot_autograd.html b/nightly/aot_autograd.html
@@ -267,7 +267,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/batch_norm.html b/nightly/batch_norm.html
@@ -267,7 +267,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/experimental.html b/nightly/experimental.html
@@ -267,7 +267,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/functorch.html b/nightly/functorch.html
@@ -267,7 +267,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/generated/functorch.combine_state_for_ensemble.html b/nightly/generated/functorch.combine_state_for_ensemble.html
@@ -267,7 +267,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/generated/functorch.compile.aot_function.html b/nightly/generated/functorch.compile.aot_function.html
@@ -267,7 +267,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/generated/functorch.compile.aot_module.html b/nightly/generated/functorch.compile.aot_module.html
@@ -267,7 +267,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/generated/functorch.compile.default_partition.html b/nightly/generated/functorch.compile.default_partition.html
@@ -267,7 +267,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/generated/functorch.compile.memory_efficient_fusion.html b/nightly/generated/functorch.compile.memory_efficient_fusion.html
@@ -267,7 +267,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/generated/functorch.compile.min_cut_rematerialization_partition.html b/nightly/generated/functorch.compile.min_cut_rematerialization_partition.html
@@ -267,7 +267,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/generated/functorch.compile.nop.html b/nightly/generated/functorch.compile.nop.html
@@ -267,7 +267,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/generated/functorch.compile.ts_compile.html b/nightly/generated/functorch.compile.ts_compile.html
@@ -267,7 +267,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/generated/functorch.functionalize.html b/nightly/generated/functorch.functionalize.html
@@ -267,7 +267,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/generated/functorch.grad.html b/nightly/generated/functorch.grad.html
@@ -267,7 +267,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/generated/functorch.grad_and_value.html b/nightly/generated/functorch.grad_and_value.html
@@ -267,7 +267,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/generated/functorch.hessian.html b/nightly/generated/functorch.hessian.html
@@ -267,7 +267,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/generated/functorch.jacfwd.html b/nightly/generated/functorch.jacfwd.html
@@ -267,7 +267,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/generated/functorch.jacrev.html b/nightly/generated/functorch.jacrev.html
@@ -267,7 +267,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/generated/functorch.jvp.html b/nightly/generated/functorch.jvp.html
@@ -267,7 +267,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/generated/functorch.make_functional.html b/nightly/generated/functorch.make_functional.html
@@ -267,7 +267,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/generated/functorch.make_functional_with_buffers.html b/nightly/generated/functorch.make_functional_with_buffers.html
@@ -267,7 +267,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/generated/functorch.vjp.html b/nightly/generated/functorch.vjp.html
@@ -267,7 +267,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/generated/functorch.vmap.html b/nightly/generated/functorch.vmap.html
@@ -267,7 +267,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/genindex.html b/nightly/genindex.html
@@ -265,7 +265,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/index.html b/nightly/index.html
@@ -266,7 +266,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/install.html b/nightly/install.html
@@ -267,7 +267,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/notebooks/aot_autograd_optimizations.html b/nightly/notebooks/aot_autograd_optimizations.html
@@ -267,7 +267,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/notebooks/ensembling.html b/nightly/notebooks/ensembling.html
@@ -267,7 +267,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/notebooks/jacobians_hessians.html b/nightly/notebooks/jacobians_hessians.html
@@ -267,7 +267,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/notebooks/minifier.html b/nightly/notebooks/minifier.html
@@ -266,7 +266,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/notebooks/neural_tangent_kernels.html b/nightly/notebooks/neural_tangent_kernels.html
@@ -267,7 +267,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/notebooks/per_sample_grads.html b/nightly/notebooks/per_sample_grads.html
@@ -267,7 +267,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/notebooks/whirlwind_tour.html b/nightly/notebooks/whirlwind_tour.html
@@ -267,7 +267,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/objects.inv b/nightly/objects.inv
diff --git a/nightly/search.html b/nightly/search.html
@@ -265,7 +265,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>
 
 

diff --git a/nightly/ux_limitations.html b/nightly/ux_limitations.html
@@ -267,7 +267,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+gite0f84f4) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git9629835) &#x25BC</a>
     </div>