std/math.zig: resolve missed optimization in rotates

This commit is contained in:
Rekai Musuka 2022-04-14 07:42:49 -03:00 committed by Andrew Kelley
parent 52c8ac1a84
commit 02a43f325b

View File

@ -548,8 +548,8 @@ pub fn rotr(comptime T: type, x: T, r: anytype) T {
} else if (@typeInfo(T).Int.signedness == .signed) {
@compileError("cannot rotate signed integer");
} else {
const ar = @mod(r, @typeInfo(T).Int.bits);
return shr(T, x, ar) | shl(T, x, @typeInfo(T).Int.bits - ar);
const ar = @intCast(Log2Int(T), @mod(r, @typeInfo(T).Int.bits));
return x >> ar | x << (1 +% ~ar);
}
}
@ -576,8 +576,8 @@ pub fn rotl(comptime T: type, x: T, r: anytype) T {
} else if (@typeInfo(T).Int.signedness == .signed) {
@compileError("cannot rotate signed integer");
} else {
const ar = @mod(r, @typeInfo(T).Int.bits);
return shl(T, x, ar) | shr(T, x, @typeInfo(T).Int.bits - ar);
const ar = @intCast(Log2Int(T), @mod(r, @typeInfo(T).Int.bits));
return x << ar | x >> 1 +% ~ar;
}
}