diff options
Diffstat (limited to 'multiplier')
| -rwxr-xr-x | multiplier/hdl/gen_wallace.py | 7 |
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() |