tilelang.tileop.gemm.gemm_mma_sm70

Classes

GemmMMASm70

Module Contents

class tilelang.tileop.gemm.gemm_mma_sm70.GemmMMASm70

Bases: tilelang.tileop.gemm.gemm_base.GemmBase

infer_layout(target, thread_nums)

Parameters:

• target (tvm.target.Target)

• thread_nums (int)

lower(layout_map, target, thread_nums, thread_var)

Parameters:

• layout_map (dict)

• 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