tiny.simd.AlignedLayout

1pub fn Layout(comptime axes: usize) type {2    if (axes == 0) @compileError("aligned arrays require at least one axis");34    return struct {5        shape_value: [axes]usize,6        memory_shape_value: [axes]usize,7        sizes: [axes + 1]usize,8        memory_sizes: [axes + 1]usize,9        vector_bytes: usize,1011        const Self = @This();1213        pub fn init(shape_value: [axes]usize) Error!Self {14            return initFor(shape_value, native_vector_bytes);15        }1617        pub fn initFor(18            shape_value: [axes]usize,19            vector_bytes: usize,20        ) Error!Self {21            if (!std.math.isPowerOfTwo(vector_bytes)) return error.InvalidVectorBytes;22            for (shape_value) |dimension| {23                if (dimension == 0) return error.ZeroDimension;24            }25            var memory_shape_value = shape_value;26            memory_shape_value[axes - 1] = try roundUp(27                memory_shape_value[axes - 1],28                vector_bytes,29            );30            return .{31                .shape_value = shape_value,32                .memory_shape_value = memory_shape_value,33                .sizes = try computeSizes(axes, shape_value),34                .memory_sizes = try computeSizes(axes, memory_shape_value),35                .vector_bytes = vector_bytes,36            };37        }3839        pub fn shape(self: *const Self) [axes]usize {40            return self.shape_value;41        }4243        pub fn memoryShape(self: *const Self) [axes]usize {44            return self.memory_shape_value;45        }4647        pub fn len(self: *const Self) usize {48            return self.sizes[0];49        }5051        pub fn memoryLen(self: *const Self) usize {52            return self.memory_sizes[0];53        }5455        pub fn memoryBytes(self: *const Self, comptime T: type) Error!usize {56            return std.math.mul(usize, self.memoryLen(), @sizeOf(T)) catch57                error.AllocationSizeOverflow;58        }5960        pub fn rowLen(self: *const Self) usize {61            return self.shape_value[axes - 1];62        }6364        pub fn rowOffset(65            self: *const Self,66            indices: [axes - 1]usize,67        ) Error!usize {68            var offset: usize = 0;69            for (indices, 0..) |index, axis| {70                if (index >= self.shape_value[axis]) return error.IndexOutOfBounds;71                offset += self.memory_sizes[axis + 1] * index;72            }73            return offset;74        }7576        pub fn truncate(self: *Self, new_shape: [axes]usize) Error!void {77            for (new_shape, self.shape_value) |new_dimension, old_dimension| {78                if (new_dimension > old_dimension) return error.ShapeExpansion;79            }80            self.shape_value = new_shape;81            self.sizes = try computeSizes(axes, new_shape);82        }83    };84}

1pub const AlignedLayout = aligned.Layout;