xref: /llvm-project/llvm/test/Transforms/InstCombine/scalable-select.ll (revision 56c091ea7106507b36015297ee9005c9d5fab0bf)
1; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
2; RUN: opt -passes=instcombine -S < %s | FileCheck %s
3
4; This test checks a regression in the select operand folding combine, in which
5; Constant::getUniqueInteger would crash for a scalable-vector zeroinitializer.
6define <vscale x 1 x i32> @select_opt(<vscale x 1 x i32> %b, <vscale x 1 x i1> %m) {
7; CHECK-LABEL: @select_opt(
8; CHECK-NEXT:    [[C:%.*]] = add nsw <vscale x 1 x i32> [[B:%.*]], splat (i32 2)
9; CHECK-NEXT:    [[D:%.*]] = select <vscale x 1 x i1> [[M:%.*]], <vscale x 1 x i32> [[C]], <vscale x 1 x i32> [[B]]
10; CHECK-NEXT:    ret <vscale x 1 x i32> [[D]]
11;
12  %head = insertelement <vscale x 1 x i32> undef, i32 2, i32 0
13  %splat = shufflevector <vscale x 1 x i32> %head, <vscale x 1 x i32> undef, <vscale x 1 x i32> zeroinitializer
14  %c = add nsw <vscale x 1 x i32> %b, %splat
15  %d = select <vscale x 1 x i1> %m, <vscale x 1 x i32> %c, <vscale x 1 x i32> %b
16  ret <vscale x 1 x i32> %d
17}
18
19define <vscale x 2 x i64> @load_scalable_of_selected_ptrs(ptr %p0, ptr %p1, i64 %idx) {
20; CHECK-LABEL: @load_scalable_of_selected_ptrs(
21; CHECK-NEXT:  entry:
22; CHECK-NEXT:    [[ARRAYIDX0:%.*]] = getelementptr inbounds i32, ptr [[P0:%.*]], i64 [[IDX:%.*]]
23; CHECK-NEXT:    [[ARRAYIDX1:%.*]] = getelementptr inbounds i32, ptr [[P1:%.*]], i64 [[IDX]]
24; CHECK-NEXT:    [[L0:%.*]] = load i32, ptr [[ARRAYIDX0]], align 4
25; CHECK-NEXT:    [[L1:%.*]] = load i32, ptr [[ARRAYIDX1]], align 4
26; CHECK-NEXT:    [[CMP:%.*]] = icmp slt i32 [[L0]], [[L1]]
27; CHECK-NEXT:    [[SEL:%.*]] = select i1 [[CMP]], ptr [[ARRAYIDX0]], ptr [[ARRAYIDX1]]
28; CHECK-NEXT:    [[RET:%.*]] = load <vscale x 2 x i64>, ptr [[SEL]], align 16
29; CHECK-NEXT:    ret <vscale x 2 x i64> [[RET]]
30;
31entry:
32  ; Test introduced in response to a TypeSize warning arising from isSafeToLoadUnconditionally via instcombine.
33  ; deref (p0[0] < p1[0] ? p0 : p1) as <vscale x 2 x i64>.
34  %arrayidx0 = getelementptr inbounds i32, ptr %p0, i64 %idx
35  %arrayidx1 = getelementptr inbounds i32, ptr %p1, i64 %idx
36  %l0 = load i32, ptr %arrayidx0
37  %l1 = load i32, ptr %arrayidx1
38  %cmp = icmp slt i32 %l0, %l1
39  %sel = select i1 %cmp, ptr %arrayidx0, ptr %arrayidx1
40  %ret = load <vscale x 2 x i64>, ptr %sel
41  ret <vscale x 2 x i64> %ret
42}
43