mirrors/zig
1
0
Fork 0
mirror of https://codeberg.org/ziglang/zig.git synced 2025-12-06 13:54:21 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

std.math: Add splat for vectors of u0s in rotl/rotr (#24822)

Browse source
This commit is contained in:
Fri3dNstuff 2025-08-16 00:45:33 +03:00 • committed by GitHub
parent 39aca6f37e
commit f758b97bfd
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 10 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
lib/std/math.zig
View file

@ -693,7 +693,7 @@ test shr {
pub fn rotr(comptime T: type, x: T, r: anytype) T { pub fn rotr(comptime T: type, x: T, r: anytype) T {
if (@typeInfo(T) == .vector) { if (@typeInfo(T) == .vector) {
const C = @typeInfo(T).vector.child; const C = @typeInfo(T).vector.child;
if (C == u0) return 0; if (C == u0) return @splat(0);
if (@typeInfo(C).int.signedness == .signed) { if (@typeInfo(C).int.signedness == .signed) {
@compileError("cannot rotate signed integers"); @compileError("cannot rotate signed integers");
@ -725,8 +725,10 @@ test rotr {
try testing.expect(rotr(u8, 0b00000001, @as(usize, 4)) == 0b00010000); try testing.expect(rotr(u8, 0b00000001, @as(usize, 4)) == 0b00010000);
try testing.expect(rotr(u8, 0b00000001, @as(isize, -1)) == 0b00000010); try testing.expect(rotr(u8, 0b00000001, @as(isize, -1)) == 0b00000010);
try testing.expect(rotr(u12, 0o7777, 1) == 0o7777); try testing.expect(rotr(u12, 0o7777, 1) == 0o7777);
try testing.expect(rotr(@Vector(1, u32), @Vector(1, u32){1}, @as(usize, 1))[0] == @as(u32, 1) << 31); try testing.expect(rotr(@Vector(1, u32), .{1}, @as(usize, 1))[0] == @as(u32, 1) << 31);
try testing.expect(rotr(@Vector(1, u32), @Vector(1, u32){1}, @as(isize, -1))[0] == @as(u32, 1) << 1); try testing.expect(rotr(@Vector(1, u32), .{1}, @as(isize, -1))[0] == @as(u32, 1) << 1);
try std.testing.expect(@reduce(.And, rotr(@Vector(2, u0), .{ 0, 0 }, @as(usize, 42)) ==
@Vector(2, u0){ 0, 0 }));
} }
/// Rotates left. Only unsigned values can be rotated. Negative shift /// Rotates left. Only unsigned values can be rotated. Negative shift
@ -734,7 +736,7 @@ test rotr {
pub fn rotl(comptime T: type, x: T, r: anytype) T { pub fn rotl(comptime T: type, x: T, r: anytype) T {
if (@typeInfo(T) == .vector) { if (@typeInfo(T) == .vector) {
const C = @typeInfo(T).vector.child; const C = @typeInfo(T).vector.child;
if (C == u0) return 0; if (C == u0) return @splat(0);
if (@typeInfo(C).int.signedness == .signed) { if (@typeInfo(C).int.signedness == .signed) {
@compileError("cannot rotate signed integers"); @compileError("cannot rotate signed integers");
@ -766,8 +768,10 @@ test rotl {
try testing.expect(rotl(u8, 0b00000001, @as(usize, 4)) == 0b00010000); try testing.expect(rotl(u8, 0b00000001, @as(usize, 4)) == 0b00010000);
try testing.expect(rotl(u8, 0b00000001, @as(isize, -1)) == 0b10000000); try testing.expect(rotl(u8, 0b00000001, @as(isize, -1)) == 0b10000000);
try testing.expect(rotl(u12, 0o7777, 1) == 0o7777); try testing.expect(rotl(u12, 0o7777, 1) == 0o7777);
try testing.expect(rotl(@Vector(1, u32), @Vector(1, u32){1 << 31}, @as(usize, 1))[0] == 1); try testing.expect(rotl(@Vector(1, u32), .{1 << 31}, @as(usize, 1))[0] == 1);
try testing.expect(rotl(@Vector(1, u32), @Vector(1, u32){1 << 31}, @as(isize, -1))[0] == @as(u32, 1) << 30); try testing.expect(rotl(@Vector(1, u32), .{1 << 31}, @as(isize, -1))[0] == @as(u32, 1) << 30);
try std.testing.expect(@reduce(.And, rotl(@Vector(2, u0), .{ 0, 0 }, @as(usize, 42)) ==
@Vector(2, u0){ 0, 0 }));
} }
/// Returns an unsigned int type that can hold the number of bits in T - 1. /// Returns an unsigned int type that can hold the number of bits in T - 1.