tilelang.ascend.op.gemm.gemm_mad¶

Attributes¶

GEMM_INST_MAD

Classes¶

GemmMAD

Base class for GEMM tile operators.

Module Contents¶

tilelang.ascend.op.gemm.gemm_mad.GEMM_INST_MAD = 'ascend.mad'¶
class tilelang.ascend.op.gemm.gemm_mad.GemmMAD¶

Bases: tilelang.tileop.gemm.gemm_base.GemmBase

Base class for GEMM tile operators.

Classifies the GEMM variant by the memory scopes of operands A and B (SS, SR, RS, TS, RR) and provides common property accessors for the underlying gemm_node IR node.

property is_blockscaled: bool¶
Return type:

bool

property unit_flag_ctrl: tvm.tirx.PrimExpr¶
Return type:

tvm.tirx.PrimExpr

infer_layout(target, thread_nums)¶
Parameters:

  • target (tvm.target.Target)

  • thread_nums (int)

lower(layout_map, target, thread_bounds, thread_index, mbar_phase_expr=None)¶
Parameters:

  • layout_map (dict)

  • target (tvm.target.Target)

  • thread_bounds (tvm.ir.Range)

  • thread_index (tvm.tirx.PrimExpr)

  • mbar_phase_expr (tvm.tirx.PrimExpr | None)