Package ai.rapids.cudf.nvcomp

Class BatchedZstdDecompressor

java.lang.Object

ai.rapids.cudf.nvcomp.BatchedDecompressor

ai.rapids.cudf.nvcomp.BatchedZstdDecompressor

public class BatchedZstdDecompressor
extends BatchedDecompressor

ZSTD decompressor that operates on multiple input buffers in a batch

Constructor Summary

Constructors

Constructor	Description
BatchedZstdDecompressor(long chunkSize)

Method Summary

Modifier and Type	Method	Description
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.
protected long	batchedDecompressGetTempSize(long numChunks, long maxUncompressedChunkBytes)	Computes the temporary storage size in bytes needed to decompress a compressed batch.

Methods inherited from class ai.rapids.cudf.nvcomp.BatchedDecompressor

decompressAsync

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Details

BatchedZstdDecompressor

public BatchedZstdDecompressor(long chunkSize)

Method Details

batchedDecompressGetTempSize

protected long batchedDecompressGetTempSize(long numChunks,
 long maxUncompressedChunkBytes)

Description copied from class: BatchedDecompressor

Computes the temporary storage size in bytes needed to decompress a compressed batch.

Specified by:

batchedDecompressGetTempSize in class BatchedDecompressor

Parameters:

numChunks - number of chunks in the batch

maxUncompressedChunkBytes - maximum uncompressed size of any chunk in bytes

Returns:

number of temporary storage bytes needed to decompress the batch

batchedDecompressAsync

protected void batchedDecompressAsync(long devInPtrs,
 long devInSizes,
 long devOutSizes,
 long batchSize,
 long tempPtr,
 long tempSize,
 long devOutPtrs,
 long stream)

Description copied from class: BatchedDecompressor

Asynchronously decompress a batch of compressed data buffers.

Specified by:

batchedDecompressAsync in class BatchedDecompressor

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