test/CodeGen/AArch64/ldst-opt.ll

   1 ; RUN: llc -mtriple=aarch64-linux-gnu -verify-machineinstrs -o - %s | FileCheck %s
   2
   3 ; This file contains tests for the AArch64 load/store optimizer.
   4
   5 %padding = type { i8*, i8*, i8*, i8* }
   6 %s.word = type { i32, i32 }
   7 %s.doubleword = type { i64, i32 }
   8 %s.quadword = type { fp128, i32 }
   9 %s.float = type { float, i32 }
  10 %s.double = type { double, i32 }
  11 %struct.word = type { %padding, %s.word }
  12 %struct.doubleword = type { %padding, %s.doubleword }
  13 %struct.quadword = type { %padding, %s.quadword }
  14 %struct.float = type { %padding, %s.float }
  15 %struct.double = type { %padding, %s.double }
  16
  17 ; Check the following transform:
  18 ;
  19 ; (ldr|str) X, [x0, #32]
  20 ;  ...
  21 ; add x0, x0, #32
  22 ;  ->
  23 ; (ldr|str) X, [x0, #32]!
  24 ;
  25 ; with X being either w1, x1, s0, d0 or q0.
  26
  27 declare void @bar_word(%s.word*, i32)
  28
  29 define void @load-pre-indexed-word(%struct.word* %ptr) nounwind {
  30 ; CHECK-LABEL: load-pre-indexed-word
  31 ; CHECK: ldr w{{[0-9]+}}, [x{{[0-9]+}}, #32]!
  32 entry:
  33   %a = getelementptr inbounds %struct.word* %ptr, i64 0, i32 1, i32 0
  34   %add = load i32* %a, align 4
  35   br label %bar
  36 bar:
  37   %c = getelementptr inbounds %struct.word* %ptr, i64 0, i32 1
  38   tail call void @bar_word(%s.word* %c, i32 %add)
  39   ret void
  40 }
  41
  42 define void @store-pre-indexed-word(%struct.word* %ptr, i32 %val) nounwind {
  43 ; CHECK-LABEL: store-pre-indexed-word
  44 ; CHECK: str w{{[0-9]+}}, [x{{[0-9]+}}, #32]!
  45 entry:
  46   %a = getelementptr inbounds %struct.word* %ptr, i64 0, i32 1, i32 0
  47   store i32 %val, i32* %a, align 4
  48   br label %bar
  49 bar:
  50   %c = getelementptr inbounds %struct.word* %ptr, i64 0, i32 1
  51   tail call void @bar_word(%s.word* %c, i32 %val)
  52   ret void
  53 }
  54
  55 declare void @bar_doubleword(%s.doubleword*, i64)
  56
  57 define void @load-pre-indexed-doubleword(%struct.doubleword* %ptr) nounwind {
  58 ; CHECK-LABEL: load-pre-indexed-doubleword
  59 ; CHECK: ldr x{{[0-9]+}}, [x{{[0-9]+}}, #32]!
  60 entry:
  61   %a = getelementptr inbounds %struct.doubleword* %ptr, i64 0, i32 1, i32 0
  62   %add = load i64* %a, align 4
  63   br label %bar
  64 bar:
  65   %c = getelementptr inbounds %struct.doubleword* %ptr, i64 0, i32 1
  66   tail call void @bar_doubleword(%s.doubleword* %c, i64 %add)
  67   ret void
  68 }
  69
  70 define void @store-pre-indexed-doubleword(%struct.doubleword* %ptr, i64 %val) nounwind {
  71 ; CHECK-LABEL: store-pre-indexed-doubleword
  72 ; CHECK: str x{{[0-9]+}}, [x{{[0-9]+}}, #32]!
  73 entry:
  74   %a = getelementptr inbounds %struct.doubleword* %ptr, i64 0, i32 1, i32 0
  75   store i64 %val, i64* %a, align 4
  76   br label %bar
  77 bar:
  78   %c = getelementptr inbounds %struct.doubleword* %ptr, i64 0, i32 1
  79   tail call void @bar_doubleword(%s.doubleword* %c, i64 %val)
  80   ret void
  81 }
  82
  83 declare void @bar_quadword(%s.quadword*, fp128)
  84
  85 define void @load-pre-indexed-quadword(%struct.quadword* %ptr) nounwind {
  86 ; CHECK-LABEL: load-pre-indexed-quadword
  87 ; CHECK: ldr q{{[0-9]+}}, [x{{[0-9]+}}, #32]!
  88 entry:
  89   %a = getelementptr inbounds %struct.quadword* %ptr, i64 0, i32 1, i32 0
  90   %add = load fp128* %a, align 4
  91   br label %bar
  92 bar:
  93   %c = getelementptr inbounds %struct.quadword* %ptr, i64 0, i32 1
  94   tail call void @bar_quadword(%s.quadword* %c, fp128 %add)
  95   ret void
  96 }
  97
  98 define void @store-pre-indexed-quadword(%struct.quadword* %ptr, fp128 %val) nounwind {
  99 ; CHECK-LABEL: store-pre-indexed-quadword
 100 ; CHECK: str q{{[0-9]+}}, [x{{[0-9]+}}, #32]!
 101 entry:
 102   %a = getelementptr inbounds %struct.quadword* %ptr, i64 0, i32 1, i32 0
 103   store fp128 %val, fp128* %a, align 4
 104   br label %bar
 105 bar:
 106   %c = getelementptr inbounds %struct.quadword* %ptr, i64 0, i32 1
 107   tail call void @bar_quadword(%s.quadword* %c, fp128 %val)
 108   ret void
 109 }
 110
 111 declare void @bar_float(%s.float*, float)
 112
 113 define void @load-pre-indexed-float(%struct.float* %ptr) nounwind {
 114 ; CHECK-LABEL: load-pre-indexed-float
 115 ; CHECK: ldr s{{[0-9]+}}, [x{{[0-9]+}}, #32]!
 116 entry:
 117   %a = getelementptr inbounds %struct.float* %ptr, i64 0, i32 1, i32 0
 118   %add = load float* %a, align 4
 119   br label %bar
 120 bar:
 121   %c = getelementptr inbounds %struct.float* %ptr, i64 0, i32 1
 122   tail call void @bar_float(%s.float* %c, float %add)
 123   ret void
 124 }
 125
 126 define void @store-pre-indexed-float(%struct.float* %ptr, float %val) nounwind {
 127 ; CHECK-LABEL: store-pre-indexed-float
 128 ; CHECK: str s{{[0-9]+}}, [x{{[0-9]+}}, #32]!
 129 entry:
 130   %a = getelementptr inbounds %struct.float* %ptr, i64 0, i32 1, i32 0
 131   store float %val, float* %a, align 4
 132   br label %bar
 133 bar:
 134   %c = getelementptr inbounds %struct.float* %ptr, i64 0, i32 1
 135   tail call void @bar_float(%s.float* %c, float %val)
 136   ret void
 137 }
 138
 139 declare void @bar_double(%s.double*, double)
 140
 141 define void @load-pre-indexed-double(%struct.double* %ptr) nounwind {
 142 ; CHECK-LABEL: load-pre-indexed-double
 143 ; CHECK: ldr d{{[0-9]+}}, [x{{[0-9]+}}, #32]!
 144 entry:
 145   %a = getelementptr inbounds %struct.double* %ptr, i64 0, i32 1, i32 0
 146   %add = load double* %a, align 4
 147   br label %bar
 148 bar:
 149   %c = getelementptr inbounds %struct.double* %ptr, i64 0, i32 1
 150   tail call void @bar_double(%s.double* %c, double %add)
 151   ret void
 152 }
 153
 154 define void @store-pre-indexed-double(%struct.double* %ptr, double %val) nounwind {
 155 ; CHECK-LABEL: store-pre-indexed-double
 156 ; CHECK: str d{{[0-9]+}}, [x{{[0-9]+}}, #32]!
 157 entry:
 158   %a = getelementptr inbounds %struct.double* %ptr, i64 0, i32 1, i32 0
 159   store double %val, double* %a, align 4
 160   br label %bar
 161 bar:
 162   %c = getelementptr inbounds %struct.double* %ptr, i64 0, i32 1
 163   tail call void @bar_double(%s.double* %c, double %val)
 164   ret void
 165 }