Skip to content
Snippets Groups Projects
Commit 95b290f3 authored by Andreas Klöckner's avatar Andreas Klöckner
Browse files

kernel_get_work_group_info: Use Device.int_ptr in cache key

parent 7e9df904
No related branches found
No related tags found
1 merge request!134Speed up enqueue
......@@ -865,13 +865,14 @@ def _add_functionality():
KernelWithCustomEnqueue.set_args = set_args
def kernel_get_work_group_info(self, param, device):
cache_key = (param, device.int_ptr)
try:
return self._wg_info_cache[param, device]
return self._wg_info_cache[cache_key]
except KeyError:
pass
result = kernel_old_get_work_group_info(self, param, device)
self._wg_info_cache[param, device] = result
self._wg_info_cache[cache_key] = result
return result
def kernel_set_args(self, *args, **kwargs):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment