mirror of
https://gitlab.com/libeigen/eigen.git
synced 2025-06-04 18:54:00 +08:00
Fix placement of "#if defined(EIGEN_GPUCC)" guard region.
Found with -Wundefined-func-template. Author: tkoeppe@google.com
This commit is contained in:
parent
4d808e834a
commit
cc407c9d4d
@ -317,6 +317,7 @@ class TensorExecutor<Expression, ThreadPoolDevice, Vectorizable, /*Tileable*/ tr
|
|||||||
|
|
||||||
// GPU: the evaluation of the expression is offloaded to a GPU.
|
// GPU: the evaluation of the expression is offloaded to a GPU.
|
||||||
#if defined(EIGEN_USE_GPU)
|
#if defined(EIGEN_USE_GPU)
|
||||||
|
#if defined(EIGEN_GPUCC)
|
||||||
|
|
||||||
template <typename Expression, bool Vectorizable, bool Tileable>
|
template <typename Expression, bool Vectorizable, bool Tileable>
|
||||||
class TensorExecutor<Expression, GpuDevice, Vectorizable, Tileable> {
|
class TensorExecutor<Expression, GpuDevice, Vectorizable, Tileable> {
|
||||||
@ -326,7 +327,6 @@ class TensorExecutor<Expression, GpuDevice, Vectorizable, Tileable> {
|
|||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
#if defined(EIGEN_GPUCC)
|
|
||||||
template <typename Evaluator, typename StorageIndex, bool Vectorizable>
|
template <typename Evaluator, typename StorageIndex, bool Vectorizable>
|
||||||
struct EigenMetaKernelEval {
|
struct EigenMetaKernelEval {
|
||||||
static __device__ EIGEN_ALWAYS_INLINE
|
static __device__ EIGEN_ALWAYS_INLINE
|
||||||
|
Loading…
x
Reference in New Issue
Block a user