tilelang.tileop.gemm.gemm_mma_sm70¶

類別¶

Module Contents¶

class tilelang.tileop.gemm.gemm_mma_sm70.GemmMMASm70¶

Bases: 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)¶

參數:

layout_map (dict)
target (tvm.target.Target)
thread_bounds (tvm.ir.Range)
thread_var (tvm.tir.Var)

is_gemm_ss()¶

回傳型別:: bool

is_gemm_sr()¶

回傳型別:: bool

is_gemm_rs()¶

回傳型別:: bool

is_gemm_rr()¶

回傳型別:: bool