cms::alpakatools::OneToManyAssocRandomAccess< I, ONES, SIZE > Class Template Reference

#include <OneToManyAssoc.h>

Inheritance diagram for cms::alpakatools::OneToManyAssocRandomAccess< I, ONES, SIZE >:

Public Types
using	Counter = typename OneToManyAssocBase< I, ONES, SIZE >::Counter
using	View = typename OneToManyAssocBase< I, ONES, SIZE >::View
Public Types inherited from cms::alpakatools::OneToManyAssocBase< I, ONES, SIZE >
using	Counter = uint32_t
using	CountersOnly = OneToManyAssocBase< I, ONES, 0 >
using	index_type = I

Public Member Functions
template<typename TAcc >
ALPAKA_FN_HOST_ACC ALPAKA_FN_INLINE void	finalize (TAcc &acc, Counter *ws=nullptr)
ALPAKA_FN_HOST_ACC ALPAKA_FN_INLINE void	finalize ()
Public Member Functions inherited from cms::alpakatools::OneToManyAssocBase< I, ONES, SIZE >
template<typename TAcc >
ALPAKA_FN_ACC ALPAKA_FN_INLINE void	add (const TAcc &acc, CountersOnly const &co)
constexpr index_type const *	begin () const
constexpr index_type const *	begin (uint32_t b) const
constexpr auto	capacity () const
template<typename TAcc >
ALPAKA_FN_ACC ALPAKA_FN_INLINE void	count (const TAcc &acc, I b)
constexpr index_type const *	end () const
constexpr index_type const *	end (uint32_t b) const
template<typename TAcc >
ALPAKA_FN_ACC ALPAKA_FN_INLINE void	fill (const TAcc &acc, I b, index_type j)
ALPAKA_FN_HOST_ACC void	initStorage (View view)
constexpr auto	nOnes () const
constexpr auto	size () const
constexpr auto	size (uint32_t b) const
constexpr auto	totOnes () const
ALPAKA_FN_HOST_ACC void	zero ()

Static Public Member Functions
template<typename TAcc , typename TQueue >
static ALPAKA_FN_INLINE void	launchFinalize (OneToManyAssocRandomAccess *h, TQueue &queue)
template<typename TAcc , typename TQueue >
static ALPAKA_FN_INLINE void	launchFinalize (View view, TQueue &queue)
Static Public Member Functions inherited from cms::alpakatools::OneToManyAssocBase< I, ONES, SIZE >
template<typename TAcc >
ALPAKA_FN_ACC static ALPAKA_FN_INLINE uint32_t	atomicDecrement (const TAcc &acc, Counter &x)
template<typename TAcc >
ALPAKA_FN_ACC static ALPAKA_FN_INLINE uint32_t	atomicIncrement (const TAcc &acc, Counter &x)
static constexpr int32_t	ctCapacity ()
static constexpr int32_t	ctNOnes ()
template<typename TAcc , typename TQueue >
static ALPAKA_FN_INLINE void	launchZero (OneToManyAssocBase *h, TQueue &queue)
template<typename TAcc , typename TQueue >
static ALPAKA_FN_INLINE void	launchZero (View view, TQueue &queue)

Additional Inherited Members
Public Attributes inherited from cms::alpakatools::OneToManyAssocBase< I, ONES, SIZE >
FlexiStorage< index_type, SIZE >	content
FlexiStorage< Counter, ONES >	off
int32_t	psws

Detailed Description

template<typename I, int32_t ONES, int32_t SIZE>
class cms::alpakatools::OneToManyAssocRandomAccess< I, ONES, SIZE >

Definition at line 209 of file OneToManyAssoc.h.

Member Typedef Documentation

Counter

template<typename I, int32_t ONES, int32_t SIZE>

using cms::alpakatools::OneToManyAssocRandomAccess< I, ONES, SIZE >::Counter = typename OneToManyAssocBase<I, ONES, SIZE>::Counter

Definition at line 211 of file OneToManyAssoc.h.

View

template<typename I, int32_t ONES, int32_t SIZE>

