Correctly move callable f in ConsumeRvalues call operator

eth-cscs · Aug 24, 2023 · 130b7d3 · 130b7d3
1 parent 9dba07b
commit 130b7d3
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/include/dlaf/common/consume_rvalues.h b/include/dlaf/common/consume_rvalues.h
@@ -34,9 +34,14 @@ struct ConsumeRvalues {
   std::decay_t<F> f;
 
   template <typename... Ts>
-  auto operator()(Ts&&... ts) -> decltype(std::move(f)(consume_rvalue(std::forward<Ts>(ts))...)) {
+  auto operator()(Ts&&... ts) && -> decltype(std::move(f)(consume_rvalue(std::forward<Ts>(ts))...)) {
     return std::move(f)(consume_rvalue(std::forward<Ts>(ts))...);
   }
+
+  template <typename... Ts>
+  auto operator()(Ts&&... ts) & -> decltype(f(consume_rvalue(std::forward<Ts>(ts))...)) {
+    return f(consume_rvalue(std::forward<Ts>(ts))...);
+  }
 };
 
 template <typename F>