device.h

Go to the documentation of this file.

 
#pragma once
 
#include <knp/core/core.h>
 
#include <string>
 
namespace knp::core
{
 
enum class DeviceType
{
    CPU,
    GPU,
    AltAI1_NPU
};
 
 
class Device
{
public:
    virtual ~Device() = default;
 
public:
    [[nodiscard]] const UID &get_uid() const { return base_.uid_; }
 
    [[nodiscard]] auto &get_tags() { return base_.tags_; }
 
public:
    [[nodiscard]] virtual DeviceType get_type() const = 0;
 
    [[nodiscard]] virtual const std::string &get_name() const = 0;
 
    [[nodiscard]] virtual float get_power() const = 0;
 
protected:
    BaseData base_;
};
 
}  // namespace knp::core