Skip to content

Commit 3a9701f

Browse files
Update docs
1 parent 640e973 commit 3a9701f

File tree

303 files changed

+137860
-8
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

303 files changed

+137860
-8
lines changed

_sources/autoapi/index.rst.txt

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
1+
API Reference
2+
=============
3+
4+
This page contains auto-generated API reference documentation [#f1]_.
5+
6+
.. toctree::
7+
:titlesonly:
8+
9+
/autoapi/tilelang/index
10+
11+
.. [#f1] Created with `sphinx-autoapi <https://github.com/readthedocs/sphinx-autoapi>`_
Lines changed: 129 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,129 @@
1+
tilelang.autotuner.capture
2+
==========================
3+
4+
.. py:module:: tilelang.autotuner.capture
5+
6+
7+
Classes
8+
-------
9+
10+
.. autoapisummary::
11+
12+
tilelang.autotuner.capture.CaptureStack
13+
tilelang.autotuner.capture.AutotuneInputsCapture
14+
15+
16+
Functions
17+
---------
18+
19+
.. autoapisummary::
20+
21+
tilelang.autotuner.capture.set_autotune_inputs
22+
tilelang.autotuner.capture.get_autotune_inputs
23+
24+
25+
Module Contents
26+
---------------
27+
28+
.. py:class:: CaptureStack
29+
30+
A simple stack implementation for capturing items in a thread-local context.
31+
Used to manage a stack of items (e.g., input tensors) for auto-tuning capture.
32+
33+
34+
.. py:attribute:: stack
35+
:value: []
36+
37+
38+
39+
.. py:method:: push(item)
40+
41+
Push an item onto the top of the stack.
42+
43+
:param item: The item to be pushed onto the stack.
44+
45+
46+
47+
.. py:method:: pop()
48+
49+
Pop and return the top item from the stack.
50+
51+
:returns: The item at the top of the stack.
52+
53+
:raises IndexError: If the stack is empty.
54+
55+
56+
57+
.. py:method:: top()
58+
59+
Return the item at the top of the stack without removing it.
60+
61+
:returns: The item at the top of the stack.
62+
63+
:raises IndexError: If the stack is empty.
64+
65+
66+
67+
.. py:method:: size()
68+
69+
Return the number of items in the stack.
70+
71+
:returns: The size of the stack.
72+
:rtype: int
73+
74+
75+
76+
.. py:method:: __len__()
77+
78+
Return the number of items in the stack (len operator support).
79+
80+
:returns: The size of the stack.
81+
:rtype: int
82+
83+
84+
85+
.. py:method:: __bool__()
86+
87+
Return True if the stack is not empty, False otherwise.
88+
89+
:returns: Whether the stack contains any items.
90+
:rtype: bool
91+
92+
93+
94+
.. py:class:: AutotuneInputsCapture(tensors)
95+
96+
.. py:attribute:: __slots__
97+
:value: 'tensors'
98+
99+
100+
101+
.. py:attribute:: tensors
102+
103+
104+
.. py:method:: __enter__()
105+
106+
107+
.. py:method:: __exit__(exc_type, exc_val, exc_tb)
108+
109+
110+
.. py:function:: set_autotune_inputs(*args)
111+
112+
Set input tensors for auto-tuning.
113+
114+
This function creates a context manager for capturing input tensors
115+
during the auto-tuning process. It supports both:
116+
set_autotune_inputs(a, b, c)
117+
set_autotune_inputs([a, b, c])
118+
119+
:param \*args: Either a single list/tuple of tensors, or multiple tensor arguments.
120+
121+
:returns: A context manager for auto-tuning inputs.
122+
:rtype: AutotuneInputsCapture
123+
124+
125+
.. py:function:: get_autotune_inputs()
126+
127+
Get the current autotune inputs from the stack.
128+
129+
Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,17 @@
1+
tilelang.autotuner
2+
==================
3+
4+
.. py:module:: tilelang.autotuner
5+
6+
7+
Submodules
8+
----------
9+
10+
.. toctree::
11+
:maxdepth: 1
12+
13+
/autoapi/tilelang/autotuner/capture/index
14+
/autoapi/tilelang/autotuner/param/index
15+
/autoapi/tilelang/autotuner/tuner/index
16+
17+

0 commit comments

Comments
 (0)