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

; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
; RUN: opt -passes=instcombine -S < %s | FileCheck %s

; This test checks a regression in the select operand folding combine, in which
; Constant::getUniqueInteger would crash for a scalable-vector zeroinitializer.
define <vscale x 1 x i32> @select_opt(<vscale x 1 x i32> %b, <vscale x 1 x i1> %m) {
; CHECK-LABEL: @select_opt(
; CHECK-NEXT:    [[C:%.*]] = add nsw <vscale x 1 x i32> [[B:%.*]], shufflevector (<vscale x 1 x i32> insertelement (<vscale x 1 x i32> undef, i32 2, i32 0), <vscale x 1 x i32> undef, <vscale x 1 x i32> zeroinitializer)
; CHECK-NEXT:    [[D:%.*]] = select <vscale x 1 x i1> [[M:%.*]], <vscale x 1 x i32> [[C]], <vscale x 1 x i32> [[B]]
; CHECK-NEXT:    ret <vscale x 1 x i32> [[D]]
;
  %head = insertelement <vscale x 1 x i32> undef, i32 2, i32 0
  %splat = shufflevector <vscale x 1 x i32> %head, <vscale x 1 x i32> undef, <vscale x 1 x i32> zeroinitializer
  %c = add nsw <vscale x 1 x i32> %b, %splat
  %d = select <vscale x 1 x i1> %m, <vscale x 1 x i32> %c, <vscale x 1 x i32> %b
  ret <vscale x 1 x i32> %d
}

define <vscale x 2 x i64> @load_scalable_of_selected_ptrs(ptr %p0, ptr %p1, i64 %idx) {
; CHECK-LABEL: @load_scalable_of_selected_ptrs(
; CHECK-NEXT:  entry:
; CHECK-NEXT:    [[ARRAYIDX0:%.*]] = getelementptr inbounds i32, ptr [[P0:%.*]], i64 [[IDX:%.*]]
; CHECK-NEXT:    [[ARRAYIDX1:%.*]] = getelementptr inbounds i32, ptr [[P1:%.*]], i64 [[IDX]]
; CHECK-NEXT:    [[L0:%.*]] = load i32, ptr [[ARRAYIDX0]], align 4
; CHECK-NEXT:    [[L1:%.*]] = load i32, ptr [[ARRAYIDX1]], align 4
; CHECK-NEXT:    [[CMP:%.*]] = icmp slt i32 [[L0]], [[L1]]
; CHECK-NEXT:    [[SEL:%.*]] = select i1 [[CMP]], ptr [[ARRAYIDX0]], ptr [[ARRAYIDX1]]
; CHECK-NEXT:    [[RET:%.*]] = load <vscale x 2 x i64>, ptr [[SEL]], align 16
; CHECK-NEXT:    ret <vscale x 2 x i64> [[RET]]
;
entry:
  ; Test introduced in response to a TypeSize warning arising from isSafeToLoadUnconditionally via instcombine.
  ; deref (p0[0] < p1[0] ? p0 : p1) as <vscale x 2 x i64>.
  %arrayidx0 = getelementptr inbounds i32, ptr %p0, i64 %idx
  %arrayidx1 = getelementptr inbounds i32, ptr %p1, i64 %idx
  %l0 = load i32, ptr %arrayidx0
  %l1 = load i32, ptr %arrayidx1
  %cmp = icmp slt i32 %l0, %l1
  %sel = select i1 %cmp, ptr %arrayidx0, ptr %arrayidx1
  %ret = load <vscale x 2 x i64>, ptr %sel
  ret <vscale x 2 x i64> %ret
}