#!/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 = ['', '', '' + title + '', '', '', '']
html += ['' + title + '
']
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 += ['', '']
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()