CuPyAsyncNumbaManager

class cubie.memory.cupy_emm.CuPyAsyncNumbaManager(context)

Bases: CuPyNumbaManager

Numba EMM plugin using CuPy MemoryAsyncPool for allocation and freeing.

Parameters:

context – CUDA context for memory management.

Notes

Uses CuPy’s asynchronous memory pool which provides stream-ordered memory operations.

initialize() → None

Initialize the async memory pool.

Returns:

None.

Return type:

None

memalloc(nbytes: int) → FakeMemoryPointer

Allocate memory from the async CuPy pool.

Parameters:

nbytes – Number of bytes to allocate.

Returns:

Numba memory pointer wrapping the CuPy allocation.

Return type:

MemoryPointer