tiny.simd.bitShuffle

1pub fn bitShuffle(comptime D: type, values: D.Vector, indices: anytype) D.Vector {2    if (D.Lane != u64 and D.Lane != i64) @compileError("bitShuffle requires 64-bit integer lanes");3    const index_info = vectorInfo(@TypeOf(indices));4    if (@typeInfo(index_info.child) != .int or @bitSizeOf(index_info.child) != 8) {5        @compileError("bitShuffle indices require 8-bit integer lanes");6    }7    if (index_info.len != D.lane_count * 8) @compileError("bitShuffle requires eight indices per lane");8    const value_lanes: [D.lane_count]D.Lane = @bitCast(values);9    const index_lanes: [index_info.len]index_info.child = @bitCast(indices);10    var result: [D.lane_count]u64 = @splat(0);11    inline for (0..D.lane_count) |lane| {12        const bits: u64 = @bitCast(value_lanes[lane]);13        inline for (0..8) |bit| {14            const index = checkedIndex(index_lanes[lane * 8 + bit]);15            std.debug.assert(index < 64);16            result[lane] |= ((bits >> @intCast(index)) & 1) << bit;17        }18    }19    return @bitCast(result);20}

1pub const bitShuffle = table.bitShuffle;