tilelang.layout.swizzle ======================= .. py:module:: tilelang.layout.swizzle .. autoapi-nested-parse:: Wrapping Layouts. Functions --------- .. autoapisummary:: tilelang.layout.swizzle.make_swizzled_layout tilelang.layout.swizzle.make_volta_swizzled_layout tilelang.layout.swizzle.make_wgmma_swizzled_layout tilelang.layout.swizzle.make_tcgen05mma_swizzled_layout tilelang.layout.swizzle.make_full_bank_swizzled_layout tilelang.layout.swizzle.make_half_bank_swizzled_layout tilelang.layout.swizzle.make_quarter_bank_swizzled_layout tilelang.layout.swizzle.make_linear_layout Module Contents --------------- .. py:function:: make_swizzled_layout(buffer, k_major = True, allow_pad = True) .. py:function:: make_volta_swizzled_layout(buffer, is_a = True, k_inner = True) .. py:function:: make_wgmma_swizzled_layout(buffer, continuity = None, k_major = True) .. py:function:: make_tcgen05mma_swizzled_layout(buffer, continuity = None, k_major = True) .. py:function:: make_full_bank_swizzled_layout(*args) :param args: buffer/BufferLoad/BufferRegion or (stride, continuous, element_size) .. rubric:: Examples make_full_bank_swizzled_layout(buffer) make_full_bank_swizzled_layout(stride, continuous, element_size) .. py:function:: make_half_bank_swizzled_layout(*args) :param args: buffer/BufferLoad/BufferRegion or (stride, continuous, element_size) .. rubric:: Examples make_half_bank_swizzled_layout(buffer) make_half_bank_swizzled_layout(stride, continuous, element_size) .. py:function:: make_quarter_bank_swizzled_layout(*args) :param args: buffer/BufferLoad/BufferRegion or (stride, continuous, element_size) .. rubric:: Examples make_quarter_bank_swizzled_layout(buffer) make_quarter_bank_swizzled_layout(stride, continuous, element_size) .. py:function:: make_linear_layout(*args) :param args: buffer/BufferLoad/BufferRegion or (stride, continuous) .. rubric:: Examples make_linear_layout(buffer) make_linear_layout(stride, continuous)