tilelang.cuda.op.gemm_sp.gemm_sp_mma¶

Attributes¶

GEMM_SP_INST_MMA_SP

Classes¶

Module Contents¶

tilelang.cuda.op.gemm_sp.gemm_sp_mma.GEMM_SP_INST_MMA_SP = 'cuda.mma.sp'¶

class tilelang.cuda.op.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(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