diff --git a/src/algebra/TinyVector.hpp b/src/algebra/TinyVector.hpp index 038741231cb32963330d8a061f54ff41a7b05238..34b73118b7473d8f9a494d17ff12be9b272b4656 100644 --- a/src/algebra/TinyVector.hpp +++ b/src/algebra/TinyVector.hpp @@ -68,6 +68,15 @@ public: return std::move(t); } + KOKKOS_INLINE_FUNCTION + TinyVector& operator*=(const T& t) + { + for (size_t i=0; i<N; ++i) { + m_values[i] *= t; + } + return *this; + } + KOKKOS_INLINE_FUNCTION friend TinyVector operator*(const T& t, const TinyVector& v) { @@ -81,9 +90,7 @@ public: KOKKOS_INLINE_FUNCTION friend TinyVector operator*(const T& t, TinyVector&& v) { - for (size_t i=0; i<N; ++i) { - v.m_values[i] *= t; - } + v *= t; return std::move(v); }