Mojo function

tt_load_b_tile

tt_load_b_tile[mma_shape: IndexList[3], swizzle: Optional[Swizzle], k_tile_idx: Int](src: TileTensor[src.dtype, src.LayoutType, src.origin, address_space=AddressSpace.SHARED, linear_idx_type=src.linear_idx_type, element_size=src.element_size]) -> SIMD[src.dtype, simd_width_of[src.dtype]()]

Single MMA tile load from SMEM with optional swizzle.

Loads one MMA_M×MMA_K sub-tile at column k_tile_idx, distributes across lanes, applies swizzle, and reads via llvm.load.

Returns:

SIMD

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other

Submit