We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a59f36c commit 834bbc7Copy full SHA for 834bbc7
1 file changed
src/cunumpy/xp.py
@@ -27,13 +27,13 @@ def __init__(
27
28
def _load_backend(self, backend: BackendType, verbose: bool = False) -> ModuleType:
29
if backend == "cupy":
30
- try:
+ if has_cupy():
31
import cupy as cp
32
33
return cp
34
- except ImportError:
+ else:
35
if verbose:
36
- print("CuPy not available.")
+ print("CuPy not available or not functional.")
37
return np
38
import numpy as np_mod
39
@@ -163,7 +163,7 @@ def to_cupy(array: Any) -> Any:
163
164
def to_cunumpy(array: Any) -> Any:
165
"""Convert an array to the currently active backend."""
166
- if array_backend.backend == "cupy":
+ if array_backend.backend == "cupy" and has_cupy():
167
return to_cupy(array)
168
return to_numpy(array)
169
0 commit comments