|
340 | 340 | <li class="toctree-l4"><a class="reference internal" href="tilelang.profiler.bench.html">tilelang.profiler.bench module</a></li>
|
341 | 341 | </ul>
|
342 | 342 | </li>
|
| 343 | +<li class="toctree-l3 has-children"><a class="reference internal" href="tilelang.quantize.html">tilelang.quantize package</a><input class="toctree-checkbox" id="toctree-checkbox-25" name="toctree-checkbox-25" role="switch" type="checkbox"/><label for="toctree-checkbox-25"><div class="visually-hidden">Toggle child pages in navigation</div><i class="icon"><svg><use href="#svg-arrow-right"></use></svg></i></label><ul> |
| 344 | +<li class="toctree-l4"><a class="reference internal" href="tilelang.quantize.lop3.html">tilelang.quantize.lop3 module</a></li> |
| 345 | +<li class="toctree-l4"><a class="reference internal" href="tilelang.quantize.quantization.html">tilelang.quantize.quantization module</a></li> |
| 346 | +<li class="toctree-l4"><a class="reference internal" href="tilelang.quantize.utils.html">tilelang.quantize.utils module</a></li> |
| 347 | +</ul> |
| 348 | +</li> |
343 | 349 | <li class="toctree-l3"><a class="reference internal" href="tilelang.testing.html">tilelang.testing package</a></li>
|
344 |
| -<li class="toctree-l3 has-children"><a class="reference internal" href="tilelang.tools.html">tilelang.tools package</a><input class="toctree-checkbox" id="toctree-checkbox-25" name="toctree-checkbox-25" role="switch" type="checkbox"/><label for="toctree-checkbox-25"><div class="visually-hidden">Toggle child pages in navigation</div><i class="icon"><svg><use href="#svg-arrow-right"></use></svg></i></label><ul> |
| 350 | +<li class="toctree-l3 has-children"><a class="reference internal" href="tilelang.tools.html">tilelang.tools package</a><input class="toctree-checkbox" id="toctree-checkbox-26" name="toctree-checkbox-26" role="switch" type="checkbox"/><label for="toctree-checkbox-26"><div class="visually-hidden">Toggle child pages in navigation</div><i class="icon"><svg><use href="#svg-arrow-right"></use></svg></i></label><ul> |
345 | 351 | <li class="toctree-l4"><a class="reference internal" href="tilelang.tools.Analyzer.html">tilelang.tools.Analyzer module</a></li>
|
346 | 352 | <li class="toctree-l4"><a class="reference internal" href="tilelang.tools.plot_layout.html">tilelang.tools.plot_layout module</a></li>
|
347 | 353 | </ul>
|
348 | 354 | </li>
|
349 |
| -<li class="toctree-l3 has-children"><a class="reference internal" href="tilelang.transform.html">tilelang.transform package</a><input class="toctree-checkbox" id="toctree-checkbox-26" name="toctree-checkbox-26" role="switch" type="checkbox"/><label for="toctree-checkbox-26"><div class="visually-hidden">Toggle child pages in navigation</div><i class="icon"><svg><use href="#svg-arrow-right"></use></svg></i></label><ul> |
| 355 | +<li class="toctree-l3 has-children"><a class="reference internal" href="tilelang.transform.html">tilelang.transform package</a><input class="toctree-checkbox" id="toctree-checkbox-27" name="toctree-checkbox-27" role="switch" type="checkbox"/><label for="toctree-checkbox-27"><div class="visually-hidden">Toggle child pages in navigation</div><i class="icon"><svg><use href="#svg-arrow-right"></use></svg></i></label><ul> |
350 | 356 | <li class="toctree-l4"><a class="reference internal" href="tilelang.transform.pass_config.html">tilelang.transform.pass_config module</a></li>
|
351 | 357 | <li class="toctree-l4"><a class="reference internal" href="tilelang.transform.simplify.html">tilelang.transform.simplify module</a></li>
|
352 | 358 | </ul>
|
353 | 359 | </li>
|
354 |
| -<li class="toctree-l3 has-children"><a class="reference internal" href="tilelang.utils.html">tilelang.utils package</a><input class="toctree-checkbox" id="toctree-checkbox-27" name="toctree-checkbox-27" role="switch" type="checkbox"/><label for="toctree-checkbox-27"><div class="visually-hidden">Toggle child pages in navigation</div><i class="icon"><svg><use href="#svg-arrow-right"></use></svg></i></label><ul> |
| 360 | +<li class="toctree-l3 has-children"><a class="reference internal" href="tilelang.utils.html">tilelang.utils package</a><input class="toctree-checkbox" id="toctree-checkbox-28" name="toctree-checkbox-28" role="switch" type="checkbox"/><label for="toctree-checkbox-28"><div class="visually-hidden">Toggle child pages in navigation</div><i class="icon"><svg><use href="#svg-arrow-right"></use></svg></i></label><ul> |
355 | 361 | <li class="toctree-l4"><a class="reference internal" href="tilelang.utils.deprecated.html">tilelang.utils.deprecated module</a></li>
|
356 | 362 | <li class="toctree-l4"><a class="reference internal" href="tilelang.utils.language.html">tilelang.utils.language module</a></li>
|
357 | 363 | <li class="toctree-l4"><a class="reference internal" href="tilelang.utils.target.html">tilelang.utils.target module</a></li>
|
@@ -1013,6 +1019,12 @@ <h1>tilelang.autotuner package<a class="headerlink" href="#tilelang-autotuner-pa
|
1013 | 1019 |
|
1014 | 1020 | </dd></dl>
|
1015 | 1021 |
|
| 1022 | +<dl class="py exception"> |
| 1023 | +<dt class="sig sig-object py" id="tilelang.autotuner.TimeoutException"> |
| 1024 | +<em class="property"><span class="pre">exception</span><span class="w"> </span></em><span class="sig-prename descclassname"><span class="pre">tilelang.autotuner.</span></span><span class="sig-name descname"><span class="pre">TimeoutException</span></span><a class="headerlink" href="#tilelang.autotuner.TimeoutException" title="Permalink to this definition">#</a></dt> |
| 1025 | +<dd><p>Bases: <code class="xref py py-class docutils literal notranslate"><span class="pre">Exception</span></code></p> |
| 1026 | +</dd></dl> |
| 1027 | + |
1016 | 1028 | <dl class="py function">
|
1017 | 1029 | <dt class="sig sig-object py" id="tilelang.autotuner.autotune">
|
1018 | 1030 | <span class="sig-prename descclassname"><span class="pre">tilelang.autotuner.</span></span><span class="sig-name descname"><span class="pre">autotune</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">configs</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">Any</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">warmup</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">int</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">25</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">rep</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">int</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">100</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">timeout</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">int</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">100</span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">→</span> <span class="sig-return-typehint"><a class="reference internal" href="#tilelang.autotuner.AutotuneResult" title="tilelang.autotuner.AutotuneResult"><span class="pre">AutotuneResult</span></a></span></span><a class="headerlink" href="#tilelang.autotuner.autotune" title="Permalink to this definition">#</a></dt>
|
@@ -1087,6 +1099,16 @@ <h1>tilelang.autotuner package<a class="headerlink" href="#tilelang-autotuner-pa
|
1087 | 1099 | </dl>
|
1088 | 1100 | </dd></dl>
|
1089 | 1101 |
|
| 1102 | +<dl class="py function"> |
| 1103 | +<dt class="sig sig-object py" id="tilelang.autotuner.run_with_timeout"> |
| 1104 | +<span class="sig-prename descclassname"><span class="pre">tilelang.autotuner.</span></span><span class="sig-name descname"><span class="pre">run_with_timeout</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">func</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">timeout</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="o"><span class="pre">**</span></span><span class="n"><span class="pre">kwargs</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.autotuner.run_with_timeout" title="Permalink to this definition">#</a></dt> |
| 1105 | +<dd></dd></dl> |
| 1106 | + |
| 1107 | +<dl class="py function"> |
| 1108 | +<dt class="sig sig-object py" id="tilelang.autotuner.timeout_handler"> |
| 1109 | +<span class="sig-prename descclassname"><span class="pre">tilelang.autotuner.</span></span><span class="sig-name descname"><span class="pre">timeout_handler</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">signum</span></span></em>, <em class="sig-param"><span class="n"><span class="pre">frame</span></span></em><span class="sig-paren">)</span><a class="headerlink" href="#tilelang.autotuner.timeout_handler" title="Permalink to this definition">#</a></dt> |
| 1110 | +<dd></dd></dl> |
| 1111 | + |
1090 | 1112 | </section>
|
1091 | 1113 | </section>
|
1092 | 1114 |
|
@@ -1223,10 +1245,13 @@ <h1>tilelang.autotuner package<a class="headerlink" href="#tilelang-autotuner-pa
|
1223 | 1245 | <li><a class="reference internal" href="#id31"><code class="docutils literal notranslate"><span class="pre">JITContext.target</span></code></a></li>
|
1224 | 1246 | </ul>
|
1225 | 1247 | </li>
|
| 1248 | +<li><a class="reference internal" href="#tilelang.autotuner.TimeoutException"><code class="docutils literal notranslate"><span class="pre">TimeoutException</span></code></a></li> |
1226 | 1249 | <li><a class="reference internal" href="#tilelang.autotuner.autotune"><code class="docutils literal notranslate"><span class="pre">autotune()</span></code></a></li>
|
1227 | 1250 | <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>
|
1228 | 1251 | <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>
|
1229 | 1252 | <li><a class="reference internal" href="#tilelang.autotuner.jit"><code class="docutils literal notranslate"><span class="pre">jit()</span></code></a></li>
|
| 1253 | +<li><a class="reference internal" href="#tilelang.autotuner.run_with_timeout"><code class="docutils literal notranslate"><span class="pre">run_with_timeout()</span></code></a></li> |
| 1254 | +<li><a class="reference internal" href="#tilelang.autotuner.timeout_handler"><code class="docutils literal notranslate"><span class="pre">timeout_handler()</span></code></a></li> |
1230 | 1255 | </ul>
|
1231 | 1256 | </li>
|
1232 | 1257 | </ul>
|
|
0 commit comments