Template Numerical Library version main:e6bb26f
Public Types | Public Member Functions | Static Public Member Functions | Protected Attributes | List of all members
TNL::Algorithms::Segments::SlicedEllpack< Device, Index, IndexAllocator, Organization, SliceSize > Class Template Reference
Collaboration diagram for TNL::Algorithms::Segments::SlicedEllpack< Device, Index, IndexAllocator, Organization, SliceSize >:
Collaboration graph
[legend]

Public Types

using ConstViewType = SlicedEllpackView< Device, std::add_const_t< Index >, Organization, SliceSize >
 
using DeviceType = Device
 
using IndexType = std::remove_const_t< Index >
 
using OffsetsContainer = Containers::Vector< Index, DeviceType, IndexType, IndexAllocator >
 
using SegmentViewType = SegmentView< IndexType, Organization >
 
template<typename Device_ , typename Index_ >
using ViewTemplate = SlicedEllpackView< Device_, Index_, Organization, SliceSize >
 
using ViewType = SlicedEllpackView< Device, Index, Organization, SliceSize >
 

Public Member Functions

template<typename SizesContainer >
 SlicedEllpack (const SizesContainer &sizes)
 
 SlicedEllpack (const SlicedEllpack &segments)=default
 
template<typename ListIndex >
 SlicedEllpack (const std::initializer_list< ListIndex > &segmentsSizes)
 
 SlicedEllpack (SlicedEllpack &&segments) noexcept=default
 
template<typename Function >
void forAllElements (Function &&f) const
 
template<typename Function >
void forAllSegments (Function &&f) const
 
template<typename Function >
void forElements (IndexType first, IndexType last, Function &&f) const
 
template<typename Function >
void forSegments (IndexType begin, IndexType end, Function &&f) const
 
const ConstViewType getConstView () const
 
__cuda_callable__ IndexType getGlobalIndex (Index segmentIdx, Index localIdx) const
 
__cuda_callable__ IndexType getSegmentsCount () const
 
__cuda_callable__ IndexType getSegmentSize (IndexType segmentIdx) const
 
__cuda_callable__ SegmentViewType getSegmentView (IndexType segmentIdx) const
 
__cuda_callable__ IndexType getSize () const
 Number segments.
 
__cuda_callable__ IndexType getStorageSize () const
 
ViewType getView ()
 
void load (File &file)
 
SlicedEllpackoperator= (const SlicedEllpack &source)=default
 
template<typename Device_ , typename Index_ , typename IndexAllocator_ , ElementsOrganization Organization_>
SlicedEllpackoperator= (const SlicedEllpack< Device_, Index_, IndexAllocator_, Organization_, SliceSize > &source)
 
template<typename Device_ , typename Index_ , typename IndexAllocator_ , ElementsOrganization Organization_>
SlicedEllpack< Device, Index, IndexAllocator, Organization, SliceSize > & operator= (const SlicedEllpack< Device_, Index_, IndexAllocator_, Organization_, SliceSize > &source)
 
template<typename Fetch >
SegmentsPrinter< SlicedEllpack, Fetch > print (Fetch &&fetch) const
 
template<typename Fetch >
auto print (Fetch &&fetch) const -> SegmentsPrinter< SlicedEllpack, Fetch >
 
template<typename Fetch , typename Reduction , typename ResultKeeper , typename Real >
void reduceAllSegments (Fetch &fetch, const Reduction &reduction, ResultKeeper &keeper, const Real &zero) const
 
template<typename Fetch , typename Reduction , typename ResultKeeper , typename Real >
void reduceSegments (IndexType first, IndexType last, Fetch &fetch, const Reduction &reduction, ResultKeeper &keeper, const Real &zero) const
 
void reset ()
 
void save (File &file) const
 
template<typename SizesHolder = OffsetsContainer>
void setSegmentsSizes (const SizesHolder &sizes)
 Set sizes of particular segments.
 

Static Public Member Functions

static constexpr ElementsOrganization getOrganization ()
 
static String getSegmentsType ()
 
static std::string getSerializationType ()
 
static constexpr int getSliceSize ()
 
static constexpr bool havePadding ()
 

Protected Attributes

IndexType alignedSize = 0
 
IndexType segmentsCount = 0
 
IndexType size = 0
 
OffsetsContainer sliceOffsets
 
OffsetsContainer sliceSegmentSizes
 

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