-
Notifications
You must be signed in to change notification settings - Fork 58
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
base: branch-23.10
Are you sure you want to change the base?
Conversation
d534deb
to
48c95ca
Compare
There was a problem hiding this 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 |
There was a problem hiding this comment.
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") |
There was a problem hiding this comment.
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()), |
There was a problem hiding this comment.
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.
No description provided.