949c12e0d5
This allows us to test the running of bare-metal binaries on gem5. The current design assumes the binaries to run are part of gem5 Resources. Change-Id: I4f0f40a4628fd5f5951d89079824369a219d6e91 Reviewed-on: https://gem5-review.googlesource.com/c/public/gem5/+/49553 Tested-by: kokoro <noreply+kokoro@google.com> Reviewed-by: Jason Lowe-Power <power.jg@gmail.com> Maintainer: Jason Lowe-Power <power.jg@gmail.com>
139 lines
4.3 KiB
Python
139 lines
4.3 KiB
Python
# Copyright (c) 2021 The Regents of the University of California
|
|
# All rights reserved.
|
|
#
|
|
# Redistribution and use in source and binary forms, with or without
|
|
# modification, are permitted provided that the following conditions are
|
|
# met: redistributions of source code must retain the above copyright
|
|
# notice, this list of conditions and the following disclaimer;
|
|
# redistributions in binary form must reproduce the above copyright
|
|
# notice, this list of conditions and the following disclaimer in the
|
|
# documentation and/or other materials provided with the distribution;
|
|
# neither the name of the copyright holders nor the names of its
|
|
# contributors may be used to endorse or promote products derived from
|
|
# this software without specific prior written permission.
|
|
#
|
|
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
|
# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
"""
|
|
A run script for a very simple Syscall-Execution running simple binaries.
|
|
The system has no cache heirarchy and is as "bare-bones" as you can get in
|
|
gem5 while still being functinal.
|
|
"""
|
|
|
|
import m5
|
|
from m5.objects import Root
|
|
|
|
import os
|
|
import sys
|
|
|
|
# This is a lame hack to get the imports working correctly.
|
|
# TODO: This needs fixed.
|
|
sys.path.append(
|
|
os.path.join(
|
|
os.path.dirname(os.path.abspath(__file__)),
|
|
os.pardir,
|
|
os.pardir,
|
|
os.pardir,
|
|
)
|
|
)
|
|
|
|
from components_library.resources.resource import Resource
|
|
from components_library.boards.simple_board import SimpleBoard
|
|
from components_library.cachehierarchies.classic.no_cache import NoCache
|
|
from components_library.memory.single_channel import SingleChannelDDR3_1600
|
|
from components_library.processors.simple_processor import SimpleProcessor
|
|
from components_library.processors.cpu_types import CPUTypes
|
|
|
|
import argparse
|
|
|
|
parser = argparse.ArgumentParser(
|
|
description="A script to run the gem5 boot test. This test boots the "
|
|
"linux kernel."
|
|
)
|
|
|
|
parser.add_argument(
|
|
"resource",
|
|
type=str,
|
|
help="The gem5 resource binary to run.",
|
|
)
|
|
|
|
parser.add_argument(
|
|
"cpu",
|
|
type=str,
|
|
choices=("kvm", "timing", "atomic", "o3"),
|
|
help="The CPU type used.",
|
|
)
|
|
|
|
parser.add_argument(
|
|
"-r",
|
|
"--resource-directory",
|
|
type=str,
|
|
required=False,
|
|
help="The directory in which resources will be downloaded or exist.",
|
|
)
|
|
|
|
parser.add_argument(
|
|
"-o",
|
|
"--override-download",
|
|
action="store_true",
|
|
help="Override a local resource if the hashes do not match.",
|
|
)
|
|
|
|
args = parser.parse_args()
|
|
|
|
def input_to_cputype(input: str) -> CPUTypes:
|
|
if input == "kvm":
|
|
return CPUTypes.KVM
|
|
elif input == "timing":
|
|
return CPUTypes.TIMING
|
|
elif input == "atomic":
|
|
return CPUTypes.ATOMIC
|
|
elif input == "o3":
|
|
return CPUTypes.O3
|
|
else:
|
|
raise NotADirectoryError("Unknown CPU type '{}'.".format(input))
|
|
|
|
# Setup the system.
|
|
cache_hierarchy = NoCache()
|
|
memory = SingleChannelDDR3_1600()
|
|
processor = SimpleProcessor(cpu_type=input_to_cputype(args.cpu), num_cores=1)
|
|
|
|
motherboard = SimpleBoard(
|
|
clk_freq="3GHz",
|
|
processor=processor,
|
|
memory=memory,
|
|
cache_hierarchy=cache_hierarchy,
|
|
)
|
|
|
|
motherboard.connect_things()
|
|
|
|
# Set the workload
|
|
binary = Resource(args.resource,
|
|
resource_directory=args.resource_directory,
|
|
override=args.override_download)
|
|
motherboard.set_workload(binary)
|
|
|
|
root = Root(full_system=False, system=motherboard)
|
|
|
|
if args.cpu == "kvm":
|
|
# TODO: This of annoying. Is there a way to fix this to happen
|
|
# automatically when running KVM?
|
|
root.sim_quantum = int(1e9)
|
|
|
|
m5.instantiate()
|
|
|
|
exit_event = m5.simulate()
|
|
print(
|
|
"Exiting @ tick {} because {}.".format(m5.curTick(), exit_event.getCause())
|
|
)
|