Skip to content

Commit f63d7e8

Browse files
committed
Update docs
1 parent fc553f5 commit f63d7e8

8 files changed

+20
-11
lines changed

api/tilelang.jit.html

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -488,6 +488,7 @@ <h2>Submodules<a class="headerlink" href="#submodules" title="Permalink to this
488488
<li class="toctree-l3"><a class="reference internal" href="tilelang.jit.kernel.html#tilelang.jit.kernel.JITKernel.kernel_source"><code class="docutils literal notranslate"><span class="pre">JITKernel.kernel_source</span></code></a></li>
489489
<li class="toctree-l3"><a class="reference internal" href="tilelang.jit.kernel.html#tilelang.jit.kernel.JITKernel.out_idx"><code class="docutils literal notranslate"><span class="pre">JITKernel.out_idx</span></code></a></li>
490490
<li class="toctree-l3"><a class="reference internal" href="tilelang.jit.kernel.html#tilelang.jit.kernel.JITKernel.params"><code class="docutils literal notranslate"><span class="pre">JITKernel.params</span></code></a></li>
491+
<li class="toctree-l3"><a class="reference internal" href="tilelang.jit.kernel.html#tilelang.jit.kernel.JITKernel.prim_func"><code class="docutils literal notranslate"><span class="pre">JITKernel.prim_func</span></code></a></li>
491492
<li class="toctree-l3"><a class="reference internal" href="tilelang.jit.kernel.html#tilelang.jit.kernel.JITKernel.run_once"><code class="docutils literal notranslate"><span class="pre">JITKernel.run_once()</span></code></a></li>
492493
<li class="toctree-l3"><a class="reference internal" href="tilelang.jit.kernel.html#id2"><code class="docutils literal notranslate"><span class="pre">JITKernel.torch_function</span></code></a></li>
493494
</ul>

api/tilelang.jit.kernel.html

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -551,6 +551,11 @@
551551
<em class="property"><span class="pre">property</span><span class="w"> </span></em><span class="sig-name descname"><span class="pre">params</span></span><em class="property"><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="pre">List</span><span class="p"><span class="pre">[</span></span><a class="reference internal" href="tilelang.engine.param.html#tilelang.engine.param.KernelParam" title="tilelang.engine.param.KernelParam"><span class="pre">KernelParam</span></a><span class="p"><span class="pre">]</span></span></em><a class="headerlink" href="#tilelang.jit.kernel.JITKernel.params" title="Permalink to this definition">#</a></dt>
552552
<dd></dd></dl>
553553

554+
<dl class="py attribute">
555+
<dt class="sig sig-object py" id="tilelang.jit.kernel.JITKernel.prim_func">
556+
<span class="sig-name descname"><span class="pre">prim_func</span></span><em class="property"><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="pre">PrimFunc</span></em><em class="property"><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="pre">None</span></em><a class="headerlink" href="#tilelang.jit.kernel.JITKernel.prim_func" title="Permalink to this definition">#</a></dt>
557+
<dd></dd></dl>
558+
554559
<dl class="py method">
555560
<dt class="sig sig-object py" id="tilelang.jit.kernel.JITKernel.run_once">
556561
<span class="sig-name descname"><span class="pre">run_once</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">func</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">Optional</span><span class="p"><span class="pre">[</span></span><span class="pre">Callable</span><span class="p"><span class="pre">]</span></span></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">None</span></span></span><a class="headerlink" href="#tilelang.jit.kernel.JITKernel.run_once" title="Permalink to this definition">#</a></dt>
@@ -637,6 +642,7 @@
637642
<li><a class="reference internal" href="#tilelang.jit.kernel.JITKernel.kernel_source"><code class="docutils literal notranslate"><span class="pre">JITKernel.kernel_source</span></code></a></li>
638643
<li><a class="reference internal" href="#tilelang.jit.kernel.JITKernel.out_idx"><code class="docutils literal notranslate"><span class="pre">JITKernel.out_idx</span></code></a></li>
639644
<li><a class="reference internal" href="#tilelang.jit.kernel.JITKernel.params"><code class="docutils literal notranslate"><span class="pre">JITKernel.params</span></code></a></li>
645+
<li><a class="reference internal" href="#tilelang.jit.kernel.JITKernel.prim_func"><code class="docutils literal notranslate"><span class="pre">JITKernel.prim_func</span></code></a></li>
640646
<li><a class="reference internal" href="#tilelang.jit.kernel.JITKernel.run_once"><code class="docutils literal notranslate"><span class="pre">JITKernel.run_once()</span></code></a></li>
641647
<li><a class="reference internal" href="#id2"><code class="docutils literal notranslate"><span class="pre">JITKernel.torch_function</span></code></a></li>
642648
</ul>

