cupy.cuda.malloc_async#
- cupy.cuda.malloc_async(size_t size) MemoryPointer#
(实验性) 从流有序内存分配器分配内存。
此方法可用作 CuPy 内存分配器。将 CUDA 流有序内存分配器用作默认分配器的最简单方法如下：
set_allocator(malloc_async)
使用此功能需要 CUDA 版本 >= 11.2 并具有支持的 GPU 和平台。如果不支持，将引发错误。
当前 CuPy 流用于分配/释放内存。
- 参数:
size (int) – 要分配的内存大小（字节）。
- 返回:
指向已分配缓冲区的指针。
- 返回类型:
警告
此功能目前处于实验阶段，可能会发生变化。
另请参阅