From 471efb1f1abe7e704f5f6dfb99a4ddd058d4c430 Mon Sep 17 00:00:00 2001 From: Varun Agrawal Date: Mon, 27 Jan 2025 18:42:43 -0500 Subject: [PATCH] add timing for hybrid pruning --- gtsam/hybrid/HybridBayesNet.cpp | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/gtsam/hybrid/HybridBayesNet.cpp b/gtsam/hybrid/HybridBayesNet.cpp index e64284a94..e052ae04c 100644 --- a/gtsam/hybrid/HybridBayesNet.cpp +++ b/gtsam/hybrid/HybridBayesNet.cpp @@ -49,6 +49,9 @@ bool HybridBayesNet::equals(const This &bn, double tol) const { // search to find the K-best leaves and then create a single pruned conditional. HybridBayesNet HybridBayesNet::prune( size_t maxNrLeaves, const std::optional &deadModeThreshold) const { +#if GTSAM_HYBRID_TIMING + gttic_(HybridPruning); +#endif // Collect all the discrete conditionals. Could be small if already pruned. const DiscreteBayesNet marginal = discreteMarginal(); @@ -152,6 +155,10 @@ HybridBayesNet HybridBayesNet::prune( // We ignore DiscreteConditional as they are already pruned and added. } +#if GTSAM_HYBRID_TIMING + gttoc_(HybridPruning); +#endif + return result; }