Skip to content

Conversation

@madsbk
Copy link
Member

@madsbk madsbk commented May 19, 2025

Implement out-of-memory protection by using a RMM resource RmmFallbackResource based on rapidsai/rmm#1665.

The idea is to use managed memory when the RMM pool raises an OOM error.

@madsbk madsbk self-assigned this May 19, 2025
@madsbk madsbk added improvement Improves an existing functionality non-breaking Introduces a non-breaking change labels May 19, 2025
@madsbk madsbk force-pushed the rmm_fallback_resource branch from 4846b68 to 2a9a7c6 Compare May 19, 2025 15:18
@madsbk madsbk marked this pull request as ready for review May 19, 2025 15:37
@madsbk madsbk requested review from a team as code owners May 19, 2025 15:37
@madsbk madsbk requested a review from pentschev May 20, 2025 12:05
@madsbk
Copy link
Member Author

madsbk commented May 20, 2025

Thanks @pentschev, I think I have addressed all your suggestion?

Copy link
Member

@pentschev pentschev left a comment

Choose a reason for hiding this comment

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

Left a couple other questions, but otherwise LGTM. Thanks @madsbk .

Copy link
Member

@pentschev pentschev left a comment

Choose a reason for hiding this comment

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

Thanks @madsbk !

@madsbk
Copy link
Member Author

madsbk commented May 20, 2025

/merge

@rapids-bot rapids-bot bot merged commit 6c3adfa into rapidsai:branch-25.06 May 20, 2025
23 checks passed
@madsbk madsbk deleted the rmm_fallback_resource branch May 20, 2025 14:49
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

improvement Improves an existing functionality non-breaking Introduces a non-breaking change

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants