[otbn] Add a --size argument to gen-binaries.py
For now, the default is unchanged at 100 but we'll probably bump it up
at some point (otherwise loops are rather boring).
Signed-off-by: Rupert Swarbrick <rswarbrick@lowrisc.org>
diff --git a/hw/ip/otbn/dv/uvm/gen-binaries.py b/hw/ip/otbn/dv/uvm/gen-binaries.py
index 7d6d00e..7616981 100755
--- a/hw/ip/otbn/dv/uvm/gen-binaries.py
+++ b/hw/ip/otbn/dv/uvm/gen-binaries.py
@@ -118,6 +118,7 @@
'if set, or build-out at the top of the '
'repository if not.'))
parser.add_argument('--seed', type=read_positive, default=0)
+ parser.add_argument('--size', type=read_positive, default=100)
parser.add_argument('--verbose', '-v', action='store_true')
parser.add_argument('--jobs', '-j', type=read_jobs, nargs='?',
const='unlimited', help='Number of parallel jobs.')
@@ -140,7 +141,8 @@
os.makedirs(args.destdir, exist_ok=True)
with open(os.path.join(args.destdir, 'build.ninja'), 'w') as ninja_handle:
- write_ninja(ninja_handle, rig_count, args.seed, toolchain, otbn_dir)
+ write_ninja(ninja_handle, rig_count,
+ args.seed, args.size, toolchain, otbn_dir)
# Handle the -j argument like Make does, defaulting to 1 thread. This
# behaves a bit more reasonably than ninja's default (# cores) if we're
@@ -159,7 +161,7 @@
return subprocess.run(cmd, cwd=args.destdir, check=False).returncode
-def write_ninja(handle: TextIO, rig_count: int, start_seed: int,
+def write_ninja(handle: TextIO, rig_count: int, start_seed: int, size: int,
toolchain: Toolchain, otbn_dir: str) -> None:
'''Write a build.ninja to build rig_count random binaries and a smoke test
@@ -167,14 +169,17 @@
OTBN tooling is found in util_dir.
'''
+ assert start_seed >= 0
+ assert size > 0
+
otbn_rig = os.path.join(otbn_dir, 'util/otbn-rig')
smoke_src_dir = os.path.join(otbn_dir, 'dv/smoke')
seeds = [start_seed + idx for idx in range(rig_count)]
handle.write('rule rig-gen\n'
- ' command = {rig} gen --size 100 --seed $seed -o $out\n'
- .format(rig=otbn_rig))
+ ' command = {rig} gen --size {size} --seed $seed -o $out\n'
+ .format(rig=otbn_rig, size=size))
handle.write('rule rig-asm\n'
' command = {rig} asm -o $seed $in\n'
