1313
1414#include "../../../Eigen/Core"
1515
16- #ifdef EIGEN_USE_SYCL
17- #undef min
18- #undef max
19- #undef isnan
20- #undef isinf
21- #undef isfinite
22- #include <SYCL/sycl.hpp>
23- #include <map>
24- #include <memory>
25- #include <utility>
26- #endif
27-
28- #include <Eigen/src/Core/util/DisableStupidWarnings.h>
16+ #if EIGEN_HAS_CXX11
2917
3018#include "../SpecialFunctions"
19+
20+ #include "../../../Eigen/src/Core/util/DisableStupidWarnings.h"
3121#include "src/util/CXX11Meta.h"
3222#include "src/util/MaxSizeVector.h"
3323
3929 * \code
4030 * #include <Eigen/CXX11/Tensor>
4131 * \endcode
32+ *
33+ * Much of the documentation can be found \ref eigen_tensors "here".
4234 */
4335
36+ #include <atomic>
37+ #include <chrono>
4438#include <cmath>
4539#include <cstddef>
4640#include <cstring>
47-
48- #ifdef _WIN32
49- typedef __int16 int16_t;
50- typedef unsigned __int16 uint16_t;
51- typedef __int32 int32_t;
52- typedef unsigned __int32 uint32_t;
53- typedef __int64 int64_t;
54- typedef unsigned __int64 uint64_t;
55- #else
56- #include <stdint.h>
57- #endif
58-
59- #if __cplusplus > 199711 || EIGEN_COMP_MSVC >= 1900
6041#include <random>
61- #endif
62-
63- #ifdef _WIN32
64- #include <windows.h>
65- #elif defined(__APPLE__)
66- #include <mach/mach_time.h>
67- #else
68- #include <time.h>
69- #endif
42+ #include <thread>
7043
71- #ifdef EIGEN_USE_THREADS
44+ #if defined( EIGEN_USE_THREADS) || defined(EIGEN_USE_SYCL)
7245#include "ThreadPool"
7346#endif
7447
7548#ifdef EIGEN_USE_GPU
76- #include <iostream>
77- #include <cuda_runtime.h>
78- #if __cplusplus >= 201103L
79- #include <atomic>
80- #include <unistd .h>
81- #endif
49+ #include <iostream>
50+ #if defined(EIGEN_USE_HIP)
51+ #include <hip/hip_runtime.h>
52+ #else
53+ #include <cuda_runtime .h>
54+ #endif
8255#endif
8356
8457#include "src/Tensor/TensorMacros.h"
@@ -88,7 +61,10 @@ typedef unsigned __int64 uint64_t;
8861#include "src/Tensor/TensorCostModel.h"
8962#include "src/Tensor/TensorDeviceDefault.h"
9063#include "src/Tensor/TensorDeviceThreadPool.h"
91- #include "src/Tensor/TensorDeviceCuda.h"
64+ #include "src/Tensor/TensorDeviceGpu.h"
65+ #ifndef gpu_assert
66+ #define gpu_assert(x)
67+ #endif
9268#include "src/Tensor/TensorDeviceSycl.h"
9369#include "src/Tensor/TensorIndexList.h"
9470#include "src/Tensor/TensorDimensionList.h"
@@ -101,18 +77,19 @@ typedef unsigned __int64 uint64_t;
10177#include "src/Tensor/TensorGlobalFunctions.h"
10278
10379#include "src/Tensor/TensorBase.h"
80+ #include "src/Tensor/TensorBlock.h"
10481
10582#include "src/Tensor/TensorEvaluator.h"
10683#include "src/Tensor/TensorExpr.h"
10784#include "src/Tensor/TensorReduction.h"
108- #include "src/Tensor/TensorReductionCuda .h"
85+ #include "src/Tensor/TensorReductionGpu .h"
10986#include "src/Tensor/TensorArgMax.h"
11087#include "src/Tensor/TensorConcatenation.h"
11188#include "src/Tensor/TensorContractionMapper.h"
11289#include "src/Tensor/TensorContractionBlocking.h"
11390#include "src/Tensor/TensorContraction.h"
11491#include "src/Tensor/TensorContractionThreadPool.h"
115- #include "src/Tensor/TensorContractionCuda .h"
92+ #include "src/Tensor/TensorContractionGpu .h"
11693#include "src/Tensor/TensorConversion.h"
11794#include "src/Tensor/TensorConvolution.h"
11895#include "src/Tensor/TensorFFT.h"
@@ -134,8 +111,15 @@ typedef unsigned __int64 uint64_t;
134111#include "src/Tensor/TensorGenerator.h"
135112#include "src/Tensor/TensorAssign.h"
136113#include "src/Tensor/TensorScan.h"
114+ #include "src/Tensor/TensorTrace.h"
115+
116+ #ifdef EIGEN_USE_SYCL
117+ #include "src/Tensor/TensorReductionSycl.h"
118+ #include "src/Tensor/TensorConvolutionSycl.h"
119+ #include "src/Tensor/TensorContractionSycl.h"
120+ #include "src/Tensor/TensorScanSycl.h"
121+ #endif
137122
138- #include "src/Tensor/TensorSycl.h"
139123#include "src/Tensor/TensorExecutor.h"
140124#include "src/Tensor/TensorDevice.h"
141125
@@ -147,6 +131,7 @@ typedef unsigned __int64 uint64_t;
147131
148132#include "src/Tensor/TensorIO.h"
149133
150- #include < Eigen/src/Core/util/ReenableStupidWarnings.h>
134+ #include "../../../ Eigen/src/Core/util/ReenableStupidWarnings.h"
151135
136+ #endif // EIGEN_HAS_CXX11
152137//#endif // EIGEN_CXX11_TENSOR_MODULE
0 commit comments