tilelang.tileop.gemm.gemm_mma

類別

GemmMMA

Module Contents

class tilelang.tileop.gemm.gemm_mma.GemmMMA

基底類別：tilelang.tileop.gemm.gemm_base.GemmBase

infer_layout(target, thread_nums)

參數:

• target (tvm.target.Target)

• thread_nums (int)

lower(layout_map, target, thread_bounds, thread_var, mbar_phase_expr=None)

參數:

• layout_map (dict)

• target (tvm.target.Target)

• thread_bounds (tvm.ir.Range)

• thread_var (tvm.tir.Var)

• mbar_phase_expr (tvm.tir.PrimExpr | None)

is_gemm_ss()

回傳值型別:

bool

is_gemm_sr()

回傳值型別:

bool

is_gemm_rs()

回傳值型別:

bool

is_gemm_rr()

回傳值型別:

bool