Skip to content

[FEA] Add JIT cache management functions - #21553

Merged
rapids-bot[bot] merged 24 commits into
NVIDIA:mainfrom
lamarrr:jit-infra
Mar 11, 2026
Merged

[FEA] Add JIT cache management functions#21553
rapids-bot[bot] merged 24 commits into
NVIDIA:mainfrom
lamarrr:jit-infra

Conversation

@lamarrr

@lamarrr lamarrr commented Feb 25, 2026

Copy link
Copy Markdown
Contributor

Description

This pull-request adds functions to aid benchmarking of CUDF's JIT functions for the cold and hot JIT compilation states.

Follows up #18023

xref Useful for benchmarking JIT in polars #21456

Checklist

  • I am familiar with the Contributing Guidelines.
  • New or existing tests cover these changes.
  • The documentation is up to date with these changes.

@lamarrr
lamarrr requested a review from a team as a code owner February 25, 2026 12:16
@github-actions github-actions Bot added the libcudf Affects libcudf (C++/CUDA) code. label Feb 25, 2026
@lamarrr lamarrr added feature request New feature or request non-breaking Non-breaking change labels Feb 25, 2026
@lamarrr
lamarrr requested a review from Matt711 February 25, 2026 12:18
Comment thread cpp/include/cudf/context.hpp Outdated

@davidwendt davidwendt 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.

Would this be useful to make available in the pylibcudf and python interface?

Co-authored-by: David Wendt <45795991+davidwendt@users.noreply.github.com>
@lamarrr

lamarrr commented Feb 25, 2026

Copy link
Copy Markdown
Contributor Author

Would this be useful to make available in the pylibcudf and python interface?

Yes, this will be especially useful for Polars benchmarking

@lamarrr

lamarrr commented Feb 27, 2026

Copy link
Copy Markdown
Contributor Author

@Matt711 has agreed to take on creating pylibcudf functions for it

@PointKernel PointKernel left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

two minor questions but non-blocking.

Comment thread cpp/src/jit/cache.hpp Outdated
Comment thread cpp/src/jit/cache.cpp Outdated
@GregoryKimball GregoryKimball moved this to Burndown in libcudf Mar 9, 2026
Comment thread cpp/src/jit/cache.hpp Outdated
Comment thread cpp/src/transform/compute_column.cu Outdated
lamarrr and others added 5 commits March 11, 2026 01:43
Co-authored-by: Yunsong Wang <12716979+PointKernel@users.noreply.github.com>
Co-authored-by: Yunsong Wang <12716979+PointKernel@users.noreply.github.com>
Co-authored-by: Nghia Truong <7416935+ttnghia@users.noreply.github.com>
Co-authored-by: Nghia Truong <7416935+ttnghia@users.noreply.github.com>
@Matt711

Matt711 commented Mar 11, 2026

Copy link
Copy Markdown
Member

pre-commit.ci autofix

@copy-pr-bot

copy-pr-bot Bot commented Mar 11, 2026

Copy link
Copy Markdown

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@Matt711

Matt711 commented Mar 11, 2026

Copy link
Copy Markdown
Member

/ok to test 768eed0

@Matt711

Matt711 commented Mar 11, 2026

Copy link
Copy Markdown
Member

/merge

@rapids-bot
rapids-bot Bot merged commit 74c6fa3 into NVIDIA:main Mar 11, 2026
114 checks passed
@vuule vuule moved this from Burndown to Landed in libcudf Mar 11, 2026
rapids-bot Bot pushed a commit that referenced this pull request Mar 30, 2026
shrshi pushed a commit to shrshi/cudf that referenced this pull request May 12, 2026
@GregoryKimball GregoryKimball removed this from libcudf Jun 2, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

feature request New feature or request libcudf Affects libcudf (C++/CUDA) code. non-breaking Non-breaking change

Projects

None yet

Development

Successfully merging this pull request may close these issues.

7 participants