From 5cd4def63e0b9017b9f63169907dff1e87b4fe97 Mon Sep 17 00:00:00 2001
From: Joseph Schuchart <joseph.schuchart@stonybrook.edu>
Date: Mon, 16 Sep 2024 10:58:42 -0400
Subject: [PATCH] MRA: Make function nodes TTValues

Signed-off-by: Joseph Schuchart <joseph.schuchart@stonybrook.edu>
---
 examples/madness/mra-device/functionnode.h   | 4 ++--
 examples/madness/mra-device/mrattg-device.cc | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/examples/madness/mra-device/functionnode.h b/examples/madness/mra-device/functionnode.h
index d27e1d6e1..da5ca7be7 100644
--- a/examples/madness/mra-device/functionnode.h
+++ b/examples/madness/mra-device/functionnode.h
@@ -7,7 +7,7 @@
 
 namespace mra {
     template <typename T, Dimension NDIM>
-    class FunctionReconstructedNode {
+    class FunctionReconstructedNode : public ttg::TTValue<FunctionReconstructedNode<T, NDIM>> {
     public: // temporarily make everything public while we figure out what we are doing
         using key_type = Key<NDIM>;
         using tensor_type = Tensor<T,NDIM>;
@@ -43,7 +43,7 @@ namespace mra {
 
 
     template <typename T, Dimension NDIM>
-    class FunctionCompressedNode {
+    class FunctionCompressedNode : public ttg::TTValue<FunctionCompressedNode<T, NDIM>> {
     public: // temporarily make everything public while we figure out what we are doing
         static constexpr bool is_function_node = true;
 
diff --git a/examples/madness/mra-device/mrattg-device.cc b/examples/madness/mra-device/mrattg-device.cc
index 75d406d2a..286e3b6f5 100644
--- a/examples/madness/mra-device/mrattg-device.cc
+++ b/examples/madness/mra-device/mrattg-device.cc
@@ -311,10 +311,10 @@ auto make_reconstruct(
         r.key = child;
         r.is_leaf = node.is_child_leaf[it.index()];
         if (r.is_leaf) {
-          sends.push_back(ttg::device::send<1>(child, r));
+          sends.push_back(ttg::device::send<1>(child, std::move(r)));
         }
         else {
-          sends.push_back(ttg::device::send<0>(child, r.coeffs));
+          sends.push_back(ttg::device::send<0>(child, std::move(r.coeffs)));
         }
     }
 #ifndef TTG_ENABLE_HOST