GPUMemoryManager.h

Go to the documentation of this file.

// This file is part of the Peano project. For conditions of distribution and
// use, please see the copyright notice at www.peano-framework.org
#ifndef GPU_MEMORY_MANAGER_H
#define GPU_MEMORY_MANAGER_H
 
#include "tarch/multicore/multicore.h"
 
#include "tarch/multicore/BooleanSemaphore.h"
#include "tarch/logging/Log.h"
 
#include <map>
#include <list>
#include <vector>
 
namespace tarch {
  namespace accelerator {
    class GPUMemoryManager;
    #if defined(GPUOffloadingSYCL)
    typedef ::sycl::queue& DeviceType;
    #else
    typedef int DeviceType;
    #endif
  }
}
 
class tarch::accelerator::GPUMemoryManager {
  private:
    static tarch::logging::Log          _log;
 
    static int                          _allocatedMemory;
    static constexpr int                _maxMemory = -1;
 
    tarch::multicore::BooleanSemaphore  _semaphore;
 
    typedef std::map< int, std::vector<void*> >  FreePages;
 
    typedef std::map< void*, int >               UsedPages;
 
    struct DeviceMemory {
      FreePages freePages;
      UsedPages usedPages;
    };
 
    std::map<int, DeviceMemory>  _deviceMemory;
 
    GPUMemoryManager() = default;
 
  public:
    GPUMemoryManager( const GPUMemoryManager& copy ) = delete;
 
    virtual ~GPUMemoryManager();
 
    static GPUMemoryManager& getInstance();
 
    template <class T>
    T* allocate( int size, DeviceType device ) {
      return reinterpret_cast<T*>( allocateRawData(size * static_cast<int>(sizeof(T)), device ) );
    }
 
    void* allocateRawData( int size, DeviceType device );
 
    void free(void* devicePointer, DeviceType device);
 
    void clear();
};
 
#endif