Skip to content
Snippets Groups Projects

Basic support for OpenCL (experimental)

Merged Stephan Seitz requested to merge seitz/pystencils:opencl-backend into master
Viewing commit c21436bc
Show latest version
2 files
+ 54
1
Preferences
Compare changes
Files
2
@@ -64,7 +64,7 @@ def make_python_function(kernel_function_node, opencl_queue, opencl_ctx, argumen
@@ -64,7 +64,7 @@ def make_python_function(kernel_function_node, opencl_queue, opencl_ctx, argumen
block_and_thread_numbers['grid']))
block_and_thread_numbers['grid']))
args = _build_numpy_argument_list(parameters, full_arguments)
args = _build_numpy_argument_list(parameters, full_arguments)
args = [a.data for a in args if hasattr(a, 'data')]
args = [a.data if hasattr(a, 'data') else a for a in args]
cache[key] = (args, block_and_thread_numbers)
cache[key] = (args, block_and_thread_numbers)
cache_values.append(kwargs) # keep objects alive such that ids remain unique
cache_values.append(kwargs) # keep objects alive such that ids remain unique
func(opencl_queue, block_and_thread_numbers['grid'], block_and_thread_numbers['block'], *args)
func(opencl_queue, block_and_thread_numbers['grid'], block_and_thread_numbers['block'], *args)