Swift SIL: some small fixes

SwiftCompilerSources/Sources/Optimizer/PassManager/PassContext.swift

@@ -157,20 +157,31 @@ extension Builder {
     self.init(insertAt: .before(insPnt), location: insPnt.location, passContext: context._bridged)
   }
 
-  /// Creates a builder which inserts _after_ `insPnt`, using the location of `insPnt`.
-  init(after insPnt: Instruction, _ context: PassContext) {
+  /// Creates a builder which inserts _after_ `insPnt`, using a custom `location`.
+  init(after insPnt: Instruction, location: Location, _ context: PassContext) {
     if let nextInst = insPnt.next {
-      self.init(insertAt: .before(nextInst), location: insPnt.location, passContext: context._bridged)
+      self.init(insertAt: .before(nextInst), location: location, passContext: context._bridged)
     } else {
-      self.init(insertAt: .atEndOf(insPnt.block), location: insPnt.location, passContext: context._bridged)
+      self.init(insertAt: .atEndOf(insPnt.block), location: location, passContext: context._bridged)
     }
   }
 
+  /// Creates a builder which inserts _after_ `insPnt`, using the location of `insPnt`.
+  init(after insPnt: Instruction, _ context: PassContext) {
+    self.init(after: insPnt, location: insPnt.location, context)
+  }
+
   /// Creates a builder which inserts at the end of `block`, using a custom `location`.
   init(atEndOf block: BasicBlock, location: Location, _ context: PassContext) {
     self.init(insertAt: .atEndOf(block), location: location, passContext: context._bridged)
   }
 
+  /// Creates a builder which inserts at the begin of `block`, using a custom `location`.
+  init(atBeginOf block: BasicBlock, location: Location, _ context: PassContext) {
+    let firstInst = block.instructions.first!
+    self.init(insertAt: .before(firstInst), location: location, passContext: context._bridged)
+  }
+
   /// Creates a builder which inserts at the begin of `block`, using the location of the first
   /// instruction of `block`.
   init(atBeginOf block: BasicBlock, _ context: PassContext) {

SwiftCompilerSources/Sources/SIL/Instruction.swift

@@ -480,9 +480,8 @@ final public class RefTailAddrInst : SingleValueInstruction, UnaryInstruction {}
 final public class KeyPathInst : SingleValueInstruction {
   public override func visitReferencedFunctions(_ cl: (Function) -> ()) {
     var results = KeyPathFunctionResults()
-    var componentIdx = 0
-    repeat {
-      componentIdx = KeyPathInst_getReferencedFunctions(bridged, componentIdx, &results)
+    for componentIdx in 0..<KeyPathInst_getNumComponents(bridged) {
+      KeyPathInst_getReferencedFunctions(bridged, componentIdx, &results)
       let numFuncs = results.numFunctions
       withUnsafePointer(to: &results) {
         $0.withMemoryRebound(to: BridgedFunction.self, capacity: numFuncs) {
@@ -492,7 +491,7 @@ final public class KeyPathInst : SingleValueInstruction {
           }
         }
       }
-    } while componentIdx >= 0
+    }
   }
 }
 

SwiftCompilerSources/Sources/SIL/Location.swift

@@ -14,4 +14,8 @@ import SILBridging
 
 public struct Location {
   let bridged: BridgedLocation
+
+  public static var autoGeneratedLocation: Location {
+    Location(bridged: SILLocation_getAutogeneratedLocation())
+  }
 }

SwiftCompilerSources/Sources/SIL/WitnessTable.swift

@@ -21,6 +21,7 @@ public struct WitnessTable : CustomStringConvertible, CustomReflectable {
     fileprivate let bridged: BridgedWitnessTableEntry
 
     public enum Kind {
+      case invalid
       case method
       case associatedType
       case associatedTypeProtocol
@@ -29,6 +30,7 @@ public struct WitnessTable : CustomStringConvertible, CustomReflectable {
 
     public var kind: Kind {
       switch SILWitnessTableEntry_getKind(bridged) {
+        case SILWitnessTableEntry_Invalid:                return .invalid
         case SILWitnessTableEntry_Method:                 return .method
         case SILWitnessTableEntry_AssociatedType:         return .associatedType
         case SILWitnessTableEntry_AssociatedTypeProtocol: return .associatedTypeProtocol

include/swift/SIL/SILBridging.h

@@ -99,6 +99,7 @@ typedef struct {
 } BridgedWitnessTableEntry;
 
 typedef enum {
+  SILWitnessTableEntry_Invalid,
   SILWitnessTableEntry_Method,
   SILWitnessTableEntry_AssociatedType,
   SILWitnessTableEntry_AssociatedTypeProtocol,
@@ -333,6 +334,8 @@ SwiftInt SILType_getCaseIdxOfEnumType(BridgedType type,
 
 BridgedSubstitutionMap SubstitutionMap_getEmpty();
 
+BridgedLocation SILLocation_getAutogeneratedLocation();
+
 BridgedBasicBlock SILArgument_getParent(BridgedArgument argument);
 SwiftInt SILArgument_isExclusiveIndirectParameter(BridgedArgument argument);
 
@@ -396,7 +399,8 @@ struct KeyPathFunctionResults {
   BridgedFunction functions[maxFunctions];
   SwiftInt numFunctions;
 };
-SwiftInt KeyPathInst_getReferencedFunctions(BridgedInstruction kpi, SwiftInt componentIdx,
+SwiftInt KeyPathInst_getNumComponents(BridgedInstruction kpi);
+void KeyPathInst_getReferencedFunctions(BridgedInstruction kpi, SwiftInt componentIdx,
                                             KeyPathFunctionResults * _Nonnull results);
 
 BridgedSubstitutionMap ApplySite_getSubstitutionMap(BridgedInstruction inst);

include/swift/SILOptimizer/PassManager/PassManager.h

@@ -61,6 +61,10 @@ class SwiftPassInvocation {
   /// Non-null if this is an instruction pass, invoked from SILCombine.
   SILCombiner *silCombiner = nullptr;
 
+  /// Change notifications, collected during a pass run.
+  SILAnalysis::InvalidationKind changeNotifications =
+      SILAnalysis::InvalidationKind::Nothing;
+
   /// All slabs, allocated by the pass.
   SILModule::SlabList allocatedSlabs;
 
@@ -126,17 +130,9 @@ class SwiftPassInvocation {
   /// Called by the SILCombiner when the instruction pass has finished.
   void finishedInstructionPassRun();
 
-  void beginTransformFunction(SILFunction *function) {
-    assert(!this->function && transform && "not running a module pass");
-    this->function = function;
-  }
+  void beginTransformFunction(SILFunction *function);
 
-  void endTransformFunction() {
-    assert(function && "beginTransformFunction not called");
-    function = nullptr;
-    assert(numBlockSetsAllocated == 0 && "Not all BasicBlockSets deallocated");
-    assert(numNodeSetsAllocated == 0 && "Not all NodeSets deallocated");
-  }
+  void endTransformFunction();
 };
 
 /// The SIL pass manager.
@@ -184,10 +180,6 @@ class SILPassManager {
   /// For invoking Swift passes.
   SwiftPassInvocation swiftPassInvocation;
 
-  /// Change notifications, collected during a bridged pass run.
-  SILAnalysis::InvalidationKind changeNotifications =
-      SILAnalysis::InvalidationKind::Nothing;
-
   /// A mask which has one bit for each pass. A one for a pass-bit means that
   /// the pass doesn't need to run, because nothing has changed since the
   /// previous run of that pass.
@@ -341,11 +333,6 @@ class SILPassManager {
     CompletedPassesMap[F].reset();
   }
 
-  void notifyPassChanges(SILAnalysis::InvalidationKind invalidationKind) {
-    changeNotifications = (SILAnalysis::InvalidationKind)
-        (changeNotifications | invalidationKind);
-  }
-
   /// Reset the state of the pass manager and remove all transformation
   /// owned by the pass manager. Analysis passes will be kept.
   void resetAndRemoveTransformations();
@@ -440,7 +427,8 @@ class SILPassManager {
 
 inline void SwiftPassInvocation::
 notifyChanges(SILAnalysis::InvalidationKind invalidationKind) {
-  passManager->notifyPassChanges(invalidationKind);
+    changeNotifications = (SILAnalysis::InvalidationKind)
+        (changeNotifications | invalidationKind);
 }
 
 } // end namespace swift

lib/SIL/Utils/SILBridging.cpp

@@ -529,6 +529,15 @@ BridgedSubstitutionMap SubstitutionMap_getEmpty() {
   return {SubstitutionMap().getOpaqueValue()};
 }
 
+//===----------------------------------------------------------------------===//
+//                                SILLocation
+//===----------------------------------------------------------------------===//
+
+BridgedLocation SILLocation_getAutogeneratedLocation() {
+  SILDebugLocation loc;
+  return *reinterpret_cast<BridgedLocation *>(&loc);
+}
+
 //===----------------------------------------------------------------------===//
 //                            SILGlobalVariable
 //===----------------------------------------------------------------------===//
@@ -625,6 +634,8 @@ std::string SILWitnessTableEntry_debugDescription(BridgedWitnessTableEntry entry
 
 SILWitnessTableEntryKind SILWitnessTableEntry_getKind(BridgedWitnessTableEntry entry) {
   switch (castToWitnessTableEntry(entry)->getKind()) {
+    case SILWitnessTable::Invalid:
+      return SILWitnessTableEntry_Invalid;
     case SILWitnessTable::Method:
       return SILWitnessTableEntry_Method;
     case SILWitnessTable::AssociatedType:
@@ -633,8 +644,6 @@ SILWitnessTableEntryKind SILWitnessTableEntry_getKind(BridgedWitnessTableEntry e
       return SILWitnessTableEntry_AssociatedTypeProtocol;
     case SILWitnessTable::BaseProtocol:
       return SILWitnessTableEntry_BaseProtocol;
-    default:
-      llvm_unreachable("wrong witness table entry kind");
   }
 }
 
@@ -877,7 +886,14 @@ SwiftInt CondBranchInst_getNumTrueArgs(BridgedInstruction cbr) {
   return castToInst<CondBranchInst>(cbr)->getNumTrueArgs();
 }
 
-SwiftInt KeyPathInst_getReferencedFunctions(BridgedInstruction kpi, SwiftInt componentIdx,
+SwiftInt KeyPathInst_getNumComponents(BridgedInstruction kpi) {
+  if (KeyPathPattern *pattern = castToInst<KeyPathInst>(kpi)->getPattern()) {
+    return (SwiftInt)pattern->getComponents().size();
+  }
+  return 0;
+}
+
+void KeyPathInst_getReferencedFunctions(BridgedInstruction kpi, SwiftInt componentIdx,
                                             KeyPathFunctionResults * _Nonnull results) {
   KeyPathPattern *pattern = castToInst<KeyPathInst>(kpi)->getPattern();
   const KeyPathPatternComponent &comp = pattern->getComponents()[componentIdx];
@@ -887,9 +903,6 @@ SwiftInt KeyPathInst_getReferencedFunctions(BridgedInstruction kpi, SwiftInt com
       assert(results->numFunctions < KeyPathFunctionResults::maxFunctions);
       results->functions[results->numFunctions++] = {func};
     }, [](SILDeclRef) {});
-
-  ++componentIdx;
-  return componentIdx < (int)pattern->getComponents().size() ? componentIdx : -1;
 }
 
 BridgedSubstitutionMap ApplySite_getSubstitutionMap(BridgedInstruction inst) {

lib/SILOptimizer/PassManager/PassManager.cpp

@@ -530,9 +530,6 @@ void SILPassManager::runPassOnFunction(unsigned TransIdx, SILFunction *F) {
     forcePrecomputeAnalyses(F);
   }
 
-  assert(changeNotifications == SILAnalysis::InvalidationKind::Nothing
-         && "change notifications not cleared");
-
   llvm::sys::TimePoint<> startTime = std::chrono::system_clock::now();
   std::chrono::nanoseconds duration(0);
 
@@ -553,24 +550,21 @@ void SILPassManager::runPassOnFunction(unsigned TransIdx, SILFunction *F) {
     // Run it!
     SFT->run();
 
-    if (CurrentPassHasInvalidated ||
-        changeNotifications != SILAnalysis::InvalidationKind::Nothing) {
+    swiftPassInvocation.finishedFunctionPassRun();
+
+    if (CurrentPassHasInvalidated) {
       // Pause time measurement while invalidating analysis and restoring the snapshot.
       duration += (std::chrono::system_clock::now() - startTime);
 
       if (runIdx < numRepeats - 1) {
         invalidateAnalysis(F, SILAnalysis::InvalidationKind::Everything);
         F->restoreFromSnapshot(SnapshotID);
-      } else if (changeNotifications != SILAnalysis::InvalidationKind::Nothing) {
-        invalidateAnalysis(F, changeNotifications);
       }
-      changeNotifications = SILAnalysis::InvalidationKind::Nothing;
 
       // Continue time measurement (including flushing deleted instructions).
       startTime = std::chrono::system_clock::now();
     }
     Mod->flushDeletedInsts();
-    swiftPassInvocation.finishedFunctionPassRun();
   }
 
   duration += (std::chrono::system_clock::now() - startTime);
@@ -1289,9 +1283,9 @@ void SwiftPassInvocation::startModulePassRun(SILModuleTransform *transform) {
 }
 
 void SwiftPassInvocation::startFunctionPassRun(SILFunctionTransform *transform) {
-  assert(!this->function && !this->transform && "a pass is already running");
-  this->function = transform->getFunction();
+  assert(!this->transform && "a pass is already running");
   this->transform = transform;
+  beginTransformFunction(transform->getFunction());
 }
 
 void SwiftPassInvocation::startInstructionPassRun(SILInstruction *inst) {
@@ -1302,13 +1296,15 @@ void SwiftPassInvocation::startInstructionPassRun(SILInstruction *inst) {
 void SwiftPassInvocation::finishedModulePassRun() {
   endPassRunChecks();
   assert(!function && transform && "not running a pass");
+  assert(changeNotifications == SILAnalysis::InvalidationKind::Nothing
+         && "unhandled change notifications at end of module pass");
   transform = nullptr;
 }
 
 void SwiftPassInvocation::finishedFunctionPassRun() {
   endPassRunChecks();
-  assert(function && transform && "not running a pass");
-  function = nullptr;
+  endTransformFunction();
+  assert(allocatedSlabs.empty() && "StackList is leaking slabs");
   transform = nullptr;
 }
 
@@ -1322,6 +1318,24 @@ void SwiftPassInvocation::endPassRunChecks() {
   assert(numNodeSetsAllocated == 0 && "Not all NodeSets deallocated");
 }
 
+void SwiftPassInvocation::beginTransformFunction(SILFunction *function) {
+  assert(!this->function && transform && "not running a pass");
+  assert(changeNotifications == SILAnalysis::InvalidationKind::Nothing
+         && "change notifications not cleared");
+  this->function = function;
+}
+
+void SwiftPassInvocation::endTransformFunction() {
+  assert(function && transform && "not running a pass");
+  if (changeNotifications != SILAnalysis::InvalidationKind::Nothing) {
+    passManager->invalidateAnalysis(function, changeNotifications);
+    changeNotifications = SILAnalysis::InvalidationKind::Nothing;
+  }
+  function = nullptr;
+  assert(numBlockSetsAllocated == 0 && "Not all BasicBlockSets deallocated");
+  assert(numNodeSetsAllocated == 0 && "Not all NodeSets deallocated");
+}
+
 //===----------------------------------------------------------------------===//
 //                            Swift Bridging
 //===----------------------------------------------------------------------===//

test/SILOptimizer/function_uses.sil

@@ -153,12 +153,26 @@ sil_default_witness_table RP {
   method #RP.default_witness: @default_witness
 }
 
+public protocol RP2 {
+}
+
+sil_default_witness_table RP2 {
+  no_default
+}
+
 protocol P {
   func witness()
+  func witness2()
 }
 
 struct SP : P {
   func witness()
+  func witness2()
+}
+
+struct SP2 : P {
+  func witness()
+  func witness2()
 }
 
 // CHECK-LABEL: Uses of witness
@@ -172,8 +186,20 @@ bb0(%0 : $*SP):
   return %7 : $()
 }
 
+sil @witness2 : $@convention(witness_method: P) (@inout SP) -> () {
+bb0(%0 : $*SP):
+  %7 = tuple ()
+  return %7 : $()
+}
+
 sil_witness_table SP: P module witness_tables {
   method #P.witness: @witness
+  method #P.witness2: @witness2
+}
+
+sil_witness_table SP2: P module witness_tables {
+  method #P.witness: @witness
+  method #P.witness2: @witness2
 }
 
 // CHECK-LABEL: Uses of hash_func