magtrack._cupy
Compatibility helpers for optional CuPy support.
This module exposes cp and cupyx objects that either proxy to the real
CuPy modules (when available) or provide lightweight NumPy/SciPy fallbacks.
Functions
is_cupy_available(→ bool)
|
Return True when the real CuPy package is importable. |
check_cupy(→ bool)
|
Perform a more thorough check to ensure CuPy and CUDA are usable. |
Module Contents
-
magtrack._cupy._cupy_available = False[source]
-
magtrack._cupy._cupy = None[source]
-
magtrack._cupy.cp[source]
-
magtrack._cupy.is_cupy_available() → bool[source]
Return True when the real CuPy package is importable.
-
magtrack._cupy.check_cupy() → bool[source]
Perform a more thorough check to ensure CuPy and CUDA are usable.