tiny.simd.AlignedArray

1pub fn Array(comptime T: type, comptime axes: usize) type {2    return struct {3        layout: Layout(axes),4        allocation: Allocation(T),56        const Self = @This();78        pub fn init(9            allocator: std.mem.Allocator,10            shape_value: [axes]usize,11        ) Error!Self {12            return initFor(allocator, shape_value, native_vector_bytes);13        }1415        pub fn initFor(16            allocator: std.mem.Allocator,17            shape_value: [axes]usize,18            vector_bytes: usize,19        ) Error!Self {20            const layout = try Layout(axes).initFor(shape_value, vector_bytes);21            _ = try layout.memoryBytes(T);22            const allocation = try Allocation(T).init(allocator, layout.memoryLen());23            @memset(allocation.values, std.mem.zeroes(T));24            return .{25                .layout = layout,26                .allocation = allocation,27            };28        }2930        pub fn deinit(self: *Self, allocator: std.mem.Allocator) void {31            self.allocation.deinit(allocator);32            self.* = undefined;33        }3435        pub fn row(self: *Self, indices: [axes - 1]usize) Error![]T {36            const offset = try self.layout.rowOffset(indices);37            return self.allocation.values[offset..][0..self.layout.rowLen()];38        }3940        pub fn constRow(41            self: *const Self,42            indices: [axes - 1]usize,43        ) Error![]const T {44            const offset = try self.layout.rowOffset(indices);45            return self.allocation.values[offset..][0..self.layout.rowLen()];46        }4748        pub fn shape(self: *const Self) [axes]usize {49            return self.layout.shape();50        }5152        pub fn memoryShape(self: *const Self) [axes]usize {53            return self.layout.memoryShape();54        }5556        pub fn len(self: *const Self) usize {57            return self.layout.len();58        }5960        pub fn memoryLen(self: *const Self) usize {61            return self.layout.memoryLen();62        }6364        pub fn data(self: *Self) []T {65            return self.allocation.values;66        }6768        pub fn constData(self: *const Self) []const T {69            return self.allocation.values;70        }7172        pub fn truncate(self: *Self, new_shape: [axes]usize) Error!void {73            try self.layout.truncate(new_shape);74        }75    };76}

1pub const AlignedArray = aligned.Array;