Add a new -sil-disable-only-function flag (#32622)

This will enable us apply -sil-disable-pass only on certain functions

Author: meg-gupta

lib/SILOptimizer/PassManager/PassManager.cpp

@@ -87,6 +87,9 @@ llvm::cl::list<std::string>
     SILDisablePass("sil-disable-pass", llvm::cl::CommaSeparated,
                      llvm::cl::desc("Disable passes "
                                     "which contain a string from this list"));
+llvm::cl::list<std::string> SILDisablePassOnlyFun(
+    "sil-disable-pass-only-function", llvm::cl::CommaSeparated,
+    llvm::cl::desc("Apply -sil-disable-pass only on this function"));
 
 llvm::cl::list<std::string> SILVerifyBeforePass(
     "sil-verify-before-pass", llvm::cl::CommaSeparated,
@@ -204,7 +207,13 @@ static bool doPrintAfter(SILTransform *T, SILFunction *F, bool Default) {
   return Default;
 }
 
-static bool isDisabled(SILTransform *T) {
+static bool isDisabled(SILTransform *T, SILFunction *F = nullptr) {
+  if (!SILDisablePassOnlyFun.empty() && F &&
+      SILDisablePassOnlyFun.end() == std::find(SILDisablePassOnlyFun.begin(),
+                                               SILDisablePassOnlyFun.end(),
+                                               F->getName()))
+    return false;
+
   for (const std::string &NamePattern : SILDisablePass) {
     if (T->getTag().find(NamePattern) != StringRef::npos
         || T->getID().find(NamePattern) != StringRef::npos) {
@@ -395,7 +404,7 @@ void SILPassManager::runPassOnFunction(unsigned TransIdx, SILFunction *F) {
     return;
   }
 
-  if (isDisabled(SFT)) {
+  if (isDisabled(SFT, F)) {
     if (SILPrintPassName)
       dumpPassInfo("(Disabled)", TransIdx, F);
     return;

test/SILOptimizer/testflag_disable_pass_only.swift

@@ -0,0 +1,25 @@
+// RUN: %target-swift-frontend -emit-sil %s -Xllvm -sil-disable-pass=AllocBoxToStack -Xllvm -sil-disable-pass-only-function='$s26testflag_disable_pass_only4foo1yyF' | %FileCheck %s
+
+// CHECK-LABEL: sil hidden [noinline] @$s26testflag_disable_pass_only4foo1yyF :
+// CHECK: alloc_box 
+// CHECK-LABEL: } // end sil function '$s26testflag_disable_pass_only4foo1yyF'
+@inline(never)
+func foo1() {
+  var a = 100
+  @inline(never)
+  func bar() {
+    print(a)
+  }
+}
+
+// CHECK-LABEL: sil hidden [noinline] @$s26testflag_disable_pass_only4foo2yyF :
+// CHECK-NOT: alloc_box 
+// CHECK-LABEL: } // end sil function '$s26testflag_disable_pass_only4foo2yyF'
+@inline(never)
+func foo2() {
+  var a = 100
+  @inline(never)
+  func bar() {
+    print(a)
+  }
+}