Keep modules with $assume (like $assert)

diff --git a/passes/hierarchy/hierarchy.cc b/passes/hierarchy/hierarchy.cc
index d655af0d7f66b6014cd1218e3648f8506a334877..bfa3890a22248aa3a28420a1a74693c03a88e5cd 100644 (file)
--- a/passes/hierarchy/hierarchy.cc
+++ b/passes/hierarchy/hierarchy.cc
@@ -315,7 +315,7 @@ bool set_keep_assert(std::map<RTLIL::Module*, bool> &cache, RTLIL::Module *mod)
        if (cache.count(mod) == 0)
                for (auto c : mod->cells()) {
                        RTLIL::Module *m = mod->design->module(c->type);
-                       if ((m != nullptr && set_keep_assert(cache, m)) || c->type == "$assert")
+                       if ((m != nullptr && set_keep_assert(cache, m)) || c->type.in("$assert", "$assume"))
                                return cache[mod] = true;
                }
        return cache[mod];