diff options
author | René Wagner <renewagner@google.com> | 2020-04-23 17:25:08 +0200 |
---|---|---|
committer | René Wagner <renewagner@google.com> | 2020-04-23 17:25:08 +0200 |
commit | 0aebe19aca58d78434cf724ef954a8383b2bf7c9 (patch) | |
tree | 2abddd4141ae0c2d26ffcc901efe585f5cabb044 /Eigen/src/Core/BooleanRedux.h | |
parent | 3c02fefec53f21d9fad537ff0d62d8dcc8162466 (diff) |
BooleanRedux.h: Add more EIGEN_DEVICE_FUNC qualifiers.
This enables operator== on Eigen matrices in device code.
Diffstat (limited to 'Eigen/src/Core/BooleanRedux.h')
-rw-r--r-- | Eigen/src/Core/BooleanRedux.h | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/Eigen/src/Core/BooleanRedux.h b/Eigen/src/Core/BooleanRedux.h index ccf519067..e32c4ac5b 100644 --- a/Eigen/src/Core/BooleanRedux.h +++ b/Eigen/src/Core/BooleanRedux.h @@ -22,7 +22,7 @@ struct all_unroller row = (UnrollCount-1) % Rows }; - static inline bool run(const Derived &mat) + EIGEN_DEVICE_FUNC static inline bool run(const Derived &mat) { return all_unroller<Derived, UnrollCount-1, Rows>::run(mat) && mat.coeff(row, col); } @@ -31,13 +31,13 @@ struct all_unroller template<typename Derived, int Rows> struct all_unroller<Derived, 0, Rows> { - static inline bool run(const Derived &/*mat*/) { return true; } + EIGEN_DEVICE_FUNC static inline bool run(const Derived &/*mat*/) { return true; } }; template<typename Derived, int Rows> struct all_unroller<Derived, Dynamic, Rows> { - static inline bool run(const Derived &) { return false; } + EIGEN_DEVICE_FUNC static inline bool run(const Derived &) { return false; } }; template<typename Derived, int UnrollCount, int Rows> @@ -48,7 +48,7 @@ struct any_unroller row = (UnrollCount-1) % Rows }; - static inline bool run(const Derived &mat) + EIGEN_DEVICE_FUNC static inline bool run(const Derived &mat) { return any_unroller<Derived, UnrollCount-1, Rows>::run(mat) || mat.coeff(row, col); } @@ -57,13 +57,13 @@ struct any_unroller template<typename Derived, int Rows> struct any_unroller<Derived, 0, Rows> { - static inline bool run(const Derived & /*mat*/) { return false; } + EIGEN_DEVICE_FUNC static inline bool run(const Derived & /*mat*/) { return false; } }; template<typename Derived, int Rows> struct any_unroller<Derived, Dynamic, Rows> { - static inline bool run(const Derived &) { return false; } + EIGEN_DEVICE_FUNC static inline bool run(const Derived &) { return false; } }; } // end namespace internal |