Added postinstall script
This commit is contained in:
36
python/postinstall.py
Executable file
36
python/postinstall.py
Executable file
@@ -0,0 +1,36 @@
|
|||||||
|
import sys
|
||||||
|
|
||||||
|
def add_input(help, default):
|
||||||
|
sys.stdout.write(help + "[" + default + "] : ")
|
||||||
|
sys.stdout.flush()
|
||||||
|
return raw_input() or default
|
||||||
|
|
||||||
|
retune = add_input("No profile was found. Do you want to create one? (y/n)", 'y')
|
||||||
|
if retune=='y':
|
||||||
|
import subprocess
|
||||||
|
#Interactive
|
||||||
|
opts = []
|
||||||
|
subprocess.call(["${CMAKE_BINARY_DIR}/python/autotune/dist/autotune","list-devices"])
|
||||||
|
opts += ['--device'] + [add_input('Device to tune for','0')]
|
||||||
|
print '----------------'
|
||||||
|
opts += ['--operations'] + [add_input('Operations to tune for','vaxpy-float32,maxpy-float32,dot-float32,gemv-float32,gemm-float32')]
|
||||||
|
print '----------------'
|
||||||
|
opts += ['--gemv-layouts'] + [add_input('GEMV Layouts', 'N,T')]
|
||||||
|
print '----------------'
|
||||||
|
opts += ['--gemm-layouts'] + [add_input('GEMM Layouts', 'NN,NT,TN,TT')]
|
||||||
|
print '----------------'
|
||||||
|
print 'Methods:'
|
||||||
|
print 'simple: Tune the operations for a specific size.'
|
||||||
|
print 'full: Build input-dependent models for the operation (Better performance, but can take up to a day)'
|
||||||
|
opts += [add_input("Method:", 'simple')]
|
||||||
|
if opts[-1] == 'simple':
|
||||||
|
print '----------------'
|
||||||
|
opts += ['--blas1-size'] + [add_input('BLAS1 size', '10e6')]
|
||||||
|
print '----------------'
|
||||||
|
opts += ['--blas2-size'] + [add_input('BLAS2 sizes (M,N)', '2560,2560').split(',')]
|
||||||
|
print '----------------'
|
||||||
|
opts += ['--blas3-size'] + [add_input('BLAS3 sizes (M,N,K)', '1024,1024,1024').split(',')]
|
||||||
|
print '----------------'
|
||||||
|
subprocess.call(["${CMAKE_BINARY_DIR}/python/autotune/dist/autotune", "tune"] + opts)
|
||||||
|
|
||||||
|
|
Reference in New Issue
Block a user