Add offset to getSpGid in case of load balance

BlueBrain · Apr 15, 2024 · 7407274 · 7407274
1 parent dd9a0f8
commit 7407274
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/neurodamus/cell_distributor.py b/neurodamus/cell_distributor.py
@@ -506,7 +506,8 @@ def getSpGid(self, gid):
         """
         manager = self._find_manager(gid)
         if manager._binfo:
-            return manager._binfo.thishost_gid(gid - manager.local_nodes.offset)
+            gid_offset = manager.local_nodes.offset
+            return manager._binfo.thishost_gid(gid - gid_offset) + gid_offset
         return gid
 
     def getPopulationInfo(self, gid):