Skip to content

Commit

Permalink
Merge pull request #4150 from silx-kit/fix_opencl_convolution_cpu
Browse files Browse the repository at this point in the history
`silx.opencl.convolution`: Fixed separable convolution on CPU that crashed in some cases
  • Loading branch information
pierrepaleo authored Aug 2, 2024
2 parents a7da263 + 3620fd5 commit 415b020
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions src/silx/opencl/convolution.py
Original file line number Diff line number Diff line change
Expand Up @@ -385,6 +385,8 @@ def _batched_convolution(self, axis, input_ref=None, output_ref=None):
self.kernel_args, input_ref, output_ref
)
ev = opencl_kernel(*opencl_kernel_args)
if self.queue.device.type == cl.device_type.CPU:
ev.wait()
if self.profile:
self.events.append(EventDescription("batched convolution", ev))

Expand Down

0 comments on commit 415b020

Please sign in to comment.