src/simd/internal/spec/combinators.go

// Copyright 2026 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package spec

// map1 returns the vector z[i] = f(x[i]).
//
// z may have more lanes than x, in which case they will be 0.
func map1[From Elt, FromW Width, To Elt, ToW Width](x Vec[From, FromW], f func(x From) To) (z Vec[To, ToW]) {
	z = makeVec[To, ToW]()
	// Loop over x, not z, because width rounding may have expanded z.
	for i, xi := range x {
		z[i] = f(xi)
	}
	return z
}

// map2 returns the vector z[i] = f(x[i], y[i]).
//
// z may have more lanes than x, in which case they will be 0.
func map2[From Elt, FromW Width, To Elt, ToW Width](x, y Vec[From, FromW], f func(x, y From) To) (z Vec[To, ToW]) {
	z = makeVec[To, ToW]()
	for i := range x {
		z[i] = f(x[i], y[i])
	}
	return z
}