[llvm] 6c9cac5 - [AMDGPU] Add MDT update missing from D98915

[Carl Ritson via llvm-commits]

Author: Carl Ritson
Date: 2021-03-20T13:38:58+09:00
New Revision: 6c9cac5da1d1c64e2df4e9c7ea21355a34595a8a

URL: https://github.com/llvm/llvm-project/commit/6c9cac5da1d1c64e2df4e9c7ea21355a34595a8a
DIFF: https://github.com/llvm/llvm-project/commit/6c9cac5da1d1c64e2df4e9c7ea21355a34595a8a.diff

LOG: [AMDGPU] Add MDT update missing from D98915

Added: 
    

Modified: 
    llvm/lib/Target/AMDGPU/SILateBranchLowering.cpp

Removed: 
    


################################################################################
diff  --git a/llvm/lib/Target/AMDGPU/SILateBranchLowering.cpp b/llvm/lib/Target/AMDGPU/SILateBranchLowering.cpp
index 42cc09f8e484..3fb96f15313d 100644
--- a/llvm/lib/Target/AMDGPU/SILateBranchLowering.cpp
+++ b/llvm/lib/Target/AMDGPU/SILateBranchLowering.cpp
@@ -207,6 +207,7 @@ bool SILateBranchLowering::runOnMachineFunction(MachineFunction &MF) {
       }
 
       MBB->addSuccessor(EmptyMBBAtEnd);
+      MDT->getBase().insertEdge(MBB, EmptyMBBAtEnd);
       BuildMI(*MBB, MI, MI->getDebugLoc(), TII->get(AMDGPU::S_BRANCH))
           .addMBB(EmptyMBBAtEnd);
       MI->eraseFromParent();