AthCUDA/AthCUDAServices/src/DeviceMemoryResourceTool.h

Go to the documentation of this file.

//
// Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
//
#ifndef ATHCUDASERVICES_DEVICEMEMORYRESOURCETOOL_H
#define ATHCUDASERVICES_DEVICEMEMORYRESOURCETOOL_H
 
// Framework include(s).
#include "AthenaBaseComps/AthAlgTool.h"
 
// AthDevice include(s).
#include "AthDeviceInterfaces/IMemoryResourceTool.h"
 
// VecMem include(s).
#include <vecmem/memory/cuda/device_memory_resource.hpp>
 
namespace AthCUDA {

class DeviceMemoryResourceTool
    : public extends<AthAlgTool, AthDevice::IMemoryResourceTool> {
 
 public:
  // Inherit the base class's constructor(s).
  using extends::extends;


  virtual StatusCode initialize() override;



  virtual std::pmr::memory_resource& mr() const override;

  virtual std::string_view deviceName() const override;

 
 private:
  std::unique_ptr<std::pmr::memory_resource> m_mr;
  std::string m_deviceName;


  Gaudi::Property<int> m_deviceID{
      this, "DeviceID", vecmem::cuda::device_memory_resource::INVALID_DEVICE,
      "ID of the device to allocate memory on (-1 to use the default device)"};

 
};  // class DeviceMemoryResourceTool
 
}  // namespace AthCUDA
 
#endif  // ATHCUDASERVICES_DEVICEMEMORYRESOURCETOOL_H