limpet_data.h

Go to the documentation of this file.

// SPDX-FileCopyrightText: Copyright (c) NumeriCor GmbH
// SPDX-License-Identifier: LicenseRef-APL-1.1
 
#ifndef LIMPET_DATA_H
#define LIMPET_DATA_H
 
#include "target.h"
#ifdef HAS_ROCM_MODEL
#include <hip/hip_runtime.h>
#endif
#ifdef HAS_CUDA_MODEL
#include <cuda_runtime.h>
#endif
#include <iostream>
 
namespace limpet {
template<typename T>
class LimpetData {
  public:
    LimpetData(Target target, std::size_t n) : _target(target), _size(n) {
      this->_data = allocate_on_target<T>(this->_target, this->_size);
      this->_count = new std::size_t(0);
    };
 
    LimpetData() : _target(Target::UNKNOWN), _size(0), _data(nullptr) {}
 
    ~LimpetData() {
      if (*this->_count == 0) {
        deallocate_on_target<T>(this->_target, this->_data);
      }
      else {
        (*this->_count)--;
      }
    }
 
#if defined HAS_ROCM_MODEL || HAS_CUDA_MODEL
    __device__ __host__
#endif
    T *data() const {
      return this->_data;
    }
 
    LimpetData<T>& operator=(const LimpetData<T> &other) {
      if (this != &other) {
      this->_data = other._data;
      this->_size = other._size;
      this->_target = other._target;
      this->_count = other._count;
      (*this->_count)++;
      }
      return *this;
    }
 
  T operator*() {
    return *this->_data;
  }
  T* operator&() {
    return this->_data;
  }
 
  private:
    Target _target;
    std::size_t _size;
    T *_data = nullptr;
    std::size_t *_count = nullptr;
    bool _copied;
};
}
 
#endif // LIMPET_DATA_H