partialexecuter: increment and check basicblock visit count after ter…

…minals
leaningtech · Feb 8, 2024 · 0efaba0 · 0efaba0
1 parent ce7efcc
commit 0efaba0
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/llvm/lib/CheerpWriter/PartialExecuter.cpp b/llvm/lib/CheerpWriter/PartialExecuter.cpp
@@ -1699,8 +1699,10 @@ void PartialInterpreter::visitOuter(FunctionData& data, llvm::Instruction& I)
 		{
 			BasicBlock* next = findNextBasicBlock(I);
 
-			if (next)
+			if (next && data.getVisitCounter(next) < MAX_NUMBER_OF_VISITS_PER_BB)
 			{
+				data.incrementVisitCounter(next);
+
 				// We know where execution should proceed
 				incomingBB = I.getParent();
 				getTopCallFrame().CurBB = next;
@@ -1709,10 +1711,7 @@ void PartialInterpreter::visitOuter(FunctionData& data, llvm::Instruction& I)
 				// Also here we have set the proper state for the execution so we can return
 				return;
 			}
-			else
-			{
-				skip = true;
-			}
+			skip = true;
 		}
 
 		if (addToCounter(I.getFunction()))