From 9430569e0817c61d6fdaa37cf5cc35445f2f6289 Mon Sep 17 00:00:00 2001 From: Richard-Qin-X Date: Fri, 14 Aug 2026 21:39:49 +0800 Subject: [MIR] Normalize explicit successor probabilities --- backend/lib/CodeGen/MIRParser/MIParser.cpp | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) (limited to 'backend/lib/CodeGen/MIRParser/MIParser.cpp') diff --git a/backend/lib/CodeGen/MIRParser/MIParser.cpp b/backend/lib/CodeGen/MIRParser/MIParser.cpp index b21e713..62f1f14 100644 --- a/backend/lib/CodeGen/MIRParser/MIParser.cpp +++ b/backend/lib/CodeGen/MIRParser/MIParser.cpp @@ -2500,6 +2500,9 @@ bool MIParser::Impl::parseBlockSuccessors(MachineBasicBlock &Block) { ExplicitSuccessorBlocks.insert(&Block); if (token().isNewlineOrEnd()) return false; + + std::vector Successors; + std::vector Probabilities; do { if (!at(MITokenKind::MachineBasicBlock)) return fail("expected a machine basic block reference"); @@ -2517,10 +2520,17 @@ bool MIParser::Impl::parseBlockSuccessors(MachineBasicBlock &Block) { if (expect(MITokenKind::RightParen, "')' after successor weight")) return true; } - auto Added = Function.cfgUpdater().addSuccessor(Block, *Successor, Probability); + Successors.push_back(Successor); + Probabilities.push_back(Probability); + } while (consume(MITokenKind::Comma)); + + BranchProbability::normalizeProbabilities(Probabilities); + for (std::size_t Index = 0; Index < Successors.size(); ++Index) { + auto Added = + Function.cfgUpdater().addSuccessor(Block, *Successors[Index], Probabilities[Index]); if (!Added) return importError(token(), Added.error()); - } while (consume(MITokenKind::Comma)); + } return false; } -- cgit v1.2.3