tilelang.ascend.op.gemm.gemm_mad ================================ .. py:module:: tilelang.ascend.op.gemm.gemm_mad Attributes ---------- .. autoapisummary:: tilelang.ascend.op.gemm.gemm_mad.GEMM_INST_MAD Classes ------- .. autoapisummary:: tilelang.ascend.op.gemm.gemm_mad.GemmMAD Module Contents --------------- .. py:data:: GEMM_INST_MAD :value: 'ascend.mad' .. py:class:: GemmMAD Bases: :py:obj:`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. .. py:property:: is_blockscaled :type: bool .. py:property:: unit_flag_ctrl :type: tvm.tirx.PrimExpr .. py:method:: infer_layout(target, thread_nums) .. py:method:: lower(layout_map, target, thread_bounds, thread_index, mbar_phase_expr = None)