tilus.ir.layout.shared_column_major

tilus.ir.layout.shared_column_major

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

Create a shared layout with column-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 column-major order.

Return type:

SharedLayout