a2c9213a31
JIRA: https://gem5.atlassian.net/browse/GEM5-543 Change-Id: I997d6a4e45319a74e21bd0d61d4af6118474c849 Signed-off-by: Giacomo Travaglini <giacomo.travaglini@arm.com> Reviewed-on: https://gem5-review.googlesource.com/c/public/gem5/+/44513 Reviewed-by: Daniel Carvalho <odanrc@yahoo.com.br> Tested-by: kokoro <noreply+kokoro@google.com>
283 lines
9.3 KiB
Python
283 lines
9.3 KiB
Python
# Copyright (c) 2005-2007 The Regents of The University of Michigan
|
|
# 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.
|
|
|
|
# Splash2 Run Script
|
|
#
|
|
|
|
import os
|
|
import argparse
|
|
import sys
|
|
|
|
import m5
|
|
from m5.objects import *
|
|
|
|
# --------------------
|
|
# Define Command Line Options
|
|
# ====================
|
|
|
|
parser = argparse.ArgumentParser()
|
|
|
|
parser.add_argument("-d", "--detailed", action="store_true")
|
|
parser.add_argument("-t", "--timing", action="store_true")
|
|
parser.add_argument("-m", "--maxtick", type=int)
|
|
parser.add_argument("-n", "--numcpus",
|
|
help="Number of cpus in total", type=int)
|
|
parser.add_argument("-f", "--frequency",
|
|
default = "1GHz",
|
|
help="Frequency of each CPU")
|
|
parser.add_argument("--l1size",
|
|
default = "32kB")
|
|
parser.add_argument("--l1latency",
|
|
default = "1ns")
|
|
parser.add_argument("--l2size",
|
|
default = "256kB")
|
|
parser.add_argument("--l2latency",
|
|
default = "10ns")
|
|
parser.add_argument("--rootdir",
|
|
help="Root directory of Splash2",
|
|
default="/dist/splash2/codes")
|
|
parser.add_argument("-b", "--benchmark",
|
|
help="Splash 2 benchmark to run")
|
|
|
|
args = parser.parse_args()
|
|
|
|
if not args.numcpus:
|
|
print("Specify the number of cpus with -n")
|
|
sys.exit(1)
|
|
|
|
# --------------------
|
|
# Define Splash2 Benchmarks
|
|
# ====================
|
|
class Cholesky(Process):
|
|
cwd = args.rootdir + '/kernels/cholesky'
|
|
executable = args.rootdir + '/kernels/cholesky/CHOLESKY'
|
|
cmd = ['CHOLESKY', '-p' + str(args.numcpus),
|
|
args.rootdir + '/kernels/cholesky/inputs/tk23.O']
|
|
|
|
class FFT(Process):
|
|
cwd = args.rootdir + '/kernels/fft'
|
|
executable = args.rootdir + '/kernels/fft/FFT'
|
|
cmd = ['FFT', '-p', str(args.numcpus), '-m18']
|
|
|
|
class LU_contig(Process):
|
|
executable = args.rootdir + '/kernels/lu/contiguous_blocks/LU'
|
|
cmd = ['LU', '-p', str(args.numcpus)]
|
|
cwd = args.rootdir + '/kernels/lu/contiguous_blocks'
|
|
|
|
class LU_noncontig(Process):
|
|
executable = args.rootdir + '/kernels/lu/non_contiguous_blocks/LU'
|
|
cmd = ['LU', '-p', str(args.numcpus)]
|
|
cwd = args.rootdir + '/kernels/lu/non_contiguous_blocks'
|
|
|
|
class Radix(Process):
|
|
executable = args.rootdir + '/kernels/radix/RADIX'
|
|
cmd = ['RADIX', '-n524288', '-p', str(args.numcpus)]
|
|
cwd = args.rootdir + '/kernels/radix'
|
|
|
|
class Barnes(Process):
|
|
executable = args.rootdir + '/apps/barnes/BARNES'
|
|
cmd = ['BARNES']
|
|
input = args.rootdir + '/apps/barnes/input.p' + str(args.numcpus)
|
|
cwd = args.rootdir + '/apps/barnes'
|
|
|
|
class FMM(Process):
|
|
executable = args.rootdir + '/apps/fmm/FMM'
|
|
cmd = ['FMM']
|
|
if str(args.numcpus) == '1':
|
|
input = args.rootdir + '/apps/fmm/inputs/input.2048'
|
|
else:
|
|
input = args.rootdir + '/apps/fmm/inputs/input.2048.p' + str(args.numcpus)
|
|
cwd = args.rootdir + '/apps/fmm'
|
|
|
|
class Ocean_contig(Process):
|
|
executable = args.rootdir + '/apps/ocean/contiguous_partitions/OCEAN'
|
|
cmd = ['OCEAN', '-p', str(args.numcpus)]
|
|
cwd = args.rootdir + '/apps/ocean/contiguous_partitions'
|
|
|
|
class Ocean_noncontig(Process):
|
|
executable = args.rootdir + '/apps/ocean/non_contiguous_partitions/OCEAN'
|
|
cmd = ['OCEAN', '-p', str(args.numcpus)]
|
|
cwd = args.rootdir + '/apps/ocean/non_contiguous_partitions'
|
|
|
|
class Raytrace(Process):
|
|
executable = args.rootdir + '/apps/raytrace/RAYTRACE'
|
|
cmd = ['RAYTRACE', '-p' + str(args.numcpus),
|
|
args.rootdir + '/apps/raytrace/inputs/teapot.env']
|
|
cwd = args.rootdir + '/apps/raytrace'
|
|
|
|
class Water_nsquared(Process):
|
|
executable = args.rootdir + '/apps/water-nsquared/WATER-NSQUARED'
|
|
cmd = ['WATER-NSQUARED']
|
|
if args.numcpus==1:
|
|
input = args.rootdir + '/apps/water-nsquared/input'
|
|
else:
|
|
input = args.rootdir + '/apps/water-nsquared/input.p' + str(args.numcpus)
|
|
cwd = args.rootdir + '/apps/water-nsquared'
|
|
|
|
class Water_spatial(Process):
|
|
executable = args.rootdir + '/apps/water-spatial/WATER-SPATIAL'
|
|
cmd = ['WATER-SPATIAL']
|
|
if args.numcpus==1:
|
|
input = args.rootdir + '/apps/water-spatial/input'
|
|
else:
|
|
input = args.rootdir + '/apps/water-spatial/input.p' + str(args.numcpus)
|
|
cwd = args.rootdir + '/apps/water-spatial'
|
|
|
|
# --------------------
|
|
# Base L1 Cache Definition
|
|
# ====================
|
|
|
|
class L1(Cache):
|
|
latency = args.l1latency
|
|
mshrs = 12
|
|
tgts_per_mshr = 8
|
|
|
|
# ----------------------
|
|
# Base L2 Cache Definition
|
|
# ----------------------
|
|
|
|
class L2(Cache):
|
|
latency = args.l2latency
|
|
mshrs = 92
|
|
tgts_per_mshr = 16
|
|
write_buffers = 8
|
|
|
|
# ----------------------
|
|
# Define the cpus
|
|
# ----------------------
|
|
|
|
busFrequency = Frequency(args.frequency)
|
|
|
|
if args.timing:
|
|
cpus = [TimingSimpleCPU(cpu_id = i,
|
|
clock=args.frequency)
|
|
for i in range(args.numcpus)]
|
|
elif args.detailed:
|
|
cpus = [DerivO3CPU(cpu_id = i,
|
|
clock=args.frequency)
|
|
for i in range(args.numcpus)]
|
|
else:
|
|
cpus = [AtomicSimpleCPU(cpu_id = i,
|
|
clock=args.frequency)
|
|
for i in range(args.numcpus)]
|
|
|
|
# ----------------------
|
|
# Create a system, and add system wide objects
|
|
# ----------------------
|
|
system = System(cpu = cpus, physmem = SimpleMemory(),
|
|
membus = SystemXBar(clock = busFrequency))
|
|
system.clock = '1GHz'
|
|
|
|
system.toL2bus = L2XBar(clock = busFrequency)
|
|
system.l2 = L2(size = args.l2size, assoc = 8)
|
|
|
|
# ----------------------
|
|
# Connect the L2 cache and memory together
|
|
# ----------------------
|
|
|
|
system.physmem.port = system.membus.master
|
|
system.l2.cpu_side = system.toL2bus.master
|
|
system.l2.mem_side = system.membus.slave
|
|
system.system_port = system.membus.slave
|
|
|
|
# ----------------------
|
|
# Connect the L2 cache and clusters together
|
|
# ----------------------
|
|
for cpu in cpus:
|
|
cpu.addPrivateSplitL1Caches(L1(size = args.l1size, assoc = 1),
|
|
L1(size = args.l1size, assoc = 4))
|
|
# connect cpu level-1 caches to shared level-2 cache
|
|
cpu.connectAllPorts(system.toL2bus, system.membus)
|
|
|
|
|
|
# ----------------------
|
|
# Define the root
|
|
# ----------------------
|
|
|
|
root = Root(full_system = False, system = system)
|
|
|
|
# --------------------
|
|
# Pick the correct Splash2 Benchmarks
|
|
# ====================
|
|
if args.benchmark == 'Cholesky':
|
|
root.workload = Cholesky()
|
|
elif args.benchmark == 'FFT':
|
|
root.workload = FFT()
|
|
elif args.benchmark == 'LUContig':
|
|
root.workload = LU_contig()
|
|
elif args.benchmark == 'LUNoncontig':
|
|
root.workload = LU_noncontig()
|
|
elif args.benchmark == 'Radix':
|
|
root.workload = Radix()
|
|
elif args.benchmark == 'Barnes':
|
|
root.workload = Barnes()
|
|
elif args.benchmark == 'FMM':
|
|
root.workload = FMM()
|
|
elif args.benchmark == 'OceanContig':
|
|
root.workload = Ocean_contig()
|
|
elif args.benchmark == 'OceanNoncontig':
|
|
root.workload = Ocean_noncontig()
|
|
elif args.benchmark == 'Raytrace':
|
|
root.workload = Raytrace()
|
|
elif args.benchmark == 'WaterNSquared':
|
|
root.workload = Water_nsquared()
|
|
elif args.benchmark == 'WaterSpatial':
|
|
root.workload = Water_spatial()
|
|
else:
|
|
print("The --benchmark environment variable was set to something "
|
|
"improper. Use Cholesky, FFT, LUContig, LUNoncontig, Radix, "
|
|
"Barnes, FMM, OceanContig, OceanNoncontig, Raytrace, WaterNSquared, "
|
|
"or WaterSpatial", file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
# --------------------
|
|
# Assign the workload to the cpus
|
|
# ====================
|
|
|
|
for cpu in cpus:
|
|
cpu.workload = root.workload
|
|
|
|
system.workload = SEWorkload.init_compatible(root.workload.executable)
|
|
|
|
# ----------------------
|
|
# Run the simulation
|
|
# ----------------------
|
|
|
|
if args.timing or args.detailed:
|
|
root.system.mem_mode = 'timing'
|
|
|
|
# instantiate configuration
|
|
m5.instantiate()
|
|
|
|
# simulate until program terminates
|
|
if args.maxtick:
|
|
exit_event = m5.simulate(args.maxtick)
|
|
else:
|
|
exit_event = m5.simulate(m5.MaxTick)
|
|
|
|
print('Exiting @ tick', m5.curTick(), 'because', exit_event.getCause())
|
|
|