xref: /llvm-project/llvm/test/Transforms/InstCombine/cast-mul-select.ll (revision 9a844a36eb9a21de27882b6193a82fda49986347)
1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
2; RUN: opt < %s -passes=instcombine -S | FileCheck %s
3; RUN: opt -passes=debugify,instcombine -S < %s | FileCheck %s -check-prefix DBGINFO
4
5; RUN: opt -passes=debugify,instcombine -S < %s --try-experimental-debuginfo-iterators | FileCheck %s -check-prefix DBGINFO
6
7target datalayout = "e-p:64:64:64-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:64:64-f32:32:32-f64:64:64-v64:64:64-v128:128:128-a0:0:64-s0:64:64-f80:128:128-n8:16:32"
8
9define i32 @mul(i32 %x, i32 %y) {
10; CHECK-LABEL: @mul(
11; CHECK-NEXT:    [[C:%.*]] = mul i32 [[X:%.*]], [[Y:%.*]]
12; CHECK-NEXT:    [[D:%.*]] = and i32 [[C]], 255
13; CHECK-NEXT:    ret i32 [[D]]
14;
15; DBGINFO-LABEL: @mul(
16; DBGINFO-NEXT:      #dbg_value(i32 [[X:%.*]], [[META9:![0-9]+]], !DIExpression(DW_OP_LLVM_convert, 32, DW_ATE_unsigned, DW_OP_LLVM_convert, 8, DW_ATE_unsigned, DW_OP_stack_value), [[META15:![0-9]+]])
17; DBGINFO-NEXT:      #dbg_value(i32 [[Y:%.*]], [[META11:![0-9]+]], !DIExpression(DW_OP_LLVM_convert, 32, DW_ATE_unsigned, DW_OP_LLVM_convert, 8, DW_ATE_unsigned, DW_OP_stack_value), [[META16:![0-9]+]])
18; DBGINFO-NEXT:    [[C:%.*]] = mul i32 [[X]], [[Y]], !dbg [[DBG17:![0-9]+]]
19; DBGINFO-NEXT:    [[D:%.*]] = and i32 [[C]], 255, !dbg [[DBG18:![0-9]+]]
20; DBGINFO-NEXT:      #dbg_value(i32 [[C]], [[META12:![0-9]+]], !DIExpression(), [[DBG17]])
21; DBGINFO-NEXT:      #dbg_value(i32 [[D]], [[META13:![0-9]+]], !DIExpression(), [[DBG18]])
22; DBGINFO-NEXT:    ret i32 [[D]], !dbg [[DBG19:![0-9]+]]
23;
24
25; Test that when zext is evaluated in different type
26; we preserve the debug information in the resulting
27; instruction.
28
29  %A = trunc i32 %x to i8
30  %B = trunc i32 %y to i8
31  %C = mul i8 %A, %B
32  %D = zext i8 %C to i32
33  ret i32 %D
34}
35
36define i32 @select1(i1 %cond, i32 %x, i32 %y, i32 %z) {
37; CHECK-LABEL: @select1(
38; CHECK-NEXT:    [[D:%.*]] = add i32 [[X:%.*]], [[Y:%.*]]
39; CHECK-NEXT:    [[E:%.*]] = select i1 [[COND:%.*]], i32 [[Z:%.*]], i32 [[D]]
40; CHECK-NEXT:    [[F:%.*]] = and i32 [[E]], 255
41; CHECK-NEXT:    ret i32 [[F]]
42;
43; DBGINFO-LABEL: @select1(
44; DBGINFO-NEXT:      #dbg_value(i32 [[X:%.*]], [[META22:![0-9]+]], !DIExpression(DW_OP_LLVM_convert, 32, DW_ATE_unsigned, DW_OP_LLVM_convert, 8, DW_ATE_unsigned, DW_OP_stack_value), [[META28:![0-9]+]])
45; DBGINFO-NEXT:      #dbg_value(i32 [[Y:%.*]], [[META23:![0-9]+]], !DIExpression(DW_OP_LLVM_convert, 32, DW_ATE_unsigned, DW_OP_LLVM_convert, 8, DW_ATE_unsigned, DW_OP_stack_value), [[META29:![0-9]+]])
46; DBGINFO-NEXT:      #dbg_value(i32 [[Z:%.*]], [[META24:![0-9]+]], !DIExpression(DW_OP_LLVM_convert, 32, DW_ATE_unsigned, DW_OP_LLVM_convert, 8, DW_ATE_unsigned, DW_OP_stack_value), [[META30:![0-9]+]])
47; DBGINFO-NEXT:    [[D:%.*]] = add i32 [[X]], [[Y]], !dbg [[DBG31:![0-9]+]]
48; DBGINFO-NEXT:      #dbg_value(!DIArgList(i32 [[X]], i32 [[Y]]), [[META25:![0-9]+]], !DIExpression(DW_OP_LLVM_arg, 0, DW_OP_LLVM_convert, 32, DW_ATE_unsigned, DW_OP_LLVM_convert, 8, DW_ATE_unsigned, DW_OP_LLVM_arg, 1, DW_OP_LLVM_convert, 32, DW_ATE_unsigned, DW_OP_LLVM_convert, 8, DW_ATE_unsigned, DW_OP_plus, DW_OP_stack_value), [[DBG31]])
49; DBGINFO-NEXT:    [[E:%.*]] = select i1 [[COND:%.*]], i32 [[Z]], i32 [[D]], !dbg [[DBG32:![0-9]+]]
50; DBGINFO-NEXT:    [[F:%.*]] = and i32 [[E]], 255, !dbg [[DBG33:![0-9]+]]
51; DBGINFO-NEXT:      #dbg_value(i32 [[E]], [[META26:![0-9]+]], !DIExpression(), [[DBG32]])
52; DBGINFO-NEXT:      #dbg_value(i32 [[F]], [[META27:![0-9]+]], !DIExpression(), [[DBG33]])
53; DBGINFO-NEXT:    ret i32 [[F]], !dbg [[DBG34:![0-9]+]]
54;
55  %A = trunc i32 %x to i8
56  %B = trunc i32 %y to i8
57  %C = trunc i32 %z to i8
58  %D = add i8 %A, %B
59  %E = select i1 %cond, i8 %C, i8 %D
60  %F = zext i8 %E to i32
61  ret i32 %F
62}
63
64define i8 @select2(i1 %cond, i8 %x, i8 %y, i8 %z) {
65; CHECK-LABEL: @select2(
66; CHECK-NEXT:    [[D:%.*]] = add i8 [[X:%.*]], [[Y:%.*]]
67; CHECK-NEXT:    [[E:%.*]] = select i1 [[COND:%.*]], i8 [[Z:%.*]], i8 [[D]]
68; CHECK-NEXT:    ret i8 [[E]]
69;
70; DBGINFO-LABEL: @select2(
71; DBGINFO-NEXT:      #dbg_value(i8 [[X:%.*]], [[META37:![0-9]+]], !DIExpression(DW_OP_LLVM_convert, 8, DW_ATE_unsigned, DW_OP_LLVM_convert, 32, DW_ATE_unsigned, DW_OP_stack_value), [[META43:![0-9]+]])
72; DBGINFO-NEXT:      #dbg_value(i8 [[Y:%.*]], [[META38:![0-9]+]], !DIExpression(DW_OP_LLVM_convert, 8, DW_ATE_unsigned, DW_OP_LLVM_convert, 32, DW_ATE_unsigned, DW_OP_stack_value), [[META44:![0-9]+]])
73; DBGINFO-NEXT:      #dbg_value(i8 [[Z:%.*]], [[META39:![0-9]+]], !DIExpression(DW_OP_LLVM_convert, 8, DW_ATE_unsigned, DW_OP_LLVM_convert, 32, DW_ATE_unsigned, DW_OP_stack_value), [[META45:![0-9]+]])
74; DBGINFO-NEXT:    [[D:%.*]] = add i8 [[X]], [[Y]], !dbg [[DBG46:![0-9]+]]
75; DBGINFO-NEXT:      #dbg_value(!DIArgList(i8 [[X]], i8 [[Y]]), [[META40:![0-9]+]], !DIExpression(DW_OP_LLVM_arg, 0, DW_OP_LLVM_convert, 8, DW_ATE_unsigned, DW_OP_LLVM_convert, 32, DW_ATE_unsigned, DW_OP_LLVM_arg, 1, DW_OP_LLVM_convert, 8, DW_ATE_unsigned, DW_OP_LLVM_convert, 32, DW_ATE_unsigned, DW_OP_plus, DW_OP_stack_value), [[DBG46]])
76; DBGINFO-NEXT:      #dbg_value(i32 poison, [[META41:![0-9]+]], !DIExpression(), [[META47:![0-9]+]])
77; DBGINFO-NEXT:    [[F:%.*]] = select i1 [[COND:%.*]], i8 [[Z]], i8 [[D]], !dbg [[META47]]
78; DBGINFO-NEXT:      #dbg_value(i8 [[F]], [[META42:![0-9]+]], !DIExpression(), [[META48:![0-9]+]])
79; DBGINFO-NEXT:    ret i8 [[F]], !dbg [[DBG49:![0-9]+]]
80;
81  %A = zext i8 %x to i32
82  %B = zext i8 %y to i32
83  %C = zext i8 %z to i32
84  %D = add i32 %A, %B
85  %E = select i1 %cond, i32 %C, i32 %D
86  %F = trunc i32 %E to i8
87  ret i8 %F
88}
89
90; The next 3 tests could be handled in instcombine, but evaluating values
91; with multiple uses may be very slow. Let some other pass deal with it.
92
93define i32 @eval_trunc_multi_use_in_one_inst(i32 %x) {
94; CHECK-LABEL: @eval_trunc_multi_use_in_one_inst(
95; CHECK-NEXT:    [[Z:%.*]] = zext i32 [[X:%.*]] to i64
96; CHECK-NEXT:    [[A:%.*]] = add nuw nsw i64 [[Z]], 15
97; CHECK-NEXT:    [[M:%.*]] = mul i64 [[A]], [[A]]
98; CHECK-NEXT:    [[T:%.*]] = trunc i64 [[M]] to i32
99; CHECK-NEXT:    ret i32 [[T]]
100;
101; DBGINFO-LABEL: @eval_trunc_multi_use_in_one_inst(
102; DBGINFO-NEXT:    [[Z:%.*]] = zext i32 [[X:%.*]] to i64, !dbg [[DBG57:![0-9]+]]
103; DBGINFO-NEXT:      #dbg_value(i64 [[Z]], [[META52:![0-9]+]], !DIExpression(), [[DBG57]])
104; DBGINFO-NEXT:    [[A:%.*]] = add nuw nsw i64 [[Z]], 15, !dbg [[DBG58:![0-9]+]]
105; DBGINFO-NEXT:      #dbg_value(i64 [[A]], [[META54:![0-9]+]], !DIExpression(), [[DBG58]])
106; DBGINFO-NEXT:    [[M:%.*]] = mul i64 [[A]], [[A]], !dbg [[DBG59:![0-9]+]]
107; DBGINFO-NEXT:      #dbg_value(i64 [[M]], [[META55:![0-9]+]], !DIExpression(), [[DBG59]])
108; DBGINFO-NEXT:    [[T:%.*]] = trunc i64 [[M]] to i32, !dbg [[DBG60:![0-9]+]]
109; DBGINFO-NEXT:      #dbg_value(i32 [[T]], [[META56:![0-9]+]], !DIExpression(), [[DBG60]])
110; DBGINFO-NEXT:    ret i32 [[T]], !dbg [[DBG61:![0-9]+]]
111;
112  %z = zext i32 %x to i64
113  %a = add nsw nuw i64 %z, 15
114  %m = mul i64 %a, %a
115  %t = trunc i64 %m to i32
116  ret i32 %t
117}
118
119define i32 @eval_zext_multi_use_in_one_inst(i32 %x) {
120; CHECK-LABEL: @eval_zext_multi_use_in_one_inst(
121; CHECK-NEXT:    [[T:%.*]] = trunc i32 [[X:%.*]] to i16
122; CHECK-NEXT:    [[A:%.*]] = and i16 [[T]], 5
123; CHECK-NEXT:    [[M:%.*]] = mul nuw nsw i16 [[A]], [[A]]
124; CHECK-NEXT:    [[R:%.*]] = zext nneg i16 [[M]] to i32
125; CHECK-NEXT:    ret i32 [[R]]
126;
127; DBGINFO-LABEL: @eval_zext_multi_use_in_one_inst(
128; DBGINFO-NEXT:    [[T:%.*]] = trunc i32 [[X:%.*]] to i16, !dbg [[DBG69:![0-9]+]]
129; DBGINFO-NEXT:      #dbg_value(i16 [[T]], [[META64:![0-9]+]], !DIExpression(), [[DBG69]])
130; DBGINFO-NEXT:    [[A:%.*]] = and i16 [[T]], 5, !dbg [[DBG70:![0-9]+]]
131; DBGINFO-NEXT:      #dbg_value(i16 [[A]], [[META66:![0-9]+]], !DIExpression(), [[DBG70]])
132; DBGINFO-NEXT:    [[M:%.*]] = mul nuw nsw i16 [[A]], [[A]], !dbg [[DBG71:![0-9]+]]
133; DBGINFO-NEXT:      #dbg_value(i16 [[M]], [[META67:![0-9]+]], !DIExpression(), [[DBG71]])
134; DBGINFO-NEXT:    [[R:%.*]] = zext nneg i16 [[M]] to i32, !dbg [[DBG72:![0-9]+]]
135; DBGINFO-NEXT:      #dbg_value(i32 [[R]], [[META68:![0-9]+]], !DIExpression(), [[DBG72]])
136; DBGINFO-NEXT:    ret i32 [[R]], !dbg [[DBG73:![0-9]+]]
137;
138  %t = trunc i32 %x to i16
139  %a = and i16 %t, 5
140  %m = mul nuw nsw i16 %a, %a
141  %r = zext i16 %m to i32
142  ret i32 %r
143}
144
145define i32 @eval_sext_multi_use_in_one_inst(i32 %x) {
146; CHECK-LABEL: @eval_sext_multi_use_in_one_inst(
147; CHECK-NEXT:    [[T:%.*]] = trunc i32 [[X:%.*]] to i16
148; CHECK-NEXT:    [[A:%.*]] = and i16 [[T]], 14
149; CHECK-NEXT:    [[M:%.*]] = mul nuw nsw i16 [[A]], [[A]]
150; CHECK-NEXT:    [[O:%.*]] = or disjoint i16 [[M]], -32768
151; CHECK-NEXT:    [[R:%.*]] = sext i16 [[O]] to i32
152; CHECK-NEXT:    ret i32 [[R]]
153;
154; DBGINFO-LABEL: @eval_sext_multi_use_in_one_inst(
155; DBGINFO-NEXT:    [[T:%.*]] = trunc i32 [[X:%.*]] to i16, !dbg [[DBG81:![0-9]+]]
156; DBGINFO-NEXT:      #dbg_value(i16 [[T]], [[META76:![0-9]+]], !DIExpression(), [[DBG81]])
157; DBGINFO-NEXT:    [[A:%.*]] = and i16 [[T]], 14, !dbg [[DBG82:![0-9]+]]
158; DBGINFO-NEXT:      #dbg_value(i16 [[A]], [[META77:![0-9]+]], !DIExpression(), [[DBG82]])
159; DBGINFO-NEXT:    [[M:%.*]] = mul nuw nsw i16 [[A]], [[A]], !dbg [[DBG83:![0-9]+]]
160; DBGINFO-NEXT:      #dbg_value(i16 [[M]], [[META78:![0-9]+]], !DIExpression(), [[DBG83]])
161; DBGINFO-NEXT:    [[O:%.*]] = or disjoint i16 [[M]], -32768, !dbg [[DBG84:![0-9]+]]
162; DBGINFO-NEXT:      #dbg_value(i16 [[O]], [[META79:![0-9]+]], !DIExpression(), [[DBG84]])
163; DBGINFO-NEXT:    [[R:%.*]] = sext i16 [[O]] to i32, !dbg [[DBG85:![0-9]+]]
164; DBGINFO-NEXT:      #dbg_value(i32 [[R]], [[META80:![0-9]+]], !DIExpression(), [[DBG85]])
165; DBGINFO-NEXT:    ret i32 [[R]], !dbg [[DBG86:![0-9]+]]
166;
167  %t = trunc i32 %x to i16
168  %a = and i16 %t, 14
169  %m = mul nuw nsw i16 %a, %a
170  %o = or i16 %m, 32768
171  %r = sext i16 %o to i32
172  ret i32 %r
173}
174
175; If we have a transform to shrink the above 3 cases, make sure it's not
176; also trying to look through multiple uses in this test and crashing.
177
178define void @PR36225(i32 %a, i32 %b, i1 %c1, i3 %v1, i3 %v2) {
179; CHECK-LABEL: @PR36225(
180; CHECK-NEXT:  entry:
181; CHECK-NEXT:    br label [[WHILE_BODY:%.*]]
182; CHECK:       while.body:
183; CHECK-NEXT:    br i1 [[C1:%.*]], label [[FOR_BODY3_US:%.*]], label [[FOR_BODY3:%.*]]
184; CHECK:       for.body3.us:
185; CHECK-NEXT:    [[TOBOOL:%.*]] = icmp eq i32 [[B:%.*]], 0
186; CHECK-NEXT:    [[SPEC_SELECT:%.*]] = select i1 [[TOBOOL]], i8 0, i8 4
187; CHECK-NEXT:    switch i3 [[V1:%.*]], label [[EXIT:%.*]] [
188; CHECK-NEXT:      i3 0, label [[FOR_END:%.*]]
189; CHECK-NEXT:      i3 -1, label [[FOR_END]]
190; CHECK-NEXT:    ]
191; CHECK:       for.body3:
192; CHECK-NEXT:    switch i3 [[V2:%.*]], label [[EXIT]] [
193; CHECK-NEXT:      i3 0, label [[FOR_END]]
194; CHECK-NEXT:      i3 -1, label [[FOR_END]]
195; CHECK-NEXT:    ]
196; CHECK:       for.end:
197; CHECK-NEXT:    [[H:%.*]] = phi i8 [ [[SPEC_SELECT]], [[FOR_BODY3_US]] ], [ [[SPEC_SELECT]], [[FOR_BODY3_US]] ], [ 0, [[FOR_BODY3]] ], [ 0, [[FOR_BODY3]] ]
198; CHECK-NEXT:    [[CONV:%.*]] = zext nneg i8 [[H]] to i32
199; CHECK-NEXT:    [[CMP:%.*]] = icmp sgt i32 [[A:%.*]], [[CONV]]
200; CHECK-NEXT:    br i1 [[CMP]], label [[EXIT]], label [[EXIT2:%.*]]
201; CHECK:       exit2:
202; CHECK-NEXT:    unreachable
203; CHECK:       exit:
204; CHECK-NEXT:    unreachable
205;
206; DBGINFO-LABEL: @PR36225(
207; DBGINFO-NEXT:  entry:
208; DBGINFO-NEXT:    br label [[WHILE_BODY:%.*]], !dbg [[DBG94:![0-9]+]]
209; DBGINFO:       while.body:
210; DBGINFO-NEXT:      #dbg_value(i32 [[B:%.*]], [[META89:![0-9]+]], !DIExpression(DW_OP_lit0, DW_OP_eq, DW_OP_stack_value), [[META95:![0-9]+]])
211; DBGINFO-NEXT:    br i1 [[C1:%.*]], label [[FOR_BODY3_US:%.*]], label [[FOR_BODY3:%.*]], !dbg [[DBG96:![0-9]+]]
212; DBGINFO:       for.body3.us:
213; DBGINFO-NEXT:    [[TOBOOL:%.*]] = icmp eq i32 [[B]], 0, !dbg [[META95]]
214; DBGINFO-NEXT:      #dbg_value(i1 [[TOBOOL]], [[META89]], !DIExpression(), [[META95]])
215; DBGINFO-NEXT:    [[SPEC_SELECT:%.*]] = select i1 [[TOBOOL]], i8 0, i8 4, !dbg [[DBG97:![0-9]+]]
216; DBGINFO-NEXT:      #dbg_value(i8 [[SPEC_SELECT]], [[META90:![0-9]+]], !DIExpression(), [[DBG97]])
217; DBGINFO-NEXT:    switch i3 [[V1:%.*]], label [[EXIT:%.*]] [
218; DBGINFO-NEXT:      i3 0, label [[FOR_END:%.*]]
219; DBGINFO-NEXT:      i3 -1, label [[FOR_END]]
220; DBGINFO-NEXT:    ], !dbg [[DBG98:![0-9]+]]
221; DBGINFO:       for.body3:
222; DBGINFO-NEXT:    switch i3 [[V2:%.*]], label [[EXIT]] [
223; DBGINFO-NEXT:      i3 0, label [[FOR_END]]
224; DBGINFO-NEXT:      i3 -1, label [[FOR_END]]
225; DBGINFO-NEXT:    ], !dbg [[DBG99:![0-9]+]]
226; DBGINFO:       for.end:
227; DBGINFO-NEXT:    [[H:%.*]] = phi i8 [ [[SPEC_SELECT]], [[FOR_BODY3_US]] ], [ [[SPEC_SELECT]], [[FOR_BODY3_US]] ], [ 0, [[FOR_BODY3]] ], [ 0, [[FOR_BODY3]] ], !dbg [[DBG100:![0-9]+]]
228; DBGINFO-NEXT:      #dbg_value(i8 [[H]], [[META91:![0-9]+]], !DIExpression(), [[DBG100]])
229; DBGINFO-NEXT:    [[CONV:%.*]] = zext nneg i8 [[H]] to i32, !dbg [[DBG101:![0-9]+]]
230; DBGINFO-NEXT:      #dbg_value(i32 [[CONV]], [[META92:![0-9]+]], !DIExpression(), [[DBG101]])
231; DBGINFO-NEXT:    [[CMP:%.*]] = icmp sgt i32 [[A:%.*]], [[CONV]], !dbg [[DBG102:![0-9]+]]
232; DBGINFO-NEXT:      #dbg_value(i1 [[CMP]], [[META93:![0-9]+]], !DIExpression(), [[DBG102]])
233; DBGINFO-NEXT:    br i1 [[CMP]], label [[EXIT]], label [[EXIT2:%.*]], !dbg [[DBG103:![0-9]+]]
234; DBGINFO:       exit2:
235; DBGINFO-NEXT:    unreachable, !dbg [[DBG104:![0-9]+]]
236; DBGINFO:       exit:
237; DBGINFO-NEXT:    unreachable, !dbg [[DBG105:![0-9]+]]
238;
239entry:
240  br label %while.body
241
242while.body:
243  %tobool = icmp eq i32 %b, 0
244  br i1 %c1, label %for.body3.us, label %for.body3
245
246for.body3.us:
247  %spec.select = select i1 %tobool, i8 0, i8 4
248  switch i3 %v1, label %exit [
249  i3 0, label %for.end
250  i3 -1, label %for.end
251  ]
252
253for.body3:
254  switch i3 %v2, label %exit [
255  i3 0, label %for.end
256  i3 -1, label %for.end
257  ]
258
259for.end:
260  %h = phi i8 [ %spec.select, %for.body3.us ], [ %spec.select, %for.body3.us ], [ 0, %for.body3 ], [ 0, %for.body3 ]
261  %conv = sext i8 %h to i32
262  %cmp = icmp sgt i32 %a, %conv
263  br i1 %cmp, label %exit, label %exit2
264
265exit2:
266  unreachable
267
268exit:
269  unreachable
270}
271
272; Check that we don't drop debug info when a zext is removed.
273define i1 @foo(i1 zeroext %b) {
274; CHECK-LABEL: @foo(
275; CHECK-NEXT:    ret i1 [[B:%.*]]
276;
277; DBGINFO-LABEL: @foo(
278; DBGINFO-NEXT:      #dbg_value(i1 [[B:%.*]], [[META108:![0-9]+]], !DIExpression(DW_OP_LLVM_convert, 1, DW_ATE_unsigned, DW_OP_LLVM_convert, 8, DW_ATE_unsigned, DW_OP_stack_value), [[META109:![0-9]+]])
279; DBGINFO-NEXT:    ret i1 [[B]], !dbg [[DBG110:![0-9]+]]
280;
281
282  %frombool = zext i1 %b to i8
283  ret i1 %b
284}
285