From 6a889b53647890b09e880c81fda000aa27d99a5b Mon Sep 17 00:00:00 2001 From: Max Luebke Date: Mon, 16 Dec 2024 19:53:13 +0100 Subject: [PATCH] feat: add dht_snaps and dht_out_dir parameters to ChemistryModule and main function --- src/Chemistry/ChemistryModule.hpp | 6 ++++++ src/poet.cpp | 2 ++ 2 files changed, 8 insertions(+) diff --git a/src/Chemistry/ChemistryModule.hpp b/src/Chemistry/ChemistryModule.hpp index 697b50744..4bf925400 100644 --- a/src/Chemistry/ChemistryModule.hpp +++ b/src/Chemistry/ChemistryModule.hpp @@ -81,6 +81,8 @@ public: bool dht_enabled; std::uint32_t dht_size_mb; + int dht_snaps; + std::string dht_out_dir; bool interp_enabled; std::uint32_t interp_bucket_size; @@ -103,6 +105,10 @@ public: if (this->dht_enabled || this->interp_enabled) { this->initializeDHT(setup.dht_size_mb, this->params.dht_species, setup.has_het_ids); + + if (setup.dht_snaps != DHT_SNAPS_DISABLED) { + this->setDHTSnapshots(setup.dht_snaps, setup.dht_out_dir); + } } if (this->interp_enabled) { diff --git a/src/poet.cpp b/src/poet.cpp index c7864b4b6..742f4c395 100644 --- a/src/poet.cpp +++ b/src/poet.cpp @@ -587,6 +587,8 @@ int main(int argc, char *argv[]) { getHasID(init_list.getInitialGrid(), 0, MPI_COMM_WORLD), run_params.use_dht, run_params.dht_size, + run_params.dht_snaps, + run_params.out_dir, run_params.use_interp, run_params.interp_bucket_entries, run_params.interp_size,