Fixed CUDA compilation errors

This commit is contained in:
Benoit Steiner
2015-11-11 14:36:33 -08:00
parent 228edfe616
commit 5cb18e5b5e
2 changed files with 7 additions and 1 deletions

View File

@@ -336,6 +336,12 @@ struct FullReducer<Self, Op, ThreadPoolDevice, true> {
};
#endif
#if defined(EIGEN_USE_GPU) && defined(__CUDACC__)
template <int B, int N, typename S, typename R, typename I>
__global__ void FullReductionKernel(R, const S, I, typename S::CoeffReturnType*);
#endif
} // end namespace internal