SE script and tests for risc-v's vector extension (#1542)
This two commits add the SE config and test script, respectively, to run the rvv tests mentioned in #1246.
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()
|
||||||
63
tests/gem5/se_mode/rvv_intrinsic_tests/test.py
Normal file
63
tests/gem5/se_mode/rvv_intrinsic_tests/test.py
Normal file
@@ -0,0 +1,63 @@
|
|||||||
|
# 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.
|
||||||
|
|
||||||
|
import os
|
||||||
|
import re
|
||||||
|
import sys
|
||||||
|
|
||||||
|
from testlib import *
|
||||||
|
|
||||||
|
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",
|
||||||
|
]
|
||||||
|
|
||||||
|
vlens = [2**x for x in range(7, 15)]
|
||||||
|
|
||||||
|
for resource in resources:
|
||||||
|
out_verifier = verifier.MatchRegex(re.compile(f"^.*{resource}: pass$"))
|
||||||
|
|
||||||
|
for vlen in vlens:
|
||||||
|
gem5_verify_config(
|
||||||
|
name=f"test-riscv-{resource}-vlen_{vlen}-O3-se-mode",
|
||||||
|
fixtures=(),
|
||||||
|
verifiers=(out_verifier,),
|
||||||
|
config=f"{config.base_dir}/configs/example/gem5_library/riscv-rvv-example.py",
|
||||||
|
config_args=[resource, f"--vlen={vlen}"],
|
||||||
|
valid_isas=(constants.all_compiled_tag,),
|
||||||
|
length=constants.quick_tag,
|
||||||
|
)
|
||||||
Reference in New Issue
Block a user