run.py revision 3360
1# Copyright (c) 2005-2006 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 m5
33from m5.objects import *
34import os, optparse, sys
35m5.AddToPath('../common')
36
37# --------------------
38# Define Command Line Options
39# ====================
40
41parser = optparse.OptionParser()
42
43parser.add_option("-d", "--detailed", action="store_true")
44parser.add_option("-t", "--timing", action="store_true")
45parser.add_option("-m", "--maxtick", type="int")
46parser.add_option("-n", "--numcpus",
47                  help="Number of cpus in total", type="int")
48parser.add_option("-f", "--frequency",
49                  default = "1GHz",
50                  help="Frequency of each CPU")
51parser.add_option("-p", "--protocol",
52                  default="moesi",
53                  help="The coherence protocol to use for the L1'a (i.e. MOESI, MOSI)")
54parser.add_option("--l1size",
55                  default = "32kB")
56parser.add_option("--l1latency",
57                  default = 1)
58parser.add_option("--l2size",
59                  default = "256kB")
60parser.add_option("--l2latency",
61                  default = 10)
62parser.add_option("--rootdir",
63                  help="ROot directory of Splash2",
64                  default="/dist/splash2/codes")
65parser.add_option("-b", "--benchmark",
66                  help="Splash 2 benchmark to run")
67
68(options, args) = parser.parse_args()
69
70if args:
71    print "Error: script doesn't take any positional arguments"
72    sys.exit(1)
73
74if not options.numcpus:
75    print "Specify the number of cpus with -n"
76    sys.exit(1)
77
78# --------------------
79# Define Splash2 Benchmarks
80# ====================
81class Cholesky(LiveProcess):
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(LiveProcess):
87        executable = options.rootdir + '/kernels/fft/FFT'
88        cmd = 'FFT -p' + str(options.numcpus) + ' -m18'
89
90class LU_contig(LiveProcess):
91        executable = options.rootdir + '/kernels/lu/contiguous_blocks/LU'
92        cmd = 'LU -p' + str(options.numcpus)
93
94class LU_noncontig(LiveProcess):
95        executable = options.rootdir + '/kernels/lu/non_contiguous_blocks/LU'
96        cmd = 'LU -p' + str(options.numcpus)
97
98class Radix(LiveProcess):
99        executable = options.rootdir + '/kernels/radix/RADIX'
100        cmd = 'RADIX -n524288 -p' + str(options.numcpus)
101
102class Barnes(LiveProcess):
103        executable = options.rootdir + '/apps/barnes/BARNES'
104        cmd = 'BARNES'
105        input = options.rootdir + '/apps/barnes/input.p' + str(options.numcpus)
106
107class FMM(LiveProcess):
108        executable = options.rootdir + '/apps/fmm/FMM'
109        cmd = 'FMM'
110        input = options.rootdir + '/apps/fmm/inputs/input.2048.p' + str(options.numcpus)
111
112class Ocean_contig(LiveProcess):
113        executable = options.rootdir + '/apps/ocean/contiguous_partitions/OCEAN'
114        cmd = 'OCEAN -p' + str(options.numcpus)
115
116class Ocean_noncontig(LiveProcess):
117        executable = options.rootdir + '/apps/ocean/non_contiguous_partitions/OCEAN'
118        cmd = 'OCEAN -p' + str(options.numcpus)
119
120class Raytrace(LiveProcess):
121        executable = options.rootdir + '/apps/raytrace/RAYTRACE'
122        cmd = 'RAYTRACE -p' + str(options.numcpus) + ' ' \
123             + options.rootdir + 'apps/raytrace/inputs/teapot.env'
124
125class Water_nsquared(LiveProcess):
126        executable = options.rootdir + '/apps/water-nsquared/WATER-NSQUARED'
127        cmd = 'WATER-NSQUARED'
128        input = options.rootdir + '/apps/water-nsquared/input.p' + str(options.numcpus)
129
130class Water_spatial(LiveProcess):
131        executable = options.rootdir + '/apps/water-spatial/WATER-SPATIAL'
132        cmd = 'WATER-SPATIAL'
133        input = options.rootdir + '/apps/water-spatial/input.p' + str(options.numcpus)
134
135
136# --------------------
137# Base L1 Cache Definition
138# ====================
139
140class L1(BaseCache):
141    latency = options.l1latency
142    block_size = 64
143    mshrs = 12
144    tgts_per_mshr = 8
145    protocol = CoherenceProtocol(protocol=options.protocol)
146
147# ----------------------
148# Base L2 Cache Definition
149# ----------------------
150
151class L2(BaseCache):
152    block_size = 64
153    latency = options.l2latency
154    mshrs = 92
155    tgts_per_mshr = 16
156    write_buffers = 8
157
158# ----------------------
159# Define the cpus
160# ----------------------
161
162busFrequency = Frequency(options.frequency)
163
164if options.timing:
165    cpus = [TimingSimpleCPU(cpu_id = i,
166                            clock=options.frequency)
167            for i in xrange(options.numcpus)]
168elif options.detailed:
169    cpus = [DerivO3CPU(cpu_id = i,
170                       clock=options.frequency)
171            for i in xrange(options.numcpus)]
172else:
173    cpus = [AtomicSimpleCPU(cpu_id = i,
174                            clock=options.frequency)
175            for i in xrange(options.numcpus)]
176
177# ----------------------
178# Create a system, and add system wide objects
179# ----------------------
180system = System(cpu = cpus, physmem = PhysicalMemory(),
181                membus = Bus(clock = busFrequency))
182
183system.toL2bus = Bus(clock = busFrequency)
184system.l2 = L2(size = options.l2size, assoc = 8)
185
186# ----------------------
187# Connect the L2 cache and memory together
188# ----------------------
189
190system.physmem.port = system.membus.port
191system.l2.cpu_side = system.toL2bus.port
192system.l2.mem_side = system.membus.port
193
194# ----------------------
195# Connect the L2 cache and clusters together
196# ----------------------
197for cpu in cpus:
198    cpu.addPrivateSplitL1Caches(L1(size = options.l1size, assoc = 1),
199                                L1(size = options.l1size, assoc = 4))
200    cpu.mem = cpu.dcache
201    # connect cpu level-1 caches to shared level-2 cache
202    cpu.connectMemPorts(system.toL2bus)
203
204
205# ----------------------
206# Define the root
207# ----------------------
208
209root = Root(system = system)
210
211# --------------------
212# Pick the correct Splash2 Benchmarks
213# ====================
214if options.benchmark == 'Cholesky':
215    root.workload = Cholesky()
216elif options.benchmark == 'FFT':
217    root.workload = FFT()
218elif options.benchmark == 'LUContig':
219    root.workload = LU_contig()
220elif options.benchmark == 'LUNoncontig':
221    root.workload = LU_noncontig()
222elif options.benchmark == 'Radix':
223    root.workload = Radix()
224elif options.benchmark == 'Barnes':
225    root.workload = Barnes()
226elif options.benchmark == 'FMM':
227    root.workload = FMM()
228elif options.benchmark == 'OceanContig':
229    root.workload = Ocean_contig()
230elif options.benchmark == 'OceanNoncontig':
231    root.workload = Ocean_noncontig()
232elif options.benchmark == 'Raytrace':
233    root.workload = Raytrace()
234elif options.benchmark == 'WaterNSquared':
235    root.workload = Water_nsquared()
236elif options.benchmark == 'WaterSpatial':
237    root.workload = Water_spatial()
238else:
239    panic("The --benchmark environment variable was set to something" \
240          +" improper.\nUse Cholesky, FFT, LUContig, LUNoncontig, Radix" \
241          +", Barnes, FMM, OceanContig,\nOceanNoncontig, Raytrace," \
242          +" WaterNSquared, or WaterSpatial\n")
243
244# --------------------
245# Assign the workload to the cpus
246# ====================
247
248for cpu in cpus:
249    cpu.workload = root.workload
250
251# ----------------------
252# Run the simulation
253# ----------------------
254
255if options.timing or options.detailed:
256    root.system.mem_mode = 'timing'
257
258# instantiate configuration
259m5.instantiate(root)
260
261# simulate until program terminates
262if options.maxtick:
263    exit_event = m5.simulate(options.maxtick)
264else:
265    exit_event = m5.simulate()
266
267print 'Exiting @ tick', m5.curTick(), 'because', exit_event.getCause()
268
269