Restrict TaggedVector operators (#795)

CExA-project · Mar 5, 2025 · 20fad08 · 20fad08
1 parent 35a2f54
commit 20fad08
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/include/ddc/detail/tagged_vector.hpp b/include/ddc/detail/tagged_vector.hpp
@@ -372,7 +372,9 @@ class TaggedVector : public TaggedVectorConversionOperators<TaggedVector<Element
         return *this;
     }
 
-    template <class OElementType>
+    template <
+            class OElementType,
+            class = std::enable_if_t<std::is_convertible_v<OElementType, ElementType>>>
     KOKKOS_FUNCTION constexpr TaggedVector& operator+=(OElementType const& rhs)
     {
         ((m_values[type_seq_rank_v<Tags, tags_seq>] += rhs), ...);
@@ -388,7 +390,9 @@ class TaggedVector : public TaggedVectorConversionOperators<TaggedVector<Element
         return *this;
     }
 
-    template <class OElementType>
+    template <
+            class OElementType,
+            class = std::enable_if_t<std::is_convertible_v<OElementType, ElementType>>>
     KOKKOS_FUNCTION constexpr TaggedVector& operator-=(OElementType const& rhs)
     {
         ((m_values[type_seq_rank_v<Tags, tags_seq>] -= rhs), ...);