AdvancedPhotonSource / tike

Repository for ptychography software
http://tike.readthedocs.io
Other
29 stars 15 forks source link

BUG: Must index eigen probes in correct device context #183

Closed carterbox closed 2 years ago

carterbox commented 2 years ago

This PR pulls in changes from one of my personal development branches to fix a bug where cupy arrays are indexed outside of their proper device context. It does this by refactoring the eigen probe updates so they use our special CUDA context map() instead of a list comprehension.