MemoryManager.hpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file MemoryManager.hpp

// This file is part of the Acts project.
//
// Copyright (C) 2020 CERN for the benefit of the Acts project
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

#pragma once

// System include(s).
#include <cstddef>
#include <vector>

namespace Acts {
namespace Cuda {

class MemoryManager {
 public:
  ~MemoryManager();


  MemoryManager(const MemoryManager&) = delete;
  MemoryManager(MemoryManager&&) = delete;

  MemoryManager& operator=(const MemoryManager&) = delete;
  MemoryManager& operator=(MemoryManager&&) = delete;



  static MemoryManager& instance();

  void setMemorySize(std::size_t sizeInBytes, int device = -1);



  std::size_t availableMemory(int device = -1) const;

  void* allocate(std::size_t sizeInBytes, int device = -1);

  void reset(int device = -1);


 private:
  MemoryManager();

  struct DeviceMemory {
    std::size_t m_size = 0;
    char* m_ptr = nullptr;
    char* m_nextAllocation = nullptr;
    std::ptrdiff_t m_maxUsage = 0;
  };

  std::vector<DeviceMemory> m_memory;

};  // class MemoryManager

}  // namespace Cuda
}  // namespace Acts