tilelang.tileop.gemm_sp¶

Submodules¶

Classes¶

Package Contents¶

class tilelang.tileop.gemm_sp.GemmSP¶

Bases: tvm.ir.base.Node, tvm.runtime.Scriptable

A: tvm.tirx.Buffer¶

E: tvm.tirx.Buffer¶

B: tvm.tirx.Buffer¶

C: tvm.tirx.Buffer¶

aRegion: tvm.tirx.BufferRegion¶

eRegion: tvm.tirx.BufferRegion¶

bRegion: tvm.tirx.BufferRegion¶

cRegion: tvm.tirx.BufferRegion¶

M: int¶

N: int¶

K: int¶

trans_A: bool¶

trans_B: bool¶

trans_E: bool¶

stride_A: int¶

stride_B: int¶

offset_A: int¶

offset_B: int¶

clear_accum: bool¶

kPack: int¶

wg_wait: int¶

policy: tilelang.ir.GemmSPWarpPolicy¶

property k_pack¶

gemm_sp_infer_layout(target, thread_bounds)¶

Parameters:

target (tvm.target.Target)
thread_bounds (tvm.ir.Range)

gemm_sp_lower(target, layout_map, thread_bounds, thread_var)¶

Parameters:

target (tvm.target.Target)
layout_map (dict)
thread_bounds (tvm.ir.Range)
thread_var (tvm.tirx.Var)

infer_layout(target, thread_nums)¶

Parameters:

target (tvm.target.Target)
thread_nums (int)

lower(target, layout_map, thread_nums, thread_var)¶

Parameters:

target (tvm.target.Target)
layout_map (dict)
thread_nums (int)
thread_var (tvm.tirx.Var)