Merge pull request #40957 from eeckstein/function-argument-effects

Infrastructure for function argument effects + two "escaping" effects

Author: eeckstein

SwiftCompilerSources/Sources/Optimizer/FunctionPasses/CMakeLists.txt

@@ -11,4 +11,5 @@ swift_compiler_sources(Optimizer
   SILPrinter.swift
   MergeCondFails.swift
   ReleaseDevirtualizer.swift
+  RunUnitTests.swift
 )

SwiftCompilerSources/Sources/Optimizer/FunctionPasses/ReleaseDevirtualizer.swift

@@ -204,7 +204,7 @@ private extension StructExtractInst {
     let structType = operand.type
 
     var nonTrivialFieldsCount = 0
-    for field in structType.getStructFields(in: function) {
+    for field in structType.getNominalFields(in: function) {
       if field.isTrivial(in: function) {
         nonTrivialFieldsCount += 1
       }

SwiftCompilerSources/Sources/Optimizer/FunctionPasses/RunUnitTests.swift

@@ -0,0 +1,24 @@
+//===--- UnitTests.swift - A pseudo pass for running the unit tests -------===//
+//
+// This source file is part of the Swift.org open source project
+//
+// Copyright (c) 2014 - 2021 Apple Inc. and the Swift project authors
+// Licensed under Apache License v2.0 with Runtime Library Exception
+//
+// See https://swift.org/LICENSE.txt for license information
+// See https://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
+//
+//===----------------------------------------------------------------------===//
+
+import SIL
+
+/// This pass should only be used by sil-opt to run all the unit tests.
+///
+let runUnitTests = FunctionPass(name: "run-unit-tests", {
+  (function: Function, context: PassContext) in
+
+  print("--- Run unit tests ---")
+  
+  print("test ProjectionPath")
+  SmallProjectionPath.runUnitTests()
+})

SwiftCompilerSources/Sources/Optimizer/PassManager/PassRegistration.swift

@@ -51,5 +51,6 @@ private func registerSwiftPasses() {
   registerPass(simplifyStrongRetainPass, { simplifyStrongRetainPass.run($0) })
   registerPass(simplifyStrongReleasePass, { simplifyStrongReleasePass.run($0) })
   registerPass(assumeSingleThreadedPass, { assumeSingleThreadedPass.run($0) })
+  registerPass(runUnitTests, { runUnitTests.run($0) })
   registerPass(releaseDevirtualizerPass, { releaseDevirtualizerPass.run($0) })
 }

SwiftCompilerSources/Sources/Optimizer/PassManager/PassUtils.swift

@@ -123,6 +123,11 @@ struct PassContext {
   func getContextSubstitutionMap(for type: Type) -> SubstitutionMap {
     SubstitutionMap(PassContext_getContextSubstitutionMap(_bridged, type.bridged))
   }
+
+  func modifyEffects(in function: Function, _ body: (inout FunctionEffects) -> ()) {
+    function._modifyEffects(body)
+    // TODO: do we need to notify any changes?
+  }
 }
 
 struct FunctionPass {

SwiftCompilerSources/Sources/SIL/CMakeLists.txt

@@ -11,12 +11,14 @@ add_swift_compiler_module(SIL
   Argument.swift
   BasicBlock.swift
   Builder.swift
+  Effects.swift
   Function.swift
   GlobalVariable.swift
   Instruction.swift
   Location.swift
   Operand.swift
   Registration.swift
+  SmallProjectionPath.swift
   SubstitutionMap.swift
   Type.swift
   Utils.swift