cupy.cuda.CFunctionAllocator#

class cupy.cuda.CFunctionAllocator(intptr_t param, intptr_t malloc_func, intptr_t free_func, owner)[source]#

使用 C 函数指针进行分配例程的分配器。

此分配器保留指向 param 对象的原始指针以及指向 malloc 和 free 函数的指针，将实际分配委托给外部源，同时仅处理资源分配和释放的时机。

malloc 应该遵循签名 void*(*malloc)(void*, size_t, int)，返回指向已分配内存的指针，给定指向 param 的指针、要分配的字节数以及应进行分配的设备 ID。

类似地，free 应该遵循签名 void(*free)(void*, void*, int)，没有返回值，接受指向 param 的指针、指向已分配内存的指针以及分配内存的设备 ID。

参数:

  • param (int) – param 的地址。

  • malloc_func (int) – malloc 的地址。

  • free_func (int) – free 的地址。

  • owner (object) – 对拥有者对象的引用，以保持 param 和函数存活。

方法

malloc(self, size_t size) → MemoryPointer#
__eq__(value, /)#

返回 self==value。

__ne__(value, /)#

返回 self!=value。

__lt__(value, /)#

返回 self<value。

__le__(value, /)#

返回 self<=value。

__gt__(value, /)#

返回 self>value。

__ge__(value, /)#

返回 self>=value。