Home | History | Annotate | Download | only in AMDGPU
      1 ; RUN: opt -S -mtriple=amdgcn-unknown-amdhsa -amdgpu-promote-alloca < %s | FileCheck %s
      2 
      3 declare void @llvm.memcpy.p0i8.p1i8.i32(i8* nocapture, i8 addrspace(1)* nocapture, i32, i32, i1) #0
      4 declare void @llvm.memcpy.p1i8.p0i8.i32(i8 addrspace(1)* nocapture, i8* nocapture, i32, i32, i1) #0
      5 
      6 declare void @llvm.memmove.p0i8.p1i8.i32(i8* nocapture, i8 addrspace(1)* nocapture, i32, i32, i1) #0
      7 declare void @llvm.memmove.p1i8.p0i8.i32(i8 addrspace(1)* nocapture, i8* nocapture, i32, i32, i1) #0
      8 
      9 declare void @llvm.memset.p0i8.i32(i8* nocapture, i8, i32, i32, i1) #0
     10 
     11 declare i32 @llvm.objectsize.i32.p0i8(i8*, i1) #1
     12 
     13 ; CHECK-LABEL: @promote_with_memcpy(
     14 ; CHECK: getelementptr inbounds [64 x [17 x i32]], [64 x [17 x i32]] addrspace(3)* @promote_with_memcpy.alloca, i32 0, i32 %{{[0-9]+}}
     15 ; CHECK: call void @llvm.memcpy.p3i8.p1i8.i32(i8 addrspace(3)* %alloca.bc, i8 addrspace(1)* %in.bc, i32 68, i32 4, i1 false)
     16 ; CHECK: call void @llvm.memcpy.p1i8.p3i8.i32(i8 addrspace(1)* %out.bc, i8 addrspace(3)* %alloca.bc, i32 68, i32 4, i1 false)
     17 define void @promote_with_memcpy(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 {
     18   %alloca = alloca [17 x i32], align 4
     19   %alloca.bc = bitcast [17 x i32]* %alloca to i8*
     20   %in.bc = bitcast i32 addrspace(1)* %in to i8 addrspace(1)*
     21   %out.bc = bitcast i32 addrspace(1)* %out to i8 addrspace(1)*
     22   call void @llvm.memcpy.p0i8.p1i8.i32(i8* %alloca.bc, i8 addrspace(1)* %in.bc, i32 68, i32 4, i1 false)
     23   call void @llvm.memcpy.p1i8.p0i8.i32(i8 addrspace(1)* %out.bc, i8* %alloca.bc, i32 68, i32 4, i1 false)
     24   ret void
     25 }
     26 
     27 ; CHECK-LABEL: @promote_with_memmove(
     28 ; CHECK: getelementptr inbounds [64 x [17 x i32]], [64 x [17 x i32]] addrspace(3)* @promote_with_memmove.alloca, i32 0, i32 %{{[0-9]+}}
     29 ; CHECK: call void @llvm.memmove.p3i8.p1i8.i32(i8 addrspace(3)* %alloca.bc, i8 addrspace(1)* %in.bc, i32 68, i32 4, i1 false)
     30 ; CHECK: call void @llvm.memmove.p1i8.p3i8.i32(i8 addrspace(1)* %out.bc, i8 addrspace(3)* %alloca.bc, i32 68, i32 4, i1 false)
     31 define void @promote_with_memmove(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 {
     32   %alloca = alloca [17 x i32], align 4
     33   %alloca.bc = bitcast [17 x i32]* %alloca to i8*
     34   %in.bc = bitcast i32 addrspace(1)* %in to i8 addrspace(1)*
     35   %out.bc = bitcast i32 addrspace(1)* %out to i8 addrspace(1)*
     36   call void @llvm.memmove.p0i8.p1i8.i32(i8* %alloca.bc, i8 addrspace(1)* %in.bc, i32 68, i32 4, i1 false)
     37   call void @llvm.memmove.p1i8.p0i8.i32(i8 addrspace(1)* %out.bc, i8* %alloca.bc, i32 68, i32 4, i1 false)
     38   ret void
     39 }
     40 
     41 ; CHECK-LABEL: @promote_with_memset(
     42 ; CHECK: getelementptr inbounds [64 x [17 x i32]], [64 x [17 x i32]] addrspace(3)* @promote_with_memset.alloca, i32 0, i32 %{{[0-9]+}}
     43 ; CHECK: call void @llvm.memset.p3i8.i32(i8 addrspace(3)* %alloca.bc, i8 7, i32 68, i32 4, i1 false)
     44 define void @promote_with_memset(i32 addrspace(1)* %out, i32 addrspace(1)* %in) #0 {
     45   %alloca = alloca [17 x i32], align 4
     46   %alloca.bc = bitcast [17 x i32]* %alloca to i8*
     47   %in.bc = bitcast i32 addrspace(1)* %in to i8 addrspace(1)*
     48   %out.bc = bitcast i32 addrspace(1)* %out to i8 addrspace(1)*
     49   call void @llvm.memset.p0i8.i32(i8* %alloca.bc, i8 7, i32 68, i32 4, i1 false)
     50   ret void
     51 }
     52 
     53 ; CHECK-LABEL: @promote_with_objectsize(
     54 ; CHECK: [[PTR:%[0-9]+]] = getelementptr inbounds [64 x [17 x i32]], [64 x [17 x i32]] addrspace(3)* @promote_with_objectsize.alloca, i32 0, i32 %{{[0-9]+}}
     55 ; CHECK: call i32 @llvm.objectsize.i32.p3i8(i8 addrspace(3)* %alloca.bc, i1 false)
     56 define void @promote_with_objectsize(i32 addrspace(1)* %out) #0 {
     57   %alloca = alloca [17 x i32], align 4
     58   %alloca.bc = bitcast [17 x i32]* %alloca to i8*
     59   %size = call i32 @llvm.objectsize.i32.p0i8(i8* %alloca.bc, i1 false)
     60   store i32 %size, i32 addrspace(1)* %out
     61   ret void
     62 }
     63 
     64 attributes #0 = { nounwind "amdgpu-max-work-group-size"="64" "amdgpu-max-waves-per-eu"="3" }
     65 attributes #1 = { nounwind readnone }
     66