Goal: Learn to use a custom type in an adjoint vector for a CoDiPack reverse mode tape.
Prerequisite: Tutorial 4 - Vector mode AD, Identifier management.
Function: Simple vector valued function
template<typename Real>
void func(
const Real* x,
size_t l,
Real* y) {
y[0] = 0.0;
y[1] = 1.0;
for(size_t i = 0; i < l; ++i) {
y[0] += x[i];
y[1] *= x[i];
}
}
Represents a concrete lvalue in the CoDiPack expression tree.
Definition: activeType.hpp:52
Full code:
#include <codi.hpp>
#include <iostream>
template<typename Real>
void func(
const Real* x,
size_t l,
Real* y) {
y[0] = 0.0;
y[1] = 1.0;
for(size_t i = 0; i < l; ++i) {
y[0] += x[i];
y[1] *= x[i];
}
}
int main(int nargs, char** args) {
x[0] = 1.0;
x[1] = 2.0;
x[2] = 3.0;
x[3] = 4.0;
x[4] = 5.0;
tape.setActive();
for(size_t i = 0; i < 5; ++i) {
tape.registerInput(x[i]);
}
func(x, 5, y);
tape.registerOutput(y[0]);
tape.registerOutput(y[1]);
tape.setPassive();
for(size_t i = 0; i < 5; ++i) {
}
std::cout << "Reverse vector mode:" << std::endl;
std::cout << "f(1 .. 5) = (" << y[0] << ", " << y[1] << ")" << std::endl;
std::cout << "df/dx (1 .. 5) = \n" << jacobian << std::endl;
tape.reset();
return 0;
}
DataExtraction< Type >::Identifier getIdentifier(Type const &v)
Extract the identifiers from a type of aggregated active types.
Definition: realTraits.hpp:216
RealReverseGen< double > RealReverse
Definition: codi.hpp:120
static Tape & getTape()
Get a reference to the tape which manages this expression.
Definition: activeType.hpp:99
T_Tape Tape
See ActiveType.
Definition: activeType.hpp:55
Allows for an arbitrary adjoint evaluation of a recorded tape.
Definition: customAdjointVectorHelper.hpp:152
Gradient & gradient(Identifier const &identifier)
Get a reference to the gradient. Checked access.
Definition: customAdjointVectorHelper.hpp:254
Gradient const & getGradient(Identifier const &identifier) const
Get a constant reference to the gradient.
Definition: customAdjointVectorHelper.hpp:239
void evaluate(Position const &start, Position const &end)
Perform a full reverse evaluation of the tape.
Definition: customAdjointVectorHelper.hpp:207
Default implementation of the Jacobian interface.
Definition: jacobian.hpp:60
The custom adjoint vector helper structure (CustomAdjointVectorHelper) allows to use a custom type for the adjoint vector, by providing a simple tape like interface. The structure creates its own adjoint vector of the correct size, which is independent of the adjoint vector created by the tape.
The biggest change is that all operations for the reverse mode have to be done on the helper structure. That is seeding, evaluation and retrieving.
Notes on using the codi::CustomAdjointVectorHelper:
- Internal adjoint vector is automatically created.
- The default tape is the global tape. (Can be changed with setTape)
- Identifiers need to be used from the variables.
- Multiple instances of the CustomAdjointVectorHelper are thread safe. (Each thread can have its own instance.)