diff --git a/ptypy/accelerate/cuda_pycuda/cufft.py b/ptypy/accelerate/cuda_pycuda/cufft.py index 3ff54d55c..5364f092d 100644 --- a/ptypy/accelerate/cuda_pycuda/cufft.py +++ b/ptypy/accelerate/cuda_pycuda/cufft.py @@ -142,6 +142,7 @@ def _load(self, array, pre_fft, post_fft, symmetric, forward): int((self.arr_shape[1] + 31) // 32), int(self.batches) ) + import skcuda.fft as cu_fft self.plan = cu_fft.Plan( self.arr_shape, array.dtype,