Update docs

Author: LeiWang1999

api/tilelang.cache.html

@@ -409,8 +409,11 @@ <h2>Submodules<a class="headerlink" href="#submodules" title="Permalink to this
 <ul>
 <li class="toctree-l1"><a class="reference internal" href="tilelang.cache.kernel_cache.html">tilelang.cache.kernel_cache module</a><ul>
 <li class="toctree-l2"><a class="reference internal" href="tilelang.cache.kernel_cache.html#tilelang.cache.kernel_cache.KernelCache"><code class="docutils literal notranslate"><span class="pre">KernelCache</span></code></a><ul>
+<li class="toctree-l3"><a class="reference internal" href="tilelang.cache.kernel_cache.html#tilelang.cache.kernel_cache.KernelCache.cache_dir"><code class="docutils literal notranslate"><span class="pre">KernelCache.cache_dir</span></code></a></li>
 <li class="toctree-l3"><a class="reference internal" href="tilelang.cache.kernel_cache.html#tilelang.cache.kernel_cache.KernelCache.cached"><code class="docutils literal notranslate"><span class="pre">KernelCache.cached()</span></code></a></li>
 <li class="toctree-l3"><a class="reference internal" href="tilelang.cache.kernel_cache.html#tilelang.cache.kernel_cache.KernelCache.clear_cache"><code class="docutils literal notranslate"><span class="pre">KernelCache.clear_cache()</span></code></a></li>
+<li class="toctree-l3"><a class="reference internal" href="tilelang.cache.kernel_cache.html#tilelang.cache.kernel_cache.KernelCache.get_cache_dir"><code class="docutils literal notranslate"><span class="pre">KernelCache.get_cache_dir()</span></code></a></li>
+<li class="toctree-l3"><a class="reference internal" href="tilelang.cache.kernel_cache.html#tilelang.cache.kernel_cache.KernelCache.set_cache_dir"><code class="docutils literal notranslate"><span class="pre">KernelCache.set_cache_dir()</span></code></a></li>
 </ul>
 </li>
 </ul>
@@ -433,6 +436,27 @@ <h2>Submodules<a class="headerlink" href="#submodules" title="Permalink to this
 <dd><p>Clears the entire kernel cache (using KernelCache class).</p>
 </dd></dl>
 
