AthCUDA/AthCUDAServices/src/HostMemoryResourceTool.h

Go to the documentation of this file.

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

class HostMemoryResourceTool
    : 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;

 
 private:
  std::unique_ptr<std::pmr::memory_resource> m_mr;
 
};  // class HostMemoryResourceTool
 
}  // namespace AthCUDA
 
#endif  // ATHCUDASERVICES_HOSTMEMORYRESOURCETOOL_H