tilus.ir.layout.global_column_major

tilus.ir.layout.global_column_major

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

Create a global layout with column-major order.

Parameters:

shape (Sequence[Expr | int]) – The shape of the global tensor. Each dimension can be an expression of grid-invariant expression, or a constant integer.

Returns:

ret – A global layout with the specified shape in column-major order.

Return type:

GlobalLayout