merge

1 files changed, 13 insertions, 3 deletions

diff --git a/tests/configs/ruby_config.py b/tests/configs/ruby_config.py
index 7b8e27613..190337e67 100644
--- a/tests/configs/ruby_config.py
+++ b/tests/configs/ruby_config.py
@@ -4,17 +4,27 @@ import subprocess
 from os.path import dirname, join as joinpath
 
 import m5
+from m5.params import *
 
-def generate(config_file, cores=1, memories=1, memory_size=1024):
+def generate(config_file, cores=1, memories=1, memory_size=1024, \
+             cache_size=32768, cache_assoc=8, dmas=1,
+             ruby_tick='1t', ports_per_cpu=2, protocol='MOESI_CMP_directory'):
     default = joinpath(dirname(__file__), '../../src/mem/ruby/config')
     ruby_config = os.environ.get('RUBY_CONFIG', default)
     args = [ "ruby", "-I", ruby_config, joinpath(ruby_config, "print_cfg.rb"),
+             "-c", str(protocol),
              "-r", joinpath(ruby_config, config_file), "-p", str(cores),
-             "-m", str(memories), "-s", str(memory_size)]
+             "-m", str(memories), "-s", str(memory_size), "-C", str(cache_size),
+             "-A", str(cache_assoc), "-D", str(dmas)]
 
     temp_config = joinpath(m5.options.outdir, "ruby.config")
     ret = subprocess.call(args, stdout=file(temp_config, "w"))
     if ret != 0:
         raise RuntimeError, "subprocess failed!"
 
-    return m5.objects.RubyMemory(config_file=temp_config, num_cpus=cores)
+    return m5.objects.RubyMemory(clock = ruby_tick,
+                                 config_file = temp_config,
+                                 num_cpus = cores,
+                                 range = AddrRange(str(memory_size)+"MB"),
+                                 num_dmas = dmas,
+                                 ports_per_core = ports_per_cpu)