From 4de3ee093eecf4e766734fddb113187df36be31a Mon Sep 17 00:00:00 2001
From: Robert O'Callahan <rocallahan@google.com>
Date: Sat, 26 Jul 2025 09:34:37 +0000
Subject: [PATCH] Mark kept FF output wires as ports directly instead of via
 the 'keep' attribute

---
 passes/techmap/abc.cc | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

diff --git a/passes/techmap/abc.cc b/passes/techmap/abc.cc
index 0c2576c1a..dfaee9415 100644
--- a/passes/techmap/abc.cc
+++ b/passes/techmap/abc.cc
@@ -291,12 +291,15 @@ void AbcModuleState::extract_cell(RTLIL::Module *module, RTLIL::Cell *cell, bool
 				return;
 		}
 
-		if (keepff)
-			for (auto &c : ff.sig_q.chunks())
-				if (c.wire != nullptr)
-					c.wire->attributes[ID::keep] = 1;
-
-		map_signal(ff.sig_q, type, map_signal(ff.sig_d));
+		int gate_id = map_signal(ff.sig_q, type, map_signal(ff.sig_d));
+		if (keepff) {
+			SigBit bit = ff.sig_q;
+			if (assign_map(bit).wire != nullptr) {
+				signal_list[gate_id].is_port = true;
+			}
+			if (bit.wire != nullptr)
+				bit.wire->attributes[ID::keep] = 1;
+		}
 
 		ff.remove();
 		return;