Merge pull request #366 from kaeldai/fix/multicore_iclamp

Adding fix for when running current clamps across multiple cores
AllenInstitute · May 7, 2024 · d1f41a1 · d1f41a1
2 parents 704e4e4 + 2b200cd
commit d1f41a1
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/bmtk/simulator/bionet/modules/iclamp.py b/bmtk/simulator/bionet/modules/iclamp.py
@@ -130,7 +130,7 @@ def __init__(self, input_type, **mod_args):
     def initialize(self, sim):
         # Get select node gids, but only for those nodes that are on the current rank (if running on multiple cores)
         select_gids = list(sim.net.get_node_set(self._node_set).gids())
-        gids_on_rank = list(set(select_gids) & set(select_gids))
+        gids_on_rank = list(set(select_gids) & set(sim.local_gids))
 
         for gid in gids_on_rank:
             cell = sim.net.get_cell_gid(gid)

diff --git a/bmtk/tests/simulator/bionet/test_iclamp.py b/bmtk/tests/simulator/bionet/test_iclamp.py
@@ -38,6 +38,10 @@ def __init__(self):
         self.cell = None
         self.vm = None
 
+    @property
+    def local_gids(self):
+        return [0]
+
     def get_node_set(self, _):
         return self