current_cupy_stream

cubie.memory.cupy_emm.current_cupy_stream(nb_stream: FakeStream) → None

Context manager that forwards a Numba stream into CuPy APIs.

Parameters:

nb_stream – Numba CUDA stream to expose to CuPy.

cubie.memory.cupy_emm.nb_stream

The Numba stream being forwarded.

cubie.memory.cupy_emm.cupy_ext_stream

CuPy external stream wrapper around the Numba stream when CuPy manages the active allocator.

Notes

This context manager only affects CuPy-backed memory managers. When the active manager is not CuPy-based, the context operates as a no-op.