configs: Remove Python 2 compatibility code in Arm configs

[gem5.git] / configs / example / arm / starter_se.py

diff --git a/configs/example/arm/starter_se.py b/configs/example/arm/starter_se.py
index 0f20ecc6451d069d6d2b1008b86b2a7658779eed..23da8e7917c4e72e99cbb62afffe292327affa6f 100644 (file)
--- a/configs/example/arm/starter_se.py
+++ b/configs/example/arm/starter_se.py
@@ -38,9 +38,6 @@ Research Starter Kit on System Modeling. More information can be found
 at: http://www.arm.com/ResearchEnablement/SystemModeling
 """
 
-from __future__ import print_function
-from __future__ import absolute_import
-
 import os
 import m5
 from m5.util import addToPath
@@ -171,6 +168,8 @@ def create(args):
               (len(processes), args.num_cores))
         sys.exit(1)
 
+    system.workload = SEWorkload.init_compatible(processes[0].executable)
+
     # Assign one workload to each CPU
     for cpu, workload in zip(system.cpu_cluster.cpus, processes):
         cpu.workload = workload
@@ -183,7 +182,7 @@ def main():
 
     parser.add_argument("commands_to_run", metavar="command(s)", nargs='*',
                         help="Command(s) to run")
-    parser.add_argument("--cpu", type=str, choices=cpu_types.keys(),
+    parser.add_argument("--cpu", type=str, choices=list(cpu_types.keys()),
                         default="atomic",
                         help="CPU model to use")
     parser.add_argument("--cpu-freq", type=str, default="4GHz")