For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).

Mojo function

short_conv_ring_commit

def short_conv_ring_commit[dtype: DType, //, target: StringSpan[ImmStaticOrigin]](x: TileTensor[dtype, Engine=x.Engine, address_space=x.address_space, linear_idx_type=x.linear_idx_type], ring: TileTensor[Engine=ring.Engine, address_space=ring.address_space, linear_idx_type=ring.linear_idx_type], input_row_offsets: TileTensor[.uint32, Engine=input_row_offsets.Engine, address_space=input_row_offsets.address_space, linear_idx_type=input_row_offsets.linear_idx_type], positions: TileTensor[.uint32, Engine=positions.Engine, address_space=positions.address_space, linear_idx_type=positions.linear_idx_type], conv_row: TileTensor[.uint32, Engine=conv_row.Engine, address_space=conv_row.address_space, linear_idx_type=conv_row.linear_idx_type], context: DeviceContext)

Writes each sequence's last R rows of x into its ring slot.

x: [total_seq_len, channels]. ring: [slots, R, channels]. Launch after every reader of the ring in the same forward.