run.py revision 11851
1# Copyright (c) 2005-2007 The Regents of The University of Michigan
2# All rights reserved.
3#
4# Redistribution and use in source and binary forms, with or without
5# modification, are permitted provided that the following conditions are
6# met: redistributions of source code must retain the above copyright
7# notice, this list of conditions and the following disclaimer;
8# redistributions in binary form must reproduce the above copyright
9# notice, this list of conditions and the following disclaimer in the
10# documentation and/or other materials provided with the distribution;
11# neither the name of the copyright holders nor the names of its
12# contributors may be used to endorse or promote products derived from
13# this software without specific prior written permission.
14#
15# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
16# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
17# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
18# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
19# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
20# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
21# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
22# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
23# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
25# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26#
27# Authors: Ron Dreslinski
28
29# Splash2 Run Script
30#
31
32import os
33import optparse
34import sys
35
36import m5
37from m5.objects import *
38
39# --------------------
40# Define Command Line Options
41# ====================
42
43parser = optparse.OptionParser()
44
45parser.add_option("-d", "--detailed", action="store_true")
46parser.add_option("-t", "--timing", action="store_true")
47parser.add_option("-m", "--maxtick", type="int")
48parser.add_option("-n", "--numcpus",
49                  help="Number of cpus in total", type="int")
50parser.add_option("-f", "--frequency",
51                  default = "1GHz",
52                  help="Frequency of each CPU")
53parser.add_option("--l1size",
54                  default = "32kB")
55parser.add_option("--l1latency",
56                  default = "1ns")
57parser.add_option("--l2size",
58                  default = "256kB")
59parser.add_option("--l2latency",
60                  default = "10ns")
61parser.add_option("--rootdir",
62                  help="Root directory of Splash2",
63                  default="/dist/splash2/codes")
64parser.add_option("-b", "--benchmark",
65                  help="Splash 2 benchmark to run")
66
67(options, args) = parser.parse_args()
68
69if args:
70    print "Error: script doesn't take any positional arguments"
71    sys.exit(1)
72
73if not options.numcpus:
74    print "Specify the number of cpus with -n"
75    sys.exit(1)
76
77# --------------------
78# Define Splash2 Benchmarks
79# ====================
80class Cholesky(Process):
81    cwd = options.rootdir + '/kernels/cholesky'
82    executable = options.rootdir + '/kernels/cholesky/CHOLESKY'
83    cmd = ['CHOLESKY', '-p' +  str(options.numcpus),
84            options.rootdir + '/kernels/cholesky/inputs/tk23.O']
85
86class FFT(Process):
87    cwd = options.rootdir + '/kernels/fft'
88    executable = options.rootdir + '/kernels/fft/FFT'
89    cmd = ['FFT', '-p', str(options.numcpus), '-m18']
90
91class LU_contig(Process):
92    executable = options.rootdir + '/kernels/lu/contiguous_blocks/LU'
93    cmd = ['LU', '-p', str(options.numcpus)]
94    cwd = options.rootdir + '/kernels/lu/contiguous_blocks'
95
96class LU_noncontig(Process):
97    executable = options.rootdir + '/kernels/lu/non_contiguous_blocks/LU'
98    cmd = ['LU', '-p', str(options.numcpus)]
99    cwd = options.rootdir + '/kernels/lu/non_contiguous_blocks'
100
101class Radix(Process):
102    executable = options.rootdir + '/kernels/radix/RADIX'
103    cmd = ['RADIX', '-n524288', '-p', str(options.numcpus)]
104    cwd = options.rootdir + '/kernels/radix'
105
106class Barnes(Process):
107    executable = options.rootdir + '/apps/barnes/BARNES'
108    cmd = ['BARNES']
109    input = options.rootdir + '/apps/barnes/input.p' + str(options.numcpus)
110    cwd = options.rootdir + '/apps/barnes'
111
112class FMM(Process):
113    executable = options.rootdir + '/apps/fmm/FMM'
114    cmd = ['FMM']
115    if str(options.numcpus) == '1':
116        input = options.rootdir + '/apps/fmm/inputs/input.2048'
117    else:
118        input = options.rootdir + '/apps/fmm/inputs/input.2048.p' + str(options.numcpus)
119    cwd = options.rootdir + '/apps/fmm'
120
121class Ocean_contig(Process):
122    executable = options.rootdir + '/apps/ocean/contiguous_partitions/OCEAN'
123    cmd = ['OCEAN', '-p', str(options.numcpus)]
124    cwd = options.rootdir + '/apps/ocean/contiguous_partitions'
125
126class Ocean_noncontig(Process):
127    executable = options.rootdir + '/apps/ocean/non_contiguous_partitions/OCEAN'
128    cmd = ['OCEAN', '-p', str(options.numcpus)]
129    cwd = options.rootdir + '/apps/ocean/non_contiguous_partitions'
130
131class Raytrace(Process):
132    executable = options.rootdir + '/apps/raytrace/RAYTRACE'
133    cmd = ['RAYTRACE', '-p' + str(options.numcpus),
134           options.rootdir + '/apps/raytrace/inputs/teapot.env']
135    cwd = options.rootdir + '/apps/raytrace'
136
137class Water_nsquared(Process):
138    executable = options.rootdir + '/apps/water-nsquared/WATER-NSQUARED'
139    cmd = ['WATER-NSQUARED']
140    if options.numcpus==1:
141        input = options.rootdir + '/apps/water-nsquared/input'
142    else:
143        input = options.rootdir + '/apps/water-nsquared/input.p' + str(options.numcpus)
144    cwd = options.rootdir + '/apps/water-nsquared'
145
146class Water_spatial(Process):
147    executable = options.rootdir + '/apps/water-spatial/WATER-SPATIAL'
148    cmd = ['WATER-SPATIAL']
149    if options.numcpus==1:
150        input = options.rootdir + '/apps/water-spatial/input'
151    else:
152        input = options.rootdir + '/apps/water-spatial/input.p' + str(options.numcpus)
153    cwd = options.rootdir + '/apps/water-spatial'
154
155# --------------------
156# Base L1 Cache Definition
157# ====================
158
159class L1(Cache):
160    latency = options.l1latency
161    mshrs = 12
162    tgts_per_mshr = 8
163
164# ----------------------
165# Base L2 Cache Definition
166# ----------------------
167
168class L2(Cache):
169    latency = options.l2latency
170    mshrs = 92
171    tgts_per_mshr = 16
172    write_buffers = 8
173
174# ----------------------
175# Define the cpus
176# ----------------------
177
178busFrequency = Frequency(options.frequency)
179
180if options.timing:
181    cpus = [TimingSimpleCPU(cpu_id = i,
182                            clock=options.frequency)
183            for i in xrange(options.numcpus)]
184elif options.detailed:
185    cpus = [DerivO3CPU(cpu_id = i,
186                       clock=options.frequency)
187            for i in xrange(options.numcpus)]
188else:
189    cpus = [AtomicSimpleCPU(cpu_id = i,
190                            clock=options.frequency)
191            for i in xrange(options.numcpus)]
192
193# ----------------------
194# Create a system, and add system wide objects
195# ----------------------
196system = System(cpu = cpus, physmem = SimpleMemory(),
197                membus = SystemXBar(clock = busFrequency))
198system.clock = '1GHz'
199
200system.toL2bus = L2XBar(clock = busFrequency)
201system.l2 = L2(size = options.l2size, assoc = 8)
202
203# ----------------------
204# Connect the L2 cache and memory together
205# ----------------------
206
207system.physmem.port = system.membus.master
208system.l2.cpu_side = system.toL2bus.master
209system.l2.mem_side = system.membus.slave
210system.system_port = system.membus.slave
211
212# ----------------------
213# Connect the L2 cache and clusters together
214# ----------------------
215for cpu in cpus:
216    cpu.addPrivateSplitL1Caches(L1(size = options.l1size, assoc = 1),
217                                L1(size = options.l1size, assoc = 4))
218    # connect cpu level-1 caches to shared level-2 cache
219    cpu.connectAllPorts(system.toL2bus, system.membus)
220
221
222# ----------------------
223# Define the root
224# ----------------------
225
226root = Root(full_system = False, system = system)
227
228# --------------------
229# Pick the correct Splash2 Benchmarks
230# ====================
231if options.benchmark == 'Cholesky':
232    root.workload = Cholesky()
233elif options.benchmark == 'FFT':
234    root.workload = FFT()
235elif options.benchmark == 'LUContig':
236    root.workload = LU_contig()
237elif options.benchmark == 'LUNoncontig':
238    root.workload = LU_noncontig()
239elif options.benchmark == 'Radix':
240    root.workload = Radix()
241elif options.benchmark == 'Barnes':
242    root.workload = Barnes()
243elif options.benchmark == 'FMM':
244    root.workload = FMM()
245elif options.benchmark == 'OceanContig':
246    root.workload = Ocean_contig()
247elif options.benchmark == 'OceanNoncontig':
248    root.workload = Ocean_noncontig()
249elif options.benchmark == 'Raytrace':
250    root.workload = Raytrace()
251elif options.benchmark == 'WaterNSquared':
252    root.workload = Water_nsquared()
253elif options.benchmark == 'WaterSpatial':
254    root.workload = Water_spatial()
255else:
256    print >> sys.stderr, """The --benchmark environment variable was set to something improper.
257Use Cholesky, FFT, LUContig, LUNoncontig, Radix, Barnes, FMM, OceanContig,
258OceanNoncontig, Raytrace, WaterNSquared, or WaterSpatial"""
259    sys.exit(1)
260
261# --------------------
262# Assign the workload to the cpus
263# ====================
264
265for cpu in cpus:
266    cpu.workload = root.workload
267
268# ----------------------
269# Run the simulation
270# ----------------------
271
272if options.timing or options.detailed:
273    root.system.mem_mode = 'timing'
274
275# instantiate configuration
276m5.instantiate()
277
278# simulate until program terminates
279if options.maxtick:
280    exit_event = m5.simulate(options.maxtick)
281else:
282    exit_event = m5.simulate(m5.MaxTick)
283
284print 'Exiting @ tick', m5.curTick(), 'because', exit_event.getCause()
285
286