Broadcast: Forward rvalue references instead of wrapping them in std:…

…:tie Signed-off-by: Joseph Schuchart <joseph.schuchart@stonybrook.edu>
devreal · Sep 5, 2024 · 6b1eaee · 6b1eaee
1 parent ded1793
commit 6b1eaee
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/ttg/ttg/device/task.h b/ttg/ttg/device/task.h
@@ -474,7 +474,11 @@ namespace ttg::device {
   template <size_t i, typename rangeT,
             ttg::Runtime Runtime = ttg::ttg_runtime>
   inline detail::send_t broadcastk(rangeT &&keylist) {
-    return detail::send_t{detail::broadcastk_coro<i>(std::tie(keylist))};
+    if constexpr (std::is_rvalue_reference_v<decltype(keylist)>) {
+      return detail::send_t{detail::broadcastk_coro<i>(std::forward<rangeT>(keylist))};
+    } else {
+      return detail::send_t{detail::broadcastk_coro<i>(std::tie(keylist))};
+    }
   }