Skip to content

Commit 6f2cf45

Browse files
Update docs
1 parent 16d1fa4 commit 6f2cf45

File tree

5 files changed

+43
-1
lines changed

5 files changed

+43
-1
lines changed

_sources/autoapi/tilelang/jit/adapter/cython/adapter/index.rst.txt

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,11 +25,25 @@ Classes
2525
tilelang.jit.adapter.cython.adapter.CythonKernelAdapter
2626

2727

28+
Functions
29+
---------
30+
31+
.. autoapisummary::
32+
33+
tilelang.jit.adapter.cython.adapter.is_symbolic_expr
34+
35+
2836
Module Contents
2937
---------------
3038

3139
.. py:data:: logger
3240
41+
.. py:function:: is_symbolic_expr(expr)
42+
43+
Check if the expression is a symbolic expression.
44+
A symbolic expression can be a simple tvm.Var, or an tvm.PrimExpr containing tvm.Var.
45+
46+
3347
.. py:class:: CythonKernelAdapter(params, result_idx, target, func_or_mod, host_mod = None, device_mod = None, kernel_global_source = None, verbose = False, pass_configs = None, compile_flags = None)
3448
3549
Bases: :py:obj:`tilelang.jit.adapter.base.BaseKernelAdapter`

autoapi/tilelang/jit/adapter/cython/adapter/index.html

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -503,13 +503,37 @@ <h2>Classes<a class="headerlink" href="#classes" title="Link to this heading">¶
503503
</table>
504504
</div>
505505
</section>
506+
<section id="functions">
507+
<h2>Functions<a class="headerlink" href="#functions" title="Link to this heading"></a></h2>
508+
<div class="table-wrapper autosummary longtable docutils container">
509+
<table class="autosummary longtable docutils align-default">
510+
<tbody>
511+
<tr class="row-odd"><td><p><a class="reference internal" href="#tilelang.jit.adapter.cython.adapter.is_symbolic_expr" title="tilelang.jit.adapter.cython.adapter.is_symbolic_expr"><code class="xref py py-obj docutils literal notranslate"><span class="pre">is_symbolic_expr</span></code></a>(expr)</p></td>
512+
<td><p>Check if the expression is a symbolic expression.</p></td>
513+
</tr>
514+
</tbody>
515+
</table>
516+
</div>
517+
</section>
506518
<section id="module-contents">
507519
<h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to this heading"></a></h2>
508520
<dl class="py data">
509521
<dt class="sig sig-object py" id="tilelang.jit.adapter.cython.adapter.logger">
510522
<span class="sig-prename descclassname"><span class="pre">tilelang.jit.adapter.cython.adapter.</span></span><span class="sig-name descname"><span class="pre">logger</span></span><a class="headerlink" href="#tilelang.jit.adapter.cython.adapter.logger" title="Link to this definition"></a></dt>
511523
<dd></dd></dl>
512524

525+
<dl class="py function">
526+
<dt class="sig sig-object py" id="tilelang.jit.adapter.cython.adapter.is_symbolic_expr">
527+
<span class="sig-prename descclassname"><span class="pre">tilelang.jit.adapter.cython.adapter.</span></span><span class="sig-name descname"><span class="pre">is_symbolic_expr</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">expr</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.jit.adapter.cython.adapter.is_symbolic_expr" title="Link to this definition"></a></dt>
528+
<dd><p>Check if the expression is a symbolic expression.
529+
A symbolic expression can be a simple tvm.Var, or an tvm.PrimExpr containing tvm.Var.</p>
530+
<dl class="field-list simple">
531+
<dt class="field-odd">Return type<span class="colon">:</span></dt>
532+
<dd class="field-odd"><p>bool</p>
533+
</dd>
534+
</dl>
535+
</dd></dl>
536+
513537
<dl class="py class">
514538
<dt class="sig sig-object py" id="tilelang.jit.adapter.cython.adapter.CythonKernelAdapter">
515539
<em class="property"><span class="pre">class</span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">tilelang.jit.adapter.cython.adapter.</span></span><span class="sig-name descname"><span class="pre">CythonKernelAdapter</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">params</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">result_idx</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">target</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">func_or_mod</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">host_mod</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">device_mod</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">kernel_global_source</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">verbose</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">False</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">pass_configs</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">compile_flags</span></span><span class="o"><span class="pre">=</span></span><span class="default_value"><span class="pre">None</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.jit.adapter.cython.adapter.CythonKernelAdapter" title="Link to this definition"></a></dt>
@@ -759,8 +783,10 @@ <h2>Module Contents<a class="headerlink" href="#module-contents" title="Link to
759783
<li><a class="reference internal" href="#">tilelang.jit.adapter.cython.adapter</a><ul>
760784
<li><a class="reference internal" href="#attributes">Attributes</a></li>
761785
<li><a class="reference internal" href="#classes">Classes</a></li>
786+
<li><a class="reference internal" href="#functions">Functions</a></li>
762787
<li><a class="reference internal" href="#module-contents">Module Contents</a><ul>
763788
<li><a class="reference internal" href="#tilelang.jit.adapter.cython.adapter.logger"><code class="docutils literal notranslate"><span class="pre">logger</span></code></a></li>
789+
<li><a class="reference internal" href="#tilelang.jit.adapter.cython.adapter.is_symbolic_expr"><code class="docutils literal notranslate"><span class="pre">is_symbolic_expr()</span></code></a></li>
764790
<li><a class="reference internal" href="#tilelang.jit.adapter.cython.adapter.CythonKernelAdapter"><code class="docutils literal notranslate"><span class="pre">CythonKernelAdapter</span></code></a><ul>
765791
<li><a class="reference internal" href="#tilelang.jit.adapter.cython.adapter.CythonKernelAdapter.target"><code class="docutils literal notranslate"><span class="pre">CythonKernelAdapter.target</span></code></a></li>
766792
<li><a class="reference internal" href="#tilelang.jit.adapter.cython.adapter.CythonKernelAdapter.ir_module"><code class="docutils literal notranslate"><span class="pre">CythonKernelAdapter.ir_module</span></code></a></li>

genindex.html

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2754,6 +2754,8 @@ <h2>I</h2>
27542754
<li><a href="autoapi/tilelang/intrinsics/wgmma_macro_generator/index.html#tilelang.intrinsics.wgmma_macro_generator.SwizzleMode.is_swizzle_32b">is_swizzle_32b() (tilelang.intrinsics.wgmma_macro_generator.SwizzleMode method)</a>
27552755
</li>
27562756
<li><a href="autoapi/tilelang/intrinsics/wgmma_macro_generator/index.html#tilelang.intrinsics.wgmma_macro_generator.SwizzleMode.is_swizzle_64b">is_swizzle_64b() (tilelang.intrinsics.wgmma_macro_generator.SwizzleMode method)</a>
2757+
</li>
2758+
<li><a href="autoapi/tilelang/jit/adapter/cython/adapter/index.html#tilelang.jit.adapter.cython.adapter.is_symbolic_expr">is_symbolic_expr() (in module tilelang.jit.adapter.cython.adapter)</a>
27572759
</li>
27582760
<li><a href="autoapi/tilelang/carver/arch/cuda/index.html#tilelang.carver.arch.cuda.is_tensorcore_supported_precision">is_tensorcore_supported_precision() (in module tilelang.carver.arch.cuda)</a>
27592761
</li>

objects.inv

12 Bytes
Binary file not shown.

searchindex.js

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)