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

Auto-tuner: Renamed "json_file" to "out"

Browse Source
This commit is contained in:
Philippe Tillet
2015-02-01 19:41:45 -05:00
parent 3b61842528
commit f0bb130416
2 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
python/autotune/pysrc/autotune.py
View File

@@ -35,8 +35,8 @@ def do_tuning(args):
device = args.device
context = atd.context(device)
context.queues.append(atd.command_queue(context, device))
if os.path.isfile(args.json_file):
json_out = json.load(open(args.json_file, 'r'))
if os.path.isfile(args.out):
json_out = json.load(open(args.out, 'r'))
else:
json_out = {}
json_out["version"] = "1.0"
@@ -166,7 +166,7 @@ def do_tuning(args):
return execute(atd.dot(LHS, RHS), sizes, Template[(A_trans, B_trans)], parameters, fname)
tune(execution_handler, 100, 2000, 3,(A_trans,B_trans), 'linear', 'linear')
json.dump(json_out, open(args.json_file,'w'))
json.dump(json_out, open(args.out,'w'))
@@ -182,7 +182,7 @@ class ArgumentsHandler:
tune_parser.add_argument("--operations", default = 'vaxpy,maxpy,dot,gemv,gemm-float32', type=str)
tune_parser.add_argument("--gemm-layouts", default='NN,NT,TN,TT', type=str)
tune_parser.add_argument("--gemv-layouts", default='N,T', type=str)
tune_parser.add_argument("--json-file", default='', type=str)
tune_parser.add_argument("--out", default='', type=str)
tune_parser.add_argument("--viennacl-src-path", default='', type=str)
tune_subparsers = tune_parser.add_subparsers(dest='method')
@@ -202,8 +202,8 @@ class ArgumentsHandler:
if self.action == 'tune':
#Retypes
self.device = devices[int(self.device)]
if not self.json_file:
self.json_file = misc_tools.sanitize_string(self.device.name) + '.json'
if not self.out:
self.out = misc_tools.sanitize_string(self.device.name) + '.json'
self.operations = self.operations.split(',')
self.gemm_layouts = self.gemm_layouts.split(',')
self.gemv_layouts = self.gemv_layouts.split(',')

2
python/postinstall.py
View File

@@ -31,6 +31,6 @@ if retune=='y':
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 +['--json', os.environ['HOME'] + '/.atidlas/devices/device' + opts[1] + '.json'])
subprocess.call(["${CMAKE_BINARY_DIR}/python/autotune/dist/autotune", "tune"] + opts +['--out', os.environ['HOME'] + '/.atidlas/devices/device' + opts[1] + '.json'])