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

Public Types

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

Public Member Functions

virtual void apply_update_async (const T *delta_x, T *jacobian_scales, cudaStream_t stream)=0
 
virtual void augment_block_diagonal_async (InvP *block_diagonal, InvP *scalar_diagonal, T mu, cudaStream_t stream)=0
 
virtual void apply_block_jacobi (T *z, const T *r, InvP *block_diagonal, cudaStream_t stream)=0
 
virtual size_t dimension () const =0
 
virtual size_t count () const =0
 
virtual void backup_parameters_async ()=0
 
virtual void restore_parameters_async ()=0
 
virtual void to_device ()=0
 
virtual const std::unordered_map< size_t, size_t > & get_global_map () const =0
 
virtual const size_t * get_hessian_ids () const =0
 
virtual void set_hessian_column (const size_t global_id, const size_t hessian_column, const size_t block_index)=0
 
virtual bool is_fixed (const size_t id) const =0
 
virtual bool is_active (const size_t id) const =0
 
virtual uint8_t * get_active_state () const =0
 
virtual const size_t * get_block_ids () const =0
 

Member Function Documentation

◆ apply_block_jacobi()

template<typename T , typename S >
virtual void graphite::BaseVertexDescriptor< T, S >::apply_block_jacobi ( T *  z,
const T *  r,
InvP *  block_diagonal,
cudaStream_t  stream 
)
pure virtual

Implemented in graphite::VertexDescriptor< T, S, VTraits >.

◆ apply_update_async()

template<typename T , typename S >
virtual void graphite::BaseVertexDescriptor< T, S >::apply_update_async ( const T *  delta_x,
T *  jacobian_scales,
cudaStream_t  stream 
)
pure virtual

Implemented in graphite::VertexDescriptor< T, S, VTraits >.

◆ augment_block_diagonal_async()

template<typename T , typename S >
virtual void graphite::BaseVertexDescriptor< T, S >::augment_block_diagonal_async ( InvP *  block_diagonal,
InvP *  scalar_diagonal,
T  mu,
cudaStream_t  stream 
)
pure virtual

Implemented in graphite::VertexDescriptor< T, S, VTraits >.

◆ backup_parameters_async()

template<typename T , typename S >
virtual void graphite::BaseVertexDescriptor< T, S >::backup_parameters_async ( )
pure virtual

Implemented in graphite::VertexDescriptor< T, S, VTraits >.

◆ count()

template<typename T , typename S >
virtual size_t graphite::BaseVertexDescriptor< T, S >::count ( ) const
pure virtual

Implemented in graphite::VertexDescriptor< T, S, VTraits >.

◆ dimension()

template<typename T , typename S >
virtual size_t graphite::BaseVertexDescriptor< T, S >::dimension ( ) const
pure virtual

Implemented in graphite::VertexDescriptor< T, S, VTraits >.

◆ get_active_state()

template<typename T , typename S >
virtual uint8_t * graphite::BaseVertexDescriptor< T, S >::get_active_state ( ) const
pure virtual

Implemented in graphite::VertexDescriptor< T, S, VTraits >.

◆ get_block_ids()

template<typename T , typename S >
virtual const size_t * graphite::BaseVertexDescriptor< T, S >::get_block_ids ( ) const
pure virtual

Implemented in graphite::VertexDescriptor< T, S, VTraits >.

◆ is_active()

template<typename T , typename S >
virtual bool graphite::BaseVertexDescriptor< T, S >::is_active ( const size_t  id) const
pure virtual

Implemented in graphite::VertexDescriptor< T, S, VTraits >.

◆ is_fixed()

template<typename T , typename S >
virtual bool graphite::BaseVertexDescriptor< T, S >::is_fixed ( const size_t  id) const
pure virtual

Implemented in graphite::VertexDescriptor< T, S, VTraits >.

◆ restore_parameters_async()

template<typename T , typename S >
virtual void graphite::BaseVertexDescriptor< T, S >::restore_parameters_async ( )
pure virtual

Implemented in graphite::VertexDescriptor< T, S, VTraits >.

◆ set_hessian_column()

template<typename T , typename S >
virtual void graphite::BaseVertexDescriptor< T, S >::set_hessian_column ( const size_t  global_id,
const size_t  hessian_column,
const size_t  block_index 
)
pure virtual

Implemented in graphite::VertexDescriptor< T, S, VTraits >.

◆ to_device()

template<typename T , typename S >
virtual void graphite::BaseVertexDescriptor< T, S >::to_device ( )
pure virtual

Implemented in graphite::VertexDescriptor< T, S, VTraits >.

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