Closed ltang320 closed 8 months ago
Hi, our stochastic engines are not compatible with MPI. You can however split the dataset and have EPIE work on a subset per MPI rank. Andrew Maiden talked about this scheme at the Ptycho Workshop last year. But this is ultimately a different type of engine and doesn't necessarily have the same convergence behavior. I don't recall how Andrew called these engines, maybe it's WASP? https://preprints.opticaopen.org/articles/preprint/WASP_Weighted_Average_of_Sequential_Projections_for_ptychographic_phase_retrieval/24894489 Timothy is working on a pull request for that engine. https://github.com/ptycho/ptypy/pull/522 Maybe @daurer can comment better on that.
Hi, our stochastic engines are not compatible with MPI. You can however split the dataset and have EPIE work on a subset per MPI rank. Andrew Maiden talked about this scheme at the Ptycho Workshop last year. But this is ultimately a different type of engine and doesn't necessarily have the same convergence behavior. I don't recall how Andrew called these engines, maybe it's WASP? https://preprints.opticaopen.org/articles/preprint/WASP_Weighted_Average_of_Sequential_Projections_for_ptychographic_phase_retrieval/24894489 Timothy is working on a pull request for that engine. #522 Maybe @daurer can comment better on that.
Oh, I see your point. @ptim0626 Hi Timothy, I see you add the WASP reconstruction engine. Can I have a test of it? By the way, can it be used with multi-modes? Thanks for all your help!
Hi @ltang320, of course you are welcome to try the WASP engine! It supports multi probe and object modes.
I should add the WASP engine codes in #522 is currently under active development and you may find errors and other unexpected behaviour, so please use if with caution and if you encounter any issue, please let us know.
Hi, I encountered an issue when I used EPIE to do the reconstruction.
The error told me, "The stochastic engines are incompatible with MPI." When I check the code in /type/type/accelerate/cuda_pycuda/ engines/stochastic.py, EPIE supports GPU. However, it doesn't support MPI.
Is it possible to make EPIE to support MPI? @bjoernenders @daurer Thanks a lot for your help!