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

Removed obsolete imports

Browse Source
This commit is contained in:
Philippe Tillet
2014-10-05 21:24:08 +02:00
parent 1770d3d60b
commit 88f10958ba
1 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
autotune/python/autotune.py
View File

@@ -15,9 +15,6 @@ from pyviennacl import atidlas
from dataset import generate_dataset from dataset import generate_dataset
from model import train_model from model import train_model
import tools import tools
import utils
import vclio
import optimize import optimize
import sys import sys
@@ -52,7 +49,13 @@ def do_tuning(config_fname, spec_fname, viennacl_root):
if operation in config: if operation in config:
p = config[operation] p = config[operation]
confdevices = p['devices'] confdevices = p['devices']
devices = utils.DEVICES_PRESETS[confdevices] if confdevices in utils.DEVICES_PRESETS else [utils.all_devices[int(i)] for i in confdevices] all_devices = [d for platform in cl.get_platforms() for d in platform.get_devices()]
DEVICES_PRESETS = {'all': all_devices,
'gpus': [d for d in all_devices if d.type==cl.device_type.GPU],
'cpus': [d for d in all_devices if d.type==cl.device_type.CPU],
'accelerators': [d for d in all_devices if d.type==cl.device_type.ACCELERATOR]
}
devices = DEVICES_PRESETS[confdevices] if confdevices in DEVICES_PRESETS else [all_devices[int(i)] for i in confdevices]
precisions = map_to_list(str, p['precision']) precisions = map_to_list(str, p['precision'])
if 'all' in precisions: if 'all' in precisions:
precisions = ['single','double'] precisions = ['single','double']
@@ -162,7 +165,7 @@ if __name__ == "__main__":
print("----------------") print("----------------")
devices = [d for platform in cl.get_platforms() for d in platform.get_devices()] devices = [d for platform in cl.get_platforms() for d in platform.get_devices()]
for (i, d) in enumerate(devices): for (i, d) in enumerate(devices):
print('Device', i, ':', utils.DEVICE_TYPE_PREFIX[d.type].upper() + ':', d.name, 'on', d.platform.name) print('Device', i, ':', tools.DEVICE_TYPE_PREFIX[d.type].upper() + ':', d.name, 'on', d.platform.name)
print("----------------") print("----------------")
else: else:
print("------") print("------")