For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /max/get-started.md).
Mojo struct
IteratorScatterGatherAmd
struct IteratorScatterGatherAmd[thread_layout: Layout, num_threads: Int = thread_layout.size(), thread_scope: ThreadScope = ThreadScope.BLOCK, block_dim_count: Int = Int(1)]
Iterator-based AMD scatter-gather for DRAM-register data movement.
Parameters
- thread_layout (
Layout): Thread organization layout. - num_threads (
Int): Total threads (defaults to thread_layout size). - thread_scope (
ThreadScope): Thread execution scope (block or warp). - block_dim_count (
Int): Number of block dimensions.
Fields
- buffer (
AMDBufferResource):
Implemented traits
Methods
__init__
def __init__(out self, tensor: LayoutTensor[address_space=tensor.address_space, element_layout=tensor.element_layout, layout_int_type=tensor.layout_int_type, linear_idx_type=tensor.linear_idx_type, masked=tensor.masked, alignment=tensor.alignment], tensor_iter: LayoutTensorIter[address_space=tensor_iter.address_space, alignment=tensor_iter.alignment, circular=tensor_iter.circular, axis=tensor_iter.axis, layout_int_type=tensor_iter.layout_int_type, linear_idx_type=tensor_iter.linear_idx_type, masked=tensor_iter.masked])
Initialize with tensor and iterator.
Args:
- tensor (
LayoutTensor[address_space=tensor.address_space, element_layout=tensor.element_layout, layout_int_type=tensor.layout_int_type, linear_idx_type=tensor.linear_idx_type, masked=tensor.masked, alignment=tensor.alignment]): Layout tensor for bounds. - tensor_iter (
LayoutTensorIter[address_space=tensor_iter.address_space, alignment=tensor_iter.alignment, circular=tensor_iter.circular, axis=tensor_iter.axis, layout_int_type=tensor_iter.layout_int_type, linear_idx_type=tensor_iter.linear_idx_type, masked=tensor_iter.masked]): Iterator for AMD buffer resource.
copy
def copy(self, dst_reg_tile: LayoutTensor[address_space=dst_reg_tile.address_space, element_layout=dst_reg_tile.element_layout, layout_int_type=dst_reg_tile.layout_int_type, linear_idx_type=dst_reg_tile.linear_idx_type, masked=dst_reg_tile.masked, alignment=dst_reg_tile.alignment], src_gmem_tile_iter: LayoutTensorIter[address_space=src_gmem_tile_iter.address_space, alignment=src_gmem_tile_iter.alignment, circular=src_gmem_tile_iter.circular, axis=src_gmem_tile_iter.axis, layout_int_type=src_gmem_tile_iter.layout_int_type, linear_idx_type=src_gmem_tile_iter.linear_idx_type, masked=src_gmem_tile_iter.masked])
Copy DRAM to registers via iterator.
Args:
- dst_reg_tile (
LayoutTensor[address_space=dst_reg_tile.address_space, element_layout=dst_reg_tile.element_layout, layout_int_type=dst_reg_tile.layout_int_type, linear_idx_type=dst_reg_tile.linear_idx_type, masked=dst_reg_tile.masked, alignment=dst_reg_tile.alignment]): Destination register tile. - src_gmem_tile_iter (
LayoutTensorIter[address_space=src_gmem_tile_iter.address_space, alignment=src_gmem_tile_iter.alignment, circular=src_gmem_tile_iter.circular, axis=src_gmem_tile_iter.axis, layout_int_type=src_gmem_tile_iter.layout_int_type, linear_idx_type=src_gmem_tile_iter.linear_idx_type, masked=src_gmem_tile_iter.masked]): Source memory iterator.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!