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

gather_elementwise_fn_wrapper

def gather_elementwise_fn_wrapper[dtype: DType, indices_type: DType, InputFnType: ImplicitlyCopyable & RegisterPassable & def[width: Int, rank: Int, element_alignment: Int](IndexList[rank]) -> SIMD[dtype, width], IndicesFnType: ImplicitlyCopyable & RegisterPassable & def[width: Int, rank: Int](IndexList[rank]) -> SIMD[indices_type, width], OutputFnType: ImplicitlyCopyable & RegisterPassable & def[width: Int, rank: Int, element_alignment: Int](IndexList[rank], SIMD[dtype, width]) -> None, *, simd_width: Int, prefetch_fn: OptionalReg[def[input_rank: Int, indices_rank: Int](IndexList[input_rank], IndexList[indices_rank]) capturing -> None] = None, target: StringSlice[StaticConstantOrigin] = StringSlice("cpu"), element_alignment: Int = 1](input_fn: InputFnType, indices_fn: IndicesFnType, output_fn: OutputFnType, axis: Axis, input_shape: IndexList[element_type=input_shape.element_type], indices_shape: IndexList[element_type=indices_shape.element_type], output_shape: IndexList[element_type=output_shape.element_type], coords: IndexList[element_type=coords.element_type], error_index_ptr: OptionalReg[UnsafePointer[Int, MutAnyOrigin]] = None) where (eq InputFnType.dtype, dtype) where (eq IndicesFnType.indices_type, indices_type) where (eq OutputFnType.dtype, dtype)