triton.experimental.gluon.language.set_auto_layout

triton.experimental.gluon.language.set_auto_layout(value, layout, _semantic=None)

Set a tensor with AutoLayout to a concrete layout

Parameters:

• value (tensor) – The input tensor.

• layout (DistribtedLayout) – The target layout.

Returns:

The tensor with the new layout.

Return type:

tensor