public final class PinnedMemoryPool extends Object implements AutoCloseable
| 修饰符和类型 | 方法和描述 |
|---|---|
静态 HostMemoryBuffer |
allocate(long bytes)
创建主机缓冲区的工厂方法,但优选指向固定内存。
|
静态 HostMemoryBuffer |
allocate(long bytes, HostMemoryAllocator hostMemoryAllocator)
创建主机缓冲区的工厂方法,但优选指向固定内存。
|
void |
close() |
静态 boolean |
configureDefaultCudfPinnedPoolSize(long size)
设置 cuDF 默认固定池的大小。
|
静态 long |
getTotalPoolSizeBytes()
获取固定内存池分配的字节数。
|
静态 void |
initialize(long poolSize)
初始化池。
|
静态 void |
initialize(long poolSize, int gpuId)
初始化池。
|
静态 void |
initialize(long poolSize, int gpuId, boolean setCudfPinnedPoolMemoryResource)
初始化池。
|
静态 boolean |
isInitialized()
检查池是否已初始化。
|
静态 void |
shutdown()
关闭 RMM pool_memory_resource,并将其引用置空。在此之后,任何正在进行的分配或释放都将失败。
|
静态 HostMemoryBuffer |
tryAllocate(long bytes)
创建固定主机内存缓冲区的工厂方法。
|
public static void initialize(long poolSize)
poolSize - 要初始化的池的大小。public static void initialize(long poolSize,
int gpuId)
poolSize - 要初始化的池的大小。gpuId - 获取内存池的 GPU ID,-1 表示使用默认值public static void initialize(long poolSize,
int gpuId,
boolean setCudfPinnedPoolMemoryResource)
poolSize - 要初始化的池的大小。gpuId - 获取内存池的 GPU ID,-1 表示使用默认值setCudfPinnedPoolMemoryResource - 如果 cuDF 应使用此固定池作为固定内存,则为 truepublic static boolean isInitialized()
public static void shutdown()
public static HostMemoryBuffer tryAllocate(long bytes)
bytes - 要分配的字节大小public static HostMemoryBuffer allocate(long bytes, HostMemoryAllocator hostMemoryAllocator)
bytes - 要分配的字节大小public static HostMemoryBuffer allocate(long bytes)
bytes - 要分配的字节大小public static long getTotalPoolSizeBytes()
public void close()
close 在接口 AutoCloseable 中public static boolean configureDefaultCudfPinnedPoolSize(long size)
size - cuDF 默认固定池的初始大小和最大大小。传递 size=0 以禁用默认池。版权所有 © 2025。保留所有权利。