[llvm] [CaptureTracking][AA] Only consider provenance captures (PR #130777)

[Nikita Popov via llvm-commits]

================
@@ -0,0 +1,42 @@
+; RUN: opt < %s -passes=aa-eval -print-all-alias-modref-info -disable-output 2>&1 | FileCheck %s
+
+declare void @capture(ptr)
+declare ptr @get_ptr()
+
+; CHECK-LABEL: address_capture
+; CHECK: NoAlias:	i32* %a, i32* %p
+; CHECK: NoModRef:  Ptr: i32* %a	<->  %p = call ptr @get_ptr()
+define void @address_capture() {
+  %a = alloca i32
+  call void @capture(ptr captures(address) %a)
+  %p = call ptr @get_ptr()
+  store i32 0, ptr %p
+  load i32, ptr %a
+  ret void
+}
+
+; CHECK-LABEL: read_only_capture
+; CHECK: MayAlias:	i32* %a, i32* %p
+; CHECK: Both ModRef:  Ptr: i32* %a	<->  %p = call ptr @get_ptr()
+; TODO: The ModRef could be just Ref.
+define void @read_only_capture() {
+  %a = alloca i32
+  call void @capture(ptr captures(address, read_provenance) %a)
+  %p = call ptr @get_ptr()
+  store i32 0, ptr %p
+  load i32, ptr %a
+  ret void
+}
+
+; CHECK-LABEL: address_capture_and_full_capture
+; CHECK: MayAlias:	i32* %a, i32* %p
+; CHECK: Both ModRef:  Ptr: i32* %a	<->  %p = call ptr @get_ptr()
+define void @address_capture_and_full_capture() {
+  %a = alloca i32
+  call void @capture(ptr captures(address) %a)
----------------
nikic wrote:

The intent of this test is to check the case where there are captures of different types, to make sure we don't incorrectly stop the traversal when seeing the captures(address).

https://github.com/llvm/llvm-project/pull/130777