triton.experimental.gluon.language.amd.cdna5.get_wmma_scale_layout
- triton.experimental.gluon.language.amd.cdna5.get_wmma_scale_layout()
Get the scale layout for WMMA 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, i.e. the number of elements of operand sharing a single scale.
- Returns:
The scale layout.
- Return type: