Updated readmeHEADmaster

1 files changed, 7 insertions, 0 deletions

diff --git a/multiplier/hdl/gen_wallace.py b/multiplier/hdl/gen_wallace.py
index 579041f..88d0b25 100755
--- a/multiplier/hdl/gen_wallace.py
+++ b/multiplier/hdl/gen_wallace.py
@@ -330,14 +330,21 @@ def main():
 	parser = argparse.ArgumentParser(prog="Multiplier Generator", description="Generates a n bit multiplier based on the bits argument provided",
 				epilog="bits sets the bit width of the multiplier, the output of the multiplier is 2 times the number of bits")
 	parser.add_argument("bits", type=int, help="The bit width of the multiplier")
+	parser.add_argument("-p", "--pipeline", help="Generates specified number of pipeline barriers")
+	parser.add_argument("-s", "--stages", nargs='+' help="Generates stages at specific reduction layers")
 	parser.add_argument("-a", "--adder", help="Generates the full and half adders for you to use", action='store_true')
 	parser.add_argument("-d", "--debug", help="Enables debug prints during generation scripting", action='store_true')
 	args = parser.parse_args()
 
 	bits = args.bits
+	pipeline = args.pipeline
+	stages = args.stages
 	debug = args.debug
 	adder  = args.adder
 
+	print(pipeline)
+	print(stages)
+
 	if (adder):
 		gen_half_adder()
 		gen_full_adder()