From 930f03e8830ed8a8023ff88207b97e757ae8496c Mon Sep 17 00:00:00 2001
From: Eddie Hung <eddie@fpgeh.com>
Date: Fri, 3 Jan 2020 15:38:18 -0800
Subject: [PATCH] Call -prep_holes before aigmap; fix topo ordering

---
 backends/aiger/xaiger.cc   |  9 +------
 passes/techmap/abc9.cc     |  7 ++---
 passes/techmap/abc9_ops.cc | 55 +++++++++++---------------------------
 3 files changed, 18 insertions(+), 53 deletions(-)

diff --git a/backends/aiger/xaiger.cc b/backends/aiger/xaiger.cc
index e9b4f07bf..7ef744d04 100644
--- a/backends/aiger/xaiger.cc
+++ b/backends/aiger/xaiger.cc
@@ -199,7 +199,7 @@ struct XAigerWriter
 				}
 			}
 
-		for (auto cell : module->selected_cells()) {
+		for (auto cell : module->cells()) {
 			if (cell->type == "$_NOT_")
 			{
 				SigBit A = sigmap(cell->getPort("\\A").as_bit());
@@ -613,16 +613,9 @@ struct XAigerWriter
 			if (holes_module) {
 				log_push();
 
-				// Move into a new (temporary) design so that "clean" will only
-				// operate (and run checks on) this one module
-				RTLIL::Design *holes_design = new RTLIL::Design;
-				module->design->modules_.erase(holes_module->name);
-				holes_design->add(holes_module);
-
 				std::stringstream a_buffer;
 				XAigerWriter writer(holes_module);
 				writer.write_aiger(a_buffer, false /*ascii_mode*/);
-				delete holes_design;
 
 				f << "a";
 				std::string buffer_str = a_buffer.str();
diff --git a/passes/techmap/abc9.cc b/passes/techmap/abc9.cc
index da56f42ea..0da815870 100644
--- a/passes/techmap/abc9.cc
+++ b/passes/techmap/abc9.cc
@@ -186,14 +186,11 @@ struct Abc9Pass : public ScriptPass
 	void script() YS_OVERRIDE
 	{
 		run("scc -set_attr abc9_scc_id {}");
-		run("abc9_ops -break_scc");
-		run("aigmap");
-
-		run("abc9_ops -prep_holes");
+		run("abc9_ops -break_scc -prep_holes");
 		run("select -set abc9_holes A:abc9_holes");
 		run("flatten -wb @abc9_holes");
 		run("techmap @abc9_holes");
-		run("aigmap @abc9_holes");
+		run("aigmap");
 		if (dff_mode)
 			run("abc9_ops -prep_dff");
 		run("opt -purge @abc9_holes");
diff --git a/passes/techmap/abc9_ops.cc b/passes/techmap/abc9_ops.cc
index 632f2bc8a..2f07de8d4 100644
--- a/passes/techmap/abc9_ops.cc
+++ b/passes/techmap/abc9_ops.cc
@@ -21,6 +21,7 @@
 #include "kernel/register.h"
 #include "kernel/sigtools.h"
 #include "kernel/utils.h"
+#include "kernel/celltypes.h"
 
 USING_YOSYS_NAMESPACE
 PRIVATE_NAMESPACE_BEGIN
@@ -194,58 +195,32 @@ void prep_holes(RTLIL::Module *module)
 
 	SigMap sigmap(module);
 
-	// TODO: Speed up toposort -- ultimately we care about
-	//       box ordering, but not individual AIG cells
 	dict<SigBit, pool<IdString>> bit_drivers, bit_users;
 	TopoSort<IdString, RTLIL::sort_by_id_str> toposort;
 	bool abc9_box_seen = false;
 
 	for (auto cell : module->selected_cells()) {
-		if (cell->type == "$_NOT_")
-		{
-			SigBit A = sigmap(cell->getPort("\\A").as_bit());
-			SigBit Y = sigmap(cell->getPort("\\Y").as_bit());
-			toposort.node(cell->name);
-			bit_users[A].insert(cell->name);
-			bit_drivers[Y].insert(cell->name);
-			continue;
-		}
-
-		if (cell->type == "$_AND_")
-		{
-			SigBit A = sigmap(cell->getPort("\\A").as_bit());
-			SigBit B = sigmap(cell->getPort("\\B").as_bit());
-			SigBit Y = sigmap(cell->getPort("\\Y").as_bit());
-			toposort.node(cell->name);
-			bit_users[A].insert(cell->name);
-			bit_users[B].insert(cell->name);
-			bit_drivers[Y].insert(cell->name);
-			continue;
-		}
-
 		if (cell->type == "$__ABC9_FF_")
 			continue;
 
-		RTLIL::Module* inst_module = design->module(cell->type);
-		if (inst_module) {
-			if (!inst_module->attributes.count("\\abc9_box_id") || cell->get_bool_attribute("\\abc9_keep"))
-				continue;
+		auto inst_module = module->design->module(cell->type);
+		bool abc9_box = inst_module && inst_module->attributes.count("\\abc9_box_id") && !cell->get_bool_attribute("\\abc9_keep");
+		abc9_box_seen = abc9_box_seen || abc9_box;
 
-			for (const auto &conn : cell->connections()) {
-				auto port_wire = inst_module->wire(conn.first);
-				// Ignore inout for the sake of topographical ordering
-				if (port_wire->port_input && !port_wire->port_output)
-					for (auto bit : sigmap(conn.second))
-						bit_users[bit].insert(cell->name);
-				if (port_wire->port_output)
-					for (auto bit : sigmap(conn.second))
-						bit_drivers[bit].insert(cell->name);
-			}
+		if (!abc9_box && !yosys_celltypes.cell_known(cell->type))
+			continue;
 
-                        abc9_box_seen = true;
+		for (auto conn : cell->connections()) {
+			if (cell->input(conn.first))
+				for (auto bit : sigmap(conn.second))
+					bit_users[bit].insert(cell->name);
 
-                        toposort.node(cell->name);
+			if (cell->output(conn.first))
+				for (auto bit : sigmap(conn.second))
+					bit_drivers[bit].insert(cell->name);
 		}
+
+		toposort.node(cell->name);
 	}
 
 	if (!abc9_box_seen)
-- 
2.30.2