diff --git a/configs/example/gem5_library/riscv-rvv-example.py b/configs/example/gem5_library/riscv-rvv-example.py new file mode 100755 index 0000000000..57a6fd7afd --- /dev/null +++ b/configs/example/gem5_library/riscv-rvv-example.py @@ -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] + +""" + +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() diff --git a/tests/gem5/se_mode/rvv_intrinsic_tests/test.py b/tests/gem5/se_mode/rvv_intrinsic_tests/test.py new file mode 100644 index 0000000000..e20018ba60 --- /dev/null +++ b/tests/gem5/se_mode/rvv_intrinsic_tests/test.py @@ -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, + )