ComputeCapability.h

#pragma once
 
#include <iostream>
#include <string>
 
namespace isce3 { namespace cuda { namespace core {

struct ComputeCapability {
    constexpr ComputeCapability(int major, int minor)
        : major {major}, minor {minor}
    {}
 
    explicit operator std::string() const
    {
        return std::to_string(major) + "." + std::to_string(minor);
    }
 
    friend std::ostream& operator<<(std::ostream& os, ComputeCapability cc)
    {
        return os << std::string(cc);
    }
 
    friend constexpr bool operator==(ComputeCapability lhs,
                                     ComputeCapability rhs) noexcept
    {
        return lhs.major == rhs.major and lhs.minor == rhs.minor;
    }
 
    friend constexpr bool operator!=(ComputeCapability lhs,
                                     ComputeCapability rhs) noexcept
    {
        return not(lhs == rhs);
    }
 
    friend constexpr bool operator<(ComputeCapability lhs,
                                    ComputeCapability rhs) noexcept
    {
        return lhs.major < rhs.major or
               (lhs.major == rhs.major and lhs.minor < rhs.minor);
    }
 
    friend constexpr bool operator>(ComputeCapability lhs,
                                    ComputeCapability rhs) noexcept
    {
        return lhs.major > rhs.major or
               (lhs.major == rhs.major and lhs.minor > rhs.minor);
    }
 
    friend constexpr bool operator<=(ComputeCapability lhs,
                                     ComputeCapability rhs) noexcept
    {
        return not(lhs > rhs);
    }
 
    friend constexpr bool operator>=(ComputeCapability lhs,
                                     ComputeCapability rhs) noexcept
    {
        return not(lhs < rhs);
    }
 
    int major; 
    int minor; 
};

constexpr ComputeCapability minComputeCapability() noexcept { return {3, 5}; }
 
}}} // namespace isce3::cuda::core