tiny.simd.replaceIf

1pub fn replaceIf(2    comptime D: type,3    input_output: []D.Lane,4    new: D.Lane,5    predicate: anytype,6) void {7    const new_vector: D.Vector = @splat(new);8    var index: usize = 0;9    while (index + D.lane_count <= input_output.len) : (index += D.lane_count) {10        const value = memory.load(D, input_output[index..]);11        const result = @select(D.Lane, predicate.call(D, value), new_vector, value);12        memory.store(D, result, input_output[index..]);13    }14    if (index != input_output.len) {15        const remaining = input_output.len - index;16        const value = memory.loadN(D, input_output[index..], remaining);17        const result = @select(D.Lane, predicate.call(D, value), new_vector, value);18        memory.storeN(D, result, input_output[index..], remaining);19    }20}

1pub const replaceIf = algo.replaceIf;