ai.rapids.cudf

类 MemoryBuffer

java.lang.Object

ai.rapids.cudf.MemoryBuffer

所有已实现的接口

AutoCloseable

直接已知子类

BaseDeviceMemoryBuffer, HostMemoryBuffer

public abstract class MemoryBuffer
extends Object
implements AutoCloseable

表示内存缓冲区的抽象类 注意：MemoryBuffer 是公共的，以便更容易使用类层次结构，不建议且不支持超出 CUDF 范围的子类化。

嵌套类摘要

嵌套类

修饰符和类型	类和描述
static interface	MemoryBuffer.EventHandler 处理此 MemoryBuffer 事件的接口。
static class	MemoryBuffer.MemoryBufferCleaner

字段摘要

字段

修饰符和类型	字段和描述
protected long	address
protected MemoryBuffer.MemoryBufferCleaner	cleaner
protected boolean	closed
protected long	id
protected long	length
protected int	refCount

构造方法摘要

构造方法

修饰符	构造方法和描述
protected	MemoryBuffer(long address, long length) 构造方法
protected	MemoryBuffer(long address, long length, MemoryBuffer.MemoryBufferCleaner cleaner) 构造方法
protected	MemoryBuffer(long address, long length, MemoryBuffer parent) 创建切片时使用的内部构造方法。

方法摘要

修饰符和类型	方法和描述
protected void	addressOutOfBoundsCheck(long address, long size, String type)
void	close() 关闭此缓冲区并释放内存
void	copyFromMemoryBuffer(long destOffset, MemoryBuffer src, long srcOffset, long length, Cuda.Stream stream) 使用指定的 CUDA 流将 src 的子集复制到此缓冲区中，从 destOffset 开始。
void	copyFromMemoryBufferAsync(long destOffset, MemoryBuffer src, long srcOffset, long length, Cuda.Stream stream) 使用指定的 CUDA 流将 src 的子集复制到此缓冲区中，从 destOffset 开始。
long	getAddress() 返回此缓冲区指向的数据位置
MemoryBuffer.EventHandler	getEventHandler() 返回此缓冲区当前的事件处理程序；如果未关联处理程序或此缓冲区已关闭，则返回 null。
long	getLength() 返回此缓冲区的大小
int	getRefCount() 获取此缓冲区当前的引用计数。
void	incRefCount() 增加此列的引用计数。
void	noWarnLeakExpected() 这是一个非常糟糕的 API，但由于 Java 和 GC，数据列的生命周期可能不明确。此 API 通知泄漏跟踪代码，此列预期会出现这种情况，并且在此发生时不应打印可怕的大警告。
MemoryBuffer.EventHandler	setEventHandler(MemoryBuffer.EventHandler newHandler) 为此缓冲区设置事件处理程序。
abstract MemoryBuffer	slice(long offset, long len) 从此缓冲区切出一部分。
String	toString()

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

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

字段详细信息

address

protected final long address

length

protected final long length

closed

protected boolean closed

refCount

protected int refCount

cleaner

protected final MemoryBuffer.MemoryBufferCleaner cleaner

id

protected final long id

构造方法详细信息

MemoryBuffer

protected MemoryBuffer(long address,
                       long length,
                       MemoryBuffer.MemoryBufferCleaner cleaner)

构造方法

参数

address - 内存位置

length - 此缓冲区的大小

cleaner - 用于清理内存。如果不需要清理，可以为 null。

MemoryBuffer

protected MemoryBuffer(long address,
                       long length)

构造方法

参数

address - 内存位置

length - 此缓冲区的大小

MemoryBuffer

protected MemoryBuffer(long address,
                       long length,
                       MemoryBuffer parent)

创建切片时使用的内部构造方法。

参数

address - 内存位置

length - 此缓冲区的大小

parent - 应该关闭的缓冲区，而不是关闭当前缓冲区。

方法详细信息

noWarnLeakExpected

public void noWarnLeakExpected()

这是一个非常糟糕的 API，但由于 Java 和 GC，数据列的生命周期可能不明确。此 API 通知泄漏跟踪代码，此列预期会出现这种情况，并且在此发生时不应打印可怕的大警告。

getLength

public final long getLength()

返回此缓冲区的大小

返回

- 大小

addressOutOfBoundsCheck

protected final void addressOutOfBoundsCheck(long address,
                                             long size,
                                             String type)

getAddress

public final long getAddress()

返回此缓冲区指向的数据位置

返回

- 数据地址

copyFromMemoryBuffer

public final void copyFromMemoryBuffer(long destOffset,
                                       MemoryBuffer src,
                                       long srcOffset,
                                       long length,
                                       Cuda.Stream stream)

使用指定的 CUDA 流将 src 的子集复制到此缓冲区中，从 destOffset 开始。复制在此方法返回时完成，但内存复制可能与在其他流上发生的操作重叠。

参数

destOffset - 在此缓冲区中开始复制的偏移量。

src - 要从中复制的内容

srcOffset - src 中开始复制的偏移量

length - 要复制的字节数

stream - 要使用的 CUDA 流

copyFromMemoryBufferAsync

public final void copyFromMemoryBufferAsync(long destOffset,
                                            MemoryBuffer src,
                                            long srcOffset,
                                            long length,
                                            Cuda.Stream stream)

使用指定的 CUDA 流将 src 的子集复制到此缓冲区中，从 destOffset 开始。复制是异步的，在此方法返回时可能尚未完成。

参数

destOffset - 在此缓冲区中开始复制的偏移量。

src - 要从中复制的内容

srcOffset - src 中开始复制的偏移量

length - 要复制的字节数

stream - 要使用的 CUDA 流

slice

public abstract MemoryBuffer slice(long offset,
                                   long len)

从此缓冲区切出一部分。请注意，这是零复制操作，所有切片必须与原始缓冲区一起关闭，内存才会释放。因此，请谨慎使用。请注意，[[DeviceMemoryBuffer]] 和 [[HostMemoryBuffer]] 支持切片并覆盖此函数。

参数

offset - 切片开始的位置。

len - 要切片的字节数

返回

原始缓冲区的一个切片，需要独立关闭

setEventHandler

public MemoryBuffer.EventHandler setEventHandler(MemoryBuffer.EventHandler newHandler)

为此缓冲区设置事件处理程序。可以使用 null 调用此方法以取消设置处理程序。

参数

newHandler - - 从此刻起使用的 EventHandler

返回

先前的事件处理程序，如果未设置则为 null。

getEventHandler

public MemoryBuffer.EventHandler getEventHandler()

返回此缓冲区当前的事件处理程序；如果未关联处理程序或此缓冲区已关闭，则返回 null。

close

public void close()

关闭此缓冲区并释放内存

指定者

由接口 AutoCloseable 中的 close 指定

toString

public String toString()

覆盖

覆盖类 Object 中的 toString

incRefCount

public void incRefCount()

增加此列的引用计数。您需要对此对象调用 close 以再次减少引用计数。

getRefCount

public int getRefCount()

获取此缓冲区当前的引用计数。