tilelang.tileop.gemm.gemm_scalar

類別

GemmScalar

CPU scalar fallback: triple nested loop gemm.

Module Contents

class tilelang.tileop.gemm.gemm_scalar.GemmScalar

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

CPU scalar fallback: triple nested loop gemm.

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)