Skip to content

Commit 1e63652

Browse files
committed
Update docs
1 parent aef85a3 commit 1e63652

8 files changed

+17
-6
lines changed

api/tilelang.contrib.html

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -430,6 +430,7 @@ <h2>Submodules<a class="headerlink" href="#submodules" title="Permalink to this
430430
<li class="toctree-l2"><a class="reference internal" href="tilelang.contrib.nvcc.html#tilelang.contrib.nvcc.compile_cuda"><code class="docutils literal notranslate"><span class="pre">compile_cuda()</span></code></a></li>
431431
<li class="toctree-l2"><a class="reference internal" href="tilelang.contrib.nvcc.html#tilelang.contrib.nvcc.find_cuda_path"><code class="docutils literal notranslate"><span class="pre">find_cuda_path()</span></code></a></li>
432432
<li class="toctree-l2"><a class="reference internal" href="tilelang.contrib.nvcc.html#tilelang.contrib.nvcc.get_cuda_version"><code class="docutils literal notranslate"><span class="pre">get_cuda_version()</span></code></a></li>
433+
<li class="toctree-l2"><a class="reference internal" href="tilelang.contrib.nvcc.html#tilelang.contrib.nvcc.get_nvcc_compiler"><code class="docutils literal notranslate"><span class="pre">get_nvcc_compiler()</span></code></a></li>
433434
<li class="toctree-l2"><a class="reference internal" href="tilelang.contrib.nvcc.html#tilelang.contrib.nvcc.have_cudagraph"><code class="docutils literal notranslate"><span class="pre">have_cudagraph()</span></code></a></li>
434435
<li class="toctree-l2"><a class="reference internal" href="tilelang.contrib.nvcc.html#tilelang.contrib.nvcc.have_fp16"><code class="docutils literal notranslate"><span class="pre">have_fp16()</span></code></a></li>
435436
<li class="toctree-l2"><a class="reference internal" href="tilelang.contrib.nvcc.html#tilelang.contrib.nvcc.have_int8"><code class="docutils literal notranslate"><span class="pre">have_int8()</span></code></a></li>

api/tilelang.contrib.nvcc.html

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -462,6 +462,12 @@
462462
</dl>
463463
</dd></dl>
464464

