1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
//===--- DifferentiationInvoker.cpp ---------------------------*- C++ -*---===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2019 - 2020 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
//
//===----------------------------------------------------------------------===//
//
// Class that represents an invoker of differentiation.
// Used to track diagnostic source locations.
//
//===----------------------------------------------------------------------===//
#include "swift/SILOptimizer/Differentiation/DifferentiationInvoker.h"
#include "swift/SIL/SILDifferentiabilityWitness.h"
#include "swift/SIL/SILFunction.h"
#include "swift/SIL/SILInstruction.h"
namespace swift {
namespace autodiff {
SourceLoc DifferentiationInvoker::getLocation() const {
switch (kind) {
case Kind::DifferentiableFunctionInst:
return getDifferentiableFunctionInst()->getLoc().getSourceLoc();
case Kind::LinearFunctionInst:
return getLinearFunctionInst()->getLoc().getSourceLoc();
case Kind::IndirectDifferentiation:
return getIndirectDifferentiation().first->getLoc().getSourceLoc();
case Kind::SILDifferentiabilityWitnessInvoker:
return getSILDifferentiabilityWitnessInvoker()
->getOriginalFunction()
->getLocation()
.getSourceLoc();
}
llvm_unreachable("Invalid invoker kind"); // silences MSVC C4715
}
void DifferentiationInvoker::print(llvm::raw_ostream &os) const {
os << "(differentiation_invoker ";
switch (kind) {
case Kind::DifferentiableFunctionInst:
os << "differentiable_function_inst=(" << *getDifferentiableFunctionInst()
<< ')';
break;
case Kind::LinearFunctionInst:
os << "linear_function_inst=(" << *getLinearFunctionInst() << ')';
break;
case Kind::IndirectDifferentiation: {
auto indDiff = getIndirectDifferentiation();
os << "indirect_differentiation=(" << *std::get<0>(indDiff) << ')';
// TODO: Enable printing parent invokers.
// May require storing a `DifferentiableInvoker *` in the
// `IndirectDifferentiation` case.
/*
SILInstruction *inst;
SILDifferentiableAttr *attr;
std::tie(inst, attr) = getIndirectDifferentiation();
auto invokerLookup = invokers.find(attr); // No access to ADContext?
assert(invokerLookup != invokers.end() && "Expected parent invoker");
*/
break;
}
case Kind::SILDifferentiabilityWitnessInvoker: {
auto witness = getSILDifferentiabilityWitnessInvoker();
os << "sil_differentiability_witness_invoker=(witness=(";
witness->print(os);
os << ") function=" << witness->getOriginalFunction()->getName();
break;
}
}
os << ')';
}
} // end namespace autodiff
} // end namespace swift
|
