Teach ValueTracking that aarch64.irg result aliases its input.

Reviewers: javed.absar, olista01

Subscribers: kristof.beyls, hiraditya, llvm-commits

Tags: #llvm

Differential Revision: https://reviews.llvm.org/D64103

llvm-svn: 365079
diff --git a/llvm/lib/Analysis/ValueTracking.cpp b/llvm/lib/Analysis/ValueTracking.cpp
index ebd2145..3ff5bf0 100644
--- a/llvm/lib/Analysis/ValueTracking.cpp
+++ b/llvm/lib/Analysis/ValueTracking.cpp
@@ -3697,7 +3697,8 @@
 bool llvm::isIntrinsicReturningPointerAliasingArgumentWithoutCapturing(
     const CallBase *Call) {
   return Call->getIntrinsicID() == Intrinsic::launder_invariant_group ||
-         Call->getIntrinsicID() == Intrinsic::strip_invariant_group;
+         Call->getIntrinsicID() == Intrinsic::strip_invariant_group ||
+         Call->getIntrinsicID() == Intrinsic::aarch64_irg;
 }
 
 /// \p PN defines a loop-variant pointer to an object.  Check if the
diff --git a/llvm/test/Analysis/ValueTracking/aarch64.irg.ll b/llvm/test/Analysis/ValueTracking/aarch64.irg.ll
new file mode 100644
index 0000000..75e198a
--- /dev/null
+++ b/llvm/test/Analysis/ValueTracking/aarch64.irg.ll
@@ -0,0 +1,18 @@
+; RUN: opt -S -instsimplify -instcombine < %s | FileCheck %s
+
+; CHECK-LABEL: define void @checkNonnullIrg()
+define void @checkNonnullIrg() {
+; CHECK:   %[[p:.*]] = call i8* @llvm.aarch64.irg(i8* nonnull
+; CHECK:   call void @use(i8* nonnull %[[p]])
+entry:
+  %0 = alloca i8, align 16
+
+  %p = call i8* @llvm.aarch64.irg(i8* %0, i64 5)
+  call void @use(i8* %p)
+
+  ret void
+}
+
+declare i8* @llvm.aarch64.irg(i8*, i64)
+
+declare void @use(i8*)