Graphite
Loading...
Searching...
No Matches
Public Member Functions | Private Types | Private Attributes | List of all members
graphite::BlockJacobiPreconditioner< T, S > Class Template Reference
Inheritance diagram for graphite::BlockJacobiPreconditioner< T, S >:
Inheritance graph
[legend]
Collaboration diagram for graphite::BlockJacobiPreconditioner< T, S >:
Collaboration graph
[legend]

Public Member Functions

virtual void update_structure (Graph< T, S > *graph, StreamPool &streams)
 
virtual void update_values (Graph< T, S > *graph, StreamPool &streams)
 
virtual void set_damping_factor (Graph< T, S > *graph, T damping_factor, StreamPool &streams)
 
void apply (Graph< T, S > *graph, T *z, const T *r, StreamPool &streams) override
 

Private Types

using P = std::conditional_t< is_low_precision< S >::value, T, S >
 

Private Attributes

size_t dimension
 
std::vector< std::pair< size_t, size_t > > block_sizes
 
std::unordered_map< BaseVertexDescriptor< T, S > *, thrust::device_vector< P > > block_diagonals
 
std::unordered_map< BaseVertexDescriptor< T, S > *, thrust::device_vector< P > > scalar_diagonals
 
std::unordered_map< BaseVertexDescriptor< T, S > *, thrust::device_vector< P > > P_inv
 
cublasHandle_t handle
 
thrust::host_vector< P * > A_ptrs
 
thrust::host_vector< P * > Ainv_ptrs
 
thrust::device_vector< P * > A_ptrs_device
 
thrust::device_vector< P * > Ainv_ptrs_device
 
thrust::device_vector< int > info
 

Member Function Documentation

◆ apply()

template<typename T , typename S >
void graphite::BlockJacobiPreconditioner< T, S >::apply ( Graph< T, S > *  graph,
T *  z,
const T *  r,
StreamPool &  streams 
)
inlineoverridevirtual

Implements graphite::Preconditioner< T, S >.

◆ set_damping_factor()

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

Implements graphite::Preconditioner< T, S >.

◆ update_structure()

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

Implements graphite::Preconditioner< T, S >.

◆ update_values()

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

Implements graphite::Preconditioner< T, S >.

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