tilus.Script.annotate_layout

Script.annotate_layout(tensor, layout)

Annotate the layout of a register tensor.

This instruction annotates the layout of a register tensor with a specified layout. The layout parameter is an instance of RegisterLayout 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) – The tensor to annotate.

• layout (RegisterLayout) – The layout to annotate the tensor with.

Return type:

None