Small improvement

diff --git a/backends/aiger/xaiger.cc b/backends/aiger/xaiger.cc
index 2bc059dc55169da065d98009d7bc98f507e136bc..5919b23029160e3f5b617af1bb64a9c11715b022 100644 (file)
--- a/backends/aiger/xaiger.cc
+++ b/backends/aiger/xaiger.cc
@@ -328,6 +328,7 @@ struct XAigerWriter
                                                        if (I != b)
                                                                alias_map[b] = I;
                                                        co_bits.emplace_back(b, cell, port_name, offset++, 0);
+                                                       unused_bits.erase(b);
                                                }
                                        }
                                        if (w->port_output) {
@@ -347,6 +348,7 @@ struct XAigerWriter
                                                for (const auto &b : rhs.bits()) {
                                                        SigBit O = sigmap(b);
                                                        ci_bits.emplace_back(O, cell, port_name, offset++);
+                                                       undriven_bits.erase(O);
                                                }
                                        }
                                }
@@ -383,10 +385,6 @@ struct XAigerWriter
                        }
                }
 
-               // Do some CI/CO post-processing:
-               // CIs cannot be undriven
-               for (const auto &c : ci_bits)
-                       undriven_bits.erase(std::get<0>(c));
                // Erase all POs that are undriven
                if (!holes_mode)
                        for (auto bit : undriven_bits)