From 1195022cc885458a0cf69de5539cdb437f3d418b Mon Sep 17 00:00:00 2001 From: Pierre Yger Date: Fri, 13 Sep 2024 14:11:04 +0200 Subject: [PATCH] WIP --- src/spikeinterface/postprocessing/correlograms.py | 4 ++-- src/spikeinterface/postprocessing/isi.py | 2 +- src/spikeinterface/sortingcomponents/matching/tdc.py | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/spikeinterface/postprocessing/correlograms.py b/src/spikeinterface/postprocessing/correlograms.py index 8a12e9b853..27d5703c9e 100644 --- a/src/spikeinterface/postprocessing/correlograms.py +++ b/src/spikeinterface/postprocessing/correlograms.py @@ -435,7 +435,7 @@ def _compute_correlograms_numba(sorting, window_size, bin_size): if HAVE_NUMBA: - @numba.jit(nopython=True, nogil=True, cache=False, parallel=True) + @numba.jit(nopython=True, nogil=True, cache=False) def _compute_correlograms_one_segment_numba( correlograms, spike_times, spike_unit_indices, window_size, bin_size, num_half_bins ): @@ -468,7 +468,7 @@ def _compute_correlograms_one_segment_numba( The size of which to bin lags, in samples. """ start_j = 0 - for i in numba.prange(spike_times.size): + for i in range(spike_times.size): for j in range(start_j, spike_times.size): if i == j: diff --git a/src/spikeinterface/postprocessing/isi.py b/src/spikeinterface/postprocessing/isi.py index 5c9e5f0346..1f635a8c84 100644 --- a/src/spikeinterface/postprocessing/isi.py +++ b/src/spikeinterface/postprocessing/isi.py @@ -182,7 +182,7 @@ def compute_isi_histograms_numba(sorting, window_ms: float = 50.0, bin_ms: float if HAVE_NUMBA: - @numba.jit(nopython=True, nogil=True, cache=False, parallel=True) + @numba.jit(nopython=True, nogil=True, cache=False) def _compute_isi_histograms_numba(ISIs, spike_trains, spike_clusters, bins): n_units = ISIs.shape[0] diff --git a/src/spikeinterface/sortingcomponents/matching/tdc.py b/src/spikeinterface/sortingcomponents/matching/tdc.py index 5c145d1f25..e66929e2b1 100644 --- a/src/spikeinterface/sortingcomponents/matching/tdc.py +++ b/src/spikeinterface/sortingcomponents/matching/tdc.py @@ -348,7 +348,7 @@ def _tdc_find_spikes(traces, d, level=0): if HAVE_NUMBA: - @jit(nopython=True, parallel=True) + @jit(nopython=True) def numba_sparse_dist(wf, templates, union_channels, possible_clusters): """ numba implementation that compute distance from template with sparsity