triton/autotune/python/utils.py

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