GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-08-12 11:49:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

Prevent the generation of unlaunchable cuda kernels when compiling in debug mode.

Browse Source
This commit is contained in:
Benoit Steiner 2014-06-09 09:43:51 -07:00
parent 2859a31ac8
commit 36a2b2e9dc
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
unsupported/Eigen/CXX11/src/Tensor/TensorDimensions.h
View File

@ -73,7 +73,7 @@ struct Sizes : internal::numeric_list<std::size_t, Indices...> {
typedef internal::numeric_list<std::size_t, Indices...> Base; typedef internal::numeric_list<std::size_t, Indices...> Base;
static const std::size_t total_size = internal::arg_prod(Indices...); static const std::size_t total_size = internal::arg_prod(Indices...);
static std::size_t TotalSize() { static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE std::size_t TotalSize() {
return internal::arg_prod(Indices...); return internal::arg_prod(Indices...);
} }
@ -119,7 +119,7 @@ template <std::size_t V1=0, std::size_t V2=0, std::size_t V3=0, std::size_t V4=0
static const size_t count = Base::count; static const size_t count = Base::count;
static const std::size_t total_size = internal::arg_prod<Base>::value; static const std::size_t total_size = internal::arg_prod<Base>::value;
static size_t TotalSize() { static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE size_t TotalSize() {
return internal::arg_prod<Base>::value; return internal::arg_prod<Base>::value;
} }
@ -156,7 +156,8 @@ namespace internal {
template<typename Index, std::size_t NumIndices, std::size_t n, bool RowMajor> template<typename Index, std::size_t NumIndices, std::size_t n, bool RowMajor>
struct tensor_index_linearization_helper struct tensor_index_linearization_helper
{ {
static inline Index run(array<Index, NumIndices> const& indices, array<Index, NumIndices> const& dimensions) static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
Index run(array<Index, NumIndices> const& indices, array<Index, NumIndices> const& dimensions)
{ {
return array_get<RowMajor ? n : (NumIndices - n - 1)>(indices) + return array_get<RowMajor ? n : (NumIndices - n - 1)>(indices) +
array_get<RowMajor ? n : (NumIndices - n - 1)>(dimensions) * array_get<RowMajor ? n : (NumIndices - n - 1)>(dimensions) *
@ -167,7 +168,8 @@ struct tensor_index_linearization_helper
template<typename Index, std::size_t NumIndices, bool RowMajor> template<typename Index, std::size_t NumIndices, bool RowMajor>
struct tensor_index_linearization_helper<Index, NumIndices, 0, RowMajor> struct tensor_index_linearization_helper<Index, NumIndices, 0, RowMajor>
{ {
static inline Index run(array<Index, NumIndices> const& indices, array<Index, NumIndices> const&) static EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE
Index run(array<Index, NumIndices> const& indices, array<Index, NumIndices> const&)
{ {
return array_get<RowMajor ? 0 : NumIndices - 1>(indices); return array_get<RowMajor ? 0 : NumIndices - 1>(indices);
} }