From 75f03437a312f65a89de7387c7c6c23d7720a7dc Mon Sep 17 00:00:00 2001 From: Thomas Date: Tue, 16 Apr 2024 07:06:51 +0200 Subject: [PATCH] Removed dependence on CuPy --- mpi4py_fft/__init__.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/mpi4py_fft/__init__.py b/mpi4py_fft/__init__.py index e2ee0b4..10664fe 100644 --- 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