From 84d7be103a2d4234f4aab91a434da85fc39b09d6 Mon Sep 17 00:00:00 2001 From: Benoit Steiner Date: Sat, 22 Jul 2017 03:19:34 +0000 Subject: Fixing Argmax that was breaking upstream TensorFlow. --- unsupported/Eigen/CXX11/src/Tensor/TensorArgMaxSycl.h | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) (limited to 'unsupported/Eigen/CXX11/src') diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorArgMaxSycl.h b/unsupported/Eigen/CXX11/src/Tensor/TensorArgMaxSycl.h index 8f76b8254..442639868 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorArgMaxSycl.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorArgMaxSycl.h @@ -40,7 +40,7 @@ struct traits > : public traits< typedef traits XprTraits; typedef typename XprTraits::StorageKind StorageKind; typedef typename XprTraits::Index Index; - typedef typename XprType::Scalar Scalar; + typedef Index Scalar; typedef typename XprType::Nested Nested; typedef typename remove_reference::type _Nested; static const int NumDimensions = XprTraits::NumDimensions; @@ -58,7 +58,8 @@ class TensorTupleReducerDeviceOp : public TensorBase::type Nested; typedef typename Eigen::internal::traits::StorageKind StorageKind; typedef typename Eigen::internal::traits::Index Index; - typedef typename XprType::CoeffReturnType CoeffReturnType; + typedef typename XprType::CoeffReturnType TupleType; + typedef Index CoeffReturnType; EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorTupleReducerDeviceOp(XprType expr, const Index return_dim, @@ -99,9 +100,9 @@ struct TensorEvaluator, Sy { typedef TensorTupleReducerDeviceOp XprType; typedef typename XprType::Index Index; - typedef typename XprType::Index Scalar; - typedef Index CoeffReturnType; - typedef typename XprType::CoeffReturnType TupleType; + typedef typename XprType::Scalar Scalar; + typedef typename XprType::CoeffReturnType CoeffReturnType; + typedef typename XprType::TupleType TupleType; typedef typename TensorEvaluator::Dimensions Dimensions; enum { -- cgit v1.2.3 From 2062ac995864d51caad71a83bfc267a6aa17f8c3 Mon Sep 17 00:00:00 2001 From: Mehdi Goli Date: Mon, 18 Sep 2017 18:17:39 +0100 Subject: Changes required for new ComputeCpp CE version. --- unsupported/Eigen/CXX11/src/Tensor/TensorDeviceSycl.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'unsupported/Eigen/CXX11/src') diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorDeviceSycl.h b/unsupported/Eigen/CXX11/src/Tensor/TensorDeviceSycl.h index 2b5749f55..6158acbd9 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorDeviceSycl.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorDeviceSycl.h @@ -46,8 +46,8 @@ struct SyclAllocator { namespace Eigen { - #define ConvertToActualTypeSycl(Scalar, buf_acc) reinterpret_cast::pointer_t>((&(*buf_acc.get_pointer()))) - #define ConvertToActualSyclOffset(Scalar, offset) offset/sizeof(Scalar) +#define ConvertToActualTypeSycl(Scalar, buf_acc) static_cast(static_cast(((buf_acc.get_pointer().get())))) +#define ConvertToActualSyclOffset(Scalar, offset) offset/sizeof(Scalar) template class MemCopyFunctor { -- cgit v1.2.3