llvm/llvm/test/CodeGen/X86/vector-gep.ll

; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py
; RUN: llc < %s -mtriple=i686-linux -mcpu=corei7-avx | FileCheck %s
; RUN: opt -passes=instsimplify -disable-output < %s

define <4 x ptr> @AGEP0(ptr %ptr) nounwind {
; CHECK-LABEL: AGEP0:
; CHECK:       # %bb.0:
; CHECK-NEXT:    vbroadcastss {{[0-9]+}}(%esp), %xmm0
; CHECK-NEXT:    vpaddd {{\.?LCPI[0-9]+_[0-9]+}}, %xmm0, %xmm0
; CHECK-NEXT:    retl
  %vecinit.i = insertelement <4 x ptr> undef, ptr %ptr, i32 0
  %vecinit2.i = insertelement <4 x ptr> %vecinit.i, ptr %ptr, i32 1
  %vecinit4.i = insertelement <4 x ptr> %vecinit2.i, ptr %ptr, i32 2
  %vecinit6.i = insertelement <4 x ptr> %vecinit4.i, ptr %ptr, i32 3
  %A2 = getelementptr i32, <4 x ptr> %vecinit6.i, <4 x i32> <i32 1, i32 2, i32 3, i32 4>
  %A3 = getelementptr i32, <4 x ptr> %A2, <4 x i32> <i32 10, i32 14, i32 19, i32 233>
  ret <4 x ptr> %A3
}

define i32 @AGEP1(<4 x ptr> %param) nounwind {
; CHECK-LABEL: AGEP1:
; CHECK:       # %bb.0:
; CHECK-NEXT:    vextractps $3, %xmm0, %eax
; CHECK-NEXT:    movl 16(%eax), %eax
; CHECK-NEXT:    retl
  %A2 = getelementptr i32, <4 x ptr> %param, <4 x i32> <i32 1, i32 2, i32 3, i32 4>
  %k = extractelement <4 x ptr> %A2, i32 3
  %v = load i32, ptr %k
  ret i32 %v
}

define i32 @AGEP2(<4 x ptr> %param, <4 x i32> %off) nounwind {
; CHECK-LABEL: AGEP2:
; CHECK:       # %bb.0:
; CHECK-NEXT:    vpslld $2, %xmm1, %xmm1
; CHECK-NEXT:    vpaddd %xmm1, %xmm0, %xmm0
; CHECK-NEXT:    vpextrd $3, %xmm0, %eax
; CHECK-NEXT:    movl (%eax), %eax
; CHECK-NEXT:    retl
  %A2 = getelementptr i32, <4 x ptr> %param, <4 x i32> %off
  %k = extractelement <4 x ptr> %A2, i32 3
  %v = load i32, ptr %k
  ret i32 %v
}

define <4 x ptr> @AGEP3(<4 x ptr> %param, <4 x i32> %off) nounwind {
; CHECK-LABEL: AGEP3:
; CHECK:       # %bb.0:
; CHECK-NEXT:    pushl %eax
; CHECK-NEXT:    vpslld $2, %xmm1, %xmm1
; CHECK-NEXT:    vpaddd %xmm1, %xmm0, %xmm0
; CHECK-NEXT:    movl %esp, %eax
; CHECK-NEXT:    vpinsrd $3, %eax, %xmm0, %xmm0
; CHECK-NEXT:    popl %eax
; CHECK-NEXT:    retl
  %A2 = getelementptr i32, <4 x ptr> %param, <4 x i32> %off
  %v = alloca i32
  %k = insertelement <4 x ptr> %A2, ptr %v, i32 3
  ret <4 x ptr> %k
}

define <4 x ptr> @AGEP4(<4 x ptr> %param, <4 x i32> %off) nounwind {
; Multiply offset by two (add it to itself).
; add the base to the offset
; CHECK-LABEL: AGEP4:
; CHECK:       # %bb.0:
; CHECK-NEXT:    vpaddd %xmm1, %xmm1, %xmm1
; CHECK-NEXT:    vpaddd %xmm1, %xmm0, %xmm0
; CHECK-NEXT:    retl
  %A = getelementptr i16, <4 x ptr> %param, <4 x i32> %off
  ret <4 x ptr> %A
}

define <4 x ptr> @AGEP5(<4 x ptr> %param, <4 x i8> %off) nounwind {
; CHECK-LABEL: AGEP5:
; CHECK:       # %bb.0:
; CHECK-NEXT:    vpmovsxbd %xmm1, %xmm1
; CHECK-NEXT:    vpaddd %xmm1, %xmm0, %xmm0
; CHECK-NEXT:    retl
  %A = getelementptr i8, <4 x ptr> %param, <4 x i8> %off
  ret <4 x ptr> %A
}


