GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-07-28 15:52:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

Disable vectorization on device only when compiling for sycl

Browse Source
This commit is contained in:
Benoit Steiner 2016-11-02 11:44:27 -07:00
parent e44519744e
commit 0585b2965d
2 changed files with 26 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
Eigen/Core
View File

@ -35,17 +35,17 @@
#undef EIGEN_INTERNAL_DEBUGGING #undef EIGEN_INTERNAL_DEBUGGING
#endif #endif
// Do not try to vectorize on CUDA and SYCL!
#ifndef EIGEN_DONT_VECTORIZE
#define EIGEN_DONT_VECTORIZE
#endif
#ifdef EIGEN_EXCEPTIONS #ifdef EIGEN_EXCEPTIONS
#undef EIGEN_EXCEPTIONS #undef EIGEN_EXCEPTIONS
#endif #endif
// All functions callable from CUDA code must be qualified with __device__ // All functions callable from CUDA code must be qualified with __device__
#ifdef __CUDACC__ #ifdef __CUDACC__
// Do not try to vectorize on CUDA and SYCL!
#ifndef EIGEN_DONT_VECTORIZE
#define EIGEN_DONT_VECTORIZE
#endif
#define EIGEN_DEVICE_FUNC __host__ __device__ #define EIGEN_DEVICE_FUNC __host__ __device__
// We need math_functions.hpp to ensure that that EIGEN_USING_STD_MATH macro // We need math_functions.hpp to ensure that that EIGEN_USING_STD_MATH macro
// works properly on the device side // works properly on the device side

21
unsupported/Eigen/CXX11/src/Tensor/TensorMeta.h
View File

@ -83,6 +83,27 @@ struct PacketType<half, GpuDevice> {
}; };
#endif #endif
#if defined(EIGEN_USE_SYCL)
template <typename T>
struct PacketType<T, SyclDevice> {
typedef T type;
static const int size = 1;
enum {
HasAdd = 0,
HasSub = 0,
HasMul = 0,
HasNegate = 0,
HasAbs = 0,
HasArg = 0,
HasAbs2 = 0,
HasMin = 0,
HasMax = 0,
HasConj = 0,
HasSetLinear = 0,
HasBlend = 0
};
};
#endif
// Tuple mimics std::pair but works on e.g. nvcc. // Tuple mimics std::pair but works on e.g. nvcc.