Device.h

#pragma once

#include <string>

#include "ComputeCapability.h"

namespace isce3 { namespace cuda { namespace core {

class Device {
public:
    Device(int id);

    int id() const noexcept { return _id; }

    std::string name() const;

    size_t totalGlobalMem() const;

    ComputeCapability computeCapability() const;

    friend bool operator==(Device lhs, Device rhs) noexcept
    {
        return lhs.id() == rhs.id();
    }

    friend bool operator!=(Device lhs, Device rhs) noexcept
    {
        return not(lhs == rhs);
    }

private:
    int _id;
};

int getDeviceCount();

Device getDevice();

void setDevice(Device d);

}}} // namespace isce3::cuda::core