Skip to main content

Mojo function

load_b_tile

load_b_tile[mma_shape: IndexList[3], swizzle: Optional[Swizzle], k_tile_idx: Int](src: LayoutTensor[src.dtype, src.layout, src.origin, address_space=src.address_space, element_layout=src.element_layout, layout_int_type=src.layout_int_type, linear_idx_type=src.linear_idx_type, masked=src.masked, alignment=src.alignment]) -> SIMD[src.dtype, simd_width_of[src.dtype]()]

Returns:

SIMD

Was this page helpful?