FlexiStorage.h

Go to the documentation of this file.

#ifndef HeterogeneousCore_CUDAUtilities_interface_FlexiStorage_h
#define HeterogeneousCore_CUDAUtilities_interface_FlexiStorage_h
 
#include <cstdint>
 
namespace cms {
  namespace cuda {
 
    template <typename I, int S>
    class FlexiStorage {
    public:
      constexpr int capacity() const { return S; }
 
      constexpr I& operator[](int i) { return m_v[i]; }
      constexpr const I& operator[](int i) const { return m_v[i]; }
 
      constexpr I* data() { return m_v; }
      constexpr I const* data() const { return m_v; }
 
    private:
      I m_v[S];
    };
 
    template <typename I>
    class FlexiStorage<I, -1> {
    public:
      constexpr void init(I* v, int s) {
        m_v = v;
        m_capacity = s;
      }
 
      constexpr int capacity() const { return m_capacity; }
 
      constexpr I& operator[](int i) { return m_v[i]; }
      constexpr const I& operator[](int i) const { return m_v[i]; }
 
      constexpr I* data() { return m_v; }
      constexpr I const* data() const { return m_v; }
 
    private:
      I* m_v;
      int m_capacity;
    };
 
  }  // namespace cuda
 
}  // namespace cms
 
#endif