llvm/test/Transforms/InstCombine/scalable-select.ll

   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.
   6 define <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
  19 define <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 ;
  31 entry:
  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 }