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; }