Script.annotate_layout

Script.annotate_layout(tensor, layout)

Annotate the layout of a register or shared tensor.

This instruction annotates the layout of a register or shared tensor with a specified layout. The layout parameter is an instance of RegisterLayout or SharedLayout that defines how the tensor’s data is organized among the threads in the thread block.

This layout will be used to guide the layout inference process.

Parameters:

• tensor (RegisterTensor | SharedTensor) – The tensor to annotate.

• layout (RegisterLayout | SharedLayout) – The layout to annotate the tensor with. The type of layout must match the type of tensor.

Return type:

None

Notes

• Thread group: Can be executed by any sized thread group.