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