You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Problem:
There are issues with cupy's/numba's JIT compilers. They simply are not robust and there are issues with version matching. When initializing a phase object with device='gpu' a JIT compile error can occur (e.g. in py4DSTEM.process.phase.Parallax(...). If the initialization is done with device='cpu' to avoid the error, then the later reconstruction step (e.g. parallax_recon.reconstruct(...)) also need to be run in device='cpu' as currently implemented.
Solution
Allow for device transfer...or figure out a way to handle the cupy shenanigans. @gvarnavi suggested the first one as cupy is ultimately out of our hands.
The text was updated successfully, but these errors were encountered:
Problem:
There are issues with cupy's/numba's JIT compilers. They simply are not robust and there are issues with version matching. When initializing a phase object with
device='gpu'
a JIT compile error can occur (e.g. inpy4DSTEM.process.phase.Parallax(...)
. If the initialization is done withdevice='cpu'
to avoid the error, then the later reconstruction step (e.g.parallax_recon.reconstruct(...)
) also need to be run indevice='cpu'
as currently implemented.Solution
Allow for device transfer...or figure out a way to handle the cupy shenanigans.
@gvarnavi suggested the first one as cupy is ultimately out of our hands.
The text was updated successfully, but these errors were encountered: