Skip to content

Commit f1f1955

Browse files
committed
hfft needs cuda for the same reason
1 parent 4a13cd7 commit f1f1955

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mne/cuda.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -216,7 +216,7 @@ def _setup_cuda_fft_multiply_repeated(n_jobs, h, n_fft, kind="FFT FIR filtering"
216216

217217
try:
218218
# do the IFFT normalization now so we don't have to later
219-
h_fft = cupy.asarray(_share_cuda_mem(cuda_dict["h_fft"], n_jobs))
219+
h_fft = cupy.asarray(_share_cuda_mem(cuda_dict["h_fft"], "cuda"))
220220
logger.info(f"Using CUDA for {kind}")
221221
except Exception as exp:
222222
logger.info(

0 commit comments

Comments
 (0)