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