1const std = @import("std");
2const builtin = @import("builtin");
3const mem = std.mem;
4const expect = std.testing.expect;
5
6test "@select vectors" {
7 if (builtin.zig_backend == .stage2_aarch64) return error.SkipZigTest;
8 if (builtin.zig_backend == .stage2_arm) return error.SkipZigTest; // TODO
9 if (builtin.zig_backend == .stage2_sparc64) return error.SkipZigTest; // TODO
10 if (builtin.zig_backend == .stage2_riscv64) return error.SkipZigTest;
11 if (builtin.zig_backend == .stage2_spirv) return error.SkipZigTest;
12
13 try comptime selectVectors();
14 try selectVectors();
15}
16
17fn selectVectors() !void {
18 var a = @Vector(4, bool){ true, false, true, false };
19 var b = @Vector(4, i32){ -1, 4, 999, -31 };
20 var c = @Vector(4, i32){ -5, 1, 0, 1234 };
21 _ = .{ &a, &b, &c };
22 const abc = @select(i32, a, b, c);
23 try expect(abc[0] == -1);
24 try expect(abc[1] == 1);
25 try expect(abc[2] == 999);
26 try expect(abc[3] == 1234);
27
28 var x = @Vector(4, bool){ false, false, false, true };
29 var y = @Vector(4, f32){ 0.001, 33.4, 836, -3381.233 };
30 var z = @Vector(4, f32){ 0.0, 312.1, -145.9, 9993.55 };
31 _ = .{ &x, &y, &z };
32 const xyz = @select(f32, x, y, z);
33 try expect(mem.eql(f32, &@as([4]f32, xyz), &[4]f32{ 0.0, 312.1, -145.9, -3381.233 }));
34
35 var vec_u0: @Vector(4, u0) = @splat(0);
36 var mask_u0 = @Vector(4, bool){ true, false, true, false };
37 var mask_empty = @Vector(0, i32){};
38 var vec_empty = @Vector(0, i32){};
39 _ = .{ &vec_u0, &mask_u0, &mask_empty, &vec_empty };
40 const sel_u0 = @select(u0, mask_u0, vec_u0, vec_u0);
41 const sel_u0_undefined = @select(u0, mask_u0, undefined, undefined);
42 comptime if (sel_u0[0] != 0) unreachable;
43 comptime if (sel_u0_undefined[1] != 0) unreachable;
44 const sel_empty = @select(i32, mask_empty, vec_empty, vec_empty);
45 const sel_empty_undefined = @select(i32, @Vector(0, bool){}, undefined, undefined);
46 comptime if (@as(u0, @bitCast(sel_empty)) != 0) unreachable;
47 comptime if (@as(u0, @bitCast(sel_empty_undefined)) != 0) unreachable;
48}
49
50test "@select arrays" {
51 if (builtin.zig_backend == .stage2_aarch64) return error.SkipZigTest;
52 if (builtin.zig_backend == .stage2_arm) return error.SkipZigTest; // TODO
53 if (builtin.zig_backend == .stage2_sparc64) return error.SkipZigTest; // TODO
54 if (builtin.zig_backend == .stage2_riscv64) return error.SkipZigTest;
55 if (builtin.zig_backend == .stage2_spirv) return error.SkipZigTest;
56
57 try comptime selectArrays();
58 try selectArrays();
59}
60
61fn selectArrays() !void {
62 var a = [4]bool{ false, true, false, true };
63 var b = [4]usize{ 0, 1, 2, 3 };
64 var c = [4]usize{ 4, 5, 6, 7 };
65 _ = .{ &a, &b, &c };
66 const abc = @select(usize, a, b, c);
67 try expect(abc[0] == 4);
68 try expect(abc[1] == 1);
69 try expect(abc[2] == 6);
70 try expect(abc[3] == 3);
71
72 var x = [4]bool{ false, false, false, true };
73 var y = [4]f32{ 0.001, 33.4, 836, -3381.233 };
74 var z = [4]f32{ 0.0, 312.1, -145.9, 9993.55 };
75 _ = .{ &x, &y, &z };
76 const xyz = @select(f32, x, y, z);
77 try expect(mem.eql(f32, &@as([4]f32, xyz), &[4]f32{ 0.0, 312.1, -145.9, -3381.233 }));
78}
79
80test "@select compare result" {
81 if (builtin.zig_backend == .stage2_aarch64) return error.SkipZigTest;
82 if (builtin.zig_backend == .stage2_riscv64) return error.SkipZigTest;
83 if (builtin.zig_backend == .stage2_llvm and builtin.cpu.arch == .hexagon) return error.SkipZigTest;
84
85 const S = struct {
86 fn min(comptime V: type, lhs: V, rhs: V) V {
87 return @select(@typeInfo(V).vector.child, lhs < rhs, lhs, rhs);
88 }
89
90 fn doTheTest() !void {
91 try expect(@reduce(.And, min(@Vector(4, f32), .{ -1, 2, -3, 4 }, .{ 1, -2, 3, -4 }) == @Vector(4, f32){ -1, -2, -3, -4 }));
92 try expect(@reduce(.And, min(@Vector(2, f64), .{ -1, 2 }, .{ 1, -2 }) == @Vector(2, f64){ -1, -2 }));
93 }
94 };
95
96 try S.doTheTest();
97 try comptime S.doTheTest();
98}