包 | 描述 |
---|---|
ai.rapids.cudf | |
ai.rapids.cudf.nvcomp |
修饰符和类型 | 字段和描述 |
---|---|
static Cuda.Stream |
Cuda.DEFAULT_STREAM |
修饰符和类型 | 方法和描述 |
---|---|
static DeviceMemoryBuffer |
Rmm.alloc(long size, Cuda.Stream stream) 分配设备内存并返回设备内存指针。
|
static CudaMemoryBuffer |
CudaMemoryBuffer.allocate(long bytes, Cuda.Stream stream) 分配供 GPU 使用的内存。
|
static DeviceMemoryBuffer |
DeviceMemoryBuffer.allocate(long bytes, Cuda.Stream stream) 分配供 GPU 使用的内存。
|
static CudaMemoryBuffer |
Rmm.allocCuda(long size, Cuda.Stream stream) 使用 `cudaMalloc` 分配设备内存并返回设备内存指针。
|
void |
HostMemoryBuffer.copyFromDeviceBuffer(BaseDeviceMemoryBuffer deviceMemoryBuffer, Cuda.Stream stream) 使用指定的流将数据从 DeviceMemoryBuffer 复制到 HostMemoryBuffer。
|
void |
HostMemoryBuffer.copyFromDeviceBufferAsync(BaseDeviceMemoryBuffer deviceMemoryBuffer, Cuda.Stream stream) 使用指定的流将数据从 DeviceMemoryBuffer 复制到 HostMemoryBuffer。
|
void |
BaseDeviceMemoryBuffer.copyFromDeviceBufferAsync(long destOffset, BaseDeviceMemoryBuffer src, long srcOffset, long length, Cuda.Stream stream) 使用指定的 CUDA 流将 src 的子集复制到此缓冲区,从 destOffset 开始。
|
void |
BaseDeviceMemoryBuffer.copyFromHostBuffer(HostMemoryBuffer src, Cuda.Stream stream) 使用 CUDA 流将整个主机缓冲区复制到此缓冲区的开头。
|
void |
BaseDeviceMemoryBuffer.copyFromHostBuffer(long destOffset, HostMemoryBuffer src, long srcOffset, long length, Cuda.Stream stream) 使用指定的 CUDA 流将 src 的子集复制到此缓冲区,从 destOffset 开始。
|
void |
BaseDeviceMemoryBuffer.copyFromHostBufferAsync(HostMemoryBuffer src, 使用 CUDA 流将整个主机缓冲区复制到此缓冲区的开头。
|
void |
BaseDeviceMemoryBuffer.copyFromHostBufferAsync(long destOffset, HostMemoryBuffer src, long srcOffset, long length, Cuda.Stream stream) 使用指定的 CUDA 流将 src 的子集复制到此缓冲区,从 destOffset 开始。
|
void |
MemoryBuffer.copyFromMemoryBuffer(long destOffset, MemoryBuffer src, long srcOffset, long length, Cuda.Stream stream) 使用指定的 CUDA 流将 src 的子集复制到此缓冲区,从 destOffset 开始。
|
void |
MemoryBuffer.copyFromMemoryBufferAsync(long destOffset, MemoryBuffer src, long srcOffset, long length, Cuda.Stream stream) 使用指定的 CUDA 流将 src 的子集复制到此缓冲区,从 destOffset 开始。
|
HostColumnVector |
ColumnView.copyToHostAsync(Cuda.Stream stream) 异步将数据复制到主机。
|
HostColumnVector |
ColumnView.copyToHostAsync(Cuda.Stream stream, HostMemoryAllocator hostMemoryAllocator) 异步将数据复制到主机。
|
long |
MultiBufferDataSource.deviceRead(long offset, DeviceMemoryBuffer dest, Cuda.Stream stream) |
long |
DataSource.deviceRead(long offset, DeviceMemoryBuffer dest, Cuda.Stream stream) 从源的给定偏移量读取数据到 dest 中。
|
static void |
Cuda.multiBufferCopyAsync(long[] destAddrs, long[] srcAddrs, long[] copySizes, Cuda.Stream stream) 将数据从多个设备缓冲区源复制到多个设备缓冲区目标。
|
void |
Cuda.Event.record(Cuda.Stream stream) 在调用时捕获流的内容。
|
构造函数和描述 |
---|
CudaMemoryBuffer(long address, long lengthInBytes, Cuda.Stream stream)
将现有的 CUDA 分配包装到设备内存缓冲区中。
|
修饰符和类型 | 方法和描述 |
---|---|
DeviceMemoryBuffer[] |
BatchedCompressor.compress(BaseDeviceMemoryBuffer[] origInputs, Cuda.Stream stream) 压缩一批缓冲区。
|
void |
BatchedDecompressor.decompressAsync(BaseDeviceMemoryBuffer[] origInputs, BaseDeviceMemoryBuffer[] outputs, Cuda.Stream stream) 异步解压缩一批缓冲区
|
static void |
BatchedLZ4Decompressor.decompressAsync(long chunkSize, BaseDeviceMemoryBuffer[] origInputs, BaseDeviceMemoryBuffer[] outputs, Cuda.Stream stream) 异步解压缩一批缓冲区
|
版权所有 © 2025。保留所有权利。