ai.rapids.cudf.nvcomp

类 BatchedDecompressor

java.lang.Object

ai.rapids.cudf.nvcomp.BatchedDecompressor

直接已知子类

BatchedLZ4Decompressor, BatchedZstdDecompressor

public abstract class BatchedDecompressor
extends Object

可对批次中的多个输入缓冲区进行操作的解压缩器

构造方法摘要

构造方法

构造方法和说明
BatchedDecompressor(long chunkSize) 构造一个批处理解压缩器实例

方法摘要

修饰符和类型	方法和说明
protected abstract void	batchedDecompressAsync(long devInPtrs, long devInSizes, long devOutSizes, long batchSize, long tempPtr, long tempSize, long devOutPtrs, long stream) 异步解压缩一批压缩数据缓冲区。
protected abstract long	batchedDecompressGetTempSize(long numChunks, long maxUncompressedChunkBytes) 计算解压缩压缩批次所需的临时存储空间大小（以字节为单位）。
void	BaseDeviceMemoryBuffer[] origInputs, BaseDeviceMemoryBuffer[] outputs, Cuda.Stream stream) 异步解压缩一批缓冲区

从类 java.lang.Object继承的方法

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

构造方法详细信息

BatchedDecompressor

public BatchedDecompressor(long chunkSize)

构造一个批处理解压缩器实例

参数

chunkSize - 最大未压缩块大小，必须与压缩时使用的值匹配

方法详细信息

decompressAsync

public void decompressAsync(BaseDeviceMemoryBuffer[] origInputs,
                            BaseDeviceMemoryBuffer[] outputs,
                            Cuda.Stream stream)

异步解压缩一批缓冲区

参数

origInputs - 要解压缩的缓冲区，将在此操作中关闭

outputs - 将包含解压缩结果的输出缓冲区，每个缓冲区的大小必须与相应的输入解压缩后的大小完全一致

stream - 要使用的 CUDA 流

batchedDecompressGetTempSize

protected abstract long batchedDecompressGetTempSize(long numChunks,
                                                     long maxUncompressedChunkBytes)

计算解压缩压缩批次所需的临时存储空间大小（以字节为单位）。

参数

numChunks - 批次中的块数

maxUncompressedChunkBytes - 任何块的最大未压缩大小（以字节为单位）

返回

解压缩批次所需的临时存储空间字节数

batchedDecompressAsync

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

异步解压缩一批压缩数据缓冲区。

参数

devInPtrs - 压缩输入缓冲区地址向量的设备地址

devInSizes - 压缩输入缓冲区大小向量的设备地址

devOutSizes - 未压缩缓冲区大小向量的设备地址

batchSize - 批次中的缓冲区数量

tempPtr - 临时解压缩空间的设备地址

tempSize - 临时解压缩空间的大小（以字节为单位）

devOutPtrs - 未压缩输出缓冲区地址向量的设备地址

stream - 要使用的 CUDA 流