From f0236d8d200022bbddf46962850649f9a393e201 Mon Sep 17 00:00:00 2001
From: nella <simontupy64@gmail.com>
Date: Thu, 27 Aug 2026 14:34:10 +0200
Subject: [PATCH] Move module seeding into Design::add.

---
 kernel/rtlil.cc         | 16 ----------
 kernel/rtlil.h          |  2 --
 kernel/rtlil_bufnorm.cc | 68 +++++++++++++++++++++++++----------------
 3 files changed, 41 insertions(+), 45 deletions(-)

diff --git a/kernel/rtlil.cc b/kernel/rtlil.cc
index eb477c5d3..622e16d7c 100644
--- a/kernel/rtlil.cc
+++ b/kernel/rtlil.cc
@@ -1226,22 +1226,6 @@ RTLIL::Module *RTLIL::Design::top_module() const
 	return module_count == 1 ? module : nullptr;
 }
 
-void RTLIL::Design::add(RTLIL::Module *module)
-{
-	log_assert(modules_.count(module->name) == 0);
-	log_assert(refcount_modules_ == 0);
-	modules_[module->name] = module;
-	module->design = this;
-
-	for (auto mon : monitors)
-		mon->notify_module_add(module);
-
-	if (yosys_xtrace) {
-		log("#X# New Module: %s\n", module);
-		log_backtrace("-X- ", yosys_xtrace-1);
-	}
-}
-
 void RTLIL::Design::add(RTLIL::Binding *binding)
 {
 	log_assert(binding != nullptr);
diff --git a/kernel/rtlil.h b/kernel/rtlil.h
index fffd09400..79466ac3b 100644
--- a/kernel/rtlil.h
+++ b/kernel/rtlil.h
@@ -2115,8 +2115,6 @@ public:
 	pool<RTLIL::Wire *> buf_norm_wire_queue;
 	pool<RTLIL::Cell *> pending_deleted_cells;
 	dict<RTLIL::Wire *, pool<RTLIL::Cell *>> buf_norm_connect_index;
-	bool buf_norm_initialized = false;
-	void bufNormalizeInit();
 	void bufNormalize();
 
 	template<typename T> void rewrite_sigspecs(T &functor);
diff --git a/kernel/rtlil_bufnorm.cc b/kernel/rtlil_bufnorm.cc
index ecdef8bc4..96705a4e9 100644
--- a/kernel/rtlil_bufnorm.cc
+++ b/kernel/rtlil_bufnorm.cc
@@ -28,6 +28,19 @@
 YOSYS_NAMESPACE_BEGIN
 
 
+static void buf_norm_seed_queues(RTLIL::Module *module)
+{
+	for (auto cell : module->cells())
+	for (auto &conn : cell->connections()) {
+		if (GetSize(conn.second) == 0 || (cell->port_dir(conn.first) != RTLIL::PD_OUTPUT && cell->port_dir(conn.first) != RTLIL::PD_INOUT))
+			continue;
+		module->buf_norm_cell_queue.insert(cell);
+		module->buf_norm_cell_port_queue.emplace(cell, conn.first);
+	}
+	for (auto wire : module->wires())
+		module->buf_norm_wire_queue.insert(wire);
+}
+
 void RTLIL::Design::bufNormalize(bool enable)
 {
 	if (!enable)
@@ -44,39 +57,24 @@ void RTLIL::Design::bufNormalize(bool enable)
 				wire->driverPort_ = IdString();
 			}
 			module->buf_norm_connect_index.clear();
-			module->buf_norm_initialized = false;
 		}
 
 		flagBufferedNormalized = false;
 		return;
 	}
 
-	flagBufferedNormalized = true;
+	if (!flagBufferedNormalized)
+	{
+		for (auto module : modules())
+			buf_norm_seed_queues(module);
+
+		flagBufferedNormalized = true;
+	}
 
 	for (auto module : modules())
 		module->bufNormalize();
 }
 
-void RTLIL::Module::bufNormalizeInit()
-{
-	// When entering buf normalized mode, we need the first module-level bufNormalize
-	// call to know about all drivers, about all module ports (whether represented by
-	// a cell or not) and about all used but undriven wires (whether represented by a
-	// cell or not). We ensure this by enqueing all cell output ports and all wires.
-
-	for (auto cell : cells())
-	for (auto &conn : cell->connections()) {
-		if (GetSize(conn.second) == 0 || (cell->port_dir(conn.first) != RTLIL::PD_OUTPUT && cell->port_dir(conn.first) != RTLIL::PD_INOUT))
-			continue;
-		buf_norm_cell_queue.insert(cell);
-		buf_norm_cell_port_queue.emplace(cell, conn.first);
-	}
-	for (auto wire : wires())
-		buf_norm_wire_queue.insert(wire);
-
-	buf_norm_initialized = true;
-}
-
 struct bit_drive_data_t {
 	int drivers = 0;
 	int inout = 0;
@@ -94,11 +92,6 @@ void RTLIL::Module::bufNormalize()
 	if (!design->flagBufferedNormalized)
 		return;
 
-	// A module added after the design entered buf normalized mode has to be
-	// enqueued before its first pass
-	if (!buf_norm_initialized)
-		bufNormalizeInit();
-
 	if (!buf_norm_cell_queue.empty() || !buf_norm_wire_queue.empty() || !connections_.empty())
 	{
 		// Ensure that every enqueued input port is represented by a cell
@@ -678,4 +671,25 @@ void RTLIL::Cell::setPort(RTLIL::IdString portname, RTLIL::SigSpec signal)
 
 }
 
+void RTLIL::Design::add(RTLIL::Module *module)
+{
+	log_assert(modules_.count(module->name) == 0);
+	log_assert(refcount_modules_ == 0);
+	modules_[module->name] = module;
+	module->design = this;
+
+	for (auto mon : monitors)
+		mon->notify_module_add(module);
+
+	if (yosys_xtrace) {
+		log("#X# New Module: %s\n", module);
+		log_backtrace("-X- ", yosys_xtrace-1);
+	}
+
+	if (flagBufferedNormalized) {
+		buf_norm_seed_queues(module);
+		module->bufNormalize();
+	}
+}
+
 YOSYS_NAMESPACE_END