BaseCUDAMemoryManager

class nvmath.BaseCUDAMemoryManager(*args, **kwargs)

内存管理器插件的协议。

另请参阅

numba.cuda.BaseCUDAMemoryManager

方法

__init__(*args, **kwargs)

abstract memalloc(size)

分配设备内存。

参数:

size – 内存缓冲区的大小，以字节为单位。

返回值:

一个对象，该对象拥有已分配的内存，并负责释放它（到操作系统或池）。该对象必须具有名为 device_ptr、device_pointer 或 ptr 的属性，用于指定指向已分配的内存缓冲区的指针。 有关示例接口，请参阅 MemoryPointer。

注意

类型为 numba.cuda.MemoryPointer 以及 cupy.cuda.MemoryPointer 的对象满足上面列出的设备内存指针对象的要求。