WIP

SpikeInterface · Sep 13, 2024 · 1195022 · 1195022
1 parent 2e3a77b
commit 1195022
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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
@@ -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