类的用途
ai.rapids.cudf.DeviceMemoryBuffer

使用 DeviceMemoryBuffer 的包

包	描述
ai.rapids.cudf
ai.rapids.cudf.nvcomp

在 ai.rapids.cudf 中使用 DeviceMemoryBuffer

在 ai.rapids.cudf 中返回 DeviceMemoryBuffer 的方法

修饰符和类型	方法和描述
static DeviceMemoryBuffer	Rmm.alloc(long size) 分配设备内存并返回指向设备内存的指针，使用流 0。
static DeviceMemoryBuffer	Rmm.alloc(long size, Cuda.Stream stream) 分配设备内存并返回指向设备内存的指针。
static DeviceMemoryBuffer	DeviceMemoryBuffer.allocate(long bytes) 分配用于 GPU 的内存。
static DeviceMemoryBuffer	DeviceMemoryBuffer.allocate(long bytes, Cuda.Stream stream) 分配用于 GPU 的内存。
static DeviceMemoryBuffer	DeviceMemoryBuffer.fromRmm(long address, long lengthInBytes, long rmmBufferAddress) 将现有的 RMM 分配包装到设备内存缓冲区中。
DeviceMemoryBuffer	ContiguousTable.getBuffer() 获取支持连续表数据的设备缓冲区。
DeviceMemoryBuffer	GatherMap.releaseBuffer() 释放底层的设备缓冲区实例。
DeviceMemoryBuffer	DeviceMemoryBuffer.slice(long offset, long len) 切分设备缓冲区的一部分。
DeviceMemoryBuffer	BaseDeviceMemoryBuffer.sliceWithCopy(long offset, long len) 切分设备缓冲区的一部分，复制它而不是进行引用计数。

在 ai.rapids.cudf 中使用类型为 DeviceMemoryBuffer 参数的方法

修饰符和类型	方法和描述
long	MultiBufferDataSource.deviceRead(long offset, DeviceMemoryBuffer dest, Cuda.Stream stream)
long	DataSource.deviceRead(long offset, DeviceMemoryBuffer dest, Cuda.Stream stream) 从源读取指定偏移量的数据到 dest 中。
static Table	Table.fromPackedTable(ByteBuffer metadata, DeviceMemoryBuffer data) 从打包表示构建一个表。
static ColumnVector	ColumnVector.fromViewWithContiguousAllocation(long columnViewAddress, DeviceMemoryBuffer buffer) 使用连续设备分配从原生 column_view 创建一个 ColumnVector。
long	ChunkedPack.next(DeviceMemoryBuffer userPtr) 将表的下一个连续块放入 `userPtr` 中。

在 ai.rapids.cudf 中使用类型为 DeviceMemoryBuffer 参数的构造方法

构造方法和描述
ColumnVector(DType type, long rows, Optional<Long> nullCount, DeviceMemoryBuffer dataBuffer, DeviceMemoryBuffer validityBuffer, DeviceMemoryBuffer offsetBuffer) 基于设备上已有的数据创建一个新的列向量。
ColumnVector(DType type, long rows, Optional<Long> nullCount, DeviceMemoryBuffer dataBuffer, DeviceMemoryBuffer validityBuffer, DeviceMemoryBuffer offsetBuffer, List<DeviceMemoryBuffer> toClose, long[] childHandles) 基于设备上已有的数据和子列创建一个新的列向量。
GatherMap(DeviceMemoryBuffer buffer) 从设备缓冲区构建一个 gather map 实例。
OffHeapState(DeviceMemoryBuffer data, DeviceMemoryBuffer valid, DeviceMemoryBuffer offsets, List<DeviceMemoryBuffer> buffers, long viewHandle) 从现有的 cudf::column_view 和缓冲区创建。
OffHeapState(long viewHandle, DeviceMemoryBuffer contiguousBuffer) 从现有的 cudf::column_view 和连续缓冲区创建。

在 ai.rapids.cudf 中使用类型参数为 DeviceMemoryBuffer 的构造方法参数

构造方法和描述
ColumnVector(DType type, long rows, Optional<Long> nullCount, DeviceMemoryBuffer dataBuffer, DeviceMemoryBuffer validityBuffer, DeviceMemoryBuffer offsetBuffer, List<DeviceMemoryBuffer> toClose, long[] childHandles) 基于设备上已有的数据和子列创建一个新的列向量。
OffHeapState(DeviceMemoryBuffer data, DeviceMemoryBuffer valid, DeviceMemoryBuffer offsets, List<DeviceMemoryBuffer> buffers, long viewHandle) 从现有的 cudf::column_view 和缓冲区创建。

在 ai.rapids.cudf.nvcomp 中使用 DeviceMemoryBuffer

在 ai.rapids.cudf.nvcomp 中返回 DeviceMemoryBuffer 的方法

修饰符和类型	方法和描述
DeviceMemoryBuffer[]	BatchedCompressor.compress(BaseDeviceMemoryBuffer[] origInputs, Cuda.Stream stream) 压缩一批缓冲区。