Apply suggestions from code review

mtrofin · web-flow · commit bce84a4544bb · 2025-10-31T15:48:16.000-07:00
diff --git a/llvm/lib/Transforms/Scalar/SimpleLoopUnswitch.cpp b/llvm/lib/Transforms/Scalar/SimpleLoopUnswitch.cpp
@@ -387,7 +387,7 @@ static void buildPartialInvariantUnswitchConditionalBranch(
                        Direction ? &NormalSucc : &UnswitchedSucc, ProfData);
   if (!ProfData)
     setExplicitlyUnknownBranchWeightsIfProfiled(
-        *BR, *BR->getParent()->getParent(), DEBUG_TYPE);
+        *BR, *BR->getFunction(), DEBUG_TYPE);
 }
 
 /// Rewrite the PHI nodes in an unswitched loop exit basic block.

Original file line number	Diff line number	Diff line change
`@@ -387,7 +387,7 @@ static void buildPartialInvariantUnswitchConditionalBranch(`
`387`	`387`	`Direction ? &NormalSucc : &UnswitchedSucc, ProfData);`
`388`	`388`	`if (!ProfData)`
`389`	`389`	`setExplicitlyUnknownBranchWeightsIfProfiled(`
`390`		`- BR, BR->getParent()->getParent(), DEBUG_TYPE);`
	`390`	`+ BR, BR->getFunction(), DEBUG_TYPE);`
`391`	`391`	`}`
`392`	`392`
`393`	`393`	`/// Rewrite the PHI nodes in an unswitched loop exit basic block.`