tilelang.tileop.gemm_sp.gemm_sp_mma¶

Classes¶

GemmSPMMA

Module Contents¶

class tilelang.tileop.gemm_sp.gemm_sp_mma.GemmSPMMA¶

Bases: tilelang.tileop.gemm_sp.gemm_sp_base.GemmSPBase

infer_layout(target, thread_nums)¶
Parameters:

  • target (tvm.target.Target)

  • thread_nums (int)

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