MRA: Make function nodes TTValues

Signed-off-by: Joseph Schuchart <[email protected]>
devreal · Sep 16, 2024 · 5cd4def · 5cd4def
1 parent c0f70d7
commit 5cd4def
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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