tiny.simd.UnrollerUnit

1pub fn UnrollerUnit(comptime InputLane: type, comptime OutputLane: type) type {2    validateLane(InputLane);3    validateLane(OutputLane);4    const LargerLane = signedLane(@max(@sizeOf(InputLane), @sizeOf(OutputLane)));5    const LargerTag = tag.ScalableTag(LargerLane);67    return struct {8        pub const input_count: usize = 1;9        pub const Input: type = InputLane;10        pub const Output: type = OutputLane;11        pub const Larger: type = LargerLane;12        pub const Tag: type = LargerTag;13        pub const InputTag: type = LargerTag.rebind(InputLane);14        pub const OutputTag: type = LargerTag.rebind(OutputLane);15        pub const InputVector: type = InputTag.Vector;16        pub const OutputVector: type = OutputTag.Vector;17        pub const XVector: type = InputVector;18        pub const YVector: type = OutputVector;19        pub const max_unit_lanes: usize = LargerTag.lane_count;20        pub const actual_lanes: usize = LargerTag.lane_count;2122        pub fn maxUnitLanes() usize {23            return max_unit_lanes;24        }2526        pub fn actualLanes() usize {27            return actual_lanes;28        }2930        pub fn x0Init() InputVector {31            return @splat(0);32        }3334        pub fn yInit() OutputVector {35            return @splat(0);36        }3738        pub fn load(index: isize, input: []const Input) InputVector {39            return loadFull(InputTag, index, input);40        }4142        pub fn maskLoad(index: isize, input: []const Input, places: isize) InputVector {43            return maskLoadOr(@splat(0), index, input, places);44        }4546        pub fn maskLoadOr(47            inactive: InputVector,48            index: isize,49            input: []const Input,50            places: isize,51        ) InputVector {52            return loadPartial(InputTag, inactive, index, input, places);53        }5455        pub fn storeAndShortCircuit(56            index: isize,57            output: []Output,58            value: OutputVector,59        ) bool {60            storeFull(OutputTag, index, output, value);61            return true;62        }6364        pub fn maskStore(65            index: isize,66            output: []Output,67            value: OutputVector,68            places: isize,69        ) usize {70            return storePartial(OutputTag, index, output, value, places);71        }7273        pub fn reduceFinal(_: OutputVector, _: []Output) usize {74            return 0;75        }7677        pub fn reduceUnrolled(78            _: OutputVector,79            _: OutputVector,80            _: OutputVector,81            value: OutputVector,82        ) OutputVector {83            return value;84        }85    };86}

1pub const UnrollerUnit = unroller.UnrollerUnit;