product.hpp File Reference

#include <graphite/ops/common.hpp>

Include dependency graph for product.hpp:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Functions
template<typename T , typename G , size_t I, size_t N, typename M , size_t E, typename F , typename VT , std::size_t... Is>
__device__ void	graphite::ops::compute_Jcol_ad (Dual< T, G > *error, const size_t col, const size_t factor_id, const size_t vertex_id, const M *obs, const typename F::ConstraintDataType *constraint_data, size_t *ids, const size_t *hessian_ids, VT args, std::index_sequence< Is... >)
template<typename T , typename S , size_t I, size_t N, size_t E, size_t D, typename F , std::size_t... Is>
__global__ void	graphite::ops::compute_Jv_kernel (T *y, const T *x, const size_t *active_ids, const size_t *ids, const size_t *hessian_ids, const size_t num_threads, const S *jacs, const uint8_t *active_state, std::index_sequence< Is... >)
template<typename T , typename S , size_t I, size_t N, typename M , size_t E, typename F , typename VT , std::size_t... Is>
__global__ void	graphite::ops::compute_Jv_dynamic_manual2 (T *y, T *x, const M *obs, const T *jacobian_scales, const typename F::ConstraintDataType *constraint_data, const size_t *active_ids, const size_t *ids, const size_t *hessian_ids, const size_t num_factors, VT args, const uint8_t *active_state, std::index_sequence< Is... >)
template<typename T , typename S , typename F , std::size_t... Is>
void	graphite::ops::launch_kernel_compute_Jv (F *f, T *out, T *in, std::array< const size_t *, F::get_num_vertices()> &hessian_ids, std::array< S *, F::get_num_vertices()> &jacs, const T *jacobian_scales, const size_t num_factors, StreamPool &streams, std::index_sequence< Is... >)
template<typename T , typename S , typename F >
void	graphite::ops::compute_Jv (F *f, T *out, T *in, const T *jacobian_scales, StreamPool &streams)
template<typename T , typename S , size_t I, size_t N, size_t E, size_t D, typename F , std::size_t... Is>
__global__ void	graphite::ops::compute_JtPv_kernel (T *y, const T *x, const size_t *active_ids, const size_t *ids, const size_t *hessian_ids, const size_t num_threads, const S *jacs, const uint8_t *active_state, const S *pmat, const S *chi2_derivative, const std::index_sequence< Is... >)
template<typename T , typename S , size_t I, size_t N, typename M , size_t E, size_t D, typename F , typename VT , std::size_t... Is>
__global__ void	graphite::ops::compute_JtPv_dynamic_kernel (T *y, const T *x, const size_t *active_ids, const size_t *ids, const size_t *hessian_ids, const size_t num_threads, const VT args, const M *obs, const T *jacobian_scales, const typename F::ConstraintDataType *constraint_data, const uint8_t *active_state, const S *pmat, const S *chi2_derivative, const std::index_sequence< Is... >)
template<typename T , typename S , typename F , std::size_t... Is>
void	graphite::ops::launch_kernel_compute_JtPv (F *f, T *out, T *in, std::array< const size_t *, F::get_num_vertices()> &hessian_ids, std::array< S *, F::get_num_vertices()> &jacs, const T *jacobian_scales, const size_t num_factors, StreamPool &streams, std::index_sequence< Is... >)
template<typename T , typename S , typename F >
void	graphite::ops::compute_Jtv (F *f, T *out, T *in, const T *jacobian_scales, StreamPool &streams)