diff options
| -rw-r--r-- | src/mem/ruby/network/simple/SimpleNetwork.py | 2 | ||||
| -rw-r--r-- | src/mem/ruby/network/simple/Switch.hh | 2 |
2 files changed, 3 insertions, 1 deletions
diff --git a/src/mem/ruby/network/simple/SimpleNetwork.py b/src/mem/ruby/network/simple/SimpleNetwork.py index 217dc20ec..8d0442b7d 100644 --- a/src/mem/ruby/network/simple/SimpleNetwork.py +++ b/src/mem/ruby/network/simple/SimpleNetwork.py @@ -42,6 +42,6 @@ class SimpleNetwork(RubyNetwork): class Switch(BasicRouter): type = 'Switch' - cxx_class = 'Switch' + cxx_header = 'mem/ruby/network/simple/Switch.hh' virt_nets = Param.Int(Parent.number_of_virtual_networks, "number of virtual networks") diff --git a/src/mem/ruby/network/simple/Switch.hh b/src/mem/ruby/network/simple/Switch.hh index 05e90f278..2757e6511 100644 --- a/src/mem/ruby/network/simple/Switch.hh +++ b/src/mem/ruby/network/simple/Switch.hh @@ -42,6 +42,8 @@ #include <iostream> #include <vector> +#include "mem/packet.hh" +#include "mem/ruby/common/TypeDefines.hh" #include "mem/ruby/network/BasicRouter.hh" #include "params/Switch.hh" |