tilelang.layout.swizzle

Wrapping Layouts.

Functions

make_swizzled_layout(buffer[, k_major, allow_pad])
make_wgmma_swizzled_layout(buffer[, continuity, k_major])
make_full_bank_swizzled_layout(*args)
make_half_bank_swizzled_layout(*args)
make_quarter_bank_swizzled_layout(*args)
make_linear_layout(*args)

Module Contents

tilelang.layout.swizzle.make_swizzled_layout(buffer, k_major=True, allow_pad=True)

Parameters:

• buffer (tvm.tir.Buffer)

• k_major (bool)

• allow_pad (bool)

tilelang.layout.swizzle.make_wgmma_swizzled_layout(buffer, continuity=None, k_major=True)

Parameters:

• buffer (tvm.tir.Buffer)

• continuity (int)

• k_major (bool)

tilelang.layout.swizzle.make_full_bank_swizzled_layout(*args)

Parameters:

args – buffer or (stride, continuous, element_size)

Examples

make_full_bank_swizzled_layout(buffer) make_full_bank_swizzled_layout(stride, continuous, element_size)

tilelang.layout.swizzle.make_half_bank_swizzled_layout(*args)

Parameters:

args – buffer or (stride, continuous, element_size)

Examples

make_half_bank_swizzled_layout(buffer) make_half_bank_swizzled_layout(stride, continuous, element_size)

tilelang.layout.swizzle.make_quarter_bank_swizzled_layout(*args)

Parameters:

args – buffer or (stride, continuous, element_size)

Examples

make_quarter_bank_swizzled_layout(buffer) make_quarter_bank_swizzled_layout(stride, continuous, element_size)

tilelang.layout.swizzle.make_linear_layout(*args)

Parameters:

args – buffer or (stride, continuous)

Examples

make_linear_layout(buffer) make_linear_layout(stride, continuous)