sinabs 3.1.1.dev1__py3-none-any.whl → 3.1.2__py3-none-any.whl

sinabs/backend/dynapcnn/dynapcnn_visualizer.py

@@ -418,7 +418,9 @@ class DynapcnnVisualizer:
                 + " should contain value `dvs`. "
             )
 
-        last_layer = dynapcnn_network.chip_layers_ordering[-1]
+        last_layer = dynapcnn_network.layer2core_map[
+            dynapcnn_network.exit_layer_ids[-1]
+        ]
 
         if not config.cnn_layers[last_layer].monitor_enable:
             raise ValueError(
@@ -568,7 +570,9 @@ class DynapcnnVisualizer:
             dynapcnn_network (DynapcnnNetwork): DynapcnnNetwork object
 
         """
-        last_layer = dynapcnn_network.chip_layers_ordering[-1]
+        last_layer = dynapcnn_network.layer2core_map[
+            dynapcnn_network.exit_layer_ids[-1]
+        ]
         config = dynapcnn_network.samna_config
         model_output_feature_count = config.cnn_layers[
             last_layer

sinabs/backend/dynapcnn/io.py

@@ -133,8 +133,11 @@ def get_device_map() -> Dict:
     # Group by device_type_name
     device_groups = groupby(devices, lambda x: x.device_type_name)
     # Switch keys from samna's device_type_name to device_type names
+    # -- guarantee is a supported device
     device_groups = {
-        device_type_map[k]: sort_devices(list(v)) for k, v in device_groups
+        device_type_map[k]: sort_devices(list(v))
+        for k, v in device_groups
+        if k in device_type_map
     }
     # Flat map
     for dev_type, dev_list in device_groups.items():

sinabs/nir.py

@@ -46,8 +46,8 @@ def _import_sinabs_module(
             groups=node.groups,
             bias=True,
         )
-        conv.weight.data = torch.tensor(node.weight).float()
-        conv.bias.data = torch.tensor(node.bias).float()
+        conv.weight.data = node.weight.detach().clone().to(float)
+        conv.bias.data = node.bias.detach().clone().to(float)
         return conv
 
     elif isinstance(node, nir.Conv2d):
@@ -184,6 +184,7 @@ def _extract_sinabs_module(module: torch.nn.Module) -> Optional[nir.NIRNode]:
             return nir.Affine(module.weight.detach(), module.bias.detach())
     elif isinstance(module, torch.nn.Conv1d):
         return nir.Conv1d(
+            input_shape=None,
             weight=module.weight.detach(),
             stride=module.stride,
             padding=module.padding,
@@ -191,7 +192,7 @@ def _extract_sinabs_module(module: torch.nn.Module) -> Optional[nir.NIRNode]:
             groups=module.groups,
             bias=(
                 module.bias.detach()
-                if module.bias
+                if isinstance(module.bias, torch.Tensor)
                 else torch.zeros((module.weight.shape[0]))
             ),
         )

{sinabs-3.1.1.dev1.dist-info → sinabs-3.1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sinabs
-Version: 3.1.1.dev1
+Version: 3.1.2
 Summary: SynSense Spiking Neural Network simulator for deep neural networks (DNNs).
 Author: SynSense (formerly AiCTX)
 Author-email: support@synsense.ai
@@ -87,7 +87,7 @@ Sinabs is published under Apache v2.0. See the LICENSE file for details.
 
 Contributing to Sinabs
 ------------------------
-Checkout the [contributing](https://sinabs.readthedocs.io/en/develop/about/contributing.html) page for more info.
+Checkout the [contributing](https://sinabs.readthedocs.io/develop/about/contributing.html) page for more info.
 
 
 Citation

{sinabs-3.1.1.dev1.dist-info → sinabs-3.1.2.dist-info}/RECORD

@@ -4,7 +4,7 @@ sinabs/conversion.py,sha256=wEACb-IMrsNcjoNvGjoaQSINHYs66j0XOeQSmk7nhTM,2011
 sinabs/from_torch.py,sha256=_jVoGP3aQcdCUIc39ujxhCQmbIDlDtzLHz6VY3zmX-k,5777
 sinabs/hooks.py,sha256=u6bzKXodhUHDLi_ZaUNbD8hogRX7RD4QoRouSUaKEzY,15962
 sinabs/network.py,sha256=Zn7fQ0WTU_7vgBwfVloGINrXivZTNIBsmUGLWAbX81A,9542
-sinabs/nir.py,sha256=r72RZ2WNuhnHvQ2MaSJ04J-Bes1mAdzoU9LKbJupZzE,8695
+sinabs/nir.py,sha256=LUUlWbV2jXJPWby5SoHFl0_c5tMVaQlsw-U_s-WjecY,8761
 sinabs/synopcounter.py,sha256=VFefA5ix0JqCr2DPf8hNB19Y0Pe8tTvSrCuspFKa4Ic,12779
 sinabs/utils.py,sha256=1ctTxNT2ONYVK5d99VhZvXwxaOG4pD2lZ_PGRPPO6bQ,12114
 sinabs/validate_memory_speck.py,sha256=WKzB3o76sddUx9griloLQ6-ijDBPqFFXshdp9AjEiAM,5002
@@ -24,11 +24,11 @@ sinabs/backend/dynapcnn/dvs_layer.py,sha256=b9OryaOv5X5ZOxGOEZhabHZf3jZoykDuB295
 sinabs/backend/dynapcnn/dynapcnn_layer.py,sha256=Pf-Qhl24LwKyu-7sJXDRYQ_3DN_7Z3MQyj3lNsGKTaE,8917
 sinabs/backend/dynapcnn/dynapcnn_layer_utils.py,sha256=9aPk9YuAgbfI_Xworfy-SbJDQSRVFksqaLU3ZaCmS74,13153
 sinabs/backend/dynapcnn/dynapcnn_network.py,sha256=klYI82qV1IqhDyD_OkOs5G3F-FysTtJJRgQbFEpDrfg,34721
-sinabs/backend/dynapcnn/dynapcnn_visualizer.py,sha256=MRewU6519dAtAMxf-JlFBrlynJTZeLiDfB0d85-mMFQ,24262
+sinabs/backend/dynapcnn/dynapcnn_visualizer.py,sha256=XspRBAWgeka0wtYOUPhXCks1VR9LRK0yO52tlV8xydg,24360
 sinabs/backend/dynapcnn/dynapcnnnetwork_module.py,sha256=aHLTAe59h_r0rZ0bHeSuIDG811MFP6_tOtyxiZIVVfA,14487
 sinabs/backend/dynapcnn/exceptions.py,sha256=MHvsM8a5c5ukHkH59OVYU0AeivVeNgdn9Tgqb07WOtU,3523
 sinabs/backend/dynapcnn/flipdims.py,sha256=I0I1nakrF0ngWBh-2SHHg7OkCOxotqukwHOQ45GWyCs,860
-sinabs/backend/dynapcnn/io.py,sha256=8m86xm0BcLUc8AGLaA30FyY5OXFu0NrpOaPk4zLCfJU,9918
+sinabs/backend/dynapcnn/io.py,sha256=-XtP801uVragUlBf-6PjR5UHY6B0fcy1pUg0M6QlQKo,9999
 sinabs/backend/dynapcnn/mapping.py,sha256=sI0GKh2FkyciYFJuuCLCcitOzlEjfEEEbSG_OiJXICw,8683
 sinabs/backend/dynapcnn/nir_graph_extractor.py,sha256=1PbQzDokxW_0mzG_tdMSS1toHczW-2lnKGMUkdmCug8,37724
 sinabs/backend/dynapcnn/sinabs_edges_handler.py,sha256=pEfGUXwhQy7cnmofK5MCulSVFqMBDPRRoQVM5MTf3Qk,41085
@@ -56,10 +56,10 @@ sinabs/layers/to_spike.py,sha256=ORkEGuHucWCqB3P1Ia8XJsPtpDAugCTSo7Bwr3xVpzE,329
 sinabs/layers/functional/__init__.py,sha256=v0c7DHizKg8jfelmFYeMMg9vDafKvzoenakc4SPpj84,91
 sinabs/layers/functional/alif.py,sha256=ycJ7rlcBAd-lq5GCDZrcNPeV-7fztt3uy43XhBtTKHI,4599
 sinabs/layers/functional/lif.py,sha256=QRjiWDCBaJFk4J7RRMgktMaLCyN6xEXAKvC9Bu_PICU,4259
-sinabs-3.1.1.dev1.dist-info/licenses/AUTHORS,sha256=rSozYAiy4aFshSqHdw7bV-CyqtIyWhcPo26R5JR6-MY,1847
-sinabs-3.1.1.dev1.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-sinabs-3.1.1.dev1.dist-info/METADATA,sha256=TBK76i04bGarg88j-cJoYoakH9PFUqHTKJV5Vrx3KYk,3918
-sinabs-3.1.1.dev1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sinabs-3.1.1.dev1.dist-info/pbr.json,sha256=DsGf0oO1zpKuMU-ZkpK0AxRzUJVUeBhbXdrMY2BeHvs,47
-sinabs-3.1.1.dev1.dist-info/top_level.txt,sha256=QOXGzf0ZeDjRnJ9OgAjkk6h5jrh66cwrwvtPJTyfDk8,7
-sinabs-3.1.1.dev1.dist-info/RECORD,,
+sinabs-3.1.2.dist-info/licenses/AUTHORS,sha256=rSozYAiy4aFshSqHdw7bV-CyqtIyWhcPo26R5JR6-MY,1847
+sinabs-3.1.2.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+sinabs-3.1.2.dist-info/METADATA,sha256=KvGjdE-mfmK1kht5iuMPoX0rmlRFPEx9-gcHLqs-Qe0,3910
+sinabs-3.1.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sinabs-3.1.2.dist-info/pbr.json,sha256=Bk2ezb0sWaVYu7NrXyhkvYjDDRegj9bE_D4h05EOcs4,47
+sinabs-3.1.2.dist-info/top_level.txt,sha256=QOXGzf0ZeDjRnJ9OgAjkk6h5jrh66cwrwvtPJTyfDk8,7
+sinabs-3.1.2.dist-info/RECORD,,

sinabs-3.1.2.dist-info/pbr.json

@@ -0,0 +1 @@
+{"git_version": "15f4caa", "is_release": false}

sinabs-3.1.1.dev1.dist-info/pbr.json

@@ -1 +0,0 @@
-{"git_version": "dbf7eee", "is_release": false}