56
57import devices
58from devices import AtomicCluster, KvmCluster
59
60
61default_dtb = 'armv8_gem5_v1_big_little_2_2.dtb'
62default_kernel = 'vmlinux4.3.aarch64'
63default_disk = 'aarch64-ubuntu-trusty-headless.img'
64default_rcs = 'bootscript.rcS'
65
66default_mem_size= "2GB"
67
68def _to_ticks(value):
69 """Helper function to convert a latency from string format to Ticks"""
70
71 return m5.ticks.fromSeconds(m5.util.convert.anyToLatency(value))
72
73def _using_pdes(root):
74 """Determine if the simulator is using multiple parallel event queues"""
75
76 for obj in root.descendants():
77 if not m5.proxy.isproxy(obj.eventq_index) and \
78 obj.eventq_index != root.eventq_index:
79 return True
80
81 return False
82
83
84class BigCluster(devices.CpuCluster):
85 def __init__(self, system, num_cpus, cpu_clock,
86 cpu_voltage="1.0V"):
87 cpu_config = [ CpuConfig.get("O3_ARM_v7a_3"), devices.L1I, devices.L1D,
88 devices.WalkCache, devices.L2 ]
89 super(BigCluster, self).__init__(system, num_cpus, cpu_clock,
90 cpu_voltage, *cpu_config)
91
92class LittleCluster(devices.CpuCluster):
93 def __init__(self, system, num_cpus, cpu_clock,
94 cpu_voltage="1.0V"):
95 cpu_config = [ CpuConfig.get("MinorCPU"), devices.L1I, devices.L1D,
96 devices.WalkCache, devices.L2 ]
97 super(LittleCluster, self).__init__(system, num_cpus, cpu_clock,
98 cpu_voltage, *cpu_config)
99
100class Ex5BigCluster(devices.CpuCluster):
101 def __init__(self, system, num_cpus, cpu_clock,
102 cpu_voltage="1.0V"):
103 cpu_config = [ CpuConfig.get("ex5_big"), ex5_big.L1I, ex5_big.L1D,
104 ex5_big.WalkCache, ex5_big.L2 ]
105 super(Ex5BigCluster, self).__init__(system, num_cpus, cpu_clock,
106 cpu_voltage, *cpu_config)
107
108class Ex5LittleCluster(devices.CpuCluster):
109 def __init__(self, system, num_cpus, cpu_clock,
110 cpu_voltage="1.0V"):
111 cpu_config = [ CpuConfig.get("ex5_LITTLE"), ex5_LITTLE.L1I,
112 ex5_LITTLE.L1D, ex5_LITTLE.WalkCache, ex5_LITTLE.L2 ]
113 super(Ex5LittleCluster, self).__init__(system, num_cpus, cpu_clock,
114 cpu_voltage, *cpu_config)
115
116def createSystem(caches, kernel, bootscript, disks=[]):
117 sys = devices.SimpleSystem(caches, default_mem_size,
118 kernel=SysPaths.binary(kernel),
119 readfile=bootscript,
120 machine_type="DTOnly")
121
122 sys.mem_ctrls = SimpleMemory(range=sys._mem_range)
123 sys.mem_ctrls.port = sys.membus.master
124
125 sys.connect()
126
127 # Attach disk images
128 if disks:
129 def cow_disk(image_file):
130 image = CowDiskImage()
131 image.child.image_file = SysPaths.disk(image_file)
132 return image
133
134 sys.disk_images = [ cow_disk(f) for f in disks ]
135 sys.pci_vio_block = [ PciVirtIO(vio=VirtIOBlock(image=img))
136 for img in sys.disk_images ]
137 for dev in sys.pci_vio_block:
138 sys.attach_pci(dev)
139
140 sys.realview.setupBootLoader(sys.membus, sys, SysPaths.binary)
141
142 return sys
143
144cpu_types = {
145 "atomic" : (AtomicCluster, AtomicCluster),
146 "timing" : (BigCluster, LittleCluster),
147 "exynos" : (Ex5BigCluster, Ex5LittleCluster),
148}
149
150# Only add the KVM CPU if it has been compiled into gem5
151if devices.have_kvm:
152 cpu_types["kvm"] = (KvmCluster, KvmCluster)
153
154
155def addOptions(parser):
156 parser.add_argument("--restore-from", type=str, default=None,
157 help="Restore from checkpoint")
158 parser.add_argument("--dtb", type=str, default=default_dtb,
159 help="DTB file to load")
160 parser.add_argument("--kernel", type=str, default=default_kernel,
161 help="Linux kernel")
162 parser.add_argument("--disk", action="append", type=str, default=[],
163 help="Disks to instantiate")
164 parser.add_argument("--bootscript", type=str, default=default_rcs,
165 help="Linux bootscript")
166 parser.add_argument("--cpu-type", type=str, choices=cpu_types.keys(),
167 default="timing",
168 help="CPU simulation mode. Default: %(default)s")
169 parser.add_argument("--kernel-init", type=str, default="/sbin/init",
170 help="Override init")
171 parser.add_argument("--big-cpus", type=int, default=1,
172 help="Number of big CPUs to instantiate")
173 parser.add_argument("--little-cpus", type=int, default=1,
174 help="Number of little CPUs to instantiate")
175 parser.add_argument("--caches", action="store_true", default=False,
176 help="Instantiate caches")
177 parser.add_argument("--last-cache-level", type=int, default=2,
178 help="Last level of caches (e.g. 3 for L3)")
179 parser.add_argument("--big-cpu-clock", type=str, default="2GHz",
180 help="Big CPU clock frequency")
181 parser.add_argument("--little-cpu-clock", type=str, default="1GHz",
182 help="Little CPU clock frequency")
183 parser.add_argument("--sim-quantum", type=str, default="1ms",
184 help="Simulation quantum for parallel simulation. " \
185 "Default: %(default)s")
186 return parser
187
188def build(options):
189 m5.ticks.fixGlobalFrequency()
190
191 kernel_cmd = [
192 "earlyprintk=pl011,0x1c090000",
193 "console=ttyAMA0",
194 "lpj=19988480",
195 "norandmaps",
196 "loglevel=8",
197 "mem=%s" % default_mem_size,
198 "root=/dev/vda1",
199 "rw",
200 "init=%s" % options.kernel_init,
201 "vmalloc=768MB",
202 ]
203
204 root = Root(full_system=True)
205
206 disks = [default_disk] if len(options.disk) == 0 else options.disk
207 system = createSystem(options.caches,
208 options.kernel,
209 options.bootscript,
210 disks=disks)
211
212 root.system = system
213 system.boot_osflags = " ".join(kernel_cmd)
214
215 if options.big_cpus + options.little_cpus == 0:
216 m5.util.panic("Empty CPU clusters")
217
218 big_model, little_model = cpu_types[options.cpu_type]
219
220 all_cpus = []
221 # big cluster
222 if options.big_cpus > 0:
223 system.bigCluster = big_model(system, options.big_cpus,
224 options.big_cpu_clock)
225 system.mem_mode = system.bigCluster.memoryMode()
226 all_cpus += system.bigCluster.cpus
227
228 # little cluster
229 if options.little_cpus > 0:
230 system.littleCluster = little_model(system, options.little_cpus,
231 options.little_cpu_clock)
232 system.mem_mode = system.littleCluster.memoryMode()
233 all_cpus += system.littleCluster.cpus
234
235 # Figure out the memory mode
236 if options.big_cpus > 0 and options.little_cpus > 0 and \
237 system.littleCluster.memoryMode() != system.littleCluster.memoryMode():
238 m5.util.panic("Memory mode missmatch among CPU clusters")
239
240
241 # create caches
242 system.addCaches(options.caches, options.last_cache_level)
243 if not options.caches:
244 if options.big_cpus > 0 and system.bigCluster.requireCaches():
245 m5.util.panic("Big CPU model requires caches")
246 if options.little_cpus > 0 and system.littleCluster.requireCaches():
247 m5.util.panic("Little CPU model requires caches")
248
249 # Create a KVM VM and do KVM-specific configuration
250 if issubclass(big_model, KvmCluster):
251 _build_kvm(system, all_cpus)
252
253 # Linux device tree
254 system.dtb_filename = SysPaths.binary(options.dtb)
255
256 return root
257
258def _build_kvm(system, cpus):
259 system.kvm_vm = KvmVM()
260
261 # Assign KVM CPUs to their own event queues / threads. This
262 # has to be done after creating caches and other child objects
263 # since these mustn't inherit the CPU event queue.
264 if len(cpus) > 1:
265 device_eq = 0
266 first_cpu_eq = 1
267 for idx, cpu in enumerate(cpus):
268 # Child objects usually inherit the parent's event
269 # queue. Override that and use the same event queue for
270 # all devices.
271 for obj in cpu.descendants():
272 obj.eventq_index = device_eq
273 cpu.eventq_index = first_cpu_eq + idx
274
275
276
277def instantiate(options, checkpoint_dir=None):
278 # Setup the simulation quantum if we are running in PDES-mode
279 # (e.g., when using KVM)
280 root = Root.getInstance()
281 if root and _using_pdes(root):
282 m5.util.inform("Running in PDES mode with a %s simulation quantum.",
283 options.sim_quantum)
284 root.sim_quantum = _to_ticks(options.sim_quantum)
285
286 # Get and load from the chkpt or simpoint checkpoint
287 if options.restore_from:
288 if checkpoint_dir and not os.path.isabs(options.restore_from):
289 cpt = os.path.join(checkpoint_dir, options.restore_from)
290 else:
291 cpt = options.restore_from
292
293 m5.util.inform("Restoring from checkpoint %s", cpt)
294 m5.instantiate(cpt)
295 else:
296 m5.instantiate()
297
298
299def run(checkpoint_dir=m5.options.outdir):
300 # start simulation (and drop checkpoints when requested)
301 while True:
302 event = m5.simulate()
303 exit_msg = event.getCause()
304 if exit_msg == "checkpoint":
305 print "Dropping checkpoint at tick %d" % m5.curTick()
306 cpt_dir = os.path.join(checkpoint_dir, "cpt.%d" % m5.curTick())
307 m5.checkpoint(cpt_dir)
308 print "Checkpoint done."
309 else:
310 print exit_msg, " @ ", m5.curTick()
311 break
312
313 sys.exit(event.getCode())
314
315
316def main():
317 parser = argparse.ArgumentParser(
318 description="Generic ARM big.LITTLE configuration")
319 addOptions(parser)
320 options = parser.parse_args()
321 root = build(options)
322 instantiate(options)
323 run()
324
325
326if __name__ == "__m5_main__":
327 main()
| 55
56import devices
57from devices import AtomicCluster, KvmCluster
58
59
60default_dtb = 'armv8_gem5_v1_big_little_2_2.dtb'
61default_kernel = 'vmlinux4.3.aarch64'
62default_disk = 'aarch64-ubuntu-trusty-headless.img'
63default_rcs = 'bootscript.rcS'
64
65default_mem_size= "2GB"
66
67def _to_ticks(value):
68 """Helper function to convert a latency from string format to Ticks"""
69
70 return m5.ticks.fromSeconds(m5.util.convert.anyToLatency(value))
71
72def _using_pdes(root):
73 """Determine if the simulator is using multiple parallel event queues"""
74
75 for obj in root.descendants():
76 if not m5.proxy.isproxy(obj.eventq_index) and \
77 obj.eventq_index != root.eventq_index:
78 return True
79
80 return False
81
82
83class BigCluster(devices.CpuCluster):
84 def __init__(self, system, num_cpus, cpu_clock,
85 cpu_voltage="1.0V"):
86 cpu_config = [ CpuConfig.get("O3_ARM_v7a_3"), devices.L1I, devices.L1D,
87 devices.WalkCache, devices.L2 ]
88 super(BigCluster, self).__init__(system, num_cpus, cpu_clock,
89 cpu_voltage, *cpu_config)
90
91class LittleCluster(devices.CpuCluster):
92 def __init__(self, system, num_cpus, cpu_clock,
93 cpu_voltage="1.0V"):
94 cpu_config = [ CpuConfig.get("MinorCPU"), devices.L1I, devices.L1D,
95 devices.WalkCache, devices.L2 ]
96 super(LittleCluster, self).__init__(system, num_cpus, cpu_clock,
97 cpu_voltage, *cpu_config)
98
99class Ex5BigCluster(devices.CpuCluster):
100 def __init__(self, system, num_cpus, cpu_clock,
101 cpu_voltage="1.0V"):
102 cpu_config = [ CpuConfig.get("ex5_big"), ex5_big.L1I, ex5_big.L1D,
103 ex5_big.WalkCache, ex5_big.L2 ]
104 super(Ex5BigCluster, self).__init__(system, num_cpus, cpu_clock,
105 cpu_voltage, *cpu_config)
106
107class Ex5LittleCluster(devices.CpuCluster):
108 def __init__(self, system, num_cpus, cpu_clock,
109 cpu_voltage="1.0V"):
110 cpu_config = [ CpuConfig.get("ex5_LITTLE"), ex5_LITTLE.L1I,
111 ex5_LITTLE.L1D, ex5_LITTLE.WalkCache, ex5_LITTLE.L2 ]
112 super(Ex5LittleCluster, self).__init__(system, num_cpus, cpu_clock,
113 cpu_voltage, *cpu_config)
114
115def createSystem(caches, kernel, bootscript, disks=[]):
116 sys = devices.SimpleSystem(caches, default_mem_size,
117 kernel=SysPaths.binary(kernel),
118 readfile=bootscript,
119 machine_type="DTOnly")
120
121 sys.mem_ctrls = SimpleMemory(range=sys._mem_range)
122 sys.mem_ctrls.port = sys.membus.master
123
124 sys.connect()
125
126 # Attach disk images
127 if disks:
128 def cow_disk(image_file):
129 image = CowDiskImage()
130 image.child.image_file = SysPaths.disk(image_file)
131 return image
132
133 sys.disk_images = [ cow_disk(f) for f in disks ]
134 sys.pci_vio_block = [ PciVirtIO(vio=VirtIOBlock(image=img))
135 for img in sys.disk_images ]
136 for dev in sys.pci_vio_block:
137 sys.attach_pci(dev)
138
139 sys.realview.setupBootLoader(sys.membus, sys, SysPaths.binary)
140
141 return sys
142
143cpu_types = {
144 "atomic" : (AtomicCluster, AtomicCluster),
145 "timing" : (BigCluster, LittleCluster),
146 "exynos" : (Ex5BigCluster, Ex5LittleCluster),
147}
148
149# Only add the KVM CPU if it has been compiled into gem5
150if devices.have_kvm:
151 cpu_types["kvm"] = (KvmCluster, KvmCluster)
152
153
154def addOptions(parser):
155 parser.add_argument("--restore-from", type=str, default=None,
156 help="Restore from checkpoint")
157 parser.add_argument("--dtb", type=str, default=default_dtb,
158 help="DTB file to load")
159 parser.add_argument("--kernel", type=str, default=default_kernel,
160 help="Linux kernel")
161 parser.add_argument("--disk", action="append", type=str, default=[],
162 help="Disks to instantiate")
163 parser.add_argument("--bootscript", type=str, default=default_rcs,
164 help="Linux bootscript")
165 parser.add_argument("--cpu-type", type=str, choices=cpu_types.keys(),
166 default="timing",
167 help="CPU simulation mode. Default: %(default)s")
168 parser.add_argument("--kernel-init", type=str, default="/sbin/init",
169 help="Override init")
170 parser.add_argument("--big-cpus", type=int, default=1,
171 help="Number of big CPUs to instantiate")
172 parser.add_argument("--little-cpus", type=int, default=1,
173 help="Number of little CPUs to instantiate")
174 parser.add_argument("--caches", action="store_true", default=False,
175 help="Instantiate caches")
176 parser.add_argument("--last-cache-level", type=int, default=2,
177 help="Last level of caches (e.g. 3 for L3)")
178 parser.add_argument("--big-cpu-clock", type=str, default="2GHz",
179 help="Big CPU clock frequency")
180 parser.add_argument("--little-cpu-clock", type=str, default="1GHz",
181 help="Little CPU clock frequency")
182 parser.add_argument("--sim-quantum", type=str, default="1ms",
183 help="Simulation quantum for parallel simulation. " \
184 "Default: %(default)s")
185 return parser
186
187def build(options):
188 m5.ticks.fixGlobalFrequency()
189
190 kernel_cmd = [
191 "earlyprintk=pl011,0x1c090000",
192 "console=ttyAMA0",
193 "lpj=19988480",
194 "norandmaps",
195 "loglevel=8",
196 "mem=%s" % default_mem_size,
197 "root=/dev/vda1",
198 "rw",
199 "init=%s" % options.kernel_init,
200 "vmalloc=768MB",
201 ]
202
203 root = Root(full_system=True)
204
205 disks = [default_disk] if len(options.disk) == 0 else options.disk
206 system = createSystem(options.caches,
207 options.kernel,
208 options.bootscript,
209 disks=disks)
210
211 root.system = system
212 system.boot_osflags = " ".join(kernel_cmd)
213
214 if options.big_cpus + options.little_cpus == 0:
215 m5.util.panic("Empty CPU clusters")
216
217 big_model, little_model = cpu_types[options.cpu_type]
218
219 all_cpus = []
220 # big cluster
221 if options.big_cpus > 0:
222 system.bigCluster = big_model(system, options.big_cpus,
223 options.big_cpu_clock)
224 system.mem_mode = system.bigCluster.memoryMode()
225 all_cpus += system.bigCluster.cpus
226
227 # little cluster
228 if options.little_cpus > 0:
229 system.littleCluster = little_model(system, options.little_cpus,
230 options.little_cpu_clock)
231 system.mem_mode = system.littleCluster.memoryMode()
232 all_cpus += system.littleCluster.cpus
233
234 # Figure out the memory mode
235 if options.big_cpus > 0 and options.little_cpus > 0 and \
236 system.littleCluster.memoryMode() != system.littleCluster.memoryMode():
237 m5.util.panic("Memory mode missmatch among CPU clusters")
238
239
240 # create caches
241 system.addCaches(options.caches, options.last_cache_level)
242 if not options.caches:
243 if options.big_cpus > 0 and system.bigCluster.requireCaches():
244 m5.util.panic("Big CPU model requires caches")
245 if options.little_cpus > 0 and system.littleCluster.requireCaches():
246 m5.util.panic("Little CPU model requires caches")
247
248 # Create a KVM VM and do KVM-specific configuration
249 if issubclass(big_model, KvmCluster):
250 _build_kvm(system, all_cpus)
251
252 # Linux device tree
253 system.dtb_filename = SysPaths.binary(options.dtb)
254
255 return root
256
257def _build_kvm(system, cpus):
258 system.kvm_vm = KvmVM()
259
260 # Assign KVM CPUs to their own event queues / threads. This
261 # has to be done after creating caches and other child objects
262 # since these mustn't inherit the CPU event queue.
263 if len(cpus) > 1:
264 device_eq = 0
265 first_cpu_eq = 1
266 for idx, cpu in enumerate(cpus):
267 # Child objects usually inherit the parent's event
268 # queue. Override that and use the same event queue for
269 # all devices.
270 for obj in cpu.descendants():
271 obj.eventq_index = device_eq
272 cpu.eventq_index = first_cpu_eq + idx
273
274
275
276def instantiate(options, checkpoint_dir=None):
277 # Setup the simulation quantum if we are running in PDES-mode
278 # (e.g., when using KVM)
279 root = Root.getInstance()
280 if root and _using_pdes(root):
281 m5.util.inform("Running in PDES mode with a %s simulation quantum.",
282 options.sim_quantum)
283 root.sim_quantum = _to_ticks(options.sim_quantum)
284
285 # Get and load from the chkpt or simpoint checkpoint
286 if options.restore_from:
287 if checkpoint_dir and not os.path.isabs(options.restore_from):
288 cpt = os.path.join(checkpoint_dir, options.restore_from)
289 else:
290 cpt = options.restore_from
291
292 m5.util.inform("Restoring from checkpoint %s", cpt)
293 m5.instantiate(cpt)
294 else:
295 m5.instantiate()
296
297
298def run(checkpoint_dir=m5.options.outdir):
299 # start simulation (and drop checkpoints when requested)
300 while True:
301 event = m5.simulate()
302 exit_msg = event.getCause()
303 if exit_msg == "checkpoint":
304 print "Dropping checkpoint at tick %d" % m5.curTick()
305 cpt_dir = os.path.join(checkpoint_dir, "cpt.%d" % m5.curTick())
306 m5.checkpoint(cpt_dir)
307 print "Checkpoint done."
308 else:
309 print exit_msg, " @ ", m5.curTick()
310 break
311
312 sys.exit(event.getCode())
313
314
315def main():
316 parser = argparse.ArgumentParser(
317 description="Generic ARM big.LITTLE configuration")
318 addOptions(parser)
319 options = parser.parse_args()
320 root = build(options)
321 instantiate(options)
322 run()
323
324
325if __name__ == "__m5_main__":
326 main()
|