1; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
2; RUN: llc < %s -mattr=+avx512f | FileCheck %s  --check-prefix=CHECK --check-prefix=KNL
3; RUN: llc < %s -mattr=+avx512f,+avx512vl,+avx512bw,+avx512dq | FileCheck %s  --check-prefix=CHECK --check-prefix=SKX
4
5target datalayout = "e-m:e-i64:64-f80:128-n8:16:32:64-S128"
6target triple = "x86_64-unknown-linux-gnu"
7
8; Note that the kmovs should really *not* appear in the output, this is an
9; artifact of the current poor lowering. This is tracked by PR28175.
10
11define i64 @foo64(i1 zeroext %i) #0 {
12; CHECK-LABEL: foo64:
13; CHECK:       # BB#0:
14; CHECK-NEXT:    # kill: %EDI<def> %EDI<kill> %RDI<def>
15; CHECK-NEXT:    orq $-2, %rdi
16; CHECK-NEXT:    movq %rdi, %rax
17; CHECK-NEXT:    retq
18  br label %bb
19
20bb:
21  %z = zext i1 %i to i64
22  %v = or i64 %z, -2
23  br label %end
24
25end:
26  ret i64 %v
27}
28
29define i16 @foo16(i1 zeroext %i) #0 {
30; CHECK-LABEL: foo16:
31; CHECK:       # BB#0:
32; CHECK-NEXT:    orl $65534, %edi # imm = 0xFFFE
33; CHECK-NEXT:    movl %edi, %eax
34; CHECK-NEXT:    retq
35  br label %bb
36
37bb:
38  %z = zext i1 %i to i16
39  %v = or i16 %z, -2
40  br label %end
41
42end:
43  ret i16 %v
44}
45
46; This code is still not optimal
47define i16 @foo16_1(i1 zeroext %i, i32 %j) #0 {
48; KNL-LABEL: foo16_1:
49; KNL:       # BB#0:
50; KNL-NEXT:    kmovw %edi, %k0
51; KNL-NEXT:    kmovw %k0, %eax
52; KNL-NEXT:    andl $1, %eax
53; KNL-NEXT:    orl $2, %eax
54; KNL-NEXT:    # kill: %AX<def> %AX<kill> %EAX<kill>
55; KNL-NEXT:    retq
56;
57; SKX-LABEL: foo16_1:
58; SKX:       # BB#0:
59; SKX-NEXT:    kmovd %edi, %k0
60; SKX-NEXT:    kmovw %k0, %eax
61; SKX-NEXT:    andl $1, %eax
62; SKX-NEXT:    orl $2, %eax
63; SKX-NEXT:    # kill: %AX<def> %AX<kill> %EAX<kill>
64; SKX-NEXT:    retq
65  br label %bb
66
67bb:
68  %z = zext i1 %i to i16
69  %v = or i16 %z, 2
70  br label %end
71
72end:
73  ret i16 %v
74}
75
76define i32 @foo32(i1 zeroext %i) #0 {
77; CHECK-LABEL: foo32:
78; CHECK:       # BB#0:
79; CHECK-NEXT:    orl $-2, %edi
80; CHECK-NEXT:    movl %edi, %eax
81; CHECK-NEXT:    retq
82  br label %bb
83
84bb:
85  %z = zext i1 %i to i32
86  %v = or i32 %z, -2
87  br label %end
88
89end:
90  ret i32 %v
91}
92
93define i8 @foo8(i1 zeroext %i) #0 {
94; CHECK-LABEL: foo8:
95; CHECK:       # BB#0:
96; CHECK-NEXT:    orb $-2, %dil
97; CHECK-NEXT:    movl %edi, %eax
98; CHECK-NEXT:    retq
99  br label %bb
100
101bb:
102  %z = zext i1 %i to i8
103  %v = or i8 %z, -2
104  br label %end
105
106end:
107  ret i8 %v
108}
109
110