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。保留所有权利。