mirror of
https://github.com/apple/swift.git
synced 2025-12-21 12:14:44 +01:00
87f2f41d51
To add a module pass in `Passes.def` use the new `SWIFT_MODULE_PASS` macro. On the swift side, create a `ModulePass`. It’s run function receives a `ModulePassContext`, which provides access to all functions of a module. But it doesn't provide any APIs to modify functions. In order to modify a function, a module pass must use `ModulePassContext.transform(function:)`.
67 lines
1.9 KiB
Swift
67 lines
1.9 KiB
Swift
//===--- Passes.swift ---- instruction and function passes ----------------===//
|
|
//
|
|
// This source file is part of the Swift.org open source project
|
|
//
|
|
// Copyright (c) 2014 - 2022 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
|
|
import OptimizerBridging
|
|
|
|
struct FunctionPass {
|
|
|
|
let name: String
|
|
let runFunction: (Function, PassContext) -> ()
|
|
|
|
public init(name: String,
|
|
_ runFunction: @escaping (Function, PassContext) -> ()) {
|
|
self.name = name
|
|
self.runFunction = runFunction
|
|
}
|
|
|
|
func run(_ bridgedCtxt: BridgedFunctionPassCtxt) {
|
|
let function = bridgedCtxt.function.function
|
|
let context = PassContext(_bridged: bridgedCtxt.passContext)
|
|
runFunction(function, context)
|
|
}
|
|
}
|
|
|
|
struct InstructionPass<InstType: Instruction> {
|
|
|
|
let name: String
|
|
let runFunction: (InstType, PassContext) -> ()
|
|
|
|
public init(name: String,
|
|
_ runFunction: @escaping (InstType, PassContext) -> ()) {
|
|
self.name = name
|
|
self.runFunction = runFunction
|
|
}
|
|
|
|
func run(_ bridgedCtxt: BridgedInstructionPassCtxt) {
|
|
let inst = bridgedCtxt.instruction.getAs(InstType.self)
|
|
let context = PassContext(_bridged: bridgedCtxt.passContext)
|
|
runFunction(inst, context)
|
|
}
|
|
}
|
|
|
|
struct ModulePass {
|
|
|
|
let name: String
|
|
let runFunction: (ModulePassContext) -> ()
|
|
|
|
public init(name: String,
|
|
_ runFunction: @escaping (ModulePassContext) -> ()) {
|
|
self.name = name
|
|
self.runFunction = runFunction
|
|
}
|
|
|
|
func run(_ bridgedCtxt: BridgedPassContext) {
|
|
let context = ModulePassContext(_bridged: bridgedCtxt)
|
|
runFunction(context)
|
|
}
|
|
}
|