cupy.cuda.ManagedMemory#

class cupy.cuda.ManagedMemory(size_t size)[源代码]#

在CUDA设备上的托管内存（统一内存）分配。

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

参数:

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

方法

advise(self, int advise, Device dev)#

（实验性）关于此内存使用的建议。

参数:

  • advics (int) – 建议应用于此内存。

  • dev (cupy.cuda.Device) – 应用建议的设备。

prefetch(self, stream)#

(实验性) 预取内存。

参数:

stream (cupy.cuda.Stream) – CUDA 流。

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

类型:

大小