triton.experimental.gluon.language.amd.cdna4.get_mfma_scale_layout

triton.experimental.gluon.language.amd.cdna4.get_mfma_scale_layout()

Get the scale layout for MFMA scaled operands.

Parameters:

• dot_operand_layout (DotOperandLayout) – The dot operand layout.

• shape (List[int]) – The shape of the scale tensor.

• scale_factor (int) – The scale factor.

Returns:

The scale layout.

Return type:

layout (DistributedLinearLayout)