authorgravatar for 102751849+Fri3dNstuff@users.noreply.github.comFri3dNstuff <102751849+Fri3dNstuff@users.noreply.github.com> 2025-08-16 00:45:33+03:00
committergravatar for noreply@github.comGitHub <noreply@github.com> 2025-08-15 23:45:33+02:00
logf758b97bfd5e28e4ff1c8d9eca34de9b22990f85
tree08a72700ec77de33452b1c8e10be7b2297e7d458
parent39aca6f37e83e263236339f9e67bace2a098cd16
signaturebadge-check Signed by PGP key B5690EEEBB952194

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

1 files changed, 10 insertions(+), 6 deletions(-)

lib/std/math.zig+10-6
......@@ -693,7 +693,7 @@ test shr {
693693pub fn rotr(comptime T: type, x: T, r: anytype) T {
694694 if (@typeInfo(T) == .vector) {
695695 const C = @typeInfo(T).vector.child;
696 if (C == u0) return 0;
696 if (C == u0) return @splat(0);
697697
698698 if (@typeInfo(C).int.signedness == .signed) {
699699 @compileError("cannot rotate signed integers");
......@@ -725,8 +725,10 @@ test rotr {
725725 try testing.expect(rotr(u8, 0b00000001, @as(usize, 4)) == 0b00010000);
726726 try testing.expect(rotr(u8, 0b00000001, @as(isize, -1)) == 0b00000010);
727727 try testing.expect(rotr(u12, 0o7777, 1) == 0o7777);
728 try testing.expect(rotr(@Vector(1, u32), @Vector(1, u32){1}, @as(usize, 1))[0] == @as(u32, 1) << 31);
729 try testing.expect(rotr(@Vector(1, u32), @Vector(1, u32){1}, @as(isize, -1))[0] == @as(u32, 1) << 1);
728 try testing.expect(rotr(@Vector(1, u32), .{1}, @as(usize, 1))[0] == @as(u32, 1) << 31);
729 try testing.expect(rotr(@Vector(1, u32), .{1}, @as(isize, -1))[0] == @as(u32, 1) << 1);
730 try std.testing.expect(@reduce(.And, rotr(@Vector(2, u0), .{ 0, 0 }, @as(usize, 42)) ==
731 @Vector(2, u0){ 0, 0 }));
730732}
731733
732734/// Rotates left. Only unsigned values can be rotated. Negative shift
......@@ -734,7 +736,7 @@ test rotr {
734736pub fn rotl(comptime T: type, x: T, r: anytype) T {
735737 if (@typeInfo(T) == .vector) {
736738 const C = @typeInfo(T).vector.child;
737 if (C == u0) return 0;
739 if (C == u0) return @splat(0);
738740
739741 if (@typeInfo(C).int.signedness == .signed) {
740742 @compileError("cannot rotate signed integers");
......@@ -766,8 +768,10 @@ test rotl {
766768 try testing.expect(rotl(u8, 0b00000001, @as(usize, 4)) == 0b00010000);
767769 try testing.expect(rotl(u8, 0b00000001, @as(isize, -1)) == 0b10000000);
768770 try testing.expect(rotl(u12, 0o7777, 1) == 0o7777);
769 try testing.expect(rotl(@Vector(1, u32), @Vector(1, u32){1 << 31}, @as(usize, 1))[0] == 1);
770 try testing.expect(rotl(@Vector(1, u32), @Vector(1, u32){1 << 31}, @as(isize, -1))[0] == @as(u32, 1) << 30);
771 try testing.expect(rotl(@Vector(1, u32), .{1 << 31}, @as(usize, 1))[0] == 1);
772 try testing.expect(rotl(@Vector(1, u32), .{1 << 31}, @as(isize, -1))[0] == @as(u32, 1) << 30);
773 try std.testing.expect(@reduce(.And, rotl(@Vector(2, u0), .{ 0, 0 }, @as(usize, 42)) ==
774 @Vector(2, u0){ 0, 0 }));
771775}
772776
773777/// Returns an unsigned int type that can hold the number of bits in T - 1.