From 0d6b01761c1db42ae13c2bf91ffdea4b6ebd9f69 Mon Sep 17 00:00:00 2001
From: Florian Hahn <florian.hahn@arm.com>
Date: Thu, 7 Jun 2018 11:09:05 +0000
Subject: [PATCH] [Mem2Reg] Avoid replacing load with itself in
 promoteSingleBlockAlloca.

We do the same thing in rewriteSingleStoreAlloca.

Fixes PR37632.

Reviewers: chandlerc, davide, efriedma

Reviewed By: davide

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

llvm-svn: 334187
---
 .../Utils/PromoteMemoryToRegister.cpp         |  5 +++++
 .../pr37632-unreachable-list-of-stores.ll     | 20 +++++++++++++++++++
 2 files changed, 25 insertions(+)
 create mode 100644 llvm/test/Transforms/Mem2Reg/pr37632-unreachable-list-of-stores.ll

diff --git a/llvm/lib/Transforms/Utils/PromoteMemoryToRegister.cpp b/llvm/lib/Transforms/Utils/PromoteMemoryToRegister.cpp
index e161690f5bfe..86e15bbd7f22 100644
--- a/llvm/lib/Transforms/Utils/PromoteMemoryToRegister.cpp
+++ b/llvm/lib/Transforms/Utils/PromoteMemoryToRegister.cpp
@@ -511,6 +511,11 @@ static bool promoteSingleBlockAlloca(AllocaInst *AI, const AllocaInfo &Info,
           !isKnownNonZero(ReplVal, DL, 0, AC, LI, &DT))
         addAssumeNonNull(AC, LI);
 
+      // If the replacement value is the load, this must occur in unreachable
+      // code.
+      if (ReplVal == LI)
+        ReplVal = UndefValue::get(LI->getType());
+
       LI->replaceAllUsesWith(ReplVal);
     }
 
diff --git a/llvm/test/Transforms/Mem2Reg/pr37632-unreachable-list-of-stores.ll b/llvm/test/Transforms/Mem2Reg/pr37632-unreachable-list-of-stores.ll
new file mode 100644
index 000000000000..bb4c0f1d57f7
--- /dev/null
+++ b/llvm/test/Transforms/Mem2Reg/pr37632-unreachable-list-of-stores.ll
@@ -0,0 +1,20 @@
+; RUN: opt -mem2reg < %s -S | FileCheck %s
+
+
+; CHECK-LABEL: void @patatino()
+; CHECK-NEXT: ret void
+
+; CHECK-LABEL: cantreachme:
+; CHECK-NEXT: %dec = add nsw i32 undef, -1
+; CHECK-NEXT: br label %cantreachme
+
+define void @patatino() {
+  %a = alloca i32, align 4
+  ret void
+cantreachme:
+  %dec = add nsw i32 %tmp, -1
+  store i32 %dec, i32* %a
+  store i32 %tmp, i32* %a
+  %tmp = load i32, i32* %a
+  br label %cantreachme
+}