Update docs

Author: LeiWang1999

api/tilelang.autotuner.html

@@ -815,6 +815,12 @@ <h1>tilelang.autotuner package<a class="headerlink" href="#tilelang-autotuner-pa
 </dl>
 </dd></dl>
 
+<dl class="py function">
+<dt class="sig sig-object py" id="tilelang.autotuner.get_available_cpu_count">
+<span class="sig-prename descclassname"><span class="pre">tilelang.autotuner.</span></span><span class="sig-name descname"><span class="pre">get_available_cpu_count</span></span><span class="sig-paren">(</span><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.autotuner.get_available_cpu_count" title="Permalink to this definition">#</a></dt>
+<dd><p>Gets the number of CPU cores available to the current process.</p>
+</dd></dl>
+
 <dl class="py function">
 <dt class="sig sig-object py" id="tilelang.autotuner.jit">
 <span class="sig-prename descclassname"><span class="pre">tilelang.autotuner.</span></span><span class="sig-name descname"><span class="pre">jit</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">out_idx</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">List</span><span class="p"><span class="pre">[</span></span><span class="pre">int</span><span class="p"><span class="pre">]</span></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>, <em class="sig-param"><span class="n"><span class="pre">supply_type</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><a class="reference internal" href="tilelang.utils.tensor.html#tilelang.utils.tensor.TensorSupplyType" title="tilelang.utils.tensor.TensorSupplyType"><span class="pre">TensorSupplyType</span></a></span><span class="w"> </span><span class="o"><span class="pre">=</span></span><span class="w"> </span><span class="default_value"><span class="pre">TensorSupplyType.Auto</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">ref_prog</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>, <em class="sig-param"><span class="n"><span class="pre">supply_prog</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>, <em class="sig-param"><span class="n"><span class="pre">rtol</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">float</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">0.01</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">atol</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">float</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">0.01</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">max_mismatched_ratio</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">float</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">0.01</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">skip_check</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">bool</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">False</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">cache_input_tensors</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">bool</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">True</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">target</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">Literal</span><span class="p"><span class="pre">[</span></span><span class="s"><span class="pre">'auto'</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="s"><span class="pre">'cuda'</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="s"><span class="pre">'hip'</span></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">'auto'</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">Callable</span></span></span><a class="headerlink" href="#tilelang.autotuner.jit" title="Permalink to this definition">#</a></dt>
@@ -951,6 +957,7 @@ <h1>tilelang.autotuner package<a class="headerlink" href="#tilelang-autotuner-pa
 </li>
 <li><a class="reference internal" href="#tilelang.autotuner.autotune"><code class="docutils literal notranslate"><span class="pre">autotune()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.autotuner.check_tensor_list_compatibility"><code class="docutils literal notranslate"><span class="pre">check_tensor_list_compatibility()</span></code></a></li>
+<li><a class="reference internal" href="#tilelang.autotuner.get_available_cpu_count"><code class="docutils literal notranslate"><span class="pre">get_available_cpu_count()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.autotuner.jit"><code class="docutils literal notranslate"><span class="pre">jit()</span></code></a></li>
 </ul>
 </li>

api/tilelang.html

@@ -453,6 +453,7 @@ <h2>Subpackages<a class="headerlink" href="#subpackages" title="Permalink to thi
 </li>
 <li class="toctree-l3"><a class="reference internal" href="tilelang.autotuner.html#tilelang.autotuner.autotune"><code class="docutils literal notranslate"><span class="pre">autotune()</span></code></a></li>
 <li class="toctree-l3"><a class="reference internal" href="tilelang.autotuner.html#tilelang.autotuner.check_tensor_list_compatibility"><code class="docutils literal notranslate"><span class="pre">check_tensor_list_compatibility()</span></code></a></li>
+<li class="toctree-l3"><a class="reference internal" href="tilelang.autotuner.html#tilelang.autotuner.get_available_cpu_count"><code class="docutils literal notranslate"><span class="pre">get_available_cpu_count()</span></code></a></li>
 <li class="toctree-l3"><a class="reference internal" href="tilelang.autotuner.html#tilelang.autotuner.jit"><code class="docutils literal notranslate"><span class="pre">jit()</span></code></a></li>
 </ul>
 </li>

api/tilelang.language.proxy.html

@@ -450,7 +450,7 @@
 <p>Buffer proxy class for constructing tir buffer.</p>
 <dl class="py method">
 <dt class="sig sig-object py" id="tilelang.language.proxy.BufferProxy.from_ptr">
-<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">0x7f108884da50&gt;</span></span></span><a class="headerlink" href="#tilelang.language.proxy.BufferProxy.from_ptr" title="Permalink to this definition">#</a></dt>
+<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">0x7f0913b512d0&gt;</span></span></span><a class="headerlink" href="#tilelang.language.proxy.BufferProxy.from_ptr" title="Permalink to this definition">#</a></dt>
 <dd><p>Create a buffer from a pointer, shape, and data type.</p>
 <dl class="field-list simple">
 <dt class="field-odd">Parameters<span class="colon">:</span></dt>

genindex.html

@@ -1128,6 +1128,8 @@ <h2>G</h2>
         <li><a href="api/tilelang.jit.adapter.utils.html#tilelang.jit.adapter.utils.get_annotated_mod">get_annotated_mod() (in module tilelang.jit.adapter.utils)</a>
 </li>
         <li><a href="api/tilelang.carver.arch.html#tilelang.carver.arch.get_arch">get_arch() (in module tilelang.carver.arch)</a>
+</li>
+        <li><a href="api/tilelang.autotuner.html#tilelang.autotuner.get_available_cpu_count">get_available_cpu_count() (in module tilelang.autotuner)</a>
 </li>
         <li><a href="api/tilelang.carver.arch.arch_base.html#tilelang.carver.arch.arch_base.TileDevice.get_avaliable_tensorintrin_shapes">get_avaliable_tensorintrin_shapes() (tilelang.carver.arch.arch_base.TileDevice method)</a>