mem-garnet: Integration of HeteroGarnet
[gem5.git] / configs / example / garnet_synth_traffic.py
1 # Copyright (c) 2016 Georgia Institute of Technology
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 # Author: Tushar Krishna
28
29 from __future__ import print_function
30 from __future__ import absolute_import
31
32 import m5
33 from m5.objects import *
34 from m5.defines import buildEnv
35 from m5.util import addToPath
36 import os, optparse, sys
37
38 addToPath('../')
39
40 from common import Options
41 from ruby import Ruby
42
43 # Get paths we might need. It's expected this file is in m5/configs/example.
44 config_path = os.path.dirname(os.path.abspath(__file__))
45 config_root = os.path.dirname(config_path)
46 m5_root = os.path.dirname(config_root)
47
48 parser = optparse.OptionParser()
49 Options.addNoISAOptions(parser)
50
51 parser.add_option("--synthetic", type="choice", default="uniform_random",
52 choices=['uniform_random', 'tornado', 'bit_complement', \
53 'bit_reverse', 'bit_rotation', 'neighbor', \
54 'shuffle', 'transpose'])
55
56 parser.add_option("-i", "--injectionrate", type="float", default=0.1,
57 metavar="I",
58 help="Injection rate in packets per cycle per node. \
59 Takes decimal value between 0 to 1 (eg. 0.225). \
60 Number of digits after 0 depends upon --precision.")
61
62 parser.add_option("--precision", type="int", default=3,
63 help="Number of digits of precision after decimal point\
64 for injection rate")
65
66 parser.add_option("--sim-cycles", type="int", default=1000,
67 help="Number of simulation cycles")
68
69 parser.add_option("--num-packets-max", type="int", default=-1,
70 help="Stop injecting after --num-packets-max.\
71 Set to -1 to disable.")
72
73 parser.add_option("--single-sender-id", type="int", default=-1,
74 help="Only inject from this sender.\
75 Set to -1 to disable.")
76
77 parser.add_option("--single-dest-id", type="int", default=-1,
78 help="Only send to this destination.\
79 Set to -1 to disable.")
80
81 parser.add_option("--inj-vnet", type="int", default=-1,
82 help="Only inject in this vnet (0, 1 or 2).\
83 0 and 1 are 1-flit, 2 is 5-flit.\
84 Set to -1 to inject randomly in all vnets.")
85
86 #
87 # Add the ruby specific and protocol specific options
88 #
89 Ruby.define_options(parser)
90
91 (options, args) = parser.parse_args()
92
93 if args:
94 print("Error: script doesn't take any positional arguments")
95 sys.exit(1)
96
97
98 if options.inj_vnet > 2:
99 print("Error: Injection vnet %d should be 0 (1-flit), 1 (1-flit) "
100 "or 2 (5-flit) or -1 (random)" % (options.inj_vnet))
101 sys.exit(1)
102
103
104 cpus = [ GarnetSyntheticTraffic(
105 num_packets_max=options.num_packets_max,
106 single_sender=options.single_sender_id,
107 single_dest=options.single_dest_id,
108 sim_cycles=options.sim_cycles,
109 traffic_type=options.synthetic,
110 inj_rate=options.injectionrate,
111 inj_vnet=options.inj_vnet,
112 precision=options.precision,
113 num_dest=options.num_dirs) \
114 for i in range(options.num_cpus) ]
115
116 # create the desired simulated system
117 system = System(cpu = cpus, mem_ranges = [AddrRange(options.mem_size)])
118
119
120 # Create a top-level voltage domain and clock domain
121 system.voltage_domain = VoltageDomain(voltage = options.sys_voltage)
122
123 system.clk_domain = SrcClockDomain(clock = options.sys_clock,
124 voltage_domain = system.voltage_domain)
125
126 Ruby.create_system(options, False, system)
127
128 # Create a seperate clock domain for Ruby
129 system.ruby.clk_domain = SrcClockDomain(clock = options.ruby_clock,
130 voltage_domain = system.voltage_domain)
131
132 i = 0
133 for ruby_port in system.ruby._cpu_ports:
134 #
135 # Tie the cpu test ports to the ruby cpu port
136 #
137 cpus[i].test = ruby_port.slave
138 i += 1
139
140 # -----------------------
141 # run simulation
142 # -----------------------
143
144 root = Root(full_system = False, system = system)
145 root.system.mem_mode = 'timing'
146
147 # Not much point in this being higher than the L1 latency
148 m5.ticks.setGlobalFrequency('1ps')
149
150 # instantiate configuration
151 m5.instantiate()
152
153 # simulate until program terminates
154 exit_event = m5.simulate(options.abs_max_tick)
155
156 print('Exiting @ tick', m5.curTick(), 'because', exit_event.getCause())