diff options
| author |  Benoit Steiner <benoit.steiner.goog@gmail.com> | 2017-03-31 08:22:25 -0700 |
|---|---|---|
| committer | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2017-03-31 08:22:25 -0700 |
| commit | 73fcaa319fcd12328e0577042862db471488fd5c (patch) | |
| tree | 53f94c4c3e3389993eb8599ab18ac8fe2c25de07 /unsupported/Eigen/CXX11/src/Tensor/TensorArgMax.h | |
| parent | bd64ee8555559ee13f02f2921594b4bd224f9d00 (diff) | |
Gate the sycl specific code under #ifdef sycl
Diffstat (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorArgMax.h')
| -rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/TensorArgMax.h | 9 |
1 files changed, 7 insertions, 2 deletions
diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorArgMax.h b/unsupported/Eigen/CXX11/src/Tensor/TensorArgMax.h index 44b79e725..85facfb64 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorArgMax.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorArgMax.h @@ -228,7 +228,11 @@ struct TensorEvaluator<const TensorTupleReducerOp<ReduceOp, Dims, ArgType>, Devi EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE TensorEvaluator(const XprType& op, const Device& device) : m_orig_impl(op.expression(), device), m_impl(op.expression().index_tuples().reduce(op.reduce_dims(), op.reduce_op()), device), - m_return_dim(op.return_dim()), m_device(device) { + m_return_dim(op.return_dim()) +#ifdef EIGEN_USE_SYCL + ,m_device(device) +#endif + { gen_strides(m_orig_impl.dimensions(), m_strides); if (Layout == static_cast<int>(ColMajor)) { @@ -307,8 +311,9 @@ struct TensorEvaluator<const TensorTupleReducerOp<ReduceOp, Dims, ArgType>, Devi StrideDims m_strides; Index m_stride_mod; Index m_stride_div; - // required by sycl +#ifdef EIGEN_USE_SYCL const Device& m_device; +#endif }; } // end namespace Eigen |