llvm/llvm/test/Transforms/Attributor/ArgumentPromotion/reserve-tbaa.ll

; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --function-signature --check-attributes --check-globals
; RUN: opt -aa-pipeline=basic-aa -passes=attributor -attributor-manifest-internal  -attributor-annotate-decl-cs  -S < %s | FileCheck %s --check-prefixes=CHECK,TUNIT
; RUN: opt -aa-pipeline=basic-aa -passes=attributor-cgscc -attributor-manifest-internal  -attributor-annotate-decl-cs -S < %s | FileCheck %s --check-prefixes=CHECK,CGSCC

; PR17906
; When we promote two arguments in a single function with different types,
; before the fix, we used the same tag for the newly-created two loads.
; This testing case makes sure that we correctly transfer the tbaa tags from the
; original loads to the newly-created loads when promoting pointer arguments.

@a = global ptr null, align 8
@e = global ptr @a, align 8
@g = global i32 0, align 4
@c = global i64 0, align 8
@d = global i8 0, align 1

;.
; CHECK: @a = global ptr null, align 8
; CHECK: @e = global ptr @a, align 8
; CHECK: @g = global i32 0, align 4
; CHECK: @c = global i64 0, align 8
; CHECK: @d = global i8 0, align 1
;.
define internal fastcc void @fn(ptr nocapture readonly %p1, ptr nocapture readonly %p2) {
; CHECK: Function Attrs: mustprogress nofree norecurse nosync nounwind willreturn memory(readwrite, argmem: none)
; CHECK-LABEL: define {{[^@]+}}@fn
; CHECK-SAME: () #[[ATTR0:[0-9]+]] {
; CHECK-NEXT:  entry:
; CHECK-NEXT:    [[TMP0:%.*]] = load i32, ptr @g, align 4, !tbaa [[TBAA0:![0-9]+]]
; CHECK-NEXT:    [[CONV1:%.*]] = trunc i32 [[TMP0]] to i8
; CHECK-NEXT:    store i8 [[CONV1]], ptr @d, align 1, !tbaa [[TBAA4:![0-9]+]]
; CHECK-NEXT:    ret void
;
entry:
  %0 = load i64, ptr %p2, align 8, !tbaa !1
  %conv = trunc i64 %0 to i32
  %1 = load i32, ptr %p1, align 4, !tbaa !5
  %conv1 = trunc i32 %1 to i8
  store i8 %conv1, ptr @d, align 1, !tbaa !7
  ret void
}

define i32 @main() {
; TUNIT: Function Attrs: mustprogress nofree norecurse nosync nounwind willreturn
; TUNIT-LABEL: define {{[^@]+}}@main
; TUNIT-SAME: () #[[ATTR1:[0-9]+]] {
; TUNIT-NEXT:  entry:
; TUNIT-NEXT:    [[TMP0:%.*]] = load ptr, ptr @e, align 8, !tbaa [[TBAA5:![0-9]+]]
; TUNIT-NEXT:    store ptr @g, ptr [[TMP0]], align 8, !tbaa [[TBAA5]]
; TUNIT-NEXT:    [[TMP1:%.*]] = load ptr, ptr @a, align 8, !tbaa [[TBAA5]]
; TUNIT-NEXT:    store i32 1, ptr [[TMP1]], align 4, !tbaa [[TBAA0]]
; TUNIT-NEXT:    call fastcc void @fn() #[[ATTR2:[0-9]+]]
; TUNIT-NEXT:    ret i32 0
;
; CGSCC: Function Attrs: mustprogress nofree nosync nounwind willreturn
; CGSCC-LABEL: define {{[^@]+}}@main
; CGSCC-SAME: () #[[ATTR1:[0-9]+]] {
; CGSCC-NEXT:  entry:
; CGSCC-NEXT:    [[TMP0:%.*]] = load ptr, ptr @e, align 8, !tbaa [[TBAA5:![0-9]+]]
; CGSCC-NEXT:    store ptr @g, ptr [[TMP0]], align 8, !tbaa [[TBAA5]]
; CGSCC-NEXT:    [[TMP1:%.*]] = load ptr, ptr @a, align 8, !tbaa [[TBAA5]]
; CGSCC-NEXT:    store i32 1, ptr [[TMP1]], align 4, !tbaa [[TBAA0]]
; CGSCC-NEXT:    call fastcc void @fn() #[[ATTR2:[0-9]+]]
; CGSCC-NEXT:    ret i32 0
;
entry:
  %0 = load ptr, ptr @e, align 8, !tbaa !8
  store ptr @g, ptr %0, align 8, !tbaa !8
  %1 = load ptr, ptr @a, align 8, !tbaa !8
  store i32 1, ptr %1, align 4, !tbaa !5
  call fastcc void @fn(ptr @g, ptr @c)

  ret i32 0
}

!1 = !{!2, !2, i64 0}
!2 = !{!"long", !3, i64 0}
!3 = !{!"omnipotent char", !4, i64 0}
!4 = !{!"Simple C/C++ TBAA"}
!5 = !{!6, !6, i64 0}
!6 = !{!"int", !3, i64 0}
!7 = !{!3, !3, i64 0}
!8 = !{!9, !9, i64 0}
!9 = !{!"any pointer", !3, i64 0}

;.
; TUNIT: attributes #[[ATTR0]] = { mustprogress nofree norecurse nosync nounwind willreturn memory(readwrite, argmem: none) }
; TUNIT: attributes #[[ATTR1]] = { mustprogress nofree norecurse nosync nounwind willreturn }
; TUNIT: attributes #[[ATTR2]] = { nofree nosync nounwind willreturn }
;.
; CGSCC: attributes #[[ATTR0]] = { mustprogress nofree norecurse nosync nounwind willreturn memory(readwrite, argmem: none) }
; CGSCC: attributes #[[ATTR1]] = { mustprogress nofree nosync nounwind willreturn }
; CGSCC: attributes #[[ATTR2]] = { nofree nounwind willreturn }
;.
; TUNIT: [[TBAA0]] = !{[[META1:![0-9]+]], [[META1]], i64 0}
; TUNIT: [[META1]] = !{!"int", [[META2:![0-9]+]], i64 0}
; TUNIT: [[META2]] = !{!"omnipotent char", [[META3:![0-9]+]], i64 0}
; TUNIT: [[META3]] = !{!"Simple C/C++ TBAA"}
; TUNIT: [[TBAA4]] = !{[[META2]], [[META2]], i64 0}
; TUNIT: [[TBAA5]] = !{[[META6:![0-9]+]], [[META6]], i64 0}
; TUNIT: [[META6]] = !{!"any pointer", [[META2]], i64 0}
;.
; CGSCC: [[TBAA0]] = !{[[META1:![0-9]+]], [[META1]], i64 0}
; CGSCC: [[META1]] = !{!"int", [[META2:![0-9]+]], i64 0}
; CGSCC: [[META2]] = !{!"omnipotent char", [[META3:![0-9]+]], i64 0}
; CGSCC: [[META3]] = !{!"Simple C/C++ TBAA"}
; CGSCC: [[TBAA4]] = !{[[META2]], [[META2]], i64 0}
; CGSCC: [[TBAA5]] = !{[[META6:![0-9]+]], [[META6]], i64 0}
; CGSCC: [[META6]] = !{!"any pointer", [[META2]], i64 0}
;.