Fix size of region

SpiNNakerManchester · Oct 12, 2024 · 6daa42a · 6daa42a
1 parent 502f0e0
commit 6daa42a
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/spynnaker/pyNN/models/neural_projections/connectors/kernel_connector.py b/spynnaker/pyNN/models/neural_projections/connectors/kernel_connector.py
@@ -52,7 +52,7 @@
     float, int, List[float], NDArray[numpy.floating], RandomDistribution]
 
 HEIGHT, WIDTH = 0, 1
-N_KERNEL_PARAMS = 8
+N_KERNEL_PARAMS = 9
 
 
 class ConvolutionKernel(ndarray):
@@ -521,7 +521,12 @@ def gen_connector_params(
     @overrides(
         AbstractGenerateConnectorOnMachine.gen_connector_params_size_in_bytes)
     def gen_connector_params_size_in_bytes(self) -> int:
-        return N_KERNEL_PARAMS * BYTES_PER_WORD
+        size = N_KERNEL_PARAMS * BYTES_PER_WORD
+        if self._krn_weights is not None:
+            size += len(self._krn_weights) * BYTES_PER_WORD
+        if self._krn_delays is not None:
+            size += len(self._krn_delays) * BYTES_PER_WORD
+        return size
 
     @overrides(AbstractGenerateConnectorOnMachine.get_connected_vertices)
     def get_connected_vertices(