[Feature] Enable CuPy on main process
Make GPU processing available for operations on the main node (chiefly
libertem.udf.base.UDF.merge()
andlibertem.udf.base.UDF.get_results()
) throughlibertem.udf.base.UDF.xp
. This is targeted at methods like iDPC and iCoM where the result is calculated inlibertem.udf.base.UDF.get_results()
with Fourier transforms over the displacement vector field that run much faster on GPUs compared to CPUs. It can be controlled with the newmain_process_gpu
parameter forlibertem.api.Context.make_with()
. It is enabled by default to catch potential issues, to be reviewed before the next release (#1759).