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

Add parameter for specifying --heap-size-hint on spawned Julia processes #493

Merged
merged 5 commits into from
Dec 24, 2023

Conversation

MilesCranmer
Copy link
Owner

@MilesCranmer MilesCranmer commented Dec 24, 2023

Fixes #490 by @paulomontero and @eelregit.

This creates the heap_size_hint_in_bytes parameter which is forwarded to SymbolicRegression.jl. This allows the user to tell spawned Julia processes how much memory they can use before they should start aggressively garbage collecting (to avoid memory errors).

The default heap_size_hint_in_bytes should be enough as it is simply the current memory (on a single node) divided by the number of processes. However if you want to give processes more memory (for many node jobs) you might want specify this manually.

This PR also does some unrelated cleanup of the typing hints.

@MilesCranmer MilesCranmer merged commit 5bf2e55 into master Dec 24, 2023
21 checks passed
@MilesCranmer MilesCranmer deleted the heap-size-hint branch December 25, 2023 01:47
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

[BUG]: Possible memory leakage & best practices for memory scaling?
1 participant