Merge pull request #80097 from aschwaighofer/remove_cond_fails_by_message

SIL: Allow to selectively disabled cond_fails by cond_fail message

Author: aschwaighofer

lib/SILOptimizer/SILCombiner/SILCombine.cpp

@@ -43,7 +43,10 @@
 #include "llvm/ADT/SmallPtrSet.h"
 #include "llvm/ADT/SmallVector.h"
 #include "llvm/ADT/Statistic.h"
+#include "llvm/Support/CommandLine.h"
 #include "llvm/Support/Debug.h"
+#include <fstream>
+#include <set>
 
 using namespace swift;
 
@@ -703,3 +706,29 @@ void SwiftPassInvocation::eraseInstruction(SILInstruction *inst) {
     }
   }
 }
+
+static llvm::cl::opt<std::string> CondFailConfigFile(
+    "cond-fail-config-file", llvm::cl::init(""),
+    llvm::cl::desc("read the cond_fail message strings to elimimate from file"));
+
+static std::set<std::string> CondFailsToRemove;
+
+bool SILCombiner::shouldRemoveCondFail(CondFailInst &CFI) {
+  if (CondFailConfigFile.empty())
+    return false;
+
+  std::fstream fs(CondFailConfigFile);
+  if (!fs) {
+    llvm::errs() << "cannot cond_fail disablement config file\n";
+    exit(1);
+  }
+  if (CondFailsToRemove.empty()) {
+    std::string line;
+    while (std::getline(fs, line)) {
+      CondFailsToRemove.insert(line);
+    }
+    fs.close();
+  }
+  auto message = CFI.getMessage();
+  return CondFailsToRemove.find(message.str()) != CondFailsToRemove.end();
+}

lib/SILOptimizer/SILCombiner/SILCombiner.h

@@ -136,6 +136,8 @@ class SILCombiner :
 
   bool runOnFunction(SILFunction &F);
 
+  bool shouldRemoveCondFail(CondFailInst &);
+
   void clear() {
     Iteration = 0;
     Worklist.resetChecked();

lib/SILOptimizer/SILCombiner/SILCombinerMiscVisitors.cpp

@@ -537,6 +537,9 @@ SILInstruction *SILCombiner::visitCondFailInst(CondFailInst *CFI) {
   if (RemoveCondFails)
     return eraseInstFromFunction(*CFI);
 
+  if (shouldRemoveCondFail(*CFI))
+    return eraseInstFromFunction(*CFI);
+
   auto *I = dyn_cast<IntegerLiteralInst>(CFI->getOperand());
   if (!I)
     return nullptr;