37 lines
1.7 KiB
Python
Executable File
37 lines
1.7 KiB
Python
Executable File
import os, 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 +['--out', os.environ['HOME'] + '/.atidlas/devices/device' + opts[1] + '.json'])
|
|
|
|
|