import was missing

ptycho · Jan 31, 2024 · 7363883 · 7363883
1 parent 4854351
commit 7363883
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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,