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)
