; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
; RUN: opt -S -mtriple=nvptx64-nvidia-cuda -passes='require<domtree>,infer-address-spaces' %s | FileCheck %s

;;; Handle write corner case for infer-address-spaces with phi-nodes. The
;;; verifier will crash if we insert `addrspacecast` before phi-node.

declare void @llvm.assume(i1 noundef)
declare i1 @llvm.nvvm.isspacep.shared(ptr)
declare i1 @llvm.nvvm.isspacep.global(ptr)

define void @phinode_instr() {
; CHECK-LABEL: @phinode_instr(
; CHECK-NEXT:  entry:
; CHECK-NEXT:    [[PTR_1:%.*]] = load ptr, ptr null, align 8
; CHECK-NEXT:    [[TMP0:%.*]] = addrspacecast ptr [[PTR_1]] to ptr addrspace(3)
; CHECK-NEXT:    [[BOOL_1:%.*]] = tail call i1 @llvm.nvvm.isspacep.shared(ptr [[PTR_1]])
; CHECK-NEXT:    tail call void @llvm.assume(i1 [[BOOL_1]])
; CHECK-NEXT:    br label [[IF_SINK_SPLIT:%.*]]
; CHECK:       if.sink.split:
; CHECK-NEXT:    [[PTR_SINK:%.*]] = phi ptr addrspace(3) [ [[TMP0]], [[ENTRY:%.*]] ]
; CHECK-NEXT:    store i32 1, ptr addrspace(3) [[PTR_SINK]], align 4
; CHECK-NEXT:    ret void
;
entry:
  %ptr.1 = load ptr, ptr null, align 8
  %bool.1 = tail call i1 @llvm.nvvm.isspacep.shared(ptr %ptr.1)
  tail call void @llvm.assume(i1 %bool.1)
  br label %if.sink.split

if.sink.split:                                    ; preds = %entry
  %ptr.sink = phi ptr [ %ptr.1, %entry ]
  store i32 1, ptr %ptr.sink, align 4
  ret void
}

define void @phinode_argument(ptr %lhs_ptr) {
; CHECK-LABEL: @phinode_argument(
; CHECK-NEXT:  entry:
; CHECK-NEXT:    [[TMP0:%.*]] = addrspacecast ptr [[LHS_PTR:%.*]] to ptr addrspace(1)
; CHECK-NEXT:    [[BOOL_1:%.*]] = tail call i1 @llvm.nvvm.isspacep.global(ptr [[LHS_PTR]])
; CHECK-NEXT:    tail call void @llvm.assume(i1 [[BOOL_1]])
; CHECK-NEXT:    br label [[IF_SINK_SPLIT:%.*]]
; CHECK:       if.sink.split:
; CHECK-NEXT:    [[PTR_SINK:%.*]] = phi ptr addrspace(1) [ [[TMP0]], [[ENTRY:%.*]] ]
; CHECK-NEXT:    store i32 1, ptr addrspace(1) [[PTR_SINK]], align 4
; CHECK-NEXT:    ret void
;
entry:
  %bool.1 = tail call i1 @llvm.nvvm.isspacep.global(ptr %lhs_ptr)
  tail call void @llvm.assume(i1 %bool.1)
  br label %if.sink.split

if.sink.split:                                    ; preds = %entry
  %ptr.sink = phi ptr [ %lhs_ptr, %entry ]
  store i32 1, ptr %ptr.sink, align 4
  ret void
}