; The size of each element is 1 byte. No need to multiply by element size.
define <4 x ptr> @AGEP6(<4 x ptr> %param, <4 x i32> %off) nounwind {
; CHECK-LABEL: AGEP6:
; CHECK:       # %bb.0:
; CHECK-NEXT:    vpaddd %xmm1, %xmm0, %xmm0
; CHECK-NEXT:    retl
  %A = getelementptr i8, <4 x ptr> %param, <4 x i32> %off
  ret <4 x ptr> %A
}

define <4 x ptr> @AGEP7(<4 x ptr> %param, i32 %off) nounwind {
; CHECK-LABEL: AGEP7:
; CHECK:       # %bb.0:
; CHECK-NEXT:    vbroadcastss {{[0-9]+}}(%esp), %xmm1
; CHECK-NEXT:    vpaddd %xmm1, %xmm0, %xmm0
; CHECK-NEXT:    retl
  %A = getelementptr i8, <4 x ptr> %param, i32 %off
  ret <4 x ptr> %A
}

define <4 x ptr> @AGEP8(ptr %param, <4 x i32> %off) nounwind {
; Multiply offset by two (add it to itself).
; add the base to the offset
; CHECK-LABEL: AGEP8:
; CHECK:       # %bb.0:
; CHECK-NEXT:    vpaddd %xmm0, %xmm0, %xmm0
; CHECK-NEXT:    vbroadcastss {{[0-9]+}}(%esp), %xmm1
; CHECK-NEXT:    vpaddd %xmm0, %xmm1, %xmm0
; CHECK-NEXT:    retl
  %A = getelementptr i16, ptr %param, <4 x i32> %off
  ret <4 x ptr> %A
}