+<dl class="py function">
+<dt class="sig sig-object py" id="tilelang.cache.get_cache_dir">
+<span class="sig-prename descclassname"><span class="pre">tilelang.cache.</span></span><span class="sig-name descname"><span class="pre">get_cache_dir</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">Path</span></span></span><a class="headerlink" href="#tilelang.cache.get_cache_dir" title="Permalink to this definition">#</a></dt>
+<dd><p>Gets the cache directory for the kernel cache.
+.. rubric:: Example</p>
+<div class="doctest highlight-default notranslate"><div class="highlight"><pre><span></span><span class="gp">&gt;&gt;&gt; </span><span class="n">tilelang</span><span class="o">.</span><span class="n">cache</span><span class="o">.</span><span class="n">get_cache_dir</span><span class="p">()</span>
+<span class="go">PosixPath(&#39;/Users/username/.tilelang/cache&#39;)</span>
+</pre></div>
+</div>
+</dd></dl>
+
+<dl class="py function">
+<dt class="sig sig-object py" id="tilelang.cache.set_cache_dir">
+<span class="sig-prename descclassname"><span class="pre">tilelang.cache.</span></span><span class="sig-name descname"><span class="pre">set_cache_dir</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">cache_dir</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">str</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.cache.set_cache_dir" title="Permalink to this definition">#</a></dt>
+<dd><p>Sets the cache directory for the kernel cache.
+.. rubric:: Example</p>
+<div class="doctest highlight-default notranslate"><div class="highlight"><pre><span></span><span class="gp">&gt;&gt;&gt; </span><span class="n">tilelang</span><span class="o">.</span><span class="n">cache</span><span class="o">.</span><span class="n">set_cache_dir</span><span class="p">(</span><span class="s2">&quot;/path/to/cache&quot;</span><span class="p">)</span>
+</pre></div>
+</div>
+</dd></dl>
+
 </section>
 </section>
 
@@ -496,6 +520,8 @@ <h2>Submodules<a class="headerlink" href="#submodules" title="Permalink to this
 <li><a class="reference internal" href="#module-tilelang.cache">Module contents</a><ul>
 <li><a class="reference internal" href="#tilelang.cache.cached"><code class="docutils literal notranslate"><span class="pre">cached()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.cache.clear_cache"><code class="docutils literal notranslate"><span class="pre">clear_cache()</span></code></a></li>
+<li><a class="reference internal" href="#tilelang.cache.get_cache_dir"><code class="docutils literal notranslate"><span class="pre">get_cache_dir()</span></code></a></li>
+<li><a class="reference internal" href="#tilelang.cache.set_cache_dir"><code class="docutils literal notranslate"><span class="pre">set_cache_dir()</span></code></a></li>
 </ul>
 </li>
 </ul>

api/tilelang.cache.kernel_cache.html

@@ -416,6 +416,11 @@
 kernel_lib.so: The compiled kernel library
 params.pkl: The compiled kernel parameters</p>
 </div></blockquote>
+<dl class="py attribute">
+<dt class="sig sig-object py" id="tilelang.cache.kernel_cache.KernelCache.cache_dir">
+<span class="sig-name descname"><span class="pre">cache_dir</span></span><em class="property"><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="pre">Path</span></em><em class="property"><span class="w"> </span><span class="p"><span class="pre">=</span></span><span class="w"> </span><span class="pre">PosixPath('/home/t-leiwang/.tilelang/cache')</span></em><a class="headerlink" href="#tilelang.cache.kernel_cache.KernelCache.cache_dir" title="Permalink to this definition">#</a></dt>
+<dd></dd></dl>
+
 <dl class="py method">
 <dt class="sig sig-object py" id="tilelang.cache.kernel_cache.KernelCache.cached">
 <span class="sig-name descname"><span class="pre">cached</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">PrimFunc</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">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="o"><span class="pre">*</span></span><span class="n"><span class="pre">args</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">Union</span><span class="p"><span class="pre">[</span></span><span class="pre">str</span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="pre">Target</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>, <em class="sig-param"><span class="n"><span class="pre">target_host</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">Union</span><span class="p"><span class="pre">[</span></span><span class="pre">str</span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="pre">Target</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">execution_backend</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">'dlpack'</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="s"><span class="pre">'ctypes'</span></span><span class="p"><span class="pre">,</span></span><span class="w"> </span><span class="s"><span class="pre">'cython'</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">'cython'</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">verbose</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">pass_configs</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">dict</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"><a class="reference internal" href="tilelang.jit.kernel.html#tilelang.jit.kernel.JITKernel" title="tilelang.jit.kernel.JITKernel"><span class="pre">JITKernel</span></a></span></span><a class="headerlink" href="#tilelang.cache.kernel_cache.KernelCache.cached" title="Permalink to this definition">#</a></dt>
@@ -445,6 +450,18 @@
 <dd><p>Clears the entire kernel cache, including both in-memory and disk cache.</p>
 </dd></dl>
 
+<dl class="py method">
+<dt class="sig sig-object py" id="tilelang.cache.kernel_cache.KernelCache.get_cache_dir">
+<span class="sig-name descname"><span class="pre">get_cache_dir</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">Path</span></span></span><a class="headerlink" href="#tilelang.cache.kernel_cache.KernelCache.get_cache_dir" title="Permalink to this definition">#</a></dt>
+<dd><p>Gets the cache directory for the kernel cache.</p>
+</dd></dl>
+
+<dl class="py method">
+<dt class="sig sig-object py" id="tilelang.cache.kernel_cache.KernelCache.set_cache_dir">
+<span class="sig-name descname"><span class="pre">set_cache_dir</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">cache_dir</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">str</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.cache.kernel_cache.KernelCache.set_cache_dir" title="Permalink to this definition">#</a></dt>
+<dd><p>Sets the cache directory for the kernel cache.</p>
+</dd></dl>
+
 </dd></dl>
 
 </section>
@@ -506,8 +523,11 @@
             <ul>
 <li><a class="reference internal" href="#">tilelang.cache.kernel_cache module</a><ul>
 <li><a class="reference internal" href="#tilelang.cache.kernel_cache.KernelCache"><code class="docutils literal notranslate"><span class="pre">KernelCache</span></code></a><ul>
+<li><a class="reference internal" href="#tilelang.cache.kernel_cache.KernelCache.cache_dir"><code class="docutils literal notranslate"><span class="pre">KernelCache.cache_dir</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.cache.kernel_cache.KernelCache.cached"><code class="docutils literal notranslate"><span class="pre">KernelCache.cached()</span></code></a></li>
 <li><a class="reference internal" href="#tilelang.cache.kernel_cache.KernelCache.clear_cache"><code class="docutils literal notranslate"><span class="pre">KernelCache.clear_cache()</span></code></a></li>
+<li><a class="reference internal" href="#tilelang.cache.kernel_cache.KernelCache.get_cache_dir"><code class="docutils literal notranslate"><span class="pre">KernelCache.get_cache_dir()</span></code></a></li>
+<li><a class="reference internal" href="#tilelang.cache.kernel_cache.KernelCache.set_cache_dir"><code class="docutils literal notranslate"><span class="pre">KernelCache.set_cache_dir()</span></code></a></li>
 </ul>
 </li>
 </ul>

api/tilelang.html

@@ -501,6 +501,8 @@ <h2>Subpackages<a class="headerlink" href="#subpackages" title="Permalink to thi
 <li class="toctree-l2"><a class="reference internal" href="tilelang.cache.html#module-tilelang.cache">Module contents</a><ul>
 <li class="toctree-l3"><a class="reference internal" href="tilelang.cache.html#tilelang.cache.cached"><code class="docutils literal notranslate"><span class="pre">cached()</span></code></a></li>
 <li class="toctree-l3"><a class="reference internal" href="tilelang.cache.html#tilelang.cache.clear_cache"><code class="docutils literal notranslate"><span class="pre">clear_cache()</span></code></a></li>
+<li class="toctree-l3"><a class="reference internal" href="tilelang.cache.html#tilelang.cache.get_cache_dir"><code class="docutils literal notranslate"><span class="pre">get_cache_dir()</span></code></a></li>
+<li class="toctree-l3"><a class="reference internal" href="tilelang.cache.html#tilelang.cache.set_cache_dir"><code class="docutils literal notranslate"><span class="pre">set_cache_dir()</span></code></a></li>
 </ul>
 </li>
 </ul>

api/tilelang.language.logical.html

@@ -406,7 +406,7 @@
 <p>The language interface for tl programs.</p>
 <dl class="py function">
 <dt class="sig sig-object py" id="tilelang.language.logical.all_of">
-<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">0x7f0bb4ecb250&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>
+<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">0x7f138cbb8790&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>
 <dd><p>Check if all elements in the buffer are true.</p>
 <dl class="field-list simple">
 <dt class="field-odd">Parameters<span class="colon">:</span></dt>
@@ -420,7 +420,7 @@
 
 <dl class="py function">
 <dt class="sig sig-object py" id="tilelang.language.logical.any_of">
-<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">0x7f0bb4ecb250&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>
+<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">0x7f138cbb8790&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>
 <dd><p>Check if any element in the buffer is true.</p>
 <dl class="field-list simple">
 <dt class="field-odd">Parameters<span class="colon">:</span></dt>