[auto-merge] branch-25.10 to branch-25.12 [skip ci] [bot] #985
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Rebases #701 to spark-rapids-ml 25.08 and cupy 13.6 Description from #701 : Allowing running benchmarks using System Allocated Memory (SAM) on HMM and Grace Hopper systems. This depends on cupy/cupy#8442. Some additional changes: - moved custom aligned numpy allocator config to core library and into a separate module (invoking in a function leads to segfault in cuml import likely due to a key object being deallocated but haven't root caused) - set `CUPY_ENABLE_UMP` env var automatically in python worker run time if sam is enabled (i.e. no separate config needed) - minor patches to benchmarking scripts (e.g. to fix verbose mode). - for algo's that concatenate input batches, memadvise numpy allocations to stay on host. Otherwise, these migrate to device during concatenation and can lead to OOM wrt headroom calculation - reserve device memory for sam rmm allocations during cuml fit, in addition to non-rmm allocations by libraries (this speeds up kmeans cuml fit by 13x over original making it 4x faster than uvm) --------- Signed-off-by: Rong Ou <[email protected]> Signed-off-by: Erik Ordentlich <[email protected]> Co-authored-by: Rong Ou <[email protected]>
|
SUCCESS - auto-merge |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
auto-merge triggered by github actions on
branch-25.10to create a PR keepingbranch-25.12up-to-date. If this PR is unable to be merged due to conflicts, it will remain open until manually fix.