From 151820fe5a4e0420506eb1dcf5cddcbd9179a50d Mon Sep 17 00:00:00 2001 From: Andrew Rowley Date: Sat, 12 Oct 2024 13:20:40 +0100 Subject: [PATCH] Get 2d size correct --- .../models/neural_projections/connectors/kernel_connector.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/spynnaker/pyNN/models/neural_projections/connectors/kernel_connector.py b/spynnaker/pyNN/models/neural_projections/connectors/kernel_connector.py index 22e630eef9..f60549d91e 100644 --- a/spynnaker/pyNN/models/neural_projections/connectors/kernel_connector.py +++ b/spynnaker/pyNN/models/neural_projections/connectors/kernel_connector.py @@ -523,9 +523,9 @@ def gen_connector_params( def gen_connector_params_size_in_bytes(self) -> int: size = N_KERNEL_PARAMS * BYTES_PER_WORD if self._krn_weights is not None: - size += len(self._krn_weights) * BYTES_PER_WORD + size += sum(len(x) for x in self._krn_weights) * BYTES_PER_WORD if self._krn_delays is not None: - size += len(self._krn_delays) * BYTES_PER_WORD + size += sum(len(x) for x in self._krn_delays) * BYTES_PER_WORD return size @overrides(AbstractGenerateConnectorOnMachine.get_connected_vertices)