Skip to content

Commit dabb69a

Browse files
committed
make --enableMemH and --randomizeCosts options use smart syntax
1 parent e098c1c commit dabb69a

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

test/rev-model-options-config.py

+6-1
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
import os
1212
import argparse
1313
import sst
14+
import ast
1415

1516
DEBUG_L1 = 0
1617
DEBUG_MEM = 0
@@ -23,15 +24,19 @@
2324
parser.add_argument("--numCores", type=int, help="Number of Rev Cores per RevCPU", default=1)
2425
parser.add_argument("--numHarts", type=int, help="Number of HARTs per Rev Core", default=1)
2526
parser.add_argument("--program", help="The program executable to run in the simulation", default="a.out")
26-
parser.add_argument("--enableMemH", type=int, choices=[0, 1], help="Enable (1) or disable (0) memHierarchy backend", default=0)
27+
parser.add_argument("--enableMemH", nargs='?', help="Enable memHierarchy backend", default="False")
2728
parser.add_argument("--verbose", type=int, help="Verbosity level", default=2)
2829
parser.add_argument("--machine", help="Machine type/configuration", default="[CORES:RV64GC]")
2930
parser.add_argument("--args", help="Command line arguments to pass to the target executable", default="")
3031
parser.add_argument("--startSymbol", help="ELF Symbol Rev should begin execution at", default="[0:main]")
32+
parser.add_argument("--randomizeCosts", nargs='?', help="Randomize costs of instructions", default="False")
3133

3234
# Parse arguments
3335
args = parser.parse_args()
3436

37+
args.enableMemH = True if args.enableMemH is None or ast.literal_eval(args.enableMemH) else False
38+
args.randomizeCosts = True if args.randomizeCosts is None or ast.literal_eval(args.randomizeCosts) else False
39+
3540
# Print arguments nicely
3641
print("Rev SST Simulation Configuration:")
3742
for arg in vars(args):

0 commit comments

Comments
 (0)