def test_simple_cache(self): import pycl as cl from ctree.ocl import get_context_and_queue_from_devices devices = cl.clGetDeviceIDs() device = devices[-1] results1 = get_context_and_queue_from_devices([device]) results2 = get_context_and_queue_from_devices([device]) self.assertEqual(results1, results2)
def __init__(self): """__init__ Creates a context and queue that can be reused across calls to this function. """ # TODO: Need dependency injection to control ocl device selection self.desired_ocl_device = -1 devices = cl.clGetDeviceIDs() self.context, self.queue = get_context_and_queue_from_devices( [devices[self.desired_ocl_device]]) self.max_work_group_size = \ devices[self.desired_ocl_device].max_work_group_size # some variables that will be used that PEP-8 wants to see initialized # in __init__ self.kernel = None self.output = None self._c_function = None
def __init__(self): self.device = clGetDeviceIDs()[-1] self.context, self.queue = get_context_and_queue_from_devices([ self.device ])
def test_simple_cache(self): devices = cl.clGetDeviceIDs() device = devices[-1] results1 = get_context_and_queue_from_devices([device]) results2 = get_context_and_queue_from_devices([device]) self.assertEqual(results1, results2)