tiny.simd.countIf

1pub fn countIf(comptime D: type, input: []const D.Lane, predicate: anytype) usize {2    var total: usize = 0;3    var index: usize = 0;4    while (index + D.lane_count <= input.len) : (index += D.lane_count) {5        total += compare.countTrue(D, predicate.call(D, memory.load(D, input[index..])));6    }7    if (index != input.len) {8        const remaining = input.len - index;9        const matches = predicate.call(D, memory.loadN(D, input[index..], remaining));10        total += compare.countTrue(D, matches & construct.firstN(D, remaining));11    }12    std.debug.assert(total <= input.len);13    return total;14}

1pub const countIf = algo.countIf;