gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Cleaned model building ; added some informative commented code

Browse Source
This commit is contained in:
Philippe Tillet
2014-10-13 03:38:19 +02:00
parent b8d339e54d
commit 6313f060cb
5 changed files with 93 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
autotune/python/dataset.py
View File

@@ -14,7 +14,7 @@ def resample(X, draw):
break
return x.astype(int)
def generate_dataset(TemplateType, execution_handler, nTuning, nDataPoints, compute_perf, draw):
def generate_dataset(TemplateType, execution_handler, nTuning, nDataPoints, draw):
print "Getting some good profiles..."
nDim = draw().size
@@ -39,7 +39,7 @@ def generate_dataset(TemplateType, execution_handler, nTuning, nDataPoints, comp
x = resample(X, draw)
for j,y in enumerate(profiles):
T = execution_handler(x, os.devnull, y)
Y[i,j] = compute_perf(x, T)
Y[i,j] = T
idx = np.argmax(Y[i,:])
X[i,:] = x
t = np.argmax(Y[:i+1,], axis=1)