Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

re-allow redrock+GPU on large inputs #1966

Merged
merged 1 commit into from Jan 12, 2023
Merged

re-allow redrock+GPU on large inputs #1966

merged 1 commit into from Jan 12, 2023

Conversation

sbailey
Copy link
Contributor

@sbailey sbailey commented Jan 12, 2023

Redrock+GPU used to run out of memory on input files with >1k targets (e.g. large healpix), which we had worked around in zproc by checking the number of targets and falling back to CPU-only on big inputs. desihub/redrock#211 fixed the problem, so this PR removes the ntargets check from zproc. I tested by rerunning himalayas sv3 dark healpix 10152 with 2192 targets and confirming that zproc ran redrock with GPU and that it completed fine.

This PR also fixes a bug where zproc --batch ... was accidentally forcing --no-gpu unless --mpi was also specified.

@sbailey sbailey requested a review from akremin January 12, 2023 22:32
Copy link
Member

@akremin akremin left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The changes look good, and good catch on the bug fix as well. I will trust your test as representative and approve+merge.

@akremin akremin merged commit cb3f024 into main Jan 12, 2023
@akremin akremin deleted the zbig branch January 12, 2023 22:53
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
Status: Done
Development

Successfully merging this pull request may close these issues.

None yet

2 participants