triton.experimental.gluon.language.convert_layout

triton.experimental.gluon.language.convert_layout(value, layout, assert_trivial=False, _semantic=None)

Convert a tensor to a different distributed layout.

Parameters:

value (tensor) – The input tensor.
layout (DistributedLayout) – The target layout.
assert_trivial (bool) – If True, asserts that the conversion is trivial (no data movement).

Returns:

The tensor with the new layout.

Return type: