|
|
@@ -148,43 +148,53 @@ static inline half clamp(half x, half a, half b) |
|
|
|
|
|
|
|
namespace half_ops |
|
|
|
{ |
|
|
|
/* Enumerate the types for which operations with half are valid */ |
|
|
|
template<typename FROM, typename TO = void> struct valid {}; |
|
|
|
|
|
|
|
template<typename TO> struct valid<uint8_t, TO> |
|
|
|
{ typedef half from; typedef TO to; }; |
|
|
|
template<typename TO> struct valid<int8_t, TO> |
|
|
|
{ typedef half from; typedef TO to; }; |
|
|
|
template<typename TO> struct valid<uint16_t, TO> |
|
|
|
{ typedef half from; typedef TO to; }; |
|
|
|
template<typename TO> struct valid<int16_t, TO> |
|
|
|
{ typedef half from; typedef TO to; }; |
|
|
|
template<typename TO> struct valid<uint32_t, TO> |
|
|
|
{ typedef half from; typedef TO to; }; |
|
|
|
template<typename TO> struct valid<int32_t, TO> |
|
|
|
{ typedef half from; typedef TO to; }; |
|
|
|
template<typename TO> struct valid<uint64_t, TO> |
|
|
|
{ typedef half from; typedef TO to; }; |
|
|
|
template<typename TO> struct valid<int64_t, TO> |
|
|
|
{ typedef half from; typedef TO to; }; |
|
|
|
|
|
|
|
template<typename TO> struct valid<float, TO> |
|
|
|
{ typedef float from; typedef TO to; }; |
|
|
|
template<typename TO> struct valid<double, TO> |
|
|
|
{ typedef double from; typedef TO to; }; |
|
|
|
template<typename TO> struct valid<ldouble, TO> |
|
|
|
{ typedef ldouble from; typedef TO to; }; |
|
|
|
|
|
|
|
#define DECLARE_HALF_NUMERIC_OPS(op) \ |
|
|
|
/* integral + half */ \ |
|
|
|
template<typename T> static inline \ |
|
|
|
typename std::enable_if<std::is_integral<T>::value,half>::type \ |
|
|
|
operator op(T x, half h) { return (half)(int)x op h; } \ |
|
|
|
/* other + half */ \ |
|
|
|
template<typename T> \ |
|
|
|
static inline typename valid<T>::from operator op(T x, half h) \ |
|
|
|
{ return (typename valid<T>::from)x op (typename valid<T>::from)h; } \ |
|
|
|
\ |
|
|
|
template<typename T> static inline \ |
|
|
|
typename std::enable_if<std::is_integral<T>::value,T&>::type \ |
|
|
|
operator op##=(T &x, half h) { return x = x op h; } \ |
|
|
|
/* half + other */ \ |
|
|
|
template<typename T> \ |
|
|
|
static inline typename valid<T>::from operator op(half h, T x) \ |
|
|
|
{ return (typename valid<T>::from)h op (typename valid<T>::from)x; } \ |
|
|
|
\ |
|
|
|
/* half + integral */ \ |
|
|
|
template<typename T> static inline \ |
|
|
|
typename std::enable_if<std::is_integral<T>::value,half>::type \ |
|
|
|
operator op(half h, T x) { return h op (half)(int)x; } \ |
|
|
|
/* other += half */ \ |
|
|
|
template<typename T> \ |
|
|
|
static inline typename valid<T,T>::to& operator op##=(T& x, half h) \ |
|
|
|
{ return x = (typename valid<T>::from)x op (typename valid<T>::from)h; } \ |
|
|
|
\ |
|
|
|
template<typename T> static inline \ |
|
|
|
typename std::enable_if<std::is_integral<T>::value,half&>::type \ |
|
|
|
operator op##=(half &h, T x) { return h = h op x; } \ |
|
|
|
\ |
|
|
|
/* floating point + half */ \ |
|
|
|
template<typename T> static inline \ |
|
|
|
typename std::enable_if<std::is_floating_point<T>::value,T>::type \ |
|
|
|
operator op(T x, half h) { return x op (T)h; } \ |
|
|
|
\ |
|
|
|
template<typename T> static inline \ |
|
|
|
typename std::enable_if<std::is_floating_point<T>::value,T&>::type \ |
|
|
|
operator op##=(T &x, half h) { return x = x op h; } \ |
|
|
|
\ |
|
|
|
/* half + floating point */ \ |
|
|
|
template<typename T> static inline \ |
|
|
|
typename std::enable_if<std::is_floating_point<T>::value,T>::type \ |
|
|
|
operator op(half h, T x) { return (T)h op x; } \ |
|
|
|
\ |
|
|
|
template<typename T> static inline \ |
|
|
|
typename std::enable_if<std::is_floating_point<T>::value,half&>::type \ |
|
|
|
operator op##=(half &h, T x) { return h = h op x; } |
|
|
|
/* half += other */ \ |
|
|
|
template<typename T> \ |
|
|
|
static inline typename valid<T,half>::to& operator op##=(half& h, T x) \ |
|
|
|
{ return h = (typename valid<T>::from)h op (typename valid<T>::from)x; } |
|
|
|
|
|
|
|
DECLARE_HALF_NUMERIC_OPS(+) |
|
|
|
DECLARE_HALF_NUMERIC_OPS(-) |
|
|
@@ -194,25 +204,15 @@ DECLARE_HALF_NUMERIC_OPS(/) |
|
|
|
#undef DECLARE_HALF_NUMERIC_OPS |
|
|
|
|
|
|
|
#define DECLARE_HALF_BOOL_OPS(op) \ |
|
|
|
/* integral == half */ \ |
|
|
|
template<typename T> static inline \ |
|
|
|
typename std::enable_if<std::is_integral<T>::value,bool>::type \ |
|
|
|
operator op(T x, half h) { return (half)(int)x op h; } \ |
|
|
|
\ |
|
|
|
/* half == integral */ \ |
|
|
|
template<typename T> static inline \ |
|
|
|
typename std::enable_if<std::is_integral<T>::value,bool>::type \ |
|
|
|
operator op(half h, T x) { return h op (half)(int)x; } \ |
|
|
|
\ |
|
|
|
/* floating point == half */ \ |
|
|
|
template<typename T> static inline \ |
|
|
|
typename std::enable_if<std::is_floating_point<T>::value,bool>::type \ |
|
|
|
operator op(T x, half h) { return x op (T)h; } \ |
|
|
|
/* half == other */ \ |
|
|
|
template<typename T> \ |
|
|
|
static inline typename valid<T,bool>::to operator op(half h, T x) \ |
|
|
|
{ return (typename valid<T>::from)h op (typename valid<T>::from)x; } \ |
|
|
|
\ |
|
|
|
/* half == floating point */ \ |
|
|
|
template<typename T> static inline \ |
|
|
|
typename std::enable_if<std::is_floating_point<T>::value,bool>::type \ |
|
|
|
operator op(half h, T x) { return (T)h op x; } |
|
|
|
/* other == half */ \ |
|
|
|
template<typename T> \ |
|
|
|
static inline typename valid<T,bool>::to operator op(T x, half h) \ |
|
|
|
{ return (typename valid<T>::from)x op (typename valid<T>::from)h; } |
|
|
|
|
|
|
|
DECLARE_HALF_BOOL_OPS(==) |
|
|
|
DECLARE_HALF_BOOL_OPS(!=) |
|
|
|