Added "memory -bram"

1 files changed, 11 insertions, 2 deletions

diff --git a/passes/memory/memory.cc b/passes/memory/memory.cc
index 502e21022..866efae77 100644
--- a/passes/memory/memory.cc
+++ b/passes/memory/memory.cc
@@ -31,7 +31,7 @@ struct MemoryPass : public Pass {
 	{
 		//   |---v---|---v---|---v---|---v---|---v---|---v---|---v---|---v---|---v---|---v---|
 		log("\n");
-		log("    memory [-nomap] [selection]\n");
+		log("    memory [-nomap] [-bram <bram_rules>] [selection]\n");
 		log("\n");
 		log("This pass calls all the other memory_* passes in a useful order:\n");
 		log("\n");
@@ -40,7 +40,8 @@ struct MemoryPass : public Pass {
 		log("    memory_share\n");
 		log("    opt_clean\n");
 		log("    memory_collect\n");
-		log("    memory_map          (skipped if called with -nomap)\n");
+		log("    memory_bram -rules <bram_rules>     (when called with -bram)\n");
+		log("    memory_map                          (skipped if called with -nomap)\n");
 		log("\n");
 		log("This converts memories to word-wide DFFs and address decoders\n");
 		log("or multiport memory blocks if called with the -nomap option.\n");
@@ -49,6 +50,7 @@ struct MemoryPass : public Pass {
 	virtual void execute(std::vector<std::string> args, RTLIL::Design *design)
 	{
 		bool flag_nomap = false;
+		string memory_bram_opts;
 
 		log_header("Executing MEMORY pass.\n");
 		log_push();
@@ -59,6 +61,10 @@ struct MemoryPass : public Pass {
 				flag_nomap = true;
 				continue;
 			}
+			if (argidx+1 < args.size() && args[argidx] == "-bram") {
+				memory_bram_opts += " -rules " + args[++argidx];
+				continue;
+			}
 			break;
 		}
 		extra_args(args, argidx, design);
@@ -69,6 +75,9 @@ struct MemoryPass : public Pass {
 		Pass::call(design, "opt_clean");
 		Pass::call(design, "memory_collect");
 
+		if (!memory_bram_opts.empty())
+			Pass::call(design, "memory_bram" + memory_bram_opts);
+
 		if (!flag_nomap)
 			Pass::call(design, "memory_map");