test/Transforms/Scalarizer/vector-gep.ll

   1 ; RUN: opt -S -scalarizer %s | FileCheck %s
   2 ; RUN: opt -S -passes='function(scalarizer)' %s | FileCheck %s
   3
   4 ; Check that the scalarizer can handle vector GEPs with scalar indices
   5
   6 @vec = global <4 x i16*> <i16* null, i16* null, i16* null, i16* null>
   7 @index = global i16 1
   8 @ptr = global [4 x i16] [i16 1, i16 2, i16 3, i16 4]
   9 @ptrptr = global i16* null
  10
  11 ; constant index
  12 define void @test1() {
  13 bb:
  14   %0 = load <4 x i16*>, <4 x i16*>* @vec
  15   %1 = getelementptr i16, <4 x i16*> %0, i16 1
  16
  17   ret void
  18 }
  19
  20 ;CHECK-LABEL: @test1
  21 ;CHECK: %[[I0:.i[0-9]*]] = extractelement <4 x i16*> %0, i32 0
  22 ;CHECK: getelementptr i16, i16* %[[I0]], i16 1
  23 ;CHECK: %[[I1:.i[0-9]*]] = extractelement <4 x i16*> %0, i32 1
  24 ;CHECK: getelementptr i16, i16* %[[I1]], i16 1
  25 ;CHECK: %[[I2:.i[0-9]*]] = extractelement <4 x i16*> %0, i32 2
  26 ;CHECK: getelementptr i16, i16* %[[I2]], i16 1
  27 ;CHECK: %[[I3:.i[0-9]*]] = extractelement <4 x i16*> %0, i32 3
  28 ;CHECK: getelementptr i16, i16* %[[I3]], i16 1
  29
  30 ; non-constant index
  31 define void @test2() {
  32 bb:
  33   %0 = load <4 x i16*>, <4 x i16*>* @vec
  34   %index = load i16, i16* @index
  35   %1 = getelementptr i16, <4 x i16*> %0, i16 %index
  36
  37   ret void
  38 }
  39
  40 ;CHECK-LABEL: @test2
  41 ;CHECK: %0 = load <4 x i16*>, <4 x i16*>* @vec
  42 ;CHECK: %[[I0:.i[0-9]*]] = extractelement <4 x i16*> %0, i32 0
  43 ;CHECK: %[[I1:.i[0-9]*]] = extractelement <4 x i16*> %0, i32 1
  44 ;CHECK: %[[I2:.i[0-9]*]] = extractelement <4 x i16*> %0, i32 2
  45 ;CHECK: %[[I3:.i[0-9]*]] = extractelement <4 x i16*> %0, i32 3
  46 ;CHECK: %index = load i16, i16* @index
  47 ;CHECK: %.splatinsert = insertelement <4 x i16> undef, i16 %index, i32 0
  48 ;CHECK: %.splat = shufflevector <4 x i16> %.splatinsert, <4 x i16> undef, <4 x i32> zeroinitializer
  49 ;CHECK: %.splat[[I0]] = extractelement <4 x i16> %.splat, i32 0
  50 ;CHECK: getelementptr i16, i16* %[[I0]], i16 %.splat[[I0]]
  51 ;CHECK: %.splat[[I1]] = extractelement <4 x i16> %.splat, i32 1
  52 ;CHECK: getelementptr i16, i16* %[[I1]], i16 %.splat[[I1]]
  53 ;CHECK: %.splat[[I2]] = extractelement <4 x i16> %.splat, i32 2
  54 ;CHECK: getelementptr i16, i16* %[[I2]], i16 %.splat[[I2]]
  55 ;CHECK: %.splat[[I3]] = extractelement <4 x i16> %.splat, i32 3
  56 ;CHECK: getelementptr i16, i16* %[[I3]], i16 %.splat[[I3]]
  57
  58
  59 ; Check that the scalarizer can handle vector GEPs with scalar pointer
  60
  61 ; constant pointer
  62 define void @test3() {
  63 bb:
  64   %0 = bitcast [4 x i16]* @ptr to i16*
  65   %1 = getelementptr i16, i16* %0, <4 x i16> <i16 0, i16 1, i16 2, i16 3>
  66
  67   ret void
  68 }
  69
  70 ;CHECK-LABEL: @test3
  71 ;CHECK: %0 = bitcast [4 x i16]* @ptr to i16*
  72 ;CHECK: %.splatinsert = insertelement <4 x i16*> undef, i16* %0, i32 0
  73 ;CHECK: %.splat = shufflevector <4 x i16*> %.splatinsert, <4 x i16*> undef, <4 x i32> zeroinitializer
  74 ;CHECK: %.splat[[I0:.i[0-9]*]] = extractelement <4 x i16*> %.splat, i32 0
  75 ;CHECK: getelementptr i16, i16* %.splat[[I0]], i16 0
  76 ;CHECK: %.splat[[I1:.i[0-9]*]] = extractelement <4 x i16*> %.splat, i32 1
  77 ;CHECK: getelementptr i16, i16* %.splat[[I1]], i16 1
  78 ;CHECK: %.splat[[I2:.i[0-9]*]] = extractelement <4 x i16*> %.splat, i32 2
  79 ;CHECK: getelementptr i16, i16* %.splat[[I2]], i16 2
  80 ;CHECK: %.splat[[I3:.i[0-9]*]] = extractelement <4 x i16*> %.splat, i32 3
  81 ;CHECK: getelementptr i16, i16* %.splat[[I3]], i16 3
  82
  83 ; non-constant pointer
  84 define void @test4() {
  85 bb:
  86   %0 = load i16*, i16** @ptrptr
  87   %1 = getelementptr i16, i16* %0, <4 x i16> <i16 0, i16 1, i16 2, i16 3>
  88
  89   ret void
  90 }
  91
  92 ;CHECK-LABEL: @test4
  93 ;CHECK: %0 = load i16*, i16** @ptrptr
  94 ;CHECK: %.splatinsert = insertelement <4 x i16*> undef, i16* %0, i32 0
  95 ;CHECK: %.splat = shufflevector <4 x i16*> %.splatinsert, <4 x i16*> undef, <4 x i32> zeroinitializer
  96 ;CHECK: %.splat[[I0:.i[0-9]*]] = extractelement <4 x i16*> %.splat, i32 0
  97 ;CHECK: getelementptr i16, i16* %.splat[[I0]], i16 0
  98 ;CHECK: %.splat[[I1:.i[0-9]*]] = extractelement <4 x i16*> %.splat, i32 1
  99 ;CHECK: getelementptr i16, i16* %.splat[[I1]], i16 1
 100 ;CHECK: %.splat[[I2:.i[0-9]*]] = extractelement <4 x i16*> %.splat, i32 2
 101 ;CHECK: getelementptr i16, i16* %.splat[[I2]], i16 2
 102 ;CHECK: %.splat[[I3:.i[0-9]*]] = extractelement <4 x i16*> %.splat, i32 3
 103 ;CHECK: getelementptr i16, i16* %.splat[[I3]], i16 3
 104
 105 ; constant index, inbounds
 106 define void @test5() {
 107 bb:
 108   %0 = load <4 x i16*>, <4 x i16*>* @vec
 109   %1 = getelementptr inbounds i16, <4 x i16*> %0, i16 1
 110
 111   ret void
 112 }
 113
 114 ;CHECK-LABEL: @test5
 115 ;CHECK: %[[I0:.i[0-9]*]] = extractelement <4 x i16*> %0, i32 0
 116 ;CHECK: getelementptr inbounds i16, i16* %[[I0]], i16 1
 117 ;CHECK: %[[I1:.i[0-9]*]] = extractelement <4 x i16*> %0, i32 1
 118 ;CHECK: getelementptr inbounds i16, i16* %[[I1]], i16 1
 119 ;CHECK: %[[I2:.i[0-9]*]] = extractelement <4 x i16*> %0, i32 2
 120 ;CHECK: getelementptr inbounds i16, i16* %[[I2]], i16 1
 121 ;CHECK: %[[I3:.i[0-9]*]] = extractelement <4 x i16*> %0, i32 3
 122 ;CHECK: getelementptr inbounds i16, i16* %[[I3]], i16 1
 123