BUG: Test fail with one GPU because of multi-device - Githubissues

AdvancedPhotonSource / tike

Repository for ptychography software

http://tike.readthedocs.io

Other

29 stars 15 forks source link

BUG: Test fail with one GPU because of multi-device #85

Closed carterbox closed 4 years ago

carterbox commented 4 years ago

Purpose

Squash a bug where the tests fail due to there not being enough GPUs because the tests always test the multi-gpu code path (maybe that's fine?).

Approach

The Pool class now limits the number of workers to the number of available GPUs and emits a warning.

Pre-Merge Checklists

Submitter

[ ] Write a helpfully descriptive pull request title.
[ ] Organize changes into logically grouped commits with descriptive commit messages.
[ ] Document all new functions.
[ ] 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.