kikuchipy icon indicating copy to clipboard operation
kikuchipy copied to clipboard

Use of GPUs

Open hakonanes opened this issue 2 years ago • 2 comments

We should try to take advantage of GPUs by writing some GPU kernels. I don't have an NVIDIA GPU available, so my choice would be to use PyOpenCL instead of CuPy.

@drowenhorst-nrl have written some kernels in PyEBSDIndex that we could take inspiration from, e.g. the static background subtraction, used in one of the Radon transform functions. Such a kernel could be an alternative to our background subtraction.

In general, I think more per pattern operations in the kikuchipy.pattern module could be replaced by PyOpenCL kernels, like image rescaling. We have CPU acceleration from Numba here, but it would be good to test GPU acceleration as well.

Other resources:

hakonanes avatar Nov 19 '21 12:11 hakonanes