diff options
-rw-r--r-- | configs/common/Options.py | 2 | ||||
-rw-r--r-- | configs/common/Simulation.py | 2 |
2 files changed, 4 insertions, 0 deletions
diff --git a/configs/common/Options.py b/configs/common/Options.py index e5069dc84..39355293d 100644 --- a/configs/common/Options.py +++ b/configs/common/Options.py @@ -133,6 +133,8 @@ def addCommonOptions(parser): simulate (default: run forever)""") parser.add_option("--work-item-id", action="store", type="int", help="the specific work id for exit & checkpointing") + parser.add_option("--num-work-ids", action="store", type="int", + help="Number of distinct work item types") parser.add_option("--work-begin-cpu-id-exit", action="store", type="int", help="exit when work starts on the specified cpu") parser.add_option("--work-end-exit-count", action="store", type="int", diff --git a/configs/common/Simulation.py b/configs/common/Simulation.py index c5d7aee40..5d4398c80 100644 --- a/configs/common/Simulation.py +++ b/configs/common/Simulation.py @@ -93,6 +93,8 @@ def setMemClass(options): def setWorkCountOptions(system, options): if options.work_item_id != None: system.work_item_id = options.work_item_id + if options.num_work_ids != None: + system.num_work_ids = options.num_work_ids if options.work_begin_cpu_id_exit != None: system.work_begin_cpu_id_exit = options.work_begin_cpu_id_exit if options.work_end_exit_count != None: |