diff options
Diffstat (limited to 'src/preprocessing/passes/real_to_int.h')
-rw-r--r-- | src/preprocessing/passes/real_to_int.h | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/src/preprocessing/passes/real_to_int.h b/src/preprocessing/passes/real_to_int.h index 9f0eb529f..d26547372 100644 --- a/src/preprocessing/passes/real_to_int.h +++ b/src/preprocessing/passes/real_to_int.h @@ -22,6 +22,7 @@ #include <vector> +#include "context/cdhashmap.h" #include "expr/node.h" #include "preprocessing/preprocessing_pass.h" @@ -29,10 +30,10 @@ namespace cvc5 { namespace preprocessing { namespace passes { -using NodeMap = std::unordered_map<Node, Node, NodeHashFunction>; - class RealToInt : public PreprocessingPass { + using NodeMap = context::CDHashMap<Node, Node, NodeHashFunction>; + public: RealToInt(PreprocessingPassContext* preprocContext); @@ -42,6 +43,8 @@ class RealToInt : public PreprocessingPass private: Node realToIntInternal(TNode n, NodeMap& cache, std::vector<Node>& var_eq); + /** Cache for the above method */ + NodeMap d_cache; }; } // namespace passes |