Skip to content

[Data] Support rapidsmpf 26.2 in gpu shuffle#62062

Merged
goutamvenkat-anyscale merged 2 commits into
ray-project:masterfrom
leewyang:rapidsmpf_26.2
Mar 26, 2026
Merged

[Data] Support rapidsmpf 26.2 in gpu shuffle#62062
goutamvenkat-anyscale merged 2 commits into
ray-project:masterfrom
leewyang:rapidsmpf_26.2

Conversation

@leewyang

Copy link
Copy Markdown
Contributor

Description

This PR adds support for rapidsmpf 26.02, which had a breaking API change.

Related issues

Related to #61371

@leewyang leewyang requested a review from a team as a code owner March 25, 2026 21:40

@gemini-code-assist gemini-code-assist Bot left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

Code Review

This pull request updates the rapidsmpf_backend.py file to introduce version-dependent imports for MemoryType, BufferResource, and LimitAvailableMemory. This change ensures compatibility with different versions of the rapidsmpf library, specifically handling API changes introduced in version 26.2.0 and later. I have no feedback to provide.

@goutamvenkat-anyscale goutamvenkat-anyscale added the data Ray Data-related issues label Mar 25, 2026
Signed-off-by: Lee Yang <leey@nvidia.com>
Signed-off-by: Lee Yang <leey@nvidia.com>
@goutamvenkat-anyscale goutamvenkat-anyscale added the go add ONLY when ready to merge, run all tests label Mar 26, 2026
@goutamvenkat-anyscale goutamvenkat-anyscale enabled auto-merge (squash) March 26, 2026 03:11
@goutamvenkat-anyscale goutamvenkat-anyscale merged commit 028e164 into ray-project:master Mar 26, 2026
9 checks passed
@leewyang leewyang deleted the rapidsmpf_26.2 branch March 31, 2026 21:41
Lucas61000 pushed a commit to Lucas61000/ray that referenced this pull request May 15, 2026
## Description
This PR adds support for rapidsmpf 26.02, which had a breaking API
change.

## Related issues
Related to ray-project#61371

---------

Signed-off-by: Lee Yang <leey@nvidia.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

data Ray Data-related issues go add ONLY when ready to merge, run all tests

2 participants