Add IsInfiniteJumpThreadingBudget for testing

Author: meg-gupta

lib/SILOptimizer/Transforms/SimplifyCFG.cpp

@@ -57,6 +57,11 @@ llvm::cl::opt<bool> EnableOSSARewriteTerminator(
         "Enable OSSA simplify-cfg with non-trivial terminator rewriting "
         "(staging)."));
 
+llvm::cl::opt<bool> IsInfiniteJumpThreadingBudget(
+    "sil-infinite-jump-threading-budget",
+    llvm::cl::desc(
+        "Use infinite budget for jump threading. Useful for testing purposes"));
+
 STATISTIC(NumBlocksDeleted, "Number of unreachable blocks removed");
 STATISTIC(NumBlocksMerged, "Number of blocks merged together");
 STATISTIC(NumJumpThreads, "Number of jumps threaded");
@@ -1068,7 +1073,7 @@ bool SimplifyCFG::tryJumpThreading(BranchInst *BI) {
   // major second order simplifications.  Here we only do it if there are
   // "constant" arguments to the branch or if we know how to fold something
   // given the duplication.
-  int ThreadingBudget = 0;
+  int ThreadingBudget = IsInfiniteJumpThreadingBudget ? INT_MAX : 0;
 
   for (unsigned i : indices(BI->getArgs())) {
     SILValue Arg = BI->getArg(i);