1c1
< # Copyright (c) 2014-2015 ARM Limited
---
> # Copyright (c) 2014-2015, 2018 ARM Limited
39a40
> import math
57a59,63
> dram_generators = {
> "DRAM" : lambda x: x.createDram,
> "DRAM_ROTATE" : lambda x: x.createDramRot,
> }
>
70c76
< choices=["DRAM", "DRAM_ROTATE"],
---
> choices=dram_generators.keys(),
130,134d135
< # this is where we go off piste, and print the traffic generator
< # configuration that we will later use, crazy but it works
< cfg_file_name = "configs/dram/sweep.cfg"
< cfg_file = open(cfg_file_name, 'w')
<
162,185d162
< # now we create the state by iterating over the stride size from burst
< # size to the max stride, and from using only a single bank up to the
< # number of banks available
< nxt_state = 0
< for bank in range(1, nbr_banks + 1):
< for stride_size in range(burst_size, max_stride + 1, burst_size):
< cfg_file.write("STATE %d %d %s %d 0 %d %d "
< "%d %d %d %d %d %d %d %d %d\n" %
< (nxt_state, period, options.mode, options.rd_perc,
< max_addr, burst_size, itt, itt, 0, stride_size,
< page_size, nbr_banks, bank, options.addr_map,
< options.mem_ranks))
< nxt_state = nxt_state + 1
<
< cfg_file.write("INIT 0\n")
<
< # go through the states one by one
< for state in range(1, nxt_state):
< cfg_file.write("TRANSITION %d %d 1\n" % (state - 1, state))
<
< cfg_file.write("TRANSITION %d %d 1\n" % (nxt_state - 1, nxt_state - 1))
<
< cfg_file.close()
<
187c164
< system.tgen = TrafficGen(config_file = cfg_file_name)
---
> system.tgen = PyTrafficGen()
207d183
< m5.simulate(nxt_state * period)
208a185,200
> def trace():
> generator = dram_generators[options.mode](system.tgen)
> for bank in range(1, nbr_banks + 1):
> for stride_size in range(burst_size, max_stride + 1, burst_size):
> num_seq_pkts = int(math.ceil(float(stride_size) / burst_size))
> yield generator(period,
> 0, max_addr, burst_size, int(itt), int(itt),
> options.rd_perc, 0,
> num_seq_pkts, page_size, nbr_banks, bank,
> options.addr_map, options.mem_ranks)
> yield system.tgen.createExit(0)
>
> system.tgen.start(trace())
>
> m5.simulate()
>