llvm/test/Transforms/SROA/address-spaces.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
   2 ; RUN: opt < %s -passes='sroa<preserve-cfg>' -S | FileCheck %s --check-prefixes=CHECK,CHECK-PRESERVE-CFG
   3 ; RUN: opt < %s -passes='sroa<modify-cfg>' -S | FileCheck %s --check-prefixes=CHECK,CHECK-MODIFY-CFG
   4 target datalayout = "e-p:64:64:64-p1:16:16:16-p3:32:32:32-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:32:64-f32:32:32-f64:64:64-v64:64:64-v128:128:128-a0:0:64-n8:16:32:64"
   5
   6 declare void @llvm.memcpy.p0.p0.i32(ptr nocapture, ptr nocapture readonly, i32, i1)
   7 declare void @llvm.memcpy.p1.p0.i32(ptr addrspace(1) nocapture, ptr nocapture readonly, i32, i1)
   8 declare void @llvm.memcpy.p0.p1.i32(ptr nocapture, ptr addrspace(1) nocapture readonly, i32, i1)
   9 declare void @llvm.memcpy.p1.p1.i32(ptr addrspace(1) nocapture, ptr addrspace(1) nocapture readonly, i32, i1)
  10
  11
  12 ; Make sure an illegal bitcast isn't introduced
  13 define void @test_address_space_1_1(ptr addrspace(1) %a, ptr addrspace(1) %b) {
  14 ; CHECK-LABEL: @test_address_space_1_1(
  15 ; CHECK-NEXT:    [[AA_0_COPYLOAD:%.*]] = load <2 x i64>, ptr addrspace(1) [[A:%.*]], align 2
  16 ; CHECK-NEXT:    store <2 x i64> [[AA_0_COPYLOAD]], ptr addrspace(1) [[B:%.*]], align 2
  17 ; CHECK-NEXT:    ret void
  18 ;
  19   %aa = alloca <2 x i64>, align 16
  20   call void @llvm.memcpy.p0.p1.i32(ptr align 2 %aa, ptr addrspace(1) align 2 %a, i32 16, i1 false)
  21   call void @llvm.memcpy.p1.p0.i32(ptr addrspace(1) align 2 %b, ptr align 2 %aa, i32 16, i1 false)
  22   ret void
  23 }
  24
  25 define void @test_address_space_1_0(ptr addrspace(1) %a, ptr %b) {
  26 ; CHECK-LABEL: @test_address_space_1_0(
  27 ; CHECK-NEXT:    [[AA_0_COPYLOAD:%.*]] = load <2 x i64>, ptr addrspace(1) [[A:%.*]], align 2
  28 ; CHECK-NEXT:    store <2 x i64> [[AA_0_COPYLOAD]], ptr [[B:%.*]], align 2
  29 ; CHECK-NEXT:    ret void
  30 ;
  31   %aa = alloca <2 x i64>, align 16
  32   call void @llvm.memcpy.p0.p1.i32(ptr align 2 %aa, ptr addrspace(1) align 2 %a, i32 16, i1 false)
  33   call void @llvm.memcpy.p0.p0.i32(ptr align 2 %b, ptr align 2 %aa, i32 16, i1 false)
  34   ret void
  35 }
  36
  37 define void @test_address_space_0_1(ptr %a, ptr addrspace(1) %b) {
  38 ; CHECK-LABEL: @test_address_space_0_1(
  39 ; CHECK-NEXT:    [[AA_0_COPYLOAD:%.*]] = load <2 x i64>, ptr [[A:%.*]], align 2
  40 ; CHECK-NEXT:    store <2 x i64> [[AA_0_COPYLOAD]], ptr addrspace(1) [[B:%.*]], align 2
  41 ; CHECK-NEXT:    ret void
  42 ;
  43   %aa = alloca <2 x i64>, align 16
  44   call void @llvm.memcpy.p0.p0.i32(ptr align 2 %aa, ptr align 2 %a, i32 16, i1 false)
  45   call void @llvm.memcpy.p1.p0.i32(ptr addrspace(1) align 2 %b, ptr align 2 %aa, i32 16, i1 false)
  46   ret void
  47 }
  48
  49 %struct.struct_test_27.0.13 = type { i32, float, i64, i8, [4 x i32] }
  50
  51 define void @copy_struct([5 x i64] %in.coerce, ptr addrspace(1) align 4 %ptr) {
  52 ; CHECK-LABEL: @copy_struct(
  53 ; CHECK-NEXT:  for.end:
  54 ; CHECK-NEXT:    [[IN_COERCE_FCA_0_EXTRACT:%.*]] = extractvalue [5 x i64] [[IN_COERCE:%.*]], 0
  55 ; CHECK-NEXT:    [[IN_COERCE_FCA_1_EXTRACT:%.*]] = extractvalue [5 x i64] [[IN_COERCE]], 1
  56 ; CHECK-NEXT:    [[IN_COERCE_FCA_2_EXTRACT:%.*]] = extractvalue [5 x i64] [[IN_COERCE]], 2
  57 ; CHECK-NEXT:    [[IN_COERCE_FCA_3_EXTRACT:%.*]] = extractvalue [5 x i64] [[IN_COERCE]], 3
  58 ; CHECK-NEXT:    [[IN_SROA_2_4_EXTRACT_SHIFT:%.*]] = lshr i64 [[IN_COERCE_FCA_2_EXTRACT]], 32
  59 ; CHECK-NEXT:    [[IN_SROA_2_4_EXTRACT_TRUNC:%.*]] = trunc i64 [[IN_SROA_2_4_EXTRACT_SHIFT]] to i32
  60 ; CHECK-NEXT:    store i32 [[IN_SROA_2_4_EXTRACT_TRUNC]], ptr addrspace(1) [[PTR:%.*]], align 4
  61 ; CHECK-NEXT:    [[IN_SROA_4_20_PTR_SROA_IDX:%.*]] = getelementptr inbounds i8, ptr addrspace(1) [[PTR]], i16 4
  62 ; CHECK-NEXT:    store i64 [[IN_COERCE_FCA_3_EXTRACT]], ptr addrspace(1) [[IN_SROA_4_20_PTR_SROA_IDX]], align 4
  63 ; CHECK-NEXT:    [[IN_SROA_5_20_PTR_SROA_IDX:%.*]] = getelementptr inbounds i8, ptr addrspace(1) [[PTR]], i16 12
  64 ; CHECK-NEXT:    store i32 undef, ptr addrspace(1) [[IN_SROA_5_20_PTR_SROA_IDX]], align 4
  65 ; CHECK-NEXT:    ret void
  66 ;
  67 for.end:
  68   %in = alloca %struct.struct_test_27.0.13, align 8
  69   store [5 x i64] %in.coerce, ptr %in, align 8
  70   %scevgep9 = getelementptr %struct.struct_test_27.0.13, ptr %in, i32 0, i32 4, i32 0
  71   call void @llvm.memcpy.p1.p0.i32(ptr addrspace(1) align 4 %ptr, ptr align 4 %scevgep9, i32 16, i1 false)
  72   ret void
  73 }
  74
  75 %union.anon = type { ptr }
  76
  77 @g = common global i32 0, align 4
  78 @l = common addrspace(3) global i32 0, align 4
  79
  80 ; If pointers from different address spaces have different sizes, make sure an
  81 ; illegal bitcast isn't introduced
  82 define void @pr27557() {
  83 ; CHECK-LABEL: @pr27557(
  84 ; CHECK-NEXT:    [[DOTSROA_0:%.*]] = alloca ptr, align 8
  85 ; CHECK-NEXT:    store ptr @g, ptr [[DOTSROA_0]], align 8
  86 ; CHECK-NEXT:    store ptr addrspace(3) @l, ptr [[DOTSROA_0]], align 8
  87 ; CHECK-NEXT:    ret void
  88 ;
  89   %1 = alloca %union.anon, align 8
  90   store ptr @g, ptr %1, align 8
  91   store ptr addrspace(3) @l, ptr %1, align 8
  92   ret void
  93 }
  94
  95 @l2 = common addrspace(2) global i32 0, align 4
  96
  97 ; If pointers from different address spaces have the same size, that pointer
  98 ; should be promoted through the pair of `ptrtoint`/`inttoptr`.
  99 define ptr @pr27557.alt() {
 100 ; CHECK-LABEL: @pr27557.alt(
 101 ; CHECK-NEXT:    ret ptr inttoptr (i64 ptrtoint (ptr addrspace(2) @l2 to i64) to ptr)
 102 ;
 103   %1 = alloca %union.anon, align 8
 104   store ptr addrspace(2) @l2, ptr %1, align 8
 105   %2 = load ptr, ptr %1, align 8
 106   ret ptr %2
 107 }
 108
 109 ; Make sure pre-splitting doesn't try to introduce an illegal bitcast
 110 define float @presplit(ptr addrspace(1) %p) {
 111 ; CHECK-LABEL: @presplit(
 112 ; CHECK-NEXT:  entry:
 113 ; CHECK-NEXT:    [[L1:%.*]] = load i32, ptr addrspace(1) [[P:%.*]], align 4
 114 ; CHECK-NEXT:    [[P_SROA_IDX:%.*]] = getelementptr inbounds i8, ptr addrspace(1) [[P]], i16 4
 115 ; CHECK-NEXT:    [[L2:%.*]] = load i32, ptr addrspace(1) [[P_SROA_IDX]], align 4
 116 ; CHECK-NEXT:    [[TMP0:%.*]] = bitcast i32 [[L1]] to float
 117 ; CHECK-NEXT:    [[TMP1:%.*]] = bitcast i32 [[L2]] to float
 118 ; CHECK-NEXT:    [[RET:%.*]] = fadd float [[TMP0]], [[TMP1]]
 119 ; CHECK-NEXT:    ret float [[RET]]
 120 ;
 121 entry:
 122   %b = alloca i64
 123   %b.gep2 = getelementptr [2 x float], ptr %b, i32 0, i32 1
 124   %l = load i64, ptr addrspace(1) %p
 125   store i64 %l, ptr %b
 126   %f1 = load float, ptr %b
 127   %f2 = load float, ptr %b.gep2
 128   %ret = fadd float %f1, %f2
 129   ret float %ret
 130 }
 131
 132 ; Test load from and store to non-zero address space.
 133 define void @test_load_store_diff_addr_space(ptr addrspace(1) %complex1, ptr addrspace(1) %complex2) {
 134 ; CHECK-LABEL: @test_load_store_diff_addr_space(
 135 ; CHECK-NEXT:    [[V13:%.*]] = load i32, ptr addrspace(1) [[COMPLEX1:%.*]], align 4
 136 ; CHECK-NEXT:    [[COMPLEX1_SROA_IDX:%.*]] = getelementptr inbounds i8, ptr addrspace(1) [[COMPLEX1]], i16 4
 137 ; CHECK-NEXT:    [[V14:%.*]] = load i32, ptr addrspace(1) [[COMPLEX1_SROA_IDX]], align 4
 138 ; CHECK-NEXT:    [[TMP1:%.*]] = bitcast i32 [[V13]] to float
 139 ; CHECK-NEXT:    [[TMP2:%.*]] = bitcast i32 [[V14]] to float
 140 ; CHECK-NEXT:    [[SUM:%.*]] = fadd float [[TMP1]], [[TMP2]]
 141 ; CHECK-NEXT:    [[TMP3:%.*]] = bitcast float [[SUM]] to i32
 142 ; CHECK-NEXT:    [[TMP4:%.*]] = bitcast float [[SUM]] to i32
 143 ; CHECK-NEXT:    store i32 [[TMP3]], ptr addrspace(1) [[COMPLEX2:%.*]], align 4
 144 ; CHECK-NEXT:    [[COMPLEX2_SROA_IDX:%.*]] = getelementptr inbounds i8, ptr addrspace(1) [[COMPLEX2]], i16 4
 145 ; CHECK-NEXT:    store i32 [[TMP4]], ptr addrspace(1) [[COMPLEX2_SROA_IDX]], align 4
 146 ; CHECK-NEXT:    ret void
 147 ;
 148   %a = alloca i64
 149   %a.gep2 = getelementptr [2 x float], ptr %a, i32 0, i32 1
 150   %v1 = load i64, ptr addrspace(1) %complex1
 151   store i64 %v1, ptr %a
 152   %f1 = load float, ptr %a
 153   %f2 = load float, ptr %a.gep2
 154   %sum = fadd float %f1, %f2
 155   store float %sum, ptr %a
 156   store float %sum, ptr %a.gep2
 157   %v2 = load i64, ptr %a
 158   store i64 %v2, ptr addrspace(1) %complex2
 159   ret void
 160 }
 161 ;; NOTE: These prefixes are unused and the list is autogenerated. Do not add tests below this line:
 162 ; CHECK-MODIFY-CFG: {{.*}}
 163 ; CHECK-PRESERVE-CFG: {{.*}}