Skip to main content

Mojo function

strided_load

strided_load[type: DType, //, simd_width: Int](addr: UnsafePointer[SIMD[type, 1], address_space, exclusive, alignment], stride: Int, mask: SIMD[bool, simd_width] = 1) -> SIMD[$0, $1]

Loads values from addr according to a specific stride.

Parameters:

  • type (DType): DType of value, the value to store.
  • simd_width (Int): The width of the SIMD vectors.

Args:

  • addr (UnsafePointer[SIMD[type, 1], address_space, exclusive, alignment]): The memory location to load data from.
  • stride (Int): How many lanes to skip before loading again.
  • mask (SIMD[bool, simd_width]): A binary vector which prevents memory access to certain lanes of value.

Returns:

A vector containing the loaded data.