tilelang.tileop.gemm.gemm_mma

Classes

GemmMMA

Module Contents

class tilelang.tileop.gemm.gemm_mma.GemmMMA

Bases: tilelang.tileop.gemm.gemm_base.GemmBase

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