mim::plug::autodiff Namespace Reference

The automatic differentiation Plugin More...

Classes
class	Eval
	This pass is the heart of AD. More...

Enumerations
%autodiff.zero_repl
enum class	zero_repl : flags_t
%autodiff.eval_pass
enum class	eval_pass : flags_t

Functions
void	register_normalizers (Normalizers &normalizers)
Helpers
const Pi *	pullback_type (const Def *E, const Def *A)
	computes pb type E* -> A* E - type of the expression (return type for a function) A - type of the argument (point of orientation resp.
const Def *	id_pullback (const Def *)
const Pi *	autodiff_type_fun_pi (const Pi *)
const Def *	autodiff_type_fun (const Def *)

Variables
static constexpr plugin_t	Plugin_Id = 0x9655014a1c70000

%autodiff.Tangent
enum class	Tangent : flags_t
const Def *	tangent_type_fun (const Def *)
const Def *	normalize_Tangent (const Def *, const Def *, const Def *arg)

%autodiff.AD
enum class	AD : flags_t
enum class	ad : flags_t
const Def *	normalize_ad (const Def *, const Def *, const Def *)
	Currently this normalizer does nothin.
const Def *	normalize_AD (const Def *, const Def *, const Def *arg)

%autodiff.zero
enum class	zero : flags_t
const Def *	zero_def (const Def *T)
const Def *	zero_pullback (const Def *E, const Def *A)
const Def *	normalize_zero (const Def *, const Def *, const Def *)
	Currently this normalizer does nothing.

%autodiff.add
enum class	add : flags_t
const Def *	normalize_add (const Def *type, const Def *callee, const Def *arg)
	Currently resolved the full addition.

%autodiff.sum
enum class	sum : flags_t
const Def *	op_sum (const Def *T, Defs)
const Def *	normalize_sum (const Def *type, const Def *callee, const Def *arg)

Detailed Description

The automatic differentiation Plugin

Enumeration Type Documentation

AD

enum class mim::plug::autodiff::AD : flags_t

strong

Definition at line 22 of file autogen.h.

ad

enum class mim::plug::autodiff::ad : flags_t

strong

Definition at line 30 of file autogen.h.

add

enum class mim::plug::autodiff::add : flags_t

strong

Definition at line 46 of file autogen.h.

eval_pass

enum class mim::plug::autodiff::eval_pass : flags_t

strong

Definition at line 69 of file autogen.h.

sum

enum class mim::plug::autodiff::sum : flags_t

strong

Definition at line 54 of file autogen.h.

Tangent

enum class mim::plug::autodiff::Tangent : flags_t

strong

Definition at line 14 of file autogen.h.

zero

enum class mim::plug::autodiff::zero : flags_t

strong

Definition at line 38 of file autogen.h.

zero_repl

enum class mim::plug::autodiff::zero_repl : flags_t

strong

Definition at line 62 of file autogen.h.

Function Documentation

autodiff_type_fun()

const Def * mim::plug::autodiff::autodiff_type_fun

(

const Def *

ty

)

Definition at line 111 of file autodiff.cpp.

References autodiff_type_fun(), autodiff_type_fun_pi(), mim::Axm::isa(), mim::Idx::isa(), mim::Def::node_name(), and mim::Def::world().

Referenced by mim::plug::autodiff::Eval::augment_(), mim::plug::autodiff::Eval::augment_lam(), autodiff_type_fun(), autodiff_type_fun_pi(), and normalize_AD().

autodiff_type_fun_pi()

const Pi * mim::plug::autodiff::autodiff_type_fun_pi

(

const Pi *

pi

)

Definition at line 88 of file autodiff.cpp.

References autodiff_type_fun(), mim::Pi::codom(), mim::Pi::dom(), mim::Pi::isa_cn(), and mim::Def::world().

Referenced by autodiff_type_fun(), and mim::plug::autodiff::Eval::derive_().

id_pullback()

const Def * mim::plug::autodiff::id_pullback

(

const Def *

A

)

Definition at line 30 of file autodiff.cpp.

References pullback_type(), and mim::Def::world().

Referenced by mim::plug::autodiff::Eval::derive_().

normalize_AD()

const Def * mim::plug::autodiff::normalize_AD	(	const Def *	,
		const Def *	,
		const Def *	arg )

Definition at line 13 of file normalizers.cpp.

References autodiff_type_fun().

normalize_ad()

const Def * mim::plug::autodiff::normalize_ad	(	const Def *	,
		const Def *	,
		const Def *	)

Currently this normalizer does nothin.

TODO: Maybe we want to handle trivial lookup replacements here.

Definition at line 11 of file normalizers.cpp.

normalize_add()

const Def * mim::plug::autodiff::normalize_add	(	const Def *	type,
		const Def *	callee,
		const Def *	arg )

Currently resolved the full addition.

There is no benefit in keeping additions around longer than necessary.

Definition at line 28 of file normalizers.cpp.

References mim::plug::core::add, mim::Idx::as_lit(), mim::Axm::isa(), mim::Idx::isa(), and mim::Def::projs().

normalize_sum()

const Def * mim::plug::autodiff::normalize_sum	(	const Def *	type,
		const Def *	callee,
		const Def *	arg )

Definition at line 81 of file normalizers.cpp.

References mim::Def::projs().

normalize_Tangent()

const Def * mim::plug::autodiff::normalize_Tangent	(	const Def *	,
		const Def *	,
		const Def *	arg )

Definition at line 19 of file normalizers.cpp.

References tangent_type_fun().

normalize_zero()

const Def * mim::plug::autodiff::normalize_zero	(	const Def *	,
		const Def *	,
		const Def *	)

Currently this normalizer does nothing.

We usually want to keep zeros as long as possible to avoid unnecessary allocations. A high-level addition with zero can be shortened directly.

Definition at line 24 of file normalizers.cpp.

op_sum()

const Def * mim::plug::autodiff::op_sum	(	const Def *	T,
		Defs	defs )

Definition at line 167 of file autodiff.cpp.

References mim::World::app(), and mim::Def::world().

Referenced by mim::plug::autodiff::Eval::augment_tuple().

pullback_type()

const Pi * mim::plug::autodiff::pullback_type	(	const Def *	E,
		const Def *	A )

computes pb type E* -> A* E - type of the expression (return type for a function) A - type of the argument (point of orientation resp.

derivative - argument type for partial pullbacks)

Definition at line 60 of file autodiff.cpp.

References tangent_type_fun(), and mim::Def::world().

Referenced by mim::plug::autodiff::Eval::augment_extract(), mim::plug::autodiff::Eval::augment_lam(), mim::plug::autodiff::Eval::augment_pack(), mim::plug::autodiff::Eval::augment_tuple(), id_pullback(), and zero_pullback().

register_normalizers()

void mim::plug::autodiff::register_normalizers

(

Normalizers &

normalizers

)

Referenced by mim_get_plugin().

tangent_type_fun()

const Def * mim::plug::autodiff::tangent_type_fun

(

const Def *

ty

)

Definition at line 55 of file autodiff.cpp.

Referenced by mim::plug::autodiff::Eval::augment_pack(), mim::plug::autodiff::Eval::augment_tuple(), normalize_Tangent(), pullback_type(), and zero_pullback().

zero_def()

const Def * mim::plug::autodiff::zero_def

(

const Def *

T

)

Definition at line 138 of file autodiff.cpp.

References mim::Idx::isa(), mim::Def::node_name(), and mim::Def::world().

Referenced by reg_stages().

zero_pullback()

const Def * mim::plug::autodiff::zero_pullback	(	const Def *	E,
		const Def *	A )

Definition at line 42 of file autodiff.cpp.

References pullback_type(), tangent_type_fun(), and mim::Def::world().

Referenced by mim::plug::autodiff::Eval::augment_lam(), mim::plug::autodiff::Eval::augment_lit(), and mim::plug::autodiff::Eval::derive_().

Variable Documentation

Plugin_Id

plugin_t mim::plug::autodiff::Plugin_Id = 0x9655014a1c70000

staticconstexpr

Definition at line 10 of file autogen.h.