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

Zarr Compressor Benchmark #276

Draft
wants to merge 3 commits into
base: branch-23.10
Choose a base branch
from

Conversation

madsbk
Copy link
Member

@madsbk madsbk commented Sep 5, 2023

No description provided.

@madsbk madsbk added improvement Improves an existing functionality non-breaking Introduces a non-breaking change labels Sep 5, 2023
Copy link

@akshaysubr akshaysubr left a comment

Choose a reason for hiding this comment

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

Wanted to chime in with a couple of minor comments that I had as I was going through this PR


t0 = clock()
compressor.codec.decode(a, out=dst)
decode_time = clock() - t0

Choose a reason for hiding this comment

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

This probably needs some GPU synchronization before measuring end time, doesn't it?

compressors = {
"lz4-default": Compressor(device=Device.CPU, codec=numcodecs.LZ4()),
"lz4-blosc": Compressor(
device=Device.CPU, codec=numcodecs.blosc.Blosc(cname="lz4")

Choose a reason for hiding this comment

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

The default for Blosc is shuffle=SHUFFLE which changes the compression ratio a fair bit compared to LZ4. Might be good to indicate that in the str naming?

"snappy-nvcomp": Compressor(device=Device.GPU, codec=kvikio.zarr.Snappy()),
"cascaded-nvcomp": Compressor(device=Device.GPU, codec=kvikio.zarr.Cascaded()),
"gdeflate-nvcomp": Compressor(device=Device.GPU, codec=kvikio.zarr.Gdeflate()),
"bitcomp-nvcomp": Compressor(device=Device.GPU, codec=kvikio.zarr.Bitcomp()),

Choose a reason for hiding this comment

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

Bitcomp requires the dtype which currently isn't an option here but is an option in BitcompManager. Might be good to expose that and add it here if possible.

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