MemoryResourcesAdaptorTool.h

Go to the documentation of this file.

//
// Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
//
#ifndef ATHDEVICECOMPS_MEMORYRESOURCESADAPTORTOOL_H
#define ATHDEVICECOMPS_MEMORYRESOURCESADAPTORTOOL_H
 
// Framework include(s).
#include "AthenaBaseComps/AthAlgTool.h"
#include "GaudiKernel/ToolHandle.h"
 
// AthDevice include(s).
#include "AthDeviceInterfaces/IMemoryResourceTool.h"
#include "AthDeviceInterfaces/IMemoryResourcesTool.h"
 
namespace AthDevice {

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


  virtual StatusCode initialize() override;



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

  virtual std::pmr::memory_resource* hostMR() const override;

 
 private:
  ToolHandle<IMemoryResourceTool> m_mainMRTool{
      this, "MainMRTool", "",
      "Tool providing the 'main' / 'device' memory resource"};

  ToolHandle<IMemoryResourceTool> m_hostMRTool{
      this, "HostMRTool", "",
      "Tool providing the 'host' (accessible) memory resource"};
 
};  // class MemoryResourcesAdaptorTool
 
}  // namespace AthDevice
 
#endif  // ATHDEVICECOMPS_MEMORYRESOURCESADAPTORTOOL_H