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¶

abstractmethod infer_layout(target, thread_nums)¶

参数:

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

abstractmethod lower(target, thread_nums, thread_var)¶

参数:

target (tvm.target.Target)
thread_nums (int)
thread_var (tvm.tir.Var)

is_gemm_ss()¶

返回类型:: bool

is_gemm_sr()¶

返回类型:: bool

is_gemm_rs()¶

返回类型:: bool

is_gemm_rr()¶

返回类型:: bool

property M: int¶

返回类型:: int

property N: int¶

返回类型:: int

property K: int¶

返回类型:: int

property trans_A: bool¶

返回类型:: bool

property trans_B: bool¶

返回类型:: bool

property trans_E: bool¶

返回类型:: bool

property e_dtype: str¶

返回类型:: str

property in_dtype: str¶

返回类型:: str

property accum_dtype: str¶

返回类型:: str

property A: tvm.tir.Buffer¶

返回类型:: tvm.tir.Buffer

property E: tvm.tir.Buffer¶

返回类型:: tvm.tir.Buffer

property B: tvm.tir.Buffer¶

返回类型:: tvm.tir.Buffer

property C: tvm.tir.Buffer¶

返回类型:: tvm.tir.Buffer

property ARegion: tvm.tir.PrimExpr¶

返回类型:: tvm.tir.PrimExpr

property ERegion: tvm.tir.PrimExpr¶

返回类型:: tvm.tir.PrimExpr

property BRegion: tvm.tir.PrimExpr¶

返回类型:: tvm.tir.PrimExpr

property CRegion: tvm.tir.PrimExpr¶

返回类型:: tvm.tir.PrimExpr

property stride_A: int¶

返回类型:: int

property stride_B: int¶

返回类型:: int

property offset_A: int¶

返回类型:: int

property offset_B: int¶

返回类型:: int

property clear_accum: bool¶

返回类型:: bool

property k_pack: int¶

返回类型:: int

property wg_wait: int¶

返回类型:: int

property policy: tilelang.tileop.base.GemmWarpPolicy¶

返回类型:: tilelang.tileop.base.GemmWarpPolicy