BatchedZstdDecompressor#
- public class BatchedZstdDecompressor extends BatchedDecompressor#
ZSTD decompressor that operates on multiple input buffers in a batch.
Constructor Summary
Method Summary
Asynchronously decompress a batch of compressed data buffers.
Computes the temporary storage size in bytes needed to decompress a compressed batch.
Methods inherited from class
ai.rapids.cudf.nvcomp.BatchedDecompressor:decompressAsyncConstructors
- public BatchedZstdDecompressor(long chunkSize)#
Protected Methods
- protected long batchedDecompressGetTempSize(
- long numChunks,
- long maxUncompressedChunkBytes,
- long maxTotalSize,
- long stream
Computes the temporary storage size in bytes needed to decompress a compressed batch.
- Parameters:
numChunks – number of chunks in the batch
maxUncompressedChunkBytes – maximum uncompressed size of any chunk in bytes
maxTotalSize – Upper bound on the total uncompressed size of all chunks
stream – CUDA stream to use
- Returns:
number of temporary storage bytes needed to decompress the batch
- protected void batchedDecompressAsync(
- long devInPtrs,
- long devInSizes,
- long devOutSizes,
- long batchSize,
- long tempPtr,
- long tempSize,
- long devOutPtrs,
- long stream
Asynchronously decompress a batch of compressed data buffers.
- Parameters:
devInPtrs – device address of compressed input buffer addresses vector
devInSizes – device address of compressed input buffer sizes vector
devOutSizes – device address of uncompressed buffer sizes vector
batchSize – number of buffers in the batch
tempPtr – device address of the temporary decompression space
tempSize – size of the temporary decompression space in bytes
devOutPtrs – device address of uncompressed output buffer addresses vector
stream – CUDA stream to use