hfft needs cuda for the same reason

mne-tools · Dec 4, 2024 · f1f1955 · f1f1955
1 parent 4a13cd7
commit f1f1955
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mne/cuda.py b/mne/cuda.py
@@ -216,7 +216,7 @@ def _setup_cuda_fft_multiply_repeated(n_jobs, h, n_fft, kind="FFT FIR filtering"
 
             try:
                 # do the IFFT normalization now so we don't have to later
-                h_fft = cupy.asarray(_share_cuda_mem(cuda_dict["h_fft"], n_jobs))
+                h_fft = cupy.asarray(_share_cuda_mem(cuda_dict["h_fft"], "cuda"))
                 logger.info(f"Using CUDA for {kind}")
             except Exception as exp:
                 logger.info(