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(target, thread_nums, thread_var)¶
Parameters:

  • target (tvm.target.Target)

  • thread_nums (int)

  • thread_var (tvm.tir.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 in_dtype: str¶
Return type:

str

property accum_dtype: str¶
Return type:

str

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

tvm.tir.Buffer

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

tvm.tir.Buffer

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

tvm.tir.Buffer

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

tvm.tir.Buffer

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

tvm.tir.PrimExpr

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

tvm.tir.PrimExpr

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

tvm.tir.PrimExpr

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

tvm.tir.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.GemmWarpPolicy¶
Return type:

tilelang.ir.GemmWarpPolicy