configs: add example RVV SE parametrized config
Change-Id: I0776c5751da8b80340166ab518593686d141a4dd
This commit is contained in:
120
configs/example/gem5_library/riscv-rvv-example.py
Executable file
120
configs/example/gem5_library/riscv-rvv-example.py
Executable file
@@ -0,0 +1,120 @@
|
||||
# Copyright (c) 2024 Barcelona Supercomputing Center
|
||||
#
|
||||
# Redistribution and use in source and binary forms, with or without
|
||||
# modification, are permitted provided that the following conditions are met:
|
||||
#
|
||||
# 1. Redistributions of source code must retain the above copyright notice, this
|
||||
# list of conditions and the following disclaimer.
|
||||
#
|
||||
# 2. 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.
|
||||
#
|
||||
# 3. Neither the name of the copyright holder 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 HOLDER 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.
|
||||
|
||||
"""
|
||||
This script demonstrates how to run RISC-V vector-enabled binaries in SE mode
|
||||
with gem5. It accepts the number of CORES, VLEN, and ELEN as optional
|
||||
parameters, as well as the resource name to run. If no resource name is
|
||||
provided, a list of available resources will be displayed. If one is given the
|
||||
simulation will then execute the specified resource binary with the selected
|
||||
parameters until completion.
|
||||
|
||||
|
||||
Usage
|
||||
-----
|
||||
|
||||
# Compile gem5 for RISC-V
|
||||
scons build/RISCV/gem5.opt
|
||||
|
||||
# Run the simulation
|
||||
./build/RISCV/gem5.opt configs/example/gem5_library/riscv-rvv-example.py \
|
||||
[-c CORES] [-v VLEN] [-e ELEN] <resource>
|
||||
|
||||
"""
|
||||
|
||||
import argparse
|
||||
|
||||
from m5.objects import RiscvO3CPU
|
||||
|
||||
from gem5.components.boards.simple_board import SimpleBoard
|
||||
from gem5.components.cachehierarchies.classic.private_l1_private_l2_cache_hierarchy import (
|
||||
PrivateL1PrivateL2CacheHierarchy,
|
||||
)
|
||||
from gem5.components.memory import SingleChannelDDR3_1600
|
||||
from gem5.components.processors.base_cpu_core import BaseCPUCore
|
||||
from gem5.components.processors.base_cpu_processor import BaseCPUProcessor
|
||||
from gem5.isas import ISA
|
||||
from gem5.resources.resource import obtain_resource
|
||||
from gem5.simulate.simulator import Simulator
|
||||
from gem5.utils.requires import requires
|
||||
|
||||
|
||||
class RVVCore(BaseCPUCore):
|
||||
def __init__(self, elen, vlen, cpu_id):
|
||||
super().__init__(core=RiscvO3CPU(cpu_id=cpu_id), isa=ISA.RISCV)
|
||||
self.core.isa[0].elen = elen
|
||||
self.core.isa[0].vlen = vlen
|
||||
|
||||
|
||||
requires(isa_required=ISA.RISCV)
|
||||
|
||||
resources = [
|
||||
"rvv-branch",
|
||||
"rvv-index",
|
||||
"rvv-matmul",
|
||||
"rvv-memcpy",
|
||||
"rvv-reduce",
|
||||
"rvv-saxpy",
|
||||
"rvv-sgemm",
|
||||
"rvv-strcmp",
|
||||
"rvv-strcpy",
|
||||
"rvv-strlen",
|
||||
"rvv-strlen-fault",
|
||||
"rvv-strncpy",
|
||||
]
|
||||
|
||||
parser = argparse.ArgumentParser()
|
||||
parser.add_argument("resource", type=str, choices=resources)
|
||||
parser.add_argument("-c", "--cores", required=False, type=int, default=1)
|
||||
parser.add_argument("-v", "--vlen", required=False, type=int, default=256)
|
||||
parser.add_argument("-e", "--elen", required=False, type=int, default=64)
|
||||
|
||||
args = parser.parse_args()
|
||||
|
||||
cache_hierarchy = PrivateL1PrivateL2CacheHierarchy(
|
||||
l1d_size="32KiB", l1i_size="32KiB", l2_size="512KiB"
|
||||
)
|
||||
|
||||
memory = SingleChannelDDR3_1600()
|
||||
|
||||
processor = BaseCPUProcessor(
|
||||
cores=[RVVCore(args.elen, args.vlen, i) for i in range(args.cores)]
|
||||
)
|
||||
|
||||
board = SimpleBoard(
|
||||
clk_freq="1GHz",
|
||||
processor=processor,
|
||||
memory=memory,
|
||||
cache_hierarchy=cache_hierarchy,
|
||||
)
|
||||
|
||||
binary = obtain_resource(args.resource)
|
||||
board.set_se_binary_workload(binary)
|
||||
|
||||
simulator = Simulator(board=board, full_system=False)
|
||||
print("Beginning simulation!")
|
||||
simulator.run()
|
||||
Reference in New Issue
Block a user