34 lines
1.3 KiB
Python
34 lines
1.3 KiB
Python
|
import pyopencl as cl
|
||
|
|
import pyviennacl as vcl
|
||
|
|
|
||
|
|
all_devices = [d for platform in cl.get_platforms() for d in platform.get_devices()]
|
||
|
|
|
||
|
|
DEVICE_TYPE_PREFIX = { cl.device_type.GPU: 'gpu',
|
||
|
|
cl.device_type.CPU: 'cpu',
|
||
|
|
cl.device_type.ACCELERATOR: 'accelerator'
|
||
|
|
}
|
||
|
|
|
||
|
|
DEVICE_TYPE_CL_NAME = { cl.device_type.GPU: 'CL_DEVICE_TYPE_GPU',
|
||
|
|
cl.device_type.CPU: 'CL_DEVICE_TYPE_CPU',
|
||
|
|
cl.device_type.ACCELERATOR: 'CL_DEVICE_TYPE_ACCELERATOR'
|
||
|
|
}
|
||
|
|
|
||
|
|
VENDOR_PREFIX = { vcl.opencl.VendorId.beignet_id: 'beignet',
|
||
|
|
vcl.opencl.VendorId.nvidia_id: 'nvidia',
|
||
|
|
vcl.opencl.VendorId.amd_id: 'amd',
|
||
|
|
vcl.opencl.VendorId.intel_id: 'intel'
|
||
|
|
}
|
||
|
|
|
||
|
|
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]
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
def sanitize_string(string, keep_chars = ['_']):
|
||
|
|
string = string.replace(' ', '_').lower()
|
||
|
|
string = "".join(c for c in string if c.isalnum() or c in keep_chars).rstrip()
|
||
|
|
return string
|