cupy.cuda.ManagedMemory#

class cupy.cuda.ManagedMemory(size_t size)[source]#

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’

类型:

size