api/tilelang.language.logical.html

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -412,7 +412,7 @@
412412
<p>The language interface for tl programs.</p>
413413
<dl class="py function">
414414
<dt class="sig sig-object py" id="tilelang.language.logical.all_of">
415-
<span class="sig-prename descclassname"><span class="pre">tilelang.language.logical.</span></span><span class="sig-name descname"><span class="pre">all_of</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="pre">buffer:</span> <span class="pre">~typing.Union[&lt;tilelang.language.proxy.TensorProxy</span> <span class="pre">object</span> <span class="pre">at</span> <span class="pre">0x7f39cb09f340&gt;,</span> <span class="pre">~tvm.tir.stmt.BufferRegion]</span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.language.logical.all_of" title="Permalink to this definition">#</a></dt>
415+
<span class="sig-prename descclassname"><span class="pre">tilelang.language.logical.</span></span><span class="sig-name descname"><span class="pre">all_of</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="pre">buffer:</span> <span class="pre">~typing.Union[&lt;tilelang.language.proxy.TensorProxy</span> <span class="pre">object</span> <span class="pre">at</span> <span class="pre">0x7f7f006d6b00&gt;,</span> <span class="pre">~tvm.tir.stmt.BufferRegion]</span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.language.logical.all_of" title="Permalink to this definition">#</a></dt>
416416
<dd><p>Check if all elements in the buffer are true.</p>
417417
<dl class="field-list simple">
418418
<dt class="field-odd">Parameters<span class="colon">:</span></dt>
@@ -426,7 +426,7 @@
426426

427427
<dl class="py function">
428428
<dt class="sig sig-object py" id="tilelang.language.logical.any_of">
429-
<span class="sig-prename descclassname"><span class="pre">tilelang.language.logical.</span></span><span class="sig-name descname"><span class="pre">any_of</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="pre">buffer:</span> <span class="pre">~typing.Union[&lt;tilelang.language.proxy.TensorProxy</span> <span class="pre">object</span> <span class="pre">at</span> <span class="pre">0x7f39cb09f340&gt;,</span> <span class="pre">~tvm.tir.stmt.BufferRegion]</span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.language.logical.any_of" title="Permalink to this definition">#</a></dt>
429+
<span class="sig-prename descclassname"><span class="pre">tilelang.language.logical.</span></span><span class="sig-name descname"><span class="pre">any_of</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="pre">buffer:</span> <span class="pre">~typing.Union[&lt;tilelang.language.proxy.TensorProxy</span> <span class="pre">object</span> <span class="pre">at</span> <span class="pre">0x7f7f006d6b00&gt;,</span> <span class="pre">~tvm.tir.stmt.BufferRegion]</span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.language.logical.any_of" title="Permalink to this definition">#</a></dt>
430430
<dd><p>Check if any element in the buffer is true.</p>
431431
<dl class="field-list simple">
432432
<dt class="field-odd">Parameters<span class="colon">:</span></dt>

api/tilelang.language.proxy.html

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -460,7 +460,7 @@
460460
<p>Buffer proxy class for constructing tir buffer.</p>
461461
<dl class="py method">
462462
<dt class="sig sig-object py" id="tilelang.language.proxy.BufferProxy.from_ptr">
463-
<span class="sig-name descname"><span class="pre">from_ptr</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">pointer_var</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">Var</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">shape</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">tuple</span><span class="p"><span class="pre">[</span></span><span class="pre">tvm.ir.expr.PrimExpr</span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="p"><span class="pre">...</span></span><span class="p"><span class="pre">]</span></span></span></em>, <em class="sig-param"><span class="n"><span class="pre">dtype</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">str</span></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">'float32'</span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">&lt;tilelang.language.proxy.BufferProxy</span> <span class="pre">object</span> <span class="pre">at</span> <span class="pre">0x7f39cb09d540&gt;</span></span></span><a class="headerlink" href="#tilelang.language.proxy.BufferProxy.from_ptr" title="Permalink to this definition">#</a></dt>
463+
<span class="sig-name descname"><span class="pre">from_ptr</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">pointer_var</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">Var</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">shape</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">tuple</span><span class="p"><span class="pre">[</span></span><span class="pre">tvm.ir.expr.PrimExpr</span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="p"><span class="pre">...</span></span><span class="p"><span class="pre">]</span></span></span></em>, <em class="sig-param"><span class="n"><span class="pre">dtype</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">str</span></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">'float32'</span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">&lt;tilelang.language.proxy.BufferProxy</span> <span class="pre">object</span> <span class="pre">at</span> <span class="pre">0x7f7f006d6ad0&gt;</span></span></span><a class="headerlink" href="#tilelang.language.proxy.BufferProxy.from_ptr" title="Permalink to this definition">#</a></dt>
464464
<dd><p>Create a buffer from a pointer, shape, and data type.</p>
465465
<dl class="field-list simple">
466466
<dt class="field-odd">Parameters<span class="colon">:</span></dt>

0 commit comments

Comments
 (0)