tilelang.tileop.gemm_sp.gemm_sp_base¶

Classes¶

GemmSPBase

Module Contents¶

class tilelang.tileop.gemm_sp.gemm_sp_base.GemmSPBase¶
gemm_sp_node: tvm.ir.base.Node¶
abstract infer_layout(target, thread_nums)¶
Parameters:

  • target (tvm.target.Target)

  • thread_nums (int)

abstract lower(layout_map, target, thread_bounds, thread_var)¶
Parameters:

  • layout_map (dict)

  • target (tvm.target.Target)

  • thread_bounds (tvm.ir.Range)

  • thread_var (tvm.tirx.Var)

is_gemm_ss()¶
Return type:

bool

is_gemm_sr()¶
Return type:

bool

is_gemm_rs()¶
Return type:

bool

is_gemm_rr()¶
Return type:

bool

property M: int¶
Return type:

int

property N: int¶
Return type:

int

property K: int¶
Return type:

int

property trans_A: bool¶
Return type:

bool

property trans_B: bool¶
Return type:

bool

property trans_E: bool¶
Return type:

bool

property e_dtype: str¶
Return type:

str

property a_dtype¶
property b_dtype¶
property accum_dtype: str¶
Return type:

str

property A: tvm.tirx.Buffer¶
Return type:

tvm.tirx.Buffer

property E: tvm.tirx.Buffer¶
Return type:

tvm.tirx.Buffer

property B: tvm.tirx.Buffer¶
Return type:

tvm.tirx.Buffer

property C: tvm.tirx.Buffer¶
Return type:

tvm.tirx.Buffer

property ARegion: tvm.tirx.PrimExpr¶
Return type:

tvm.tirx.PrimExpr

property ERegion: tvm.tirx.PrimExpr¶
Return type:

tvm.tirx.PrimExpr

property BRegion: tvm.tirx.PrimExpr¶
Return type:

tvm.tirx.PrimExpr

property CRegion: tvm.tirx.PrimExpr¶
Return type:

tvm.tirx.PrimExpr

property stride_A: int¶
Return type:

int

property stride_B: int¶
Return type:

int

property offset_A: int¶
Return type:

int

property offset_B: int¶
Return type:

int

property clear_accum: bool¶
Return type:

bool

property k_pack: int¶
Return type:

int

property wg_wait: int¶
Return type:

int

property policy: tilelang.ir.GemmSPWarpPolicy¶
Return type:

tilelang.ir.GemmSPWarpPolicy