Enhance boolean simplification to handle 16-/64-bit RDRAND
[oota-llvm.git] / test / CodeGen / X86 / bool-simplify.ll
1 ; RUN: llc < %s -march=x86-64 -mattr=+sse41,-avx,+rdrand | FileCheck %s
2
3 define i32 @foo(<2 x i64> %c, i32 %a, i32 %b) {
4   %t1 = call i32 @llvm.x86.sse41.ptestz(<2 x i64> %c, <2 x i64> %c)
5   %t2 = icmp ne i32 %t1, 0
6   %t3 = select i1 %t2, i32 %a, i32 %b
7   ret i32 %t3
8 ; CHECK: foo
9 ; CHECK: ptest
10 ; CHECK-NOT: testl
11 ; CHECK: cmov
12 ; CHECK: ret
13 }
14
15 define i32 @bar(<2 x i64> %c) {
16 entry:
17   %0 = call i32 @llvm.x86.sse41.ptestz(<2 x i64> %c, <2 x i64> %c)
18   %1 = icmp ne i32 %0, 0
19   br i1 %1, label %if-true-block, label %endif-block
20 if-true-block:                                    ; preds = %entry
21   ret i32 0
22 endif-block:                                      ; preds = %entry,
23   ret i32 1
24 ; CHECK: bar
25 ; CHECK: ptest
26 ; CHECK-NOT: testl
27 ; CHECK: jne
28 ; CHECK: ret
29 }
30
31 define i32 @bax(<2 x i64> %c) {
32   %t1 = call i32 @llvm.x86.sse41.ptestz(<2 x i64> %c, <2 x i64> %c)
33   %t2 = icmp eq i32 %t1, 1
34   %t3 = zext i1 %t2 to i32
35   ret i32 %t3
36 ; CHECK: bax
37 ; CHECK: ptest
38 ; CHECK-NOT: cmpl
39 ; CHECK: ret
40 }
41
42 define i16 @rnd16(i16 %arg) nounwind uwtable {
43   %1 = tail call { i16, i32 } @llvm.x86.rdrand.16() nounwind
44   %2 = extractvalue { i16, i32 } %1, 0
45   %3 = extractvalue { i16, i32 } %1, 1
46   %4 = icmp eq i32 %3, 0
47   %5 = select i1 %4, i16 0, i16 %arg
48   %6 = add i16 %5, %2
49   ret i16 %6
50 ; CHECK: rnd16
51 ; CHECK: rdrand
52 ; CHECK: cmov
53 ; CHECK-NOT: cmov
54 ; CHECK: ret
55 }
56
57 define i32 @rnd32(i32 %arg) nounwind uwtable {
58   %1 = tail call { i32, i32 } @llvm.x86.rdrand.32() nounwind
59   %2 = extractvalue { i32, i32 } %1, 0
60   %3 = extractvalue { i32, i32 } %1, 1
61   %4 = icmp eq i32 %3, 0
62   %5 = select i1 %4, i32 0, i32 %arg
63   %6 = add i32 %5, %2
64   ret i32 %6
65 ; CHECK: rnd32
66 ; CHECK: rdrand
67 ; CHECK: cmov
68 ; CHECK-NOT: cmov
69 ; CHECK: ret
70 }
71
72 define i64 @rnd64(i64 %arg) nounwind uwtable {
73   %1 = tail call { i64, i32 } @llvm.x86.rdrand.64() nounwind
74   %2 = extractvalue { i64, i32 } %1, 0
75   %3 = extractvalue { i64, i32 } %1, 1
76   %4 = icmp eq i32 %3, 0
77   %5 = select i1 %4, i64 0, i64 %arg
78   %6 = add i64 %5, %2
79   ret i64 %6
80 ; CHECK: rnd64
81 ; CHECK: rdrand
82 ; CHECK: cmov
83 ; CHECK-NOT: cmov
84 ; CHECK: ret
85 }
86
87 declare i32 @llvm.x86.sse41.ptestz(<2 x i64>, <2 x i64>) nounwind readnone
88 declare { i16, i32 } @llvm.x86.rdrand.16() nounwind
89 declare { i32, i32 } @llvm.x86.rdrand.32() nounwind
90 declare { i64, i32 } @llvm.x86.rdrand.64() nounwind