cf5028d55b
* Added CUDA support * Performance improvements * API improvements * Added "depth" parameter to GEMM * Android cross-compilation
185 lines
6.5 KiB
C++
185 lines
6.5 KiB
C++
#include <cmath>
|
|
#include <iostream>
|
|
#include "isaac/array.h"
|
|
#include "isaac/value_scalar.h"
|
|
#include "isaac/exception/unknown_datatype.h"
|
|
|
|
namespace isaac
|
|
{
|
|
|
|
template<class T>
|
|
void value_scalar::init(T const & s)
|
|
{
|
|
switch(dtype_)
|
|
{
|
|
// case BOOL_TYPE: values_.bool8 = s; break;
|
|
case CHAR_TYPE: values_.int8 = s; break;
|
|
case UCHAR_TYPE: values_.uint8 = s; break;
|
|
case SHORT_TYPE: values_.int16 = s; break;
|
|
case USHORT_TYPE: values_.uint16 = s; break;
|
|
case INT_TYPE: values_.int32 = s; break;
|
|
case UINT_TYPE: values_.uint32 = s; break;
|
|
case LONG_TYPE: values_.int64 = s; break;
|
|
case ULONG_TYPE: values_.uint64 = s; break;
|
|
// case HALF_TYPE: values_.float16 = s; break;
|
|
case FLOAT_TYPE: values_.float32 = s; break;
|
|
case DOUBLE_TYPE: values_.float64 = s; break;
|
|
default: throw unknown_datatype(dtype_);
|
|
}
|
|
}
|
|
|
|
#define INSTANTIATE(CLTYPE, ADTYPE) value_scalar::value_scalar(CLTYPE value, numeric_type dtype) : dtype_(dtype) { init(value); }
|
|
|
|
INSTANTIATE(cl_char, CHAR_TYPE)
|
|
INSTANTIATE(cl_uchar, UCHAR_TYPE)
|
|
INSTANTIATE(cl_short, SHORT_TYPE)
|
|
INSTANTIATE(cl_ushort, USHORT_TYPE)
|
|
INSTANTIATE(cl_int, INT_TYPE)
|
|
INSTANTIATE(cl_uint, UINT_TYPE)
|
|
INSTANTIATE(cl_long, LONG_TYPE)
|
|
INSTANTIATE(cl_ulong, ULONG_TYPE)
|
|
INSTANTIATE(cl_float, FLOAT_TYPE)
|
|
INSTANTIATE(cl_double, DOUBLE_TYPE)
|
|
|
|
#undef INSTANTIATE
|
|
|
|
value_scalar::value_scalar(numeric_type dtype) : dtype_(dtype) {}
|
|
value_scalar::value_scalar(scalar const & s) : dtype_(s.dtype()) { init(s); }
|
|
value_scalar::value_scalar(array_expression const &expr) : dtype_(expr.dtype()) { init(scalar(expr)); }
|
|
|
|
values_holder value_scalar::values() const
|
|
{ return values_; }
|
|
|
|
numeric_type value_scalar::dtype() const
|
|
{ return dtype_; }
|
|
|
|
template<class T>
|
|
T value_scalar::cast() const
|
|
{
|
|
switch(dtype_)
|
|
{
|
|
case CHAR_TYPE: return values_.int8;
|
|
case UCHAR_TYPE: return values_.uint8;
|
|
case SHORT_TYPE: return values_.int16;
|
|
case USHORT_TYPE: return values_.uint16;
|
|
case INT_TYPE: return values_.int32;
|
|
case UINT_TYPE: return values_.uint32;
|
|
case LONG_TYPE: return values_.int64;
|
|
case ULONG_TYPE: return values_.uint64;
|
|
// case HALF_TYPE: return values_.float16;
|
|
case FLOAT_TYPE: return values_.float32;
|
|
case DOUBLE_TYPE: return values_.float64;
|
|
default: throw unknown_datatype(dtype_); //unreachable
|
|
}
|
|
}
|
|
|
|
#define INSTANTIATE(type) value_scalar::operator type() const { return cast<type>(); }
|
|
INSTANTIATE(bool)
|
|
INSTANTIATE(cl_char)
|
|
INSTANTIATE(cl_uchar)
|
|
INSTANTIATE(cl_short)
|
|
INSTANTIATE(cl_ushort)
|
|
INSTANTIATE(cl_int)
|
|
INSTANTIATE(cl_uint)
|
|
INSTANTIATE(cl_long)
|
|
INSTANTIATE(cl_ulong)
|
|
INSTANTIATE(cl_float)
|
|
INSTANTIATE(cl_double)
|
|
#undef INSTANTIATE
|
|
|
|
value_scalar int8(cl_char v) { return value_scalar(v); }
|
|
value_scalar uint8(cl_uchar v) { return value_scalar(v); }
|
|
value_scalar int16(cl_short v) { return value_scalar(v); }
|
|
value_scalar uint16(cl_ushort v) { return value_scalar(v); }
|
|
value_scalar int32(cl_int v) { return value_scalar(v); }
|
|
value_scalar uint32(cl_uint v) { return value_scalar(v); }
|
|
value_scalar int64(cl_long v) { return value_scalar(v); }
|
|
value_scalar uint64(cl_ulong v) { return value_scalar(v); }
|
|
value_scalar float32(cl_float v) { return value_scalar(v); }
|
|
value_scalar float64(cl_double v) { return value_scalar(v); }
|
|
|
|
#define VALUE(type, OP, x, y) (type)x OP y
|
|
#define INSTANTIATE(NAME, LDEC, RDEC, OP)\
|
|
value_scalar NAME(LDEC, RDEC)\
|
|
{\
|
|
switch(x.dtype())\
|
|
{\
|
|
case CHAR_TYPE: return VALUE(cl_char, OP, x, y);\
|
|
case UCHAR_TYPE: return VALUE(cl_uchar, OP, x, y);\
|
|
case SHORT_TYPE: return VALUE(cl_short, OP, x, y);\
|
|
case USHORT_TYPE: return VALUE(cl_ushort, OP, x, y);\
|
|
case INT_TYPE: return VALUE(cl_int, OP, x, y);\
|
|
case UINT_TYPE: return VALUE(cl_uint, OP, x, y);\
|
|
case LONG_TYPE: return VALUE(cl_long, OP, x, y);\
|
|
case ULONG_TYPE: return VALUE(cl_ulong, OP, x, y);\
|
|
case FLOAT_TYPE: return VALUE(cl_float, OP, x, y);\
|
|
case DOUBLE_TYPE: return VALUE(cl_double, OP, x, y);\
|
|
default: throw unknown_datatype(x.dtype());\
|
|
}\
|
|
}
|
|
|
|
#define INSTANTIATE_ALL(NAME, EXPR)\
|
|
INSTANTIATE(NAME, value_scalar const & x, cl_char y, EXPR)\
|
|
INSTANTIATE(NAME, value_scalar const & x, cl_uchar y, EXPR)\
|
|
INSTANTIATE(NAME, value_scalar const & x, cl_short y, EXPR)\
|
|
INSTANTIATE(NAME, value_scalar const & x, cl_ushort y, EXPR)\
|
|
INSTANTIATE(NAME, value_scalar const & x, cl_int y, EXPR)\
|
|
INSTANTIATE(NAME, value_scalar const & x, cl_uint y, EXPR)\
|
|
INSTANTIATE(NAME, value_scalar const & x, cl_long y, EXPR)\
|
|
INSTANTIATE(NAME, value_scalar const & x, cl_ulong y, EXPR)\
|
|
INSTANTIATE(NAME, value_scalar const & x, cl_float y, EXPR)\
|
|
INSTANTIATE(NAME, value_scalar const & x, cl_double y, EXPR)\
|
|
INSTANTIATE(NAME, cl_char y, value_scalar const & x, EXPR)\
|
|
INSTANTIATE(NAME, cl_uchar y, value_scalar const & x, EXPR)\
|
|
INSTANTIATE(NAME, cl_short y, value_scalar const & x, EXPR)\
|
|
INSTANTIATE(NAME, cl_ushort y, value_scalar const & x, EXPR)\
|
|
INSTANTIATE(NAME, cl_int y, value_scalar const & x, EXPR)\
|
|
INSTANTIATE(NAME, cl_uint y, value_scalar const & x, EXPR)\
|
|
INSTANTIATE(NAME, cl_long y, value_scalar const & x, EXPR)\
|
|
INSTANTIATE(NAME, cl_ulong y, value_scalar const & x, EXPR)\
|
|
INSTANTIATE(NAME, cl_float y, value_scalar const & x, EXPR)\
|
|
INSTANTIATE(NAME, cl_double y, value_scalar const & x, EXPR)
|
|
|
|
INSTANTIATE_ALL(operator+, +)
|
|
INSTANTIATE_ALL(operator-, -)
|
|
INSTANTIATE_ALL(operator*, *)
|
|
INSTANTIATE_ALL(operator/, /)
|
|
INSTANTIATE_ALL(operator>, > )
|
|
INSTANTIATE_ALL(operator>=, >=)
|
|
INSTANTIATE_ALL(operator<, < )
|
|
INSTANTIATE_ALL(operator<=, <=)
|
|
INSTANTIATE_ALL(operator==, ==)
|
|
INSTANTIATE_ALL(operator!=, !=)
|
|
|
|
#undef VALUE
|
|
#define VALUE(type, OP, x, y) OP((type)x,(type)y)
|
|
INSTANTIATE_ALL(max, std::max)
|
|
INSTANTIATE_ALL(min, std::min)
|
|
|
|
#undef VALUE
|
|
#define VALUE(type, OP, x, y) OP((type)x, y)
|
|
INSTANTIATE_ALL(pow, std::pow)
|
|
|
|
#undef INSTANTIATE_ALL
|
|
#undef INSTANTIATE
|
|
|
|
std::ostream & operator<<(std::ostream & os, value_scalar const & s)
|
|
{
|
|
switch(s.dtype())
|
|
{
|
|
case CHAR_TYPE: return os << static_cast<cl_char>(s);
|
|
case UCHAR_TYPE: return os << static_cast<cl_uchar>(s);
|
|
case SHORT_TYPE: return os << static_cast<cl_short>(s);
|
|
case USHORT_TYPE: return os << static_cast<cl_ushort>(s);
|
|
case INT_TYPE: return os << static_cast<cl_int>(s);
|
|
case UINT_TYPE: return os << static_cast<cl_uint>(s);
|
|
case LONG_TYPE: return os << static_cast<cl_long>(s);
|
|
case ULONG_TYPE: return os << static_cast<cl_ulong>(s);
|
|
case FLOAT_TYPE: return os << static_cast<cl_float>(s);
|
|
case DOUBLE_TYPE: return os << static_cast<cl_double>(s);
|
|
default: throw unknown_datatype(s.dtype());;
|
|
}
|
|
}
|
|
|
|
}
|