[BranchFolding] Set correct mem refs

Merge MBBICommon and MBBI's MMOs.

Differential Revision: http://reviews.llvm.org/D15990

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@257253 91177308-0d34-0410-b5e6-96231b3b80d8

diff --git a/lib/CodeGen/BranchFolding.cpp b/lib/CodeGen/BranchFolding.cpp
index 60803490a5de61001c9828d3b09e3945a5ffe99e..df5cac5a9f7abb17a7fddd93cb64b39daf9145b4 100644 (file)
--- a/lib/CodeGen/BranchFolding.cpp
+++ b/lib/CodeGen/BranchFolding.cpp
@@ -780,7 +780,7 @@ removeMMOsFromMemoryOperations(MachineBasicBlock::iterator MBBIStartPos,
     assert(MBBICommon->isIdenticalTo(&*MBBI) && "Expected matching MIIs!");
 
     if (MBBICommon->mayLoad() || MBBICommon->mayStore())
     assert(MBBICommon->isIdenticalTo(&*MBBI) && "Expected matching MIIs!");
 
     if (MBBICommon->mayLoad() || MBBICommon->mayStore())

-      MBBICommon->setMemRefs(MBBI->mergeMemRefsWith(*MBBI));
+      MBBICommon->setMemRefs(MBBICommon->mergeMemRefsWith(*MBBI));

 
     ++MBBI;
     ++MBBICommon;
 
     ++MBBI;
     ++MBBICommon;

diff --git a/test/CodeGen/AArch64/branch-folder-merge-mmos.ll b/test/CodeGen/AArch64/branch-folder-merge-mmos.ll
new file mode 100644 (file)
index 0000000..c62c16a
--- /dev/null
+++ b/test/CodeGen/AArch64/branch-folder-merge-mmos.ll
@@ -0,0 +1,53 @@
+; RUN: opt < %s -O3 | llc -march=aarch64 -mtriple=aarch64-none-linux-gnu -stop-after branch-folder -o /dev/null | FileCheck %s
+target datalayout = "e-m:e-i64:64-i128:128-n32:64-S128"
+
+; Function Attrs: nounwind
+define void @test(float* %test_arr) #0 {
+entry:
+  %test_arr.addr = alloca float*, align 8
+  store float* %test_arr, float** %test_arr.addr, align 8
+  %0 = load float*, float** %test_arr.addr, align 8
+  %arrayidx = getelementptr inbounds float, float* %0, i64 0
+  %1 = load float, float* %arrayidx, align 4
+  %2 = load float*, float** %test_arr.addr, align 8
+  %arrayidx1 = getelementptr inbounds float, float* %2, i64 1
+  %3 = load float, float* %arrayidx1, align 4
+  %sub = fsub float %1, %3
+  %4 = load float*, float** %test_arr.addr, align 8
+  %arrayidx2 = getelementptr inbounds float, float* %4, i64 0
+  store float %sub, float* %arrayidx2, align 4
+  ret void
+}
+
+; Function Attrs: nounwind
+define void @foo(i32 %a, i32 %b, float* %foo_arr) #0 {
+; CHECK: (load 4 from %ir.arrayidx1.i2), (load 4 from %ir.arrayidx1.i)
+entry:
+  %a.addr = alloca i32, align 4
+  %b.addr = alloca i32, align 4
+  %foo_arr.addr = alloca float*, align 8
+  store i32 %a, i32* %a.addr, align 4
+  store i32 %b, i32* %b.addr, align 4
+  store float* %foo_arr, float** %foo_arr.addr, align 8
+  %0 = load i32, i32* %a.addr, align 4
+  %cmp = icmp sgt i32 %0, 0
+  br i1 %cmp, label %if.then, label %if.end
+
+if.then:                                          ; preds = %entry
+  %1 = load float*, float** %foo_arr.addr, align 8
+  call void @test(float* %1)
+  br label %if.end3
+
+if.end:                                           ; preds = %entry
+  %2 = load i32, i32* %b.addr, align 4
+  %cmp1 = icmp sgt i32 %2, 0
+  br i1 %cmp1, label %if.then2, label %if.end3
+
+if.then2:                                         ; preds = %if.end
+  %3 = load float*, float** %foo_arr.addr, align 8
+  call void @test(float* %3)
+  br label %if.end3
+
+if.end3:                                          ; preds = %if.then, %if.then2, %if.end
+  ret void
+}