autogen.h

Go to the documentation of this file.

#pragma once
 
#include <thorin/axiom.h>
#include <thorin/plugin.h>
 
/// @namespace thorin::plug::autodiff @ref autodiff 
namespace thorin {
namespace plug::autodiff {
 
static constexpr plugin_t Plugin_Id = 0x9655014a1c70000;
 
/// @name %%autodiff.Tangent
///@{
#ifdef DOXYGEN // see https://github.com/doxygen/doxygen/issues/9668
enum Tangent : flags_t {
#else
enum class Tangent : flags_t {
#endif
};
 
Ref normalize_Tangent(Ref, Ref, Ref);
///@}
 
/// @name %%autodiff.AD
///@{
#ifdef DOXYGEN // see https://github.com/doxygen/doxygen/issues/9668
enum AD : flags_t {
#else
enum class AD : flags_t {
#endif
};
 
Ref normalize_AD(Ref, Ref, Ref);
///@}
 
/// @name %%autodiff.ad
///@{
#ifdef DOXYGEN // see https://github.com/doxygen/doxygen/issues/9668
enum ad : flags_t {
#else
enum class ad : flags_t {
#endif
};
 
Ref normalize_ad(Ref, Ref, Ref);
///@}
 
/// @name %%autodiff.zero
///@{
#ifdef DOXYGEN // see https://github.com/doxygen/doxygen/issues/9668
enum zero : flags_t {
#else
enum class zero : flags_t {
#endif
};
 
Ref normalize_zero(Ref, Ref, Ref);
///@}
 
/// @name %%autodiff.add
///@{
#ifdef DOXYGEN // see https://github.com/doxygen/doxygen/issues/9668
enum add : flags_t {
#else
enum class add : flags_t {
#endif
};
 
Ref normalize_add(Ref, Ref, Ref);
///@}
 
/// @name %%autodiff.sum
///@{
#ifdef DOXYGEN // see https://github.com/doxygen/doxygen/issues/9668
enum sum : flags_t {
#else
enum class sum : flags_t {
#endif
};
 
Ref normalize_sum(Ref, Ref, Ref);
///@}
 
/// @name %%autodiff.ad_eval_pass
///@{
#ifdef DOXYGEN // see https://github.com/doxygen/doxygen/issues/9668
enum ad_eval_pass : flags_t {
#else
enum class ad_eval_pass : flags_t {
#endif
};
 
///@}
 
/// @name %%autodiff.ad_zero_pass
///@{
#ifdef DOXYGEN // see https://github.com/doxygen/doxygen/issues/9668
enum ad_zero_pass : flags_t {
#else
enum class ad_zero_pass : flags_t {
#endif
};
 
///@}
 
/// @name %%autodiff.ad_zero_cleanup_pass
///@{
#ifdef DOXYGEN // see https://github.com/doxygen/doxygen/issues/9668
enum ad_zero_cleanup_pass : flags_t {
#else
enum class ad_zero_cleanup_pass : flags_t {
#endif
};
 
///@}
 
/// @name %%autodiff.ad_ext_cleanup_pass
///@{
#ifdef DOXYGEN // see https://github.com/doxygen/doxygen/issues/9668
enum ad_ext_cleanup_pass : flags_t {
#else
enum class ad_ext_cleanup_pass : flags_t {
#endif
};
 
///@}
 
void register_normalizers(Normalizers& normalizers);
 
#define THORIN_autodiff_NORMALIZER_IMPL \
    void register_normalizers(Normalizers& normalizers) {\
        normalizers[flags_t(Annex::Base<Tangent>)] = &normalize_Tangent; \
        normalizers[flags_t(Annex::Base<AD>)] = &normalize_AD; \
        normalizers[flags_t(Annex::Base<ad>)] = &normalize_ad; \
        normalizers[flags_t(Annex::Base<zero>)] = &normalize_zero; \
        normalizers[flags_t(Annex::Base<add>)] = &normalize_add; \
        normalizers[flags_t(Annex::Base<sum>)] = &normalize_sum; \
    }
} // namespace plug::autodiff
 
#ifndef DOXYGEN // don't include in Doxygen documentation
template<> constexpr flags_t Annex::Base<plug::autodiff::Tangent> = 0x9655014a1c70000;
template<> constexpr size_t Annex::Num<plug::autodiff::Tangent> = 0;
template<> constexpr flags_t Annex::Base<plug::autodiff::AD> = 0x9655014a1c70100;
template<> constexpr size_t Annex::Num<plug::autodiff::AD> = 0;
template<> constexpr flags_t Annex::Base<plug::autodiff::ad> = 0x9655014a1c70200;
template<> constexpr size_t Annex::Num<plug::autodiff::ad> = 0;
template<> constexpr flags_t Annex::Base<plug::autodiff::zero> = 0x9655014a1c70300;
template<> constexpr size_t Annex::Num<plug::autodiff::zero> = 0;
template<> constexpr flags_t Annex::Base<plug::autodiff::add> = 0x9655014a1c70400;
template<> constexpr size_t Annex::Num<plug::autodiff::add> = 0;
template<> constexpr flags_t Annex::Base<plug::autodiff::sum> = 0x9655014a1c70500;
template<> constexpr size_t Annex::Num<plug::autodiff::sum> = 0;
template<> constexpr flags_t Annex::Base<plug::autodiff::ad_eval_pass> = 0x9655014a1c70600;
template<> constexpr size_t Annex::Num<plug::autodiff::ad_eval_pass> = 0;
template<> constexpr flags_t Annex::Base<plug::autodiff::ad_zero_pass> = 0x9655014a1c70700;
template<> constexpr size_t Annex::Num<plug::autodiff::ad_zero_pass> = 0;
template<> constexpr flags_t Annex::Base<plug::autodiff::ad_zero_cleanup_pass> = 0x9655014a1c70800;
template<> constexpr size_t Annex::Num<plug::autodiff::ad_zero_cleanup_pass> = 0;
template<> constexpr flags_t Annex::Base<plug::autodiff::ad_ext_cleanup_pass> = 0x9655014a1c70900;
template<> constexpr size_t Annex::Num<plug::autodiff::ad_ext_cleanup_pass> = 0;
 
template<> struct Axiom::Match<plug::autodiff::ad_eval_pass> { using type = Axiom; };
template<> struct Axiom::Match<plug::autodiff::ad_zero_pass> { using type = Axiom; };
template<> struct Axiom::Match<plug::autodiff::ad_zero_cleanup_pass> { using type = Axiom; };
template<> struct Axiom::Match<plug::autodiff::ad_ext_cleanup_pass> { using type = Axiom; };
#endif
} // namespace thorin