GpuLibraryRegistry.h

Go to the documentation of this file.

#pragma once
 
#ifdef __linux__
#include "GpuLibrary.h"
#include <memory>
#include <string>
#include <utility>
 
namespace Simulators
{
class GpuLibraryRegistry
{
  public:
    explicit GpuLibraryRegistry(std::string path = "libmaestro_gpu_simulators.so") : path(std::move(path))
    {
    }
 
    std::shared_ptr<GpuLibrary> Acquire(int device, bool mute = false)
    {
        auto library = GpuLibrary::GetInstance();
        return library->InitializeForDevice(path.c_str(), device, mute) ? library : nullptr;
    }
 
    int DeviceCount()
    {
        return GpuLibrary::GetInstance()->DiscoverDevices(path.c_str());
    }
 
  private:
    std::string path;
};
} // namespace Simulators
#endif