mirror of
https://github.com/apple/swift.git
synced 2026-06-27 12:25:55 +02:00
Andrew Trick
daa82bc783
Handle calling conventions and cleanups in all the places (hopefully). - when ExistentialSpecializer copies the specialized concrete arg into the original existential value - when ExistentialSpecializer generates a think - when SILCombine substitutes concrete values in place of the opened existential. One particularly nasty problem is the existential boxes need to be destroyed. It is not ok to simply destroy their value. The "leaks" tool does not catch this problem. Ownership SIL will make this all much more robust. Fixes <rdar://problem/50595630> Multiple leaks detected - Swift Perf
104 lines
3.5 KiB
C++
104 lines
3.5 KiB
C++
//===---- ExistentialSpecializerTransform.h - Existential Specializer -----===//
|
|
//
|
|
// This source file is part of the Swift.org open source project
|
|
//
|
|
// Copyright (c) 2014 - 2017 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This contains utilities for transforming existential args to generics.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef SWIFT_SIL_EXISTENTIALTRANSFORM_H
|
|
#define SWIFT_SIL_EXISTENTIALTRANSFORM_H
|
|
#include "FunctionSignatureOpts.h"
|
|
#include "swift/SIL/SILFunction.h"
|
|
#include "swift/SIL/SILInstruction.h"
|
|
#include "swift/SILOptimizer/Utils/Existential.h"
|
|
#include "swift/SILOptimizer/Utils/Local.h"
|
|
#include "swift/SILOptimizer/Utils/SILOptFunctionBuilder.h"
|
|
#include "swift/SILOptimizer/Utils/SpecializationMangler.h"
|
|
#include "llvm/ADT/SmallBitVector.h"
|
|
#include "llvm/Support/CommandLine.h"
|
|
#include "llvm/Support/Debug.h"
|
|
|
|
namespace swift {
|
|
|
|
/// A descriptor to carry information from ExistentialTransform analysis
|
|
/// to transformation.
|
|
struct ExistentialTransformArgumentDescriptor {
|
|
OpenedExistentialAccess AccessType;
|
|
bool isConsumed;
|
|
};
|
|
|
|
/// ExistentialTransform creates a protocol constrained generic and a thunk.
|
|
class ExistentialTransform {
|
|
/// Function Builder to create a new thunk.
|
|
SILOptFunctionBuilder &FunctionBuilder;
|
|
|
|
/// The original function to analyze and transform.
|
|
SILFunction *F;
|
|
|
|
/// The newly created inner function.
|
|
SILFunction *NewF;
|
|
|
|
/// The function signature mangler we are using.
|
|
Mangle::FunctionSignatureSpecializationMangler &Mangler;
|
|
|
|
/// List of arguments and their descriptors to specialize
|
|
llvm::SmallDenseMap<int, ExistentialTransformArgumentDescriptor>
|
|
&ExistentialArgDescriptor;
|
|
|
|
/// Argument to Generic Type Map for NewF.
|
|
llvm::SmallDenseMap<int, GenericTypeParamType *> ArgToGenericTypeMap;
|
|
|
|
/// Allocate the argument descriptors.
|
|
llvm::SmallVector<ArgumentDescriptor, 4> &ArgumentDescList;
|
|
|
|
/// Create the Devirtualized Inner Function.
|
|
void createExistentialSpecializedFunction();
|
|
|
|
/// Create new generic arguments from existential arguments.
|
|
void
|
|
convertExistentialArgTypesToGenericArgTypes(GenericSignatureBuilder &Builder);
|
|
|
|
/// Create a name for the inner function.
|
|
std::string createExistentialSpecializedFunctionName();
|
|
|
|
/// Create the new devirtualized protocol function signature.
|
|
CanSILFunctionType createExistentialSpecializedFunctionType();
|
|
|
|
/// Create the thunk.
|
|
void populateThunkBody();
|
|
|
|
public:
|
|
/// Constructor.
|
|
ExistentialTransform(
|
|
SILOptFunctionBuilder &FunctionBuilder, SILFunction *F,
|
|
Mangle::FunctionSignatureSpecializationMangler &Mangler,
|
|
llvm::SmallVector<ArgumentDescriptor, 4> &ADL,
|
|
llvm::SmallDenseMap<int, ExistentialTransformArgumentDescriptor>
|
|
&ExistentialArgDescriptor)
|
|
: FunctionBuilder(FunctionBuilder), F(F), NewF(nullptr), Mangler(Mangler),
|
|
ExistentialArgDescriptor(ExistentialArgDescriptor),
|
|
ArgumentDescList(ADL) {}
|
|
|
|
/// Return the optimized iner function.
|
|
SILFunction *getExistentialSpecializedFunction() { return NewF; }
|
|
|
|
/// External function for the optimization.
|
|
bool run() {
|
|
createExistentialSpecializedFunction();
|
|
return true;
|
|
}
|
|
};
|
|
|
|
} // end namespace swift
|
|
|
|
#endif
|