diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorSyclExprConstructor.h b/unsupported/Eigen/CXX11/src/Tensor/TensorSyclExprConstructor.h index 7b15f93fe..df1a732e7 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorSyclExprConstructor.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorSyclExprConstructor.h @@ -25,6 +25,15 @@ namespace Eigen { namespace TensorSycl { namespace internal { + +template +struct DeviceFixedSizeTensor; + +template +struct DeviceFixedSizeTensor>{ + template + static EIGEN_ALWAYS_INLINE Expr instantiate(Data& dt) {return Expr(ConvertToActualTypeSycl(typename Expr::Scalar, dt), Indices...);} +}; /// this class is used by EvalToOp in order to create an lhs expression which is /// a pointer from an accessor on device-only buffer template @@ -61,6 +70,23 @@ TENSORMAP(const) TENSORMAP() #undef TENSORMAP +/// specialisation of the \ref ExprConstructor struct when the node type is +/// TensorMap +#define TENSORMAPFIXEDSIZE(CVQual)\ +template class MakePointer_, size_t N, typename... Params>\ +struct ExprConstructor< CVQual TensorMap, Options_, MakeGlobalPointer>,\ +CVQual PlaceHolder, Options_, MakePointer_>, N>, Params...>{\ + typedef CVQual TensorMap, Options_, MakeGlobalPointer> Type;\ + Type expr;\ + template \ + ExprConstructor(FuncDetector &, const utility::tuple::Tuple &t)\ + : expr(DeviceFixedSizeTensor::instantiate(utility::tuple::get(t))){}\ +}; +TENSORMAPFIXEDSIZE(const) +TENSORMAPFIXEDSIZE() +#undef TENSORMAPFIXEDSIZE + #define UNARYCATEGORY(CVQual)\ template class UnaryCategory, typename OP, typename OrigRHSExpr, typename RHSExpr, typename... Params>\ struct ExprConstructor, CVQual UnaryCategory, Params...> {\ diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorSyclExtractFunctors.h b/unsupported/Eigen/CXX11/src/Tensor/TensorSyclExtractFunctors.h index 1293b14e2..ff8be5444 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorSyclExtractFunctors.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorSyclExtractFunctors.h @@ -41,6 +41,17 @@ template struct FunctorExtractor{ : m_dimensions(expr.dimensions()) {} }; + +#define SYCLEXTRTENSORMAPFIXEDSIZE(CVQual)\ +template class MakePointer_, typename Dev>\ +struct FunctorExtractor< TensorEvaluator , Options_, MakePointer_> , Dev> >{\ +FunctorExtractor(const TensorEvaluator , Options_, MakePointer_> , Dev>& ){}\ +}; + +SYCLEXTRTENSORMAPFIXEDSIZE(const) +SYCLEXTRTENSORMAPFIXEDSIZE() +#undef SYCLEXTRTENSORMAPFIXEDSIZE + /// specialisation of the \ref FunctorExtractor struct when the node type is /// TensorCwiseNullaryOp, TensorCwiseUnaryOp, and TensorBroadcastingOp #define SYCLEXTRFUNCUNARY(CVQual)\ diff --git a/unsupported/Eigen/CXX11/src/util/CXX11Meta.h b/unsupported/Eigen/CXX11/src/util/CXX11Meta.h index e7c1a1bae..49d315a66 100644 --- a/unsupported/Eigen/CXX11/src/util/CXX11Meta.h +++ b/unsupported/Eigen/CXX11/src/util/CXX11Meta.h @@ -36,27 +36,12 @@ struct type_list { constexpr static int count = sizeof...(tt); }; template struct type_list { constexpr static int count = sizeof...(tt) + 1; typedef t first_type; }; -#ifndef EIGEN_USE_SYCL template struct numeric_list { constexpr static std::size_t count = sizeof...(nn); }; template struct numeric_list { static const std::size_t count = sizeof...(nn) + 1; const static T first_value = n; }; -#else -template -struct numeric_list { - static constexpr std::size_t count = sizeof...(nn); - const T values[count] = {nn...}; -}; -template -struct numeric_list{ - static constexpr std::size_t count = 0; - //Array of size zero strictly forbiden in ISO C++ -}; - -#endif - /* numeric list constructors * * equivalencies: @@ -138,19 +123,9 @@ template struct get<0, type_lis template struct get> : get> {}; template struct get<0, numeric_list> { constexpr static T value = a; }; -#ifndef EIGEN_USE_SYCL template constexpr T array_get(const numeric_list&) { return get<(int)n, numeric_list>::value; } -#else -template constexpr T array_get(const numeric_list& l) { - return l.values[n]; -} -template constexpr T array_get(const numeric_list& ) { - return T(0); -} - -#endif /* always get type, regardless of dummy; good for parameter pack expansion */