using cms::alpakatools::OneToManyAssocRandomAccess< I, ONES, SIZE >::View = typename OneToManyAssocBase<I, ONES, SIZE>::View

Definition at line 212 of file OneToManyAssoc.h.

Member Function Documentation

finalize() [1/2]

template<typename I, int32_t ONES, int32_t SIZE>

template<typename TAcc >

ALPAKA_FN_HOST_ACC ALPAKA_FN_INLINE void cms::alpakatools::OneToManyAssocRandomAccess< I, ONES, SIZE >::finalize ( TAcc &  acc, Counter *  ws = nullptr  )

inline

Definition at line 215 of file OneToManyAssoc.h.

                                                                                        {
      ALPAKA_ASSERT_ACC(this->off[this->totOnes() - 1] == 0);
      blockPrefixScan(acc, this->off.data(), this->totOnes(), ws);
      ALPAKA_ASSERT_ACC(this->off[this->totOnes() - 1] == this->off[this->totOnes() - 2]);
    }

finalize() [2/2]

template<typename I, int32_t ONES, int32_t SIZE>

ALPAKA_FN_HOST_ACC ALPAKA_FN_INLINE void cms::alpakatools::OneToManyAssocRandomAccess< I, ONES, SIZE >::finalize ( void  )

inline

Definition at line 221 of file OneToManyAssoc.h.

                                                        {
      // Single thread finalize.
      for (uint32_t i = 1; static_cast<int>(i) < this->totOnes(); ++i)
        this->off[i] += this->off[i - 1];
    }

launchFinalize() [1/2]

template<typename I, int32_t ONES, int32_t SIZE>

template<typename TAcc , typename TQueue >

static ALPAKA_FN_INLINE void cms::alpakatools::OneToManyAssocRandomAccess< I, ONES, SIZE >::launchFinalize ( OneToManyAssocRandomAccess< I, ONES, SIZE > *  h, TQueue &  queue  )

inlinestatic

Definition at line 228 of file OneToManyAssoc.h.

                                                                                              {
      View view = {h, nullptr, nullptr, -1, -1};
      launchFinalize<TAcc>(view, queue);
    }

launchFinalize() [2/2]

template<typename I, int32_t ONES, int32_t SIZE>

template<typename TAcc , typename TQueue >

static ALPAKA_FN_INLINE void cms::alpakatools::OneToManyAssocRandomAccess< I, ONES, SIZE >::launchFinalize ( View  view, TQueue &  queue  )

inlinestatic

Definition at line 234 of file OneToManyAssoc.h.

                                                                          {
      // View stores a base pointer, we need to upcast back...
      auto h = static_cast<OneToManyAssocRandomAccess *>(view.assoc);
      ALPAKA_ASSERT_ACC(h);
      if constexpr (!requires_single_thread_per_block_v<TAcc>) {
        Counter *poff = (Counter *)((char *)(h) + offsetof(OneToManyAssocRandomAccess, off));
        auto nOnes = OneToManyAssocRandomAccess::ctNOnes();
        if constexpr (OneToManyAssocRandomAccess::ctNOnes() < 0) {
          ALPAKA_ASSERT_ACC(view.offStorage);
          ALPAKA_ASSERT_ACC(view.offSize > 0);
          nOnes = view.offSize;
          poff = view.offStorage;
        }
        ALPAKA_ASSERT_ACC(nOnes > 0);
        int32_t *ppsws = (int32_t *)((char *)(h) + offsetof(OneToManyAssocRandomAccess, psws));
        auto nthreads = 1024;
        auto nblocks = (nOnes + nthreads - 1) / nthreads;
        auto workDiv = cms::alpakatools::make_workdiv<TAcc>(nblocks, nthreads);
        alpaka::exec<TAcc>(queue,
                           workDiv,
                           multiBlockPrefixScan<Counter>(),
                           poff,
                           poff,
                           nOnes,
                           nblocks,
                           ppsws,
                           alpaka::getPreferredWarpSize(alpaka::getDev(queue)));
      } else {
        h->finalize();
      }
    }