define <64 x ptr> @AGEP9(ptr %param, <64 x i32> %off) nounwind {
; CHECK-LABEL: AGEP9:
; CHECK:       # %bb.0:
; CHECK-NEXT:    pushl %ebp
; CHECK-NEXT:    movl %esp, %ebp
; CHECK-NEXT:    andl $-32, %esp
; CHECK-NEXT:    subl $160, %esp
; CHECK-NEXT:    vpaddd %xmm0, %xmm0, %xmm3
; CHECK-NEXT:    vbroadcastss 12(%ebp), %xmm5
; CHECK-NEXT:    vpaddd %xmm3, %xmm5, %xmm3
; CHECK-NEXT:    vmovdqa %xmm3, {{[-0-9]+}}(%e{{[sb]}}p) # 16-byte Spill
; CHECK-NEXT:    vextractf128 $1, %ymm0, %xmm0
; CHECK-NEXT:    vpaddd %xmm0, %xmm0, %xmm0
; CHECK-NEXT:    vpaddd %xmm0, %xmm5, %xmm0
; CHECK-NEXT:    vmovdqa %xmm0, {{[-0-9]+}}(%e{{[sb]}}p) # 16-byte Spill
; CHECK-NEXT:    vpaddd %xmm1, %xmm1, %xmm0
; CHECK-NEXT:    vpaddd %xmm0, %xmm5, %xmm0
; CHECK-NEXT:    vmovdqa %xmm0, {{[-0-9]+}}(%e{{[sb]}}p) # 16-byte Spill
; CHECK-NEXT:    vextractf128 $1, %ymm1, %xmm0
; CHECK-NEXT:    vpaddd %xmm0, %xmm0, %xmm0
; CHECK-NEXT:    vpaddd %xmm0, %xmm5, %xmm0
; CHECK-NEXT:    vmovdqa %xmm0, {{[-0-9]+}}(%e{{[sb]}}p) # 16-byte Spill
; CHECK-NEXT:    vpaddd %xmm2, %xmm2, %xmm0
; CHECK-NEXT:    vpaddd %xmm0, %xmm5, %xmm0
; CHECK-NEXT:    vmovdqa %xmm0, {{[-0-9]+}}(%e{{[sb]}}p) # 16-byte Spill
; CHECK-NEXT:    vextractf128 $1, %ymm2, %xmm0
; CHECK-NEXT:    vpaddd %xmm0, %xmm0, %xmm0
; CHECK-NEXT:    vpaddd %xmm0, %xmm5, %xmm0
; CHECK-NEXT:    vmovdqa %xmm0, {{[-0-9]+}}(%e{{[sb]}}p) # 16-byte Spill
; CHECK-NEXT:    vmovdqa 40(%ebp), %xmm0
; CHECK-NEXT:    vpaddd %xmm0, %xmm0, %xmm0
; CHECK-NEXT:    vpaddd %xmm0, %xmm5, %xmm0
; CHECK-NEXT:    vmovdqa %xmm0, {{[-0-9]+}}(%e{{[sb]}}p) # 16-byte Spill
; CHECK-NEXT:    vmovdqa 56(%ebp), %xmm0
; CHECK-NEXT:    vpaddd %xmm0, %xmm0, %xmm0
; CHECK-NEXT:    vpaddd %xmm0, %xmm5, %xmm0
; CHECK-NEXT:    vmovdqa %xmm0, {{[-0-9]+}}(%e{{[sb]}}p) # 16-byte Spill
; CHECK-NEXT:    vmovdqa 72(%ebp), %xmm0
; CHECK-NEXT:    vpaddd %xmm0, %xmm0, %xmm0
; CHECK-NEXT:    vpaddd %xmm0, %xmm5, %xmm0
; CHECK-NEXT:    vmovdqa %xmm0, (%esp) # 16-byte Spill
; CHECK-NEXT:    vmovdqa 88(%ebp), %xmm0
; CHECK-NEXT:    vpaddd %xmm0, %xmm0, %xmm0
; CHECK-NEXT:    vpaddd %xmm0, %xmm5, %xmm2
; CHECK-NEXT:    vmovdqa 104(%ebp), %xmm0
; CHECK-NEXT:    vpaddd %xmm0, %xmm0, %xmm0
; CHECK-NEXT:    vpaddd %xmm0, %xmm5, %xmm1
; CHECK-NEXT:    vmovdqa 120(%ebp), %xmm0
; CHECK-NEXT:    vpaddd %xmm0, %xmm0, %xmm0
; CHECK-NEXT:    vpaddd %xmm0, %xmm5, %xmm0
; CHECK-NEXT:    vmovdqa 136(%ebp), %xmm6
; CHECK-NEXT:    vpaddd %xmm6, %xmm6, %xmm6
; CHECK-NEXT:    vpaddd %xmm6, %xmm5, %xmm6
; CHECK-NEXT:    vmovdqa 152(%ebp), %xmm7
; CHECK-NEXT:    vpaddd %xmm7, %xmm7, %xmm7
; CHECK-NEXT:    vpaddd %xmm7, %xmm5, %xmm7
; CHECK-NEXT:    vmovdqa 168(%ebp), %xmm4
; CHECK-NEXT:    vpaddd %xmm4, %xmm4, %xmm4
; CHECK-NEXT:    vpaddd %xmm4, %xmm5, %xmm4
; CHECK-NEXT:    vmovdqa 184(%ebp), %xmm3
; CHECK-NEXT:    vpaddd %xmm3, %xmm3, %xmm3
; CHECK-NEXT:    vpaddd %xmm3, %xmm5, %xmm3
; CHECK-NEXT:    movl 8(%ebp), %eax
; CHECK-NEXT:    vmovdqa %xmm3, 240(%eax)
; CHECK-NEXT:    vmovdqa %xmm4, 224(%eax)
; CHECK-NEXT:    vmovdqa %xmm7, 208(%eax)
; CHECK-NEXT:    vmovdqa %xmm6, 192(%eax)
; CHECK-NEXT:    vmovdqa %xmm0, 176(%eax)
; CHECK-NEXT:    vmovdqa %xmm1, 160(%eax)
; CHECK-NEXT:    vmovdqa %xmm2, 144(%eax)
; CHECK-NEXT:    vmovaps (%esp), %xmm0 # 16-byte Reload
; CHECK-NEXT:    vmovaps %xmm0, 128(%eax)
; CHECK-NEXT:    vmovaps {{[-0-9]+}}(%e{{[sb]}}p), %xmm0 # 16-byte Reload
; CHECK-NEXT:    vmovaps %xmm0, 112(%eax)
; CHECK-NEXT:    vmovaps {{[-0-9]+}}(%e{{[sb]}}p), %xmm0 # 16-byte Reload
; CHECK-NEXT:    vmovaps %xmm0, 96(%eax)
; CHECK-NEXT:    vmovaps {{[-0-9]+}}(%e{{[sb]}}p), %xmm0 # 16-byte Reload
; CHECK-NEXT:    vmovaps %xmm0, 80(%eax)
; CHECK-NEXT:    vmovaps {{[-0-9]+}}(%e{{[sb]}}p), %xmm0 # 16-byte Reload
; CHECK-NEXT:    vmovaps %xmm0, 64(%eax)
; CHECK-NEXT:    vmovaps {{[-0-9]+}}(%e{{[sb]}}p), %xmm0 # 16-byte Reload
; CHECK-NEXT:    vmovaps %xmm0, 48(%eax)
; CHECK-NEXT:    vmovaps {{[-0-9]+}}(%e{{[sb]}}p), %xmm0 # 16-byte Reload
; CHECK-NEXT:    vmovaps %xmm0, 32(%eax)
; CHECK-NEXT:    vmovaps {{[-0-9]+}}(%e{{[sb]}}p), %xmm0 # 16-byte Reload
; CHECK-NEXT:    vmovaps %xmm0, 16(%eax)
; CHECK-NEXT:    vmovaps {{[-0-9]+}}(%e{{[sb]}}p), %xmm0 # 16-byte Reload
; CHECK-NEXT:    vmovaps %xmm0, (%eax)
; CHECK-NEXT:    movl %ebp, %esp
; CHECK-NEXT:    popl %ebp
; CHECK-NEXT:    vzeroupper
; CHECK-NEXT:    retl $4
  %A = getelementptr i16, ptr %param, <64 x i32> %off
  ret <64 x ptr> %A
}