DeviceManager.hpp

// Copyright (c) 2018 Graphcore Ltd. All rights reserved.
 
#ifndef poplar_DeviceManager_hpp
#define poplar_DeviceManager_hpp
 
#include <poplar/Device.hpp>
#include <poplar/OptionFlags.hpp>
 
namespace poplar {
 
namespace core {
class DeviceManagerImpl;
}
 
class DeviceManager {
public:
  DeviceManager();
  DeviceManager(const DeviceManager &);
  DeviceManager(DeviceManager &&) noexcept;
  virtual ~DeviceManager();
 
  std::size_t getNumDevices() const;
 
  std::vector<Device> getDevices(const OptionFlags &opts = {}) const;
 
  std::vector<Device> getDevices(TargetType type, unsigned requiredNumIPUs,
                                 const OptionFlags &opts = {}) const;
 
  Device getDevice(unsigned deviceManagerId,
                   const OptionFlags &opts = {}) const;
 
  std::vector<unsigned>
  getChildDeviceIds(unsigned parentId, unsigned numChildDeviceIpus = 1) const;
 
  static DeviceManager createDeviceManager();
 
private:
  std::unique_ptr<core::DeviceManagerImpl> impl;
};
 
} // end namespace poplar
 
#endif // poplar_DeviceManager_hpp