465+
<dl class="py function">
466+
<dt class="sig sig-object py" id="tilelang.contrib.nvcc.get_nvcc_compiler">
467+
<span class="sig-prename descclassname"><span class="pre">tilelang.contrib.nvcc.</span></span><span class="sig-name descname"><span class="pre">get_nvcc_compiler</span></span><span class="sig-paren">(</span><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">str</span></span></span><a class="headerlink" href="#tilelang.contrib.nvcc.get_nvcc_compiler" title="Permalink to this definition">#</a></dt>
468+
<dd><p>Get the path to the nvcc compiler</p>
469+
</dd></dl>
470+
465471
<dl class="py function">
466472
<dt class="sig sig-object py" id="tilelang.contrib.nvcc.have_cudagraph">
467473
<span class="sig-prename descclassname"><span class="pre">tilelang.contrib.nvcc.</span></span><span class="sig-name descname"><span class="pre">have_cudagraph</span></span><span class="sig-paren">(</span><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.contrib.nvcc.have_cudagraph" title="Permalink to this definition">#</a></dt>
@@ -585,6 +591,7 @@
585591
<li><a class="reference internal" href="#tilelang.contrib.nvcc.compile_cuda"><code class="docutils literal notranslate"><span class="pre">compile_cuda()</span></code></a></li>
586592
<li><a class="reference internal" href="#tilelang.contrib.nvcc.find_cuda_path"><code class="docutils literal notranslate"><span class="pre">find_cuda_path()</span></code></a></li>
587593
<li><a class="reference internal" href="#tilelang.contrib.nvcc.get_cuda_version"><code class="docutils literal notranslate"><span class="pre">get_cuda_version()</span></code></a></li>
594+
<li><a class="reference internal" href="#tilelang.contrib.nvcc.get_nvcc_compiler"><code class="docutils literal notranslate"><span class="pre">get_nvcc_compiler()</span></code></a></li>
588595
<li><a class="reference internal" href="#tilelang.contrib.nvcc.have_cudagraph"><code class="docutils literal notranslate"><span class="pre">have_cudagraph()</span></code></a></li>
589596
<li><a class="reference internal" href="#tilelang.contrib.nvcc.have_fp16"><code class="docutils literal notranslate"><span class="pre">have_fp16()</span></code></a></li>
590597
<li><a class="reference internal" href="#tilelang.contrib.nvcc.have_int8"><code class="docutils literal notranslate"><span class="pre">have_int8()</span></code></a></li>

api/tilelang.html

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -626,6 +626,7 @@ <h2>Subpackages<a class="headerlink" href="#subpackages" title="Permalink to thi
626626
<li class="toctree-l4"><a class="reference internal" href="tilelang.contrib.nvcc.html#tilelang.contrib.nvcc.compile_cuda"><code class="docutils literal notranslate"><span class="pre">compile_cuda()</span></code></a></li>
627627
<li class="toctree-l4"><a class="reference internal" href="tilelang.contrib.nvcc.html#tilelang.contrib.nvcc.find_cuda_path"><code class="docutils literal notranslate"><span class="pre">find_cuda_path()</span></code></a></li>
628628
<li class="toctree-l4"><a class="reference internal" href="tilelang.contrib.nvcc.html#tilelang.contrib.nvcc.get_cuda_version"><code class="docutils literal notranslate"><span class="pre">get_cuda_version()</span></code></a></li>
629+
<li class="toctree-l4"><a class="reference internal" href="tilelang.contrib.nvcc.html#tilelang.contrib.nvcc.get_nvcc_compiler"><code class="docutils literal notranslate"><span class="pre">get_nvcc_compiler()</span></code></a></li>
629630
<li class="toctree-l4"><a class="reference internal" href="tilelang.contrib.nvcc.html#tilelang.contrib.nvcc.have_cudagraph"><code class="docutils literal notranslate"><span class="pre">have_cudagraph()</span></code></a></li>
630631
<li class="toctree-l4"><a class="reference internal" href="tilelang.contrib.nvcc.html#tilelang.contrib.nvcc.have_fp16"><code class="docutils literal notranslate"><span class="pre">have_fp16()</span></code></a></li>
631632
<li class="toctree-l4"><a class="reference internal" href="tilelang.contrib.nvcc.html#tilelang.contrib.nvcc.have_int8"><code class="docutils literal notranslate"><span class="pre">have_int8()</span></code></a></li>

api/tilelang.language.logical.html

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -404,7 +404,7 @@
404404
<p>The language interface for tl programs.</p>
405405
<dl class="py function">
406406
<dt class="sig sig-object py" id="tilelang.language.logical.all_of">
407-
<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">0x7fc6d5046f50&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>
407+
<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">0x7f8f9bfcb010&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>
408408
<dd><p>Check if all elements in the buffer are true.</p>
409409
<dl class="field-list simple">
410410
<dt class="field-odd">Parameters<span class="colon">:</span></dt>
@@ -418,7 +418,7 @@
418418

419419
<dl class="py function">
420420
<dt class="sig sig-object py" id="tilelang.language.logical.any_of">
421-
<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">0x7fc6d5046f50&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>
421+
<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">0x7f8f9bfcb010&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>
422422
<dd><p>Check if any element in the buffer is true.</p>
423423
<dl class="field-list simple">
424424
<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
@@ -452,7 +452,7 @@
452452
<p>Buffer proxy class for constructing tir buffer.</p>
453453
<dl class="py method">
454454
<dt class="sig sig-object py" id="tilelang.language.proxy.BufferProxy.from_ptr">
455-
<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">0x7fc6d5046830&gt;</span></span></span><a class="headerlink" href="#tilelang.language.proxy.BufferProxy.from_ptr" title="Permalink to this definition">#</a></dt>
455+
<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">0x7f8f9bfcb040&gt;</span></span></span><a class="headerlink" href="#tilelang.language.proxy.BufferProxy.from_ptr" title="Permalink to this definition">#</a></dt>
456456
<dd><p>Create a buffer from a pointer, shape, and data type.</p>
457457
<dl class="field-list simple">
458458
<dt class="field-odd">Parameters<span class="colon">:</span></dt>

genindex.html

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1261,14 +1261,14 @@ <h2>G</h2>
12611261
</li>
12621262
<li><a href="api/tilelang.carver.matmul_analysis.html#tilelang.carver.matmul_analysis.get_index_map">get_index_map() (in module tilelang.carver.matmul_analysis)</a>
12631263
</li>
1264-
</ul></td>
1265-
<td style="width: 33%; vertical-align: top;"><ul>
12661264
<li><a href="api/tilelang.jit.adapter.wrapper.html#tilelang.jit.adapter.wrapper.TLCUDASourceWrapper.get_init_func">get_init_func() (tilelang.jit.adapter.wrapper.TLCUDASourceWrapper method)</a>
12671265

12681266
<ul>
12691267
<li><a href="api/tilelang.jit.adapter.wrapper.html#tilelang.jit.adapter.wrapper.TLHIPSourceWrapper.get_init_func">(tilelang.jit.adapter.wrapper.TLHIPSourceWrapper method)</a>
12701268
</li>
12711269
</ul></li>
1270+
</ul></td>
1271+
<td style="width: 33%; vertical-align: top;"><ul>
12721272
<li><a href="api/tilelang.carver.roller.node.html#tilelang.carver.roller.node.BlockAnalyzer.get_input_buffers">get_input_buffers() (tilelang.carver.roller.node.BlockAnalyzer method)</a>
12731273

12741274
<ul>
@@ -1326,6 +1326,8 @@ <h2>G</h2>
13261326
</li>
13271327
</ul></li>
13281328
<li><a href="api/tilelang.language.kernel.html#tilelang.language.kernel.KernelLaunchFrame.get_num_threads">get_num_threads() (tilelang.language.kernel.KernelLaunchFrame method)</a>
1329+
</li>
1330+
<li><a href="api/tilelang.contrib.nvcc.html#tilelang.contrib.nvcc.get_nvcc_compiler">get_nvcc_compiler() (in module tilelang.contrib.nvcc)</a>
13291331
</li>
13301332
<li><a href="api/tilelang.carver.roller.node.html#tilelang.carver.roller.node.PrimFuncNode.get_opt_shape">get_opt_shape() (tilelang.carver.roller.node.PrimFuncNode method)</a>
13311333
</li>

objects.inv

6 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)