llvm/test/CodeGen/AArch64/GlobalISel/translate-gep.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_mir_test_checks.py
   2 ; RUN: llc -mtriple=aarch64-linux-gnu -O0 -global-isel -stop-after=irtranslator -o - %s | FileCheck %s
   3
   4 %type = type [4 x {i8, i32}]
   5
   6 define ptr @translate_element_size1(i64 %arg) {
   7   ; CHECK-LABEL: name: translate_element_size1
   8   ; CHECK: bb.1 (%ir-block.0):
   9   ; CHECK-NEXT:   liveins: $x0
  10   ; CHECK-NEXT: {{  $}}
  11   ; CHECK-NEXT:   [[COPY:%[0-9]+]]:_(s64) = COPY $x0
  12   ; CHECK-NEXT:   [[C:%[0-9]+]]:_(p0) = G_CONSTANT i64 0
  13   ; CHECK-NEXT:   [[PTR_ADD:%[0-9]+]]:_(p0) = G_PTR_ADD [[C]], [[COPY]](s64)
  14   ; CHECK-NEXT:   [[COPY1:%[0-9]+]]:_(p0) = COPY [[PTR_ADD]](p0)
  15   ; CHECK-NEXT:   $x0 = COPY [[COPY1]](p0)
  16   ; CHECK-NEXT:   RET_ReallyLR implicit $x0
  17   %tmp = getelementptr i8, ptr null, i64 %arg
  18   ret ptr %tmp
  19 }
  20
  21 define ptr @first_offset_const(ptr %addr) {
  22
  23   ; CHECK-LABEL: name: first_offset_const
  24   ; CHECK: bb.1 (%ir-block.0):
  25   ; CHECK-NEXT:   liveins: $x0
  26   ; CHECK-NEXT: {{  $}}
  27   ; CHECK-NEXT:   [[COPY:%[0-9]+]]:_(p0) = COPY $x0
  28   ; CHECK-NEXT:   [[C:%[0-9]+]]:_(s64) = G_CONSTANT i64 32
  29   ; CHECK-NEXT:   [[PTR_ADD:%[0-9]+]]:_(p0) = G_PTR_ADD [[COPY]], [[C]](s64)
  30   ; CHECK-NEXT:   $x0 = COPY [[PTR_ADD]](p0)
  31   ; CHECK-NEXT:   RET_ReallyLR implicit $x0
  32   %res = getelementptr %type, ptr %addr, i32 1
  33   ret ptr %res
  34 }
  35
  36 define ptr @first_offset_trivial(ptr %addr) {
  37
  38   ; CHECK-LABEL: name: first_offset_trivial
  39   ; CHECK: bb.1 (%ir-block.0):
  40   ; CHECK-NEXT:   liveins: $x0
  41   ; CHECK-NEXT: {{  $}}
  42   ; CHECK-NEXT:   [[COPY:%[0-9]+]]:_(p0) = COPY $x0
  43   ; CHECK-NEXT:   [[COPY1:%[0-9]+]]:_(p0) = COPY [[COPY]](p0)
  44   ; CHECK-NEXT:   $x0 = COPY [[COPY1]](p0)
  45   ; CHECK-NEXT:   RET_ReallyLR implicit $x0
  46   %res = getelementptr %type, ptr %addr, i32 0
  47   ret ptr %res
  48 }
  49
  50 define ptr @first_offset_variable(ptr %addr, i64 %idx) {
  51
  52   ; CHECK-LABEL: name: first_offset_variable
  53   ; CHECK: bb.1 (%ir-block.0):
  54   ; CHECK-NEXT:   liveins: $x0, $x1
  55   ; CHECK-NEXT: {{  $}}
  56   ; CHECK-NEXT:   [[COPY:%[0-9]+]]:_(p0) = COPY $x0
  57   ; CHECK-NEXT:   [[COPY1:%[0-9]+]]:_(s64) = COPY $x1
  58   ; CHECK-NEXT:   [[C:%[0-9]+]]:_(s64) = G_CONSTANT i64 32
  59   ; CHECK-NEXT:   [[MUL:%[0-9]+]]:_(s64) = G_MUL [[COPY1]], [[C]]
  60   ; CHECK-NEXT:   [[PTR_ADD:%[0-9]+]]:_(p0) = G_PTR_ADD [[COPY]], [[MUL]](s64)
  61   ; CHECK-NEXT:   [[COPY2:%[0-9]+]]:_(p0) = COPY [[PTR_ADD]](p0)
  62   ; CHECK-NEXT:   $x0 = COPY [[COPY2]](p0)
  63   ; CHECK-NEXT:   RET_ReallyLR implicit $x0
  64   %res = getelementptr %type, ptr %addr, i64 %idx
  65   ret ptr %res
  66 }
  67
  68 define ptr @first_offset_ext(ptr %addr, i32 %idx) {
  69
  70   ; CHECK-LABEL: name: first_offset_ext
  71   ; CHECK: bb.1 (%ir-block.0):
  72   ; CHECK-NEXT:   liveins: $w1, $x0
  73   ; CHECK-NEXT: {{  $}}
  74   ; CHECK-NEXT:   [[COPY:%[0-9]+]]:_(p0) = COPY $x0
  75   ; CHECK-NEXT:   [[COPY1:%[0-9]+]]:_(s32) = COPY $w1
  76   ; CHECK-NEXT:   [[SEXT:%[0-9]+]]:_(s64) = G_SEXT [[COPY1]](s32)
  77   ; CHECK-NEXT:   [[C:%[0-9]+]]:_(s64) = G_CONSTANT i64 32
  78   ; CHECK-NEXT:   [[MUL:%[0-9]+]]:_(s64) = G_MUL [[SEXT]], [[C]]
  79   ; CHECK-NEXT:   [[PTR_ADD:%[0-9]+]]:_(p0) = G_PTR_ADD [[COPY]], [[MUL]](s64)
  80   ; CHECK-NEXT:   [[COPY2:%[0-9]+]]:_(p0) = COPY [[PTR_ADD]](p0)
  81   ; CHECK-NEXT:   $x0 = COPY [[COPY2]](p0)
  82   ; CHECK-NEXT:   RET_ReallyLR implicit $x0
  83   %res = getelementptr %type, ptr %addr, i32 %idx
  84   ret ptr %res
  85 }
  86
  87 %type1 = type [4 x [4 x i32]]
  88 define ptr @const_then_var(ptr %addr, i64 %idx) {
  89
  90   ; CHECK-LABEL: name: const_then_var
  91   ; CHECK: bb.1 (%ir-block.0):
  92   ; CHECK-NEXT:   liveins: $x0, $x1
  93   ; CHECK-NEXT: {{  $}}
  94   ; CHECK-NEXT:   [[COPY:%[0-9]+]]:_(p0) = COPY $x0
  95   ; CHECK-NEXT:   [[COPY1:%[0-9]+]]:_(s64) = COPY $x1
  96   ; CHECK-NEXT:   [[C:%[0-9]+]]:_(s64) = G_CONSTANT i64 272
  97   ; CHECK-NEXT:   [[PTR_ADD:%[0-9]+]]:_(p0) = G_PTR_ADD [[COPY]], [[C]](s64)
  98   ; CHECK-NEXT:   [[C1:%[0-9]+]]:_(s64) = G_CONSTANT i64 4
  99   ; CHECK-NEXT:   [[MUL:%[0-9]+]]:_(s64) = G_MUL [[COPY1]], [[C1]]
 100   ; CHECK-NEXT:   [[PTR_ADD1:%[0-9]+]]:_(p0) = G_PTR_ADD [[PTR_ADD]], [[MUL]](s64)
 101   ; CHECK-NEXT:   [[COPY2:%[0-9]+]]:_(p0) = COPY [[PTR_ADD1]](p0)
 102   ; CHECK-NEXT:   $x0 = COPY [[COPY2]](p0)
 103   ; CHECK-NEXT:   RET_ReallyLR implicit $x0
 104   %res = getelementptr %type1, ptr %addr, i32 4, i32 1, i64 %idx
 105   ret ptr %res
 106 }
 107
 108 define ptr @var_then_const(ptr %addr, i64 %idx) {
 109
 110   ; CHECK-LABEL: name: var_then_const
 111   ; CHECK: bb.1 (%ir-block.0):
 112   ; CHECK-NEXT:   liveins: $x0, $x1
 113   ; CHECK-NEXT: {{  $}}
 114   ; CHECK-NEXT:   [[COPY:%[0-9]+]]:_(p0) = COPY $x0
 115   ; CHECK-NEXT:   [[COPY1:%[0-9]+]]:_(s64) = COPY $x1
 116   ; CHECK-NEXT:   [[C:%[0-9]+]]:_(s64) = G_CONSTANT i64 64
 117   ; CHECK-NEXT:   [[MUL:%[0-9]+]]:_(s64) = G_MUL [[COPY1]], [[C]]
 118   ; CHECK-NEXT:   [[PTR_ADD:%[0-9]+]]:_(p0) = G_PTR_ADD [[COPY]], [[MUL]](s64)
 119   ; CHECK-NEXT:   [[C1:%[0-9]+]]:_(s64) = G_CONSTANT i64 40
 120   ; CHECK-NEXT:   [[PTR_ADD1:%[0-9]+]]:_(p0) = G_PTR_ADD [[PTR_ADD]], [[C1]](s64)
 121   ; CHECK-NEXT:   $x0 = COPY [[PTR_ADD1]](p0)
 122   ; CHECK-NEXT:   RET_ReallyLR implicit $x0
 123   %res = getelementptr %type1, ptr %addr, i64 %idx, i32 2, i32 2
 124   ret ptr %res
 125 }
 126
 127 @arr = external global [8 x i32]
 128
 129 define <2 x ptr> @vec_gep_scalar_base(<2 x i64> %offs) {
 130   ; CHECK-LABEL: name: vec_gep_scalar_base
 131   ; CHECK: bb.1.entry:
 132   ; CHECK-NEXT:   liveins: $q0
 133   ; CHECK-NEXT: {{  $}}
 134   ; CHECK-NEXT:   [[COPY:%[0-9]+]]:_(<2 x s64>) = COPY $q0
 135   ; CHECK-NEXT:   [[GV:%[0-9]+]]:_(p0) = G_GLOBAL_VALUE @arr
 136   ; CHECK-NEXT:   [[BUILD_VECTOR:%[0-9]+]]:_(<2 x p0>) = G_BUILD_VECTOR [[GV]](p0), [[GV]](p0)
 137   ; CHECK-NEXT:   [[C:%[0-9]+]]:_(s64) = G_CONSTANT i64 4
 138   ; CHECK-NEXT:   [[BUILD_VECTOR1:%[0-9]+]]:_(<2 x s64>) = G_BUILD_VECTOR [[C]](s64), [[C]](s64)
 139   ; CHECK-NEXT:   [[MUL:%[0-9]+]]:_(<2 x s64>) = G_MUL [[COPY]], [[BUILD_VECTOR1]]
 140   ; CHECK-NEXT:   [[PTR_ADD:%[0-9]+]]:_(<2 x p0>) = G_PTR_ADD [[BUILD_VECTOR]], [[MUL]](<2 x s64>)
 141   ; CHECK-NEXT:   [[COPY1:%[0-9]+]]:_(<2 x p0>) = COPY [[PTR_ADD]](<2 x p0>)
 142   ; CHECK-NEXT:   $q0 = COPY [[COPY1]](<2 x p0>)
 143   ; CHECK-NEXT:   RET_ReallyLR implicit $q0
 144 entry:
 145   %0 = getelementptr inbounds [8 x i32], ptr @arr, i64 0, <2 x i64> %offs
 146   ret <2 x ptr> %0
 147 }