reverse_pass_callback_alloc.hpp

Go to the documentation of this file.

#ifndef XPED_REVERSE_PASS_CALLBACK_ALLOC_HPP_
#define XPED_REVERSE_PASS_CALLBACK_ALLOC_HPP_
 
#include <stan/math/rev/core/chainable_alloc.hpp>
#include <stan/math/rev/core/vari.hpp>
 
namespace Xped {
namespace internal {
 
template <typename F>
struct reverse_pass_callback_alloc_vari : public stan::math::vari_base, public stan::math::chainable_alloc
{
    F rev_functor_;
 
    explicit reverse_pass_callback_alloc_vari(F&& rev_functor)
        : rev_functor_(std::forward<F>(rev_functor))
    {
        stan::math::ChainableStack::instance_->var_stack_.push_back(this);
    }
 
    inline void chain() final { rev_functor_(); }
    inline void set_zero_adjoint() final {}
    inline void init_dependent() {}
};
 
} // namespace internal
 
template <typename F>
inline void reverse_pass_callback_alloc(F&& functor)
{
    new internal::reverse_pass_callback_alloc_vari<F>(std::forward<F>(functor));
}
 
} // namespace Xped
 
#endif