triton.experimental.gluon.language.DotOperandLayout

class triton.experimental.gluon.language.DotOperandLayout(self, operand_index: int, parent: DistributedLayout, k_width: int)

Represents a layout for a dot operand.

Parameters:

• operand_index (int) – 0 for LHS and 1 for RHS of the dot operation.

• parent (DistributedLayout) – The parent layout, representing the MMA.

• k_width (int) – Number of elements per 32-bits.

__init__(self, operand_index: int, parent: DistributedLayout, k_width: int) → None

Methods

__init__(self, operand_index, parent, k_width)
format_hardware_view(self, shape)
format_tensor_view(self, shape)
mangle(self)

Attributes

cga_layout
rank
type
operand_index
parent
k_width