NEW: Add thread pool to lamino solver - Githubissues

AdvancedPhotonSource / tike

Repository for ptychography software

http://tike.readthedocs.io

Other

29 stars 15 forks source link

NEW: Add thread pool to lamino solver #118

Closed carterbox closed 3 years ago

carterbox commented 3 years ago

Purpose

Add multi-device parallelism to the laminography solver.

Approach

Ported the same map reduce approach used in the ptycho cgrad solver.

Pre-Merge Checklists

Submitter

[x] Write a helpfully descriptive pull request title.
[x] Organize changes into logically grouped commits with descriptive commit messages.
[x] Document all new functions.
[x] Write tests for new functions or explain why they are not needed.
[ ] Build the documentation successfully
[ ] Use yapf to format python code.

Reviewer

[ ] Actually read all of the code.
[ ] Run the new code yourself.
[ ] Write a summary of the changes as you understand them.
[ ] Thank the submitter.

carterbox commented 3 years ago

This is actually, not ready. When setting num_gpu to >=2 it fails because I forgot to divide theta amongst the processes.

carterbox commented 3 years ago

OK. Now, I think this PR is ready because it is actually being tested in a multi-gpu environment.