Skip to content

Commit 1465dd4

Browse files
Update docs
1 parent a2c7b70 commit 1465dd4

File tree

5 files changed

+14
-212
lines changed

5 files changed

+14
-212
lines changed

_sources/autoapi/tilelang/intrinsics/mfma_macro_generator/index.rst.txt

Lines changed: 1 addition & 76 deletions
Original file line numberDiff line numberDiff line change
@@ -198,89 +198,14 @@ Module Contents
198198

199199

200200

201-
.. py:class:: MatrixCorePreshuffleIntrinEmitter(a_dtype = 'float16', b_dtype = 'float16', accum_dtype = 'float16', a_transposed = False, b_transposed = False, block_row_warps = 2, block_col_warps = 2, warp_row_tiles = 8, warp_col_tiles = 8, chunk = 16, reduce_k = 1, num_elems_per_byte = 1, k_pack = None, is_m_first = False, a_preshuffle = False, b_preshuffle = False)
201+
.. py:class:: MatrixCorePreshuffleIntrinEmitter(a_dtype = 'float16', b_dtype = 'float16', accum_dtype = 'float16', a_transposed = False, b_transposed = False, block_row_warps = 2, block_col_warps = 2, warp_row_tiles = 8, warp_col_tiles = 8, chunk = 16, reduce_k = 1, num_elems_per_byte = 1, k_pack = None, is_m_first = False, a_preshuffle = False, b_preshuffle = False, thread_var = None)
202202
203203
Bases: :py:obj:`MatrixCoreIntrinEmitter`
204204

205205

206206
To eliminate Python syntax within TIR Macro.
207207

208208

209-
.. py:attribute:: a_dtype
210-
:value: 'float16'
211-
212-
213-
214-
.. py:attribute:: b_dtype
215-
:value: 'float16'
216-
217-
218-
219-
.. py:attribute:: accum_dtype
220-
:value: 'float16'
221-
222-
223-
224-
.. py:attribute:: a_transposed
225-
:value: False
226-
227-
228-
229-
.. py:attribute:: b_transposed
230-
:value: False
231-
232-
233-
234-
.. py:attribute:: block_row_warps
235-
:value: 2
236-
237-
238-
239-
.. py:attribute:: block_col_warps
240-
:value: 2
241-
242-
243-
244-
.. py:attribute:: warp_row_tiles
245-
:value: 8
246-
247-
248-
249-
.. py:attribute:: warp_col_tiles
250-
:value: 8
251-
252-
253-
254-
.. py:attribute:: chunk
255-
:value: 16
256-
257-
258-
259-
.. py:attribute:: warp_rows
260-
:value: 0
261-
262-
263-
264-
.. py:attribute:: warp_cols
265-
:value: 0
266-
267-
268-
269-
.. py:attribute:: reduce_k
270-
:value: 1
271-
272-
273-
274-
.. py:attribute:: threads
275-
:value: 256
276-
277-
278-
279-
.. py:attribute:: num_elems_per_byte
280-
:value: 1
281-
282-
283-
284209
.. py:method:: ldmatrix_a(A_local_buf, A_buf, ki, rk=0, pid_m=None, pid_n=None)
285210
286211

0 commit comments

Comments
 (0)