averello/swift-mirror
1
0
Fork 0
mirror of https://github.com/apple/swift.git synced 2025-12-21 12:14:44 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #35847 from rxwei/72666310-mangle-ad-thunks

Browse Source
This commit is contained in:
swift-ci
2021-02-13 05:07:32 -08:00
committed by GitHub
parent 7017362d85 f9ddecf459
commit d33c79edaf
18 changed files with 381 additions and 116 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
include/swift/SILOptimizer/Utils/DifferentiationMangler.h
View File

@@ -27,17 +27,29 @@ class DifferentiationMangler : public ASTMangler {
public:
DifferentiationMangler() {}
/// Returns the mangled name for a differentiation function of the given kind.
std::string mangle(SILFunction *originalFunction,
Demangle::AutoDiffFunctionKind kind,
AutoDiffConfig config);
std::string mangleAutoDiffFunction(StringRef originalName,
Demangle::AutoDiffFunctionKind kind,
AutoDiffConfig config);
/// Returns the mangled name for a derivative function of the given kind.
std::string mangleDerivativeFunction(SILFunction *originalFunction,
std::string mangleDerivativeFunction(StringRef originalName,
AutoDiffDerivativeFunctionKind kind,
AutoDiffConfig config);
/// Returns the mangled name for a linear map of the given kind.
std::string mangleLinearMap(SILFunction *originalFunction,
std::string mangleLinearMap(StringRef originalName,
AutoDiffLinearMapKind kind,
AutoDiffConfig config);
/// Returns the mangled name for a derivative function subset parameters
/// thunk.
std::string mangleDerivativeFunctionSubsetParametersThunk(
StringRef originalName, CanType toType,
AutoDiffDerivativeFunctionKind linearMapKind,
IndexSubset *fromParamIndices, IndexSubset *fromResultIndices,
IndexSubset *toParamIndices);
/// Returns the mangled name for a linear map subset parameters thunk.
std::string mangleLinearMapSubsetParametersThunk(
CanType fromType, AutoDiffLinearMapKind linearMapKind,
IndexSubset *fromParamIndices, IndexSubset *fromResultIndices,
IndexSubset *toParamIndices);
};
} // end namespace Mangle