Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add JNI bindings for zstd compression of NVCOMP. #15729

Merged
merged 4 commits into from
May 14, 2024

Conversation

firestarman
Copy link
Contributor

@firestarman firestarman commented May 13, 2024

LZ4 is already supported, and this PR mainly adds JNI bindings for the ZSTD compression of NVCOMP.

It extracts the common code from LZ4 compressor and decompressor and put it into two new classes BatchedCompressor and BatchedDecompressor. By extending from these two classes, it adds in the ZSTD support.

Description

Checklist

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

Signed-off-by: Firestarman <[email protected]>
@firestarman firestarman requested a review from a team as a code owner May 13, 2024 03:37
Copy link

copy-pr-bot bot commented May 13, 2024

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.

@github-actions github-actions bot added the Java Affects Java cuDF API. label May 13, 2024
@firestarman firestarman added feature request New feature or request Java Affects Java cuDF API. non-breaking Non-breaking change and removed Java Affects Java cuDF API. labels May 13, 2024
@firestarman
Copy link
Contributor Author

/ok to test

Signed-off-by: Firestarman <[email protected]>
@firestarman
Copy link
Contributor Author

/ok to test

Copy link
Member

@jlowe jlowe left a comment

Choose a reason for hiding this comment

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

Minor typo, lgtm.


public class NvcompTest {
private static final HostMemoryAllocator hostMemoryAllocator = DefaultHostMemoryAllocator.get();

private static final Logger log = LoggerFactory.getLogger(ColumnVector.class);

private final long chunkSize = 64 * 1024;
private final long targetIntermediteSize = Long.MAX_VALUE;
Copy link
Member

Choose a reason for hiding this comment

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

Suggested change
private final long targetIntermediteSize = Long.MAX_VALUE;
private final long targetIntermediateSize = Long.MAX_VALUE;

Copy link
Contributor Author

Choose a reason for hiding this comment

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

Nice catch. But I'll merge this first, and correct typo in the future PRs.

@firestarman
Copy link
Contributor Author

/merge

@rapids-bot rapids-bot bot merged commit 0f6ce63 into rapidsai:branch-24.06 May 14, 2024
70 checks passed
@firestarman firestarman deleted the zstd-comp branch May 14, 2024 01:31
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 Java Affects Java cuDF API. non-breaking Non-breaking change
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants