NVIDIA / NVIDIA/cudf

[FEA] Java: Add ai.rapids.cudf.HashJoin.getDeviceMemorySize introspection

Open
#22,965 0 comments 0 reactions 0 assignees View on GitHub
feature request
Dominant language
C++
Stars
9.8k
Forks
1.1k
Avg merge
3d 6m
Merged PRs (30d)
278

Description

**Is your feature request related to a problem? Please describe.**

Allow applications to understand how much GPU device memory they are retaining though cudf objects like `HashJoin`.
**Describe the solution you'd like**

- Add `getDeviceMemorySize` or similar method to `ai.rapids.cudf.HashJoin` similar to existing `ai.rapids.cudf.Table.getDeviceMemorySize`
- when the caller holds the source `Table` and the `HashJoin` built from that table, it should have a way to estimate total retained memory usage without double-counting shared columns, if any

**Describe alternatives you've considered**

Not knowing retaining memory. However this make it hard from utilizing available memory smartly.

**Additional context**

None

Contributor guide

Open the contributing guide

Research direction

Start with the Java API entry points ai.rapids.cudf.HashJoin and ai.rapids.cudf.Table.getDeviceMemorySize, then trace how HashJoin retains device memory. Define how the new introspection method estimates total retained memory when source Tables share columns, without double-counting them. Done means callers can query HashJoin-related GPU memory consistently with Table introspection.

Written by the indexing model from the issue text.

Assessment

Tech stack
java
Domain
api, backend
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Quiet
Clarity
Mostly clear
Newbie friendliness
52/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.