From e02509177374a5e56724b07a968714f0d770e187 Mon Sep 17 00:00:00 2001 From: Bence Bagi Date: Thu, 14 Mar 2024 19:29:55 +0000 Subject: [PATCH 1/3] Use GPU in Kilosort4Sorter --- src/spikeinterface/sorters/external/kilosort4.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/spikeinterface/sorters/external/kilosort4.py b/src/spikeinterface/sorters/external/kilosort4.py index 236db8bb5b..d96c68be4b 100644 --- a/src/spikeinterface/sorters/external/kilosort4.py +++ b/src/spikeinterface/sorters/external/kilosort4.py @@ -15,6 +15,7 @@ class Kilosort4Sorter(BaseSorter): sorter_name: str = "kilosort4" requires_locations = True + gpu_compatibility = "nvidia-required" _default_params = { "nblocks": 1, From 6e6d9bd22e7907ba79f1662933252da9ea2947d5 Mon Sep 17 00:00:00 2001 From: Bence Bagi Date: Fri, 15 Mar 2024 10:56:05 +0100 Subject: [PATCH 2/3] Use nvidia-optional instead of nvidia-required for Kilosort4 Co-authored-by: Alessio Buccino --- src/spikeinterface/sorters/external/kilosort4.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/spikeinterface/sorters/external/kilosort4.py b/src/spikeinterface/sorters/external/kilosort4.py index d96c68be4b..de5cb10f02 100644 --- a/src/spikeinterface/sorters/external/kilosort4.py +++ b/src/spikeinterface/sorters/external/kilosort4.py @@ -15,7 +15,7 @@ class Kilosort4Sorter(BaseSorter): sorter_name: str = "kilosort4" requires_locations = True - gpu_compatibility = "nvidia-required" + gpu_compatibility = "nvidia-optional" _default_params = { "nblocks": 1, From cb3f78ed83e70291c8aaa2d2bf7a698f4a1881d8 Mon Sep 17 00:00:00 2001 From: Bence Bagi Date: Fri, 15 Mar 2024 10:44:26 +0000 Subject: [PATCH 3/3] gpu_compatibility -> gpu_capability in Kilosort4Sorter --- src/spikeinterface/sorters/external/kilosort4.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/spikeinterface/sorters/external/kilosort4.py b/src/spikeinterface/sorters/external/kilosort4.py index de5cb10f02..9f7ce929c9 100644 --- a/src/spikeinterface/sorters/external/kilosort4.py +++ b/src/spikeinterface/sorters/external/kilosort4.py @@ -15,7 +15,7 @@ class Kilosort4Sorter(BaseSorter): sorter_name: str = "kilosort4" requires_locations = True - gpu_compatibility = "nvidia-optional" + gpu_capability = "nvidia-optional" _default_params = { "nblocks": 1,