3bdd58ccb4
This patch prevents the body of the script getting executed when the script is imported as a module. Change-Id: I70a50f6295f1e7a088398017f5fa9d06fe90476a Reviewed-by: Andreas Sandberg <andreas.sandberg@arm.com>
79 lines
3.3 KiB
Python
79 lines
3.3 KiB
Python
# Copyright (c) 2015 The University of Illinois Urbana Champaign
|
|
# 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.
|
|
#
|
|
# Authors: Mohammad Alian
|
|
|
|
# This is an example of an n port network switch to work in dist-gem5.
|
|
# Users can extend this to have different different topologies
|
|
|
|
import optparse
|
|
import sys
|
|
|
|
import m5
|
|
from m5.defines import buildEnv
|
|
from m5.objects import *
|
|
from m5.util import addToPath, fatal
|
|
|
|
addToPath('../')
|
|
|
|
from common import Simulation
|
|
from common import Options
|
|
|
|
def build_switch(options):
|
|
# instantiate an EtherSwitch
|
|
switch = EtherSwitch()
|
|
# instantiate distEtherLinks to connect switch ports
|
|
# to other gem5 instances
|
|
switch.portlink = [DistEtherLink(speed = options.ethernet_linkspeed,
|
|
delay = options.ethernet_linkdelay,
|
|
dist_rank = options.dist_rank,
|
|
dist_size = options.dist_size,
|
|
server_name = options.dist_server_name,
|
|
server_port = options.dist_server_port,
|
|
sync_start = options.dist_sync_start,
|
|
sync_repeat = options.dist_sync_repeat,
|
|
is_switch = True,
|
|
num_nodes = options.dist_size)
|
|
for i in xrange(options.dist_size)]
|
|
|
|
for (i, link) in enumerate(switch.portlink):
|
|
link.int0 = switch.interface[i]
|
|
|
|
return switch
|
|
|
|
def main():
|
|
# Add options
|
|
parser = optparse.OptionParser()
|
|
Options.addCommonOptions(parser)
|
|
Options.addFSOptions(parser)
|
|
(options, args) = parser.parse_args()
|
|
|
|
system = build_switch(options)
|
|
root = Root(full_system = True, system = system)
|
|
Simulation.run(options, root, None, None)
|
|
|
|
if __name__ == "__m5_main__":
|
|
main()
|