Skip to main content

function

strided_store

strided_store[type: DType, simd_width: Int, /, address_space: AddressSpace = 0](value: SIMD[type, simd_width], addr: DTypePointer[type, address_space], stride: Int, mask: SIMD[bool, simd_width])

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.
  • address_space (AddressSpace): The address space of the memory location.

Args:

  • value (SIMD[type, simd_width]): The values to store.
  • addr (DTypePointer[type, address_space]): The location to store values at.
  • stride (Int): How many lanes to skip before storing again.
  • mask (SIMD[bool, simd_width]): A binary vector which prevents memory access to certain lanes of value.

strided_store[type: DType, simd_width: Int, /, address_space: AddressSpace = 0](value: SIMD[type, simd_width], addr: DTypePointer[type, address_space], stride: Int)

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.
  • address_space (AddressSpace): The address space of the memory location.

Args:

  • value (SIMD[type, simd_width]): The values to store.
  • addr (DTypePointer[type, address_space]): The location to store values at.
  • stride (Int): How many lanes to skip before storing again.