mirror of
https://github.com/andreas-abel/nanoBench.git
synced 2025-12-15 19:10:08 +01:00
132 lines
5.9 KiB
Python
Executable File
132 lines
5.9 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import argparse
|
|
import os
|
|
import random
|
|
|
|
from plotly.offline import plot
|
|
import plotly.graph_objects as go
|
|
|
|
from cacheLib import *
|
|
|
|
import logging
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description='Tests if the L3 cache uses set dueling')
|
|
parser.add_argument("-nRuns", help="Maximum number of runs", type=int, default=25)
|
|
parser.add_argument("-loop", help="Loop count", type=int, default=10)
|
|
parser.add_argument("-length", help="Length of the acc. seq. (Default: associativity+1)", type=int)
|
|
parser.add_argument("-noClearHL", help="Do not clear higher levels", action='store_true')
|
|
parser.add_argument("-nMeasurements", help="Number of measurements", type=int, default=10)
|
|
parser.add_argument("-output", help="Output file name", default='setDueling.html')
|
|
parser.add_argument("-logLevel", help="Log level (DEBUG, INFO, WARNING, ERROR, CRITICAL)", default='INFO')
|
|
args = parser.parse_args()
|
|
|
|
logging.basicConfig(stream=sys.stdout, format='%(message)s', level=logging.getLevelName(args.logLevel))
|
|
|
|
assoc = getCacheInfo(3).assoc
|
|
nL3Sets = getCacheInfo(3).nSets
|
|
nL2Sets = getCacheInfo(2).nSets
|
|
lineSize = getCacheInfo(1).lineSize
|
|
nCBoxes = max(1, getNCBoxUnits())
|
|
nSlicesPerCBox = 1
|
|
if getCacheInfo(3).nSlices:
|
|
nSlicesPerCBox = getCacheInfo(3).nSlices // getCacheInfo(3).nCboxes
|
|
|
|
seqLength = (args.length if args.length is not None else assoc+1)
|
|
seq = ' '.join('B' + str(i) + '?' for i in range(0, seqLength))
|
|
hitSeq = ' '.join('B' + str(i) for i in range(0, assoc))
|
|
missSeq = ' '.join('B' + str(i) for i in range(0, 3*assoc))
|
|
|
|
title = cpuid.cpu_name(cpuid.CPUID()) + ', L3 Hits'
|
|
html = ['<html>', '<head>', '<title>' + title + '</title>', '<script src="https://cdn.plot.ly/plotly-latest.min.js">', '</script>', '</head>', '<body>']
|
|
html += ['<h3>' + title + '</h3>']
|
|
|
|
setsForSlice = {cBox: {cSlice: list(range(0,nL3Sets)) for cSlice in range(0, nSlicesPerCBox)} for cBox in range(0, nCBoxes)}
|
|
L3HitsDict = {cBox: {cSlice: [[] for s in range(0, nL3Sets)] for cSlice in range(0, nSlicesPerCBox)} for cBox in range(0, nCBoxes)}
|
|
|
|
prevOti = ''
|
|
i = -1
|
|
notChanged = -1
|
|
while notChanged < 10:
|
|
for doNotUseOtherCBoxes in ([False, True] if (not args.noClearHL and nL2Sets < nL3Sets) else [False]):
|
|
for useHitSeq in [False, True]:
|
|
i += 1
|
|
notChanged += 1
|
|
for cBox in range(0, nCBoxes):
|
|
for cSlice in range(0, nSlicesPerCBox):
|
|
curSets = setsForSlice[cBox][cSlice]
|
|
random.shuffle(curSets)
|
|
prevSets = curSets[:]
|
|
|
|
for si, s in enumerate(prevSets):
|
|
codeSet = (s + random.randint(1, nL3Sets - 100)) % nL3Sets
|
|
codeOffset = lineSize * codeSet
|
|
L3Hits = L3HitsDict[cBox][cSlice][s]
|
|
|
|
ec = getCodeForCacheExperiment(3, seq, '', [s], cBox, cSlice, (not args.noClearHL), doNotUseOtherCBoxes, True)
|
|
nb = runCacheExperimentCode(ec.code, ec.init, prevOti + ec.oneTimeInit, loop=args.loop, warmUpCount=0, codeOffset=codeOffset,
|
|
nMeasurements=args.nMeasurements, agg='med')
|
|
|
|
if nb['L1_MISS'] < seqLength - .2:
|
|
print('Hit in L1')
|
|
continue
|
|
|
|
if nb['L2_MISS'] < seqLength - .2:
|
|
print('Hit in L2')
|
|
continue
|
|
|
|
L3Hits.append(nb['L3_HIT'])
|
|
L3Hits.sort()
|
|
|
|
L3HitsStr = str(L3Hits) if len(L3Hits) <= 5 else '[%s, %s, ..., %s, %s]' % (L3Hits[0], L3Hits[1], L3Hits[-2], L3Hits[-1])
|
|
log.info('CBox ' + str(cBox) + ', slice: ' + str(cSlice) + ', run ' + str(i) + ', set: ' + str(si+1) + '/' + str(len(prevSets)) +
|
|
' (' + str(s) + '), ' + L3HitsStr)
|
|
|
|
if len(L3Hits) > 1:
|
|
if L3Hits[-1]-L3Hits[0] > 1:
|
|
curSets.remove(s)
|
|
notChanged = 0
|
|
else:
|
|
if useHitSeq:
|
|
ec = getCodeForCacheExperiment(3, hitSeq, '', [s], cBox, cSlice, (not args.noClearHL), doNotUseOtherCBoxes, True)
|
|
else:
|
|
ec = getCodeForCacheExperiment(3, missSeq, '', [s], cBox, cSlice, (not args.noClearHL), doNotUseOtherCBoxes, True)
|
|
prevOti = ec.oneTimeInit + 'mov R15, 100; pLoop:' + ec.code + '; dec R15; jnz pLoop; '
|
|
|
|
for cBox in range(0, nCBoxes):
|
|
for cSlice in range(0, nSlicesPerCBox):
|
|
fig = go.Figure()
|
|
|
|
title = 'CBox ' + str(cBox)
|
|
if nSlicesPerCBox > 1: title += ', Slice: ' + str(cSlice)
|
|
title += ', Sequence (accessed ' + str(args.loop) + ' times in each set): ' + seq
|
|
fig.update_layout(title_text=title)
|
|
|
|
fig.update_layout(showlegend=True)
|
|
fig.update_xaxes(title_text='Set')
|
|
|
|
yValuesMinMax = [min(s) + (max(s)-min(s))/2 for s in L3HitsDict[cBox][cSlice] if s]
|
|
fig.add_trace(go.Scatter(y=yValuesMinMax, mode='lines+markers', name='Min+(Max-Min)/2'))
|
|
|
|
yValuesMin = [min(s) for s in L3HitsDict[cBox][cSlice] if s]
|
|
fig.add_trace(go.Scatter(y=yValuesMin, mode='lines+markers', visible = 'legendonly', name='Min'))
|
|
|
|
yValuesMax = [max(s) for s in L3HitsDict[cBox][cSlice] if s]
|
|
fig.add_trace(go.Scatter(y=yValuesMax, mode='lines+markers', visible = 'legendonly', name='Max'))
|
|
|
|
html.append(plot(fig, include_plotlyjs=False, output_type='div'))
|
|
|
|
html += ['</body>', '</html>']
|
|
|
|
with open(args.output ,'w') as f:
|
|
f.write('\n'.join(html))
|
|
print('Output written to ' + args.output)
|
|
os.chown(args.output, int(os.environ['SUDO_UID']), int(os.environ['SUDO_GID']))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|