python: Apply Black formatter to Python files
The command executed was `black src configs tests util`. Change-Id: I8dfaa6ab04658fea37618127d6ac19270028d771 Reviewed-on: https://gem5-review.googlesource.com/c/public/gem5/+/47024 Maintainer: Bobby Bruce <bbruce@ucdavis.edu> Reviewed-by: Jason Lowe-Power <power.jg@gmail.com> Reviewed-by: Giacomo Travaglini <giacomo.travaglini@arm.com> Tested-by: kokoro <noreply+kokoro@google.com>
This commit is contained in:
Bobby R. Bruce
committed by
Giacomo Travaglini
Giacomo Travaglini
parent
1cfaa8da83
commit
787204c92d
@@ -40,7 +40,7 @@ from m5.objects import *
|
||||
from m5.options import *
|
||||
import argparse
|
||||
|
||||
m5.util.addToPath('../..')
|
||||
m5.util.addToPath("../..")
|
||||
|
||||
from common import MemConfig
|
||||
from common import ObjectList
|
||||
@@ -52,19 +52,20 @@ from ruby import Ruby
|
||||
import devices
|
||||
|
||||
|
||||
default_kernel = 'vmlinux.arm64'
|
||||
default_disk = 'linaro-minimal-aarch64.img'
|
||||
default_root_device = '/dev/vda1'
|
||||
default_kernel = "vmlinux.arm64"
|
||||
default_disk = "linaro-minimal-aarch64.img"
|
||||
default_root_device = "/dev/vda1"
|
||||
|
||||
|
||||
# Pre-defined CPU configurations.
|
||||
cpu_types = {
|
||||
"noncaching" : NonCachingSimpleCPU,
|
||||
"minor" : MinorCPU,
|
||||
"hpi" : HPI.HPI,
|
||||
"o3" : O3_ARM_v7a.O3_ARM_v7a_3,
|
||||
"noncaching": NonCachingSimpleCPU,
|
||||
"minor": MinorCPU,
|
||||
"hpi": HPI.HPI,
|
||||
"o3": O3_ARM_v7a.O3_ARM_v7a_3,
|
||||
}
|
||||
|
||||
|
||||
def create_cow_image(name):
|
||||
"""Helper function to create a Copy-on-Write disk image"""
|
||||
image = CowDiskImage()
|
||||
@@ -72,23 +73,31 @@ def create_cow_image(name):
|
||||
|
||||
return image
|
||||
|
||||
|
||||
def config_ruby(system, args):
|
||||
cpus = []
|
||||
for cluster in system.cpu_cluster:
|
||||
for cpu in cluster.cpus:
|
||||
cpus.append(cpu)
|
||||
|
||||
Ruby.create_system(args, True, system, system.iobus,
|
||||
system._dma_ports, system.realview.bootmem,
|
||||
cpus)
|
||||
Ruby.create_system(
|
||||
args,
|
||||
True,
|
||||
system,
|
||||
system.iobus,
|
||||
system._dma_ports,
|
||||
system.realview.bootmem,
|
||||
cpus,
|
||||
)
|
||||
|
||||
# Create a seperate clock domain for Ruby
|
||||
system.ruby.clk_domain = SrcClockDomain(
|
||||
clock = args.ruby_clock,
|
||||
voltage_domain = system.voltage_domain)
|
||||
clock=args.ruby_clock, voltage_domain=system.voltage_domain
|
||||
)
|
||||
|
||||
|
||||
def create(args):
|
||||
''' Create and configure the system object. '''
|
||||
""" Create and configure the system object. """
|
||||
|
||||
if args.script and not os.path.isfile(args.script):
|
||||
print("Error: Bootscript %s does not exist" % args.script)
|
||||
@@ -97,19 +106,25 @@ def create(args):
|
||||
cpu_class = cpu_types[args.cpu]
|
||||
mem_mode = cpu_class.memory_mode()
|
||||
|
||||
system = devices.ArmRubySystem(args.mem_size,
|
||||
mem_mode=mem_mode,
|
||||
workload=ArmFsLinux(
|
||||
object_file=
|
||||
SysPaths.binary(args.kernel)),
|
||||
readfile=args.script)
|
||||
system = devices.ArmRubySystem(
|
||||
args.mem_size,
|
||||
mem_mode=mem_mode,
|
||||
workload=ArmFsLinux(object_file=SysPaths.binary(args.kernel)),
|
||||
readfile=args.script,
|
||||
)
|
||||
|
||||
# Add CPU clusters to the system
|
||||
system.cpu_cluster = [
|
||||
devices.CpuCluster(system,
|
||||
args.num_cpus,
|
||||
args.cpu_freq, "1.0V",
|
||||
cpu_class, None, None, None),
|
||||
devices.CpuCluster(
|
||||
system,
|
||||
args.num_cpus,
|
||||
args.cpu_freq,
|
||||
"1.0V",
|
||||
cpu_class,
|
||||
None,
|
||||
None,
|
||||
None,
|
||||
)
|
||||
]
|
||||
|
||||
# Add the PCI devices we need for this system. The base system
|
||||
@@ -120,7 +135,7 @@ def create(args):
|
||||
# disk. Attach the disk image using gem5's Copy-on-Write
|
||||
# functionality to avoid writing changes to the stored copy of
|
||||
# the disk image.
|
||||
PciVirtIO(vio=VirtIOBlock(image=create_cow_image(args.disk_image))),
|
||||
PciVirtIO(vio=VirtIOBlock(image=create_cow_image(args.disk_image)))
|
||||
]
|
||||
|
||||
# Attach the PCI devices to the system. The helper method in the
|
||||
@@ -141,8 +156,9 @@ def create(args):
|
||||
system.workload.dtb_filename = args.dtb
|
||||
else:
|
||||
# No DTB specified: autogenerate DTB
|
||||
system.workload.dtb_filename = \
|
||||
os.path.join(m5.options.outdir, 'system.dtb')
|
||||
system.workload.dtb_filename = os.path.join(
|
||||
m5.options.outdir, "system.dtb"
|
||||
)
|
||||
system.generateDtb(system.workload.dtb_filename)
|
||||
|
||||
# Linux boot command flags
|
||||
@@ -189,41 +205,73 @@ def run(args):
|
||||
def main():
|
||||
parser = argparse.ArgumentParser()
|
||||
|
||||
parser.add_argument("--dtb", type=str, default=None,
|
||||
help="DTB file to load")
|
||||
parser.add_argument("--kernel", type=str, default=default_kernel,
|
||||
help="Linux kernel")
|
||||
parser.add_argument("--disk-image", type=str,
|
||||
default=default_disk,
|
||||
help="Disk to instantiate")
|
||||
parser.add_argument("--root-device", type=str,
|
||||
default=default_root_device,
|
||||
help="OS device name for root partition (default: {})"
|
||||
.format(default_root_device))
|
||||
parser.add_argument("--script", type=str, default="",
|
||||
help = "Linux bootscript")
|
||||
parser.add_argument("--cpu", choices=list(cpu_types.keys()),
|
||||
default="minor",
|
||||
help="CPU model to use")
|
||||
parser.add_argument(
|
||||
"--dtb", type=str, default=None, help="DTB file to load"
|
||||
)
|
||||
parser.add_argument(
|
||||
"--kernel", type=str, default=default_kernel, help="Linux kernel"
|
||||
)
|
||||
parser.add_argument(
|
||||
"--disk-image",
|
||||
type=str,
|
||||
default=default_disk,
|
||||
help="Disk to instantiate",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--root-device",
|
||||
type=str,
|
||||
default=default_root_device,
|
||||
help="OS device name for root partition (default: {})".format(
|
||||
default_root_device
|
||||
),
|
||||
)
|
||||
parser.add_argument(
|
||||
"--script", type=str, default="", help="Linux bootscript"
|
||||
)
|
||||
parser.add_argument(
|
||||
"--cpu",
|
||||
choices=list(cpu_types.keys()),
|
||||
default="minor",
|
||||
help="CPU model to use",
|
||||
)
|
||||
parser.add_argument("--cpu-freq", type=str, default="4GHz")
|
||||
parser.add_argument("-n", "--num-cpus", type=int, default=1)
|
||||
parser.add_argument("--checkpoint", action="store_true")
|
||||
parser.add_argument("--restore", type=str, default=None)
|
||||
|
||||
parser.add_argument("--mem-type", default="DDR3_1600_8x8",
|
||||
choices=ObjectList.mem_list.get_names(),
|
||||
help = "type of memory to use")
|
||||
parser.add_argument("--mem-channels", type=int, default=1,
|
||||
help = "number of memory channels")
|
||||
parser.add_argument("--mem-ranks", type=int, default=None,
|
||||
help = "number of memory ranks per channel")
|
||||
parser.add_argument(
|
||||
"--mem-size", action="store", type=str, default="2GiB",
|
||||
help="Specify the physical memory size (single memory)")
|
||||
parser.add_argument("--enable-dram-powerdown", action="store_true",
|
||||
help="Enable low-power states in DRAMInterface")
|
||||
parser.add_argument("--mem-channels-intlv", type=int, default=0,
|
||||
help="Memory channels interleave")
|
||||
"--mem-type",
|
||||
default="DDR3_1600_8x8",
|
||||
choices=ObjectList.mem_list.get_names(),
|
||||
help="type of memory to use",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--mem-channels", type=int, default=1, help="number of memory channels"
|
||||
)
|
||||
parser.add_argument(
|
||||
"--mem-ranks",
|
||||
type=int,
|
||||
default=None,
|
||||
help="number of memory ranks per channel",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--mem-size",
|
||||
action="store",
|
||||
type=str,
|
||||
default="2GiB",
|
||||
help="Specify the physical memory size (single memory)",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--enable-dram-powerdown",
|
||||
action="store_true",
|
||||
help="Enable low-power states in DRAMInterface",
|
||||
)
|
||||
parser.add_argument(
|
||||
"--mem-channels-intlv",
|
||||
type=int,
|
||||
default=0,
|
||||
help="Memory channels interleave",
|
||||
)
|
||||
|
||||
parser.add_argument("--num-dirs", type=int, default=1)
|
||||
parser.add_argument("--num-l2caches", type=int, default=1)
|
||||
|
||||
Reference in New Issue
Block a user