Removed dependence on CuPy

brownbaerchen · Apr 16, 2024 · 75f0343 · 75f0343
1 parent 77b96f3
commit 75f0343
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/mpi4py_fft/__init__.py b/mpi4py_fft/__init__.py
@@ -20,8 +20,12 @@
 __author__ = 'Lisandro Dalcin and Mikael Mortensen'
 
 from .distarray import DistArray, newDistArray, Function
-from .distarrayCuPy import DistArrayCuPy
 from .mpifft import PFFT
 from . import fftw
 from .fftw import fftlib
 from .io import HDF5File, NCFile, generate_xdmf
+
+try:
+    from .distarrayCuPy import DistArrayCuPy
+except ModuleNotFoundError:
+    pass