tilelang.tileop.gemm.gemm_tcgen05¶

Classes¶

GemmTCGEN5

Module Contents¶

class tilelang.tileop.gemm.gemm_tcgen05.GemmTCGEN5¶

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)