Added "nlutmap -assert"

1 files changed, 14 insertions, 0 deletions

diff --git a/passes/techmap/nlutmap.cc b/passes/techmap/nlutmap.cc
index a6d4e1a2f..6fcdf82bd 100644
--- a/passes/techmap/nlutmap.cc
+++ b/passes/techmap/nlutmap.cc
@@ -26,6 +26,7 @@ PRIVATE_NAMESPACE_BEGIN
 struct NlutmapConfig
 {
 	vector<int> luts;
+	bool assert_mode = false;
 };
 
 struct NlutmapWorker
@@ -116,6 +117,12 @@ struct NlutmapWorker
 				available_luts.back() += n_luts;
 		}
 
+		if (config.assert_mode) {
+			for (auto cell : module->cells())
+				if (cell->type == "$lut" && !mapped_cells.count(cell))
+					log_error("Insufficient number of LUTs to map all logic cells!\n");
+		}
+
 		run_abc(0);
 	}
 };
@@ -135,6 +142,9 @@ struct NlutmapPass : public Pass {
 		log("        The number of LUTs with 1, 2, 3, ... inputs that are\n");
 		log("        available in the target architecture.\n");
 		log("\n");
+		log("    -assert\n");
+		log("        Create an error if not all logic can be mapped\n");
+		log("\n");
 		log("Excess logic that does not fit into the specified LUTs is mapped back\n");
 		log("to generic logic gates ($_AND_, etc.).\n");
 		log("\n");
@@ -156,6 +166,10 @@ struct NlutmapPass : public Pass {
 					config.luts.push_back(atoi(token.c_str()));
 				continue;
 			}
+			if (args[argidx] == "-assert") {
+				config.assert_mode = true;
+				continue;
+			}
 			break;
 		}
 		extra_args(args, argidx, design);