Mojo function

run_producer

run_producer[dtype: DType, layout: Layout, block_rows: Int, block_cols: Int, warp_rows: Int, warp_cols: Int, producer_warps: Int, pipeline_stages: Int, k_tile_size: Int, simd_width: Int, warps_processed_per_producer: Int, tile_count: Int, swizzle: Optional[Swizzle]](matrix: LayoutTensor[dtype, layout, MutAnyOrigin, address_space=AddressSpace.GLOBAL], mut ring_buffer: RingBuffer[ring_buffer.dtype, ring_buffer.layout, ring_buffer.pipeline_stages, ring_buffer.block_rows, ring_buffer.block_cols, ring_buffer.warp_rows, ring_buffer.warp_cols, ring_buffer.reads_per_warp_block, ring_buffer.tile_buffers, ring_buffer.sync_strategy_type], warp_id: Scalar[DType.uint], block_idx_dim: Int)

Generic producer function for loading matrix tiles from global to shared memory.

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