evaluation tracing includes size of TA tensors, bumps TA tag to pull in ValeevGroup/tiledarray#516

Author: evaleev

INSTALL.md

@@ -17,7 +17,7 @@ While in the SeQuant source directory:
     * [libPerm](https://github.com/Krzmbrzl/libPerm), tag 8e4afd1461baffa5d829c8fed059f5a172a3b060, *if not found, SeQuant will download and build libPerm*
   * optional:
     * for building coupled-cluster evaluation tests:
-      * [TiledArray](https://github.com/ValeevGroup/tiledarray.git), tag 3f6629db047417e814b75ad5069b7f4ce26428e7
+      * [TiledArray](https://github.com/ValeevGroup/tiledarray.git), tag 4a646d0c96cd0283eb34185aaf5f5b6fcc302bb2
     * for building `stcc*` test programs
         * [Eigen](http://eigen.tuxfamily.org/), version 3
 

SeQuant/domain/eval/eval_result.hpp

@@ -522,6 +522,9 @@ class EvalResult {
     return const_cast<EvalResult&>(*this).get<T>();
   }
 
+  /// @return the size of the object in bytes
+  [[nodiscard]] virtual std::size_t size_in_bytes() const = 0;
+
  protected:
   template <typename T,
             typename = std::enable_if_t<!std::is_convertible_v<T, EvalResult>>>
@@ -614,6 +617,8 @@ class EvalScalar final : public EvalResult {
   [[nodiscard]] id_t type_id() const noexcept override {
     return id_for_type<EvalScalar<T>>();
   }
+
+  [[nodiscard]] std::size_t size_in_bytes() const final { return sizeof(T); }
 };
 
 ///
@@ -743,6 +748,14 @@ class EvalTensorTA final : public EvalResult {
     return eval_result<this_type>(
         particle_antisymmetrize_ta(get<ArrayT>(), bra_rank));
   }
+
+ private:
+  [[nodiscard]] std::size_t size_in_bytes() const final {
+    auto& v = get<ArrayT>();
+    auto local_size = TA::size_of<TA::MemorySpace::Host>(v);
+    v.world().gop.sum(local_size);
+    return local_size;
+  }
 };
 
 template <typename ArrayT,
@@ -882,6 +895,14 @@ class EvalTensorOfTensorTA final : public EvalResult {
     // not implemented yet
     return nullptr;
   }
+
+ private:
+  [[nodiscard]] std::size_t size_in_bytes() const final {
+    auto& v = get<ArrayT>();
+    auto local_size = TA::size_of<TA::MemorySpace::Host>(v);
+    v.world().gop.sum(local_size);
+    return local_size;
+  }
 };
 
 ///
@@ -975,6 +996,11 @@ class EvalTensorBTAS final : public EvalResult {
     return eval_result<EvalTensorBTAS<T>>(
         particle_antisymmetrize_btas(get<T>(), bra_rank));
   }
+
+ private:
+  [[nodiscard]] std::size_t size_in_bytes() const final {
+    abort();  // not yet implemented
+  }
 };
 
 }  // namespace sequant

external/versions.cmake

@@ -6,8 +6,8 @@ set(SEQUANT_TRACKED_VGCMAKEKIT_TAG 4f40440dcbda2a5a005fd15f27c582a5587ee779)
 set(SEQUANT_TRACKED_RANGEV3_TAG 0.12.0)
 set(SEQUANT_TRACKED_RANGEV3_PREVIOUS_TAG d800a032132512a54c291ce55a2a43e0460591c7)
 
-set(SEQUANT_TRACKED_TILEDARRAY_TAG fb5d5b80427cc937238a757ccec0cb313090528d)
-set(SEQUANT_TRACKED_TILEDARRAY_PREVIOUS_TAG c10f8d5cade2e6040d28958f04189d6ae578c00a)
+set(SEQUANT_TRACKED_TILEDARRAY_TAG 4a646d0c96cd0283eb34185aaf5f5b6fcc302bb2)
+set(SEQUANT_TRACKED_TILEDARRAY_PREVIOUS_TAG fb5d5b80427cc937238a757ccec0cb313090528d)
 
 set(SEQUANT_TRACKED_LIBPERM_TAG 8e4afd1461baffa5d829c8fed059f5a172a3b060)
 set(SEQUANT_TRACKED_LIBPERM_PREVIOUS_TAG 8e4afd1461baffa5d829c8fed059f5a172a3b060)