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 function

partial_simd_load

def partial_simd_load[dtype: DType, //, width: Int](storage: UnsafePointer[Scalar[dtype], address_space=storage.address_space], lbound: Int, rbound: Int, pad_value: Scalar[dtype]) -> SIMD[dtype, width]

Loads a vector with dynamic bound.

Out of bound data will be filled with pad value. Data is valid if lbound <= idx < rbound for idx from 0 to (simd_width-1). For example:

addr 0  1  2  3
data x 42 43  x

partial_simd_load[4](addr0, 1, 3) #gives [0 42 43 0]

Parameters:

dtype (DType): The DType of storage.
width (Int): The system simd vector size.

Args:

storage (UnsafePointer[Scalar[dtype], address_space=storage.address_space]): Pointer to the address to perform load.
lbound (Int): Lower bound of valid index within simd (inclusive).
rbound (Int): Upper bound of valid index within simd (non-inclusive).
pad_value (Scalar[dtype]): Value to fill for out of bound indices.

Returns:

SIMD[dtype, width]: The SIMD vector loaded and zero-filled.