; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 5
; RUN: opt -S -passes=infer-address-spaces %s | FileCheck %s

target triple = "nvptx64-nvidia-cuda"

define i32 @lifetime_flat_pointer() {
; CHECK-LABEL: define i32 @lifetime_flat_pointer() {
; CHECK-NEXT:    [[ALLOCA:%.*]] = alloca i32, align 4
; CHECK-NEXT:    [[TMP1:%.*]] = addrspacecast ptr [[ALLOCA]] to ptr addrspace(5)
; CHECK-NEXT:    call void @llvm.lifetime.start.p0(ptr [[ALLOCA]])
; CHECK-NEXT:    store i32 1, ptr addrspace(5) [[TMP1]], align 4
; CHECK-NEXT:    [[RET:%.*]] = load i32, ptr addrspace(5) [[TMP1]], align 4
; CHECK-NEXT:    call void @llvm.lifetime.end.p0(ptr [[ALLOCA]])
; CHECK-NEXT:    ret i32 [[RET]]
;
  %alloca = alloca i32, align 4
  %1 = addrspacecast ptr %alloca to ptr addrspace(5)
  call void @llvm.lifetime.start.p0(ptr %alloca)
  store i32 1, ptr addrspace(5) %1, align 4
  %ret = load i32, ptr addrspace(5) %1, align 4
  call void @llvm.lifetime.end.p0(ptr %alloca)
  ret i32 %ret
}

declare void @llvm.lifetime.start.p0(ptr nocapture)
declare void @llvm.lifetime.end.p0(ptr nocapture)