From 079f62881ee8291e1b290848c1081c822ea8389f Mon Sep 17 00:00:00 2001 From: Andrew Kelley Date: Wed, 4 Jan 2023 18:27:52 -0700 Subject: [PATCH] std.simd.iota: make it always called at comptime There's no reason for this to ever run at runtime; it should always be used to generate a constant. --- lib/std/simd.zig | 20 +++++++++++--------- 1 file changed, 11 insertions(+), 9 deletions(-) diff --git a/lib/std/simd.zig b/lib/std/simd.zig index 95de3cc11c..868f9864e7 100644 --- a/lib/std/simd.zig +++ b/lib/std/simd.zig @@ -86,16 +86,18 @@ pub fn VectorCount(comptime VectorType: type) type { /// Returns a vector containing the first `len` integers in order from 0 to `len`-1. /// For example, `iota(i32, 8)` will return a vector containing `.{0, 1, 2, 3, 4, 5, 6, 7}`. -pub fn iota(comptime T: type, comptime len: usize) @Vector(len, T) { - var out: [len]T = undefined; - for (out) |*element, i| { - element.* = switch (@typeInfo(T)) { - .Int => @intCast(T, i), - .Float => @intToFloat(T, i), - else => @compileError("Can't use type " ++ @typeName(T) ++ " in iota."), - }; +pub inline fn iota(comptime T: type, comptime len: usize) @Vector(len, T) { + comptime { + var out: [len]T = undefined; + for (out) |*element, i| { + element.* = switch (@typeInfo(T)) { + .Int => @intCast(T, i), + .Float => @intToFloat(T, i), + else => @compileError("Can't use type " ++ @typeName(T) ++ " in iota."), + }; + } + return @as(@Vector(len, T), out); } - return @as(@Vector(len, T), out); } /// Returns a vector containing the same elements as the input, but repeated until the desired length is reached.