tilus.ir.layout.shared_row_major

tilus.ir.layout.shared_row_major

tilus.ir.layout.shared_row_major(*shape)[source]

Create a shared layout with row-major order.

Parameters:

shape (Sequence[int]) – The shape of the shared tensor. Each dimension is a constant integer.

Returns:

ret – A shared layout with the specified shape in row-major order.

Return type:

SharedLayout