Recommended hardware for mne.chpi.compute_chpi_amplitudes?

[agerhardsson]

Looking to speed up data processing in general, but particularly for compute_chpi_amplitudes.

Does the chpi_amplitute computation use primarily CPU or GPU?

[mscheltienne (Mathieu Scheltienne)]

Very likely CPU, I believe it automatically scales up to use all cores without even exposing an n_jobs argument to control the number of workers.

Mathieu

[drammock (Dan McCloy)]

the core computation (in private func _fit_chpi_amplitudes) is numba-compatible, so mne.set_config("MNE_USE_NUMBA", "true") might yield a big improvement for you if you’re able to use Numba and currently aren’t.