Generate Python docs from pytorch/pytorch@6241006

pytorch · Oct 3, 2024 · 36aacfe · 36aacfe
1 parent adcca8a
commit 36aacfe
Show file tree

Hide file tree

Showing 41 changed files with 48 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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#x25BC</a>
     </div>
 
 

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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#x25BC</a>
     </div>
 
 
@@ -1249,6 +1249,14 @@ <h1>Source code for torch._functorch.partitioners</h1><div class="highlight"><pr
             <span class="k">return</span> <span class="kc">True</span>
         <span class="k">if</span> <span class="n">can_fuse_into_triton_kernel_wrapper_functional</span><span class="p">(</span><span class="n">a</span><span class="p">,</span> <span class="n">b</span><span class="p">):</span>
             <span class="k">return</span> <span class="kc">True</span>
+        <span class="k">if</span> <span class="p">(</span>
+            <span class="n">a</span><span class="o">.</span><span class="n">target</span> <span class="ow">is</span> <span class="n">operator</span><span class="o">.</span><span class="n">getitem</span>
+            <span class="ow">and</span> <span class="n">a</span><span class="o">.</span><span class="n">args</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span><span class="o">.</span><span class="n">target</span>
+            <span class="ow">is</span> <span class="n">torch</span><span class="o">.</span><span class="n">ops</span><span class="o">.</span><span class="n">higher_order</span><span class="o">.</span><span class="n">triton_kernel_wrapper_functional</span>
+        <span class="p">):</span>
+            <span class="c1"># if a is the output of a user triton kernel,</span>
+            <span class="c1"># then (by default) we will not be able to fuse b into it</span>
+            <span class="k">return</span> <span class="kc">False</span>
         <span class="k">return</span> <span class="n">op_types</span><span class="o">.</span><span class="n">is_fusible</span><span class="p">(</span><span class="n">a</span><span class="p">)</span> <span class="ow">and</span> <span class="n">op_types</span><span class="o">.</span><span class="n">is_fusible</span><span class="p">(</span><span class="n">b</span><span class="p">)</span>
 
     <span class="k">try</span><span class="p">:</span>

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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#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+gitd725758) &#x25BC</a>
+      <a href='https://pytorch.org/functorch/versions.html'>nightly (2.6.0a0+git6241006) &#x25BC</a>
     </div>