fixup! fixup! [InferAddressSpaces] Handle llvm.lifetime

Author: Zhenhao Yang

llvm/test/Transforms/InferAddressSpaces/NVPTX/lifetime.ll

@@ -4,24 +4,6 @@ target triple = "nvptx64-nvidia-cuda"
 
 define i32 @lifetime_flat_pointer() {
 ; CHECK-LABEL: define i32 @lifetime_flat_pointer() {
-; CHECK-NEXT:    [[ALLOCA:%.*]] = alloca i32, align 4, addrspace(5)
-; CHECK-NEXT:    call void @llvm.lifetime.start.p5(i64 4, ptr addrspace(5) [[ALLOCA]])
-; CHECK-NEXT:    store i32 1, ptr addrspace(5) [[ALLOCA]], align 4
-; CHECK-NEXT:    %ret = load i32, ptr addrspace(5) [[ALLOCA]], align 4
-; CHECK-NEXT:    call void @llvm.lifetime.end.p5(i64 4, ptr addrspace(5) [[ALLOCA]])
-; CHECK-NEXT:    ret i32 %ret
-;
-  %alloca = alloca i32, align 4, addrspace(5)
-  %flat = addrspacecast ptr addrspace(5) %alloca to ptr
-  call void @llvm.lifetime.start.p0(i64 4 , ptr %flat)
-  store i32 1, ptr %flat, align 4
-  %ret = load i32, ptr %flat, align 4
-  call void @llvm.lifetime.end.p0(i64 4 , ptr %flat)
-  ret i32 %ret
-}
-
-define i32 @lifetime_flat_pointer2() {
-; CHECK-LABEL: define i32 @lifetime_flat_pointer2() {
 ; CHECK-NEXT:    [[ALLOCA:%.*]] = alloca i32, align 4
 ; CHECK-NEXT:    [[TMP1:%.*]] = addrspacecast ptr [[ALLOCA]] to ptr addrspace(5)
 ; CHECK-NEXT:    call void @llvm.lifetime.start.p5(i64 4, ptr addrspace(5) [[TMP1]])
@@ -31,10 +13,13 @@ define i32 @lifetime_flat_pointer2() {
 ; CHECK-NEXT:    ret i32 %ret
 ;
   %alloca = alloca i32, align 4
-  call void @llvm.lifetime.start.p5(i64 4 , ptr %alloca)
-  store i32 1, ptr %alloca, align 4
-  %ret = load i32, ptr %alloca, align 4
-  call void @llvm.lifetime.end.p5(i64 4 , ptr %alloca)
+  %1 = addrspacecast ptr %alloca to ptr addrspace(5)
+  %2 = addrspacecast ptr addrspace(5) %1 to ptr
+  %3 = addrspacecast ptr addrspace(5) %1 to ptr
+  call void @llvm.lifetime.start.p0(i64 4, ptr %2)
+  store i32 1, ptr addrspace(5) %1, align 4
+  %ret = load i32, ptr addrspace(5) %1, align 4
+  call void @llvm.lifetime.end.p0(i64 4, ptr %3)
   ret i32 %ret
 }