cupy.cuda.MemoryAsync

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

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

此类提供 CUDA 内存分配的 RAII 接口。

参数:

• size (int) – 内存分配的大小（字节）。

• stream (Stream) – 分配和释放内存时使用的流。

方法

__eq__(value, /)

返回 self==value。

__ne__(value, /)

返回 self!=value。

__lt__(value, /)

返回 self

__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’

类型:

size

stream_ref