operands_and_partials.hpp

Go to the documentation of this file.

#ifndef STAN_MATH_OPENCL_REV_OPERANDS_AND_PARTIALS_HPP
#define STAN_MATH_OPENCL_REV_OPERANDS_AND_PARTIALS_HPP
#ifdef STAN_OPENCL
 
#include <stan/math/prim/functor/partials_propagator.hpp>
#include <stan/math/prim/meta.hpp>
#include <stan/math/rev/core/var.hpp>
#include <stan/math/opencl/kernel_generator.hpp>
#include <stan/math/opencl/rev/arena_type.hpp>
#include <stan/math/opencl/rev/to_arena.hpp>
 
namespace stan {
namespace math {
namespace internal {
 
template <typename Op>
class ops_partials_edge<double, var_value<Op>,
                        require_kernel_expression_lhs_t<Op>> {
 public:
  using partials_t = arena_matrix_cl<value_type_t<Op>>;
  partials_t partials_;                       // For univariate use-cases
  broadcast_array<partials_t> partials_vec_;  // For multivariate
  explicit ops_partials_edge(const var_value<Op>& ops)
      : partials_(constant(0.0, ops.vi_->rows(), ops.vi_->cols())),
        partials_vec_(partials_),
        operands_(ops) {}
  inline auto& partial() noexcept { return partials_; }
  inline auto& operand() const noexcept { return operands_; }
  var_value<Op> operands_;
  static constexpr int size() { return 0; }
};
 
}  // namespace internal
}  // namespace math
}  // namespace stan
#endif
#endif