cupy.cuda.MemoryAsync

class cupy.cuda.MemoryAsync(size_t size, stream)

在CUDA设备上的异步内存分配。

此类提供了一个 RAII 接口，用于 CUDA 内存分配。

参数:

• size (int) – 内存分配的大小，单位为字节。

• stream (Stream) – 内存分配和释放所在的流。

方法

__eq__(value, /)

返回 self==value。

__ne__(value, /)

返回 self!=value。

__lt__(value, /)

返回 self<value。

__le__(value, /)

返回 self<=value。

__gt__(value, /)

返回 self>value。

__ge__(value, /)

返回 self>=value。

属性

device

device_id

‘int’

类型:

device_id

ptr

‘intptr_t’

类型:

ptr

size

‘size_t’

类型:

大小

stream_ref