ai.rapids.cudf

类 RmmCudaAsyncMemoryResource

java.lang.Object

ai.rapids.cudf.RmmCudaAsyncMemoryResource

所有已实现的接口

RmmDeviceMemoryResource, AutoCloseable

public class RmmCudaAsyncMemoryResource
extends Object
implements RmmDeviceMemoryResource

一个使用 `cudaMallocAsync` 和 `cudaFreeAsync` 进行分配和释放的设备内存资源。

构造器摘要

构造器

构造器和描述
RmmCudaAsyncMemoryResource(long size, long releaseThreshold) 创建一个新的异步内存资源
RmmCudaAsyncMemoryResource(long size, long releaseThreshold, boolean fabric) 创建一个新的异步内存资源

方法摘要

修饰符和类型	方法和描述
void	close()
long	getHandle() 返回指向实现 rmm::mr::device_memory_resource 的底层 C++ 类的指针
long	getSize()
String	toString()

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

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

构造器详细信息

RmmCudaAsyncMemoryResource

public RmmCudaAsyncMemoryResource(long size,
                                  long releaseThreshold)

创建一个新的异步内存资源

参数

size - 池的初始大小

releaseThreshold - 将内存释放回 cuda 的阈值大小（以字节为单位）

RmmCudaAsyncMemoryResource

public RmmCudaAsyncMemoryResource(long size,
                                  long releaseThreshold,
                                  boolean fabric)

创建一个新的异步内存资源

参数

size - 池的初始大小

releaseThreshold - 将内存释放回 cuda 的阈值大小（以字节为单位）

fabric - 如果为 true，则在创建池时请求对等读写可访问的 fabric 句柄

方法详细信息

getHandle

public long getHandle()

说明复制自接口: RmmDeviceMemoryResource

返回指向实现 rmm::mr::device_memory_resource 的底层 C++ 类的指针

指定者

接口 RmmDeviceMemoryResource 中的 getHandle

getSize

public long getSize()

close

public void close()

指定者

接口 RmmDeviceMemoryResource 中的 close

指定者

接口 AutoCloseable 中的 close

toString

public String toString()

覆盖

类 Object 中的 toString