Graphite
Loading...
Searching...
No Matches
Public Member Functions | Private Attributes | List of all members
graphite::PCGSolver< T, S > Class Template Reference

Preconditioned Conjugate Gradient (PCG) solver. More...

#include <pcg.hpp>

Inheritance diagram for graphite::PCGSolver< T, S >:
Inheritance graph
[legend]
Collaboration diagram for graphite::PCGSolver< T, S >:
Collaboration graph
[legend]

Public Member Functions

 PCGSolver (size_t max_iter, T tol, T rejection_ratio, Preconditioner< T, S > *preconditioner)
 
virtual void update_structure (Graph< T, S > *graph, StreamPool &streams) override
 
virtual void update_values (Graph< T, S > *graph, StreamPool &streams) override
 
virtual void set_damping_factor (Graph< T, S > *graph, T damping_factor, StreamPool &streams) override
 
virtual bool solve (Graph< T, S > *graph, T *x, StreamPool &streams) override
 

Private Attributes

thrust::device_vector< T > v
 
thrust::device_vector< T > v1
 
thrust::device_vector< T > v2
 
thrust::device_vector< T > r
 
thrust::device_vector< T > p
 
thrust::device_vector< T > z
 
thrust::device_vector< T > diag
 
thrust::device_vector< T > x_backup
 
thrust::device_vector< T > y
 
size_t max_iter
 
T tol
 
T rejection_ratio
 
T damping_factor
 
Preconditioner< T, S > * preconditioner
 

Detailed Description

template<typename T, typename S>
class graphite::PCGSolver< T, S >

Preconditioned Conjugate Gradient (PCG) solver.

Member Function Documentation

◆ set_damping_factor()

template<typename T , typename S >
virtual void graphite::PCGSolver< T, S >::set_damping_factor ( Graph< T, S > *  graph,
T  damping_factor,
StreamPool &  streams 
)
inlineoverridevirtual

Implements graphite::Solver< T, S >.

◆ solve()

template<typename T , typename S >
virtual bool graphite::PCGSolver< T, S >::solve ( Graph< T, S > *  graph,
T *  x,
StreamPool &  streams 
)
inlineoverridevirtual

Implements graphite::Solver< T, S >.

◆ update_structure()

template<typename T , typename S >
virtual void graphite::PCGSolver< T, S >::update_structure ( Graph< T, S > *  graph,
StreamPool &  streams 
)
inlineoverridevirtual

Implements graphite::Solver< T, S >.

◆ update_values()

template<typename T , typename S >
virtual void graphite::PCGSolver< T, S >::update_values ( Graph< T, S > *  graph,
StreamPool &  streams 
)
inlineoverridevirtual

Implements graphite::Solver< T, S >.

The documentation for this class was generated from the following file: