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