zero_adjoints.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_REV_CORE_ZERO_ADJOINTS_HPP
#define STAN_MATH_REV_CORE_ZERO_ADJOINTS_HPP
 
#include <stan/math/rev/core/var.hpp>
#include <stan/math/prim/fun/Eigen.hpp>
#include <vector>
 
namespace stan {
namespace math {
 
inline void zero_adjoints() noexcept {}
 
template <typename T, require_st_arithmetic<T>* = nullptr>
inline void zero_adjoints(T& x) noexcept {}
 
inline void zero_adjoints(var& x) { x.adj() = 0; }
 
template <typename EigMat,
          require_eigen_vt<is_autodiff_scalar, EigMat>* = nullptr>
inline void zero_adjoints(EigMat& x) {
  for (size_t i = 0; i < x.size(); ++i)
    x.coeffRef(i).adj() = 0;
}
 
template <typename StdVec,
          require_std_vector_st<is_autodiff_scalar, StdVec>* = nullptr>
inline void zero_adjoints(StdVec& x) {
  for (size_t i = 0; i < x.size(); ++i) {
    zero_adjoints(x[i]);
  }
}
 
}  // namespace math
}  // namespace stan
#endif