sinabs 3.1.0__py3-none-any.whl → 3.1.1__py3-none-any.whl

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.0.dist-info → sinabs-3.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sinabs
-Version: 3.1.0
+Version: 3.1.1
 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.0.dist-info → sinabs-3.1.1.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
@@ -28,7 +28,7 @@ sinabs/backend/dynapcnn/dynapcnn_visualizer.py,sha256=MRewU6519dAtAMxf-JlFBrlynJ
 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.0.dist-info/licenses/AUTHORS,sha256=rSozYAiy4aFshSqHdw7bV-CyqtIyWhcPo26R5JR6-MY,1847
-sinabs-3.1.0.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-sinabs-3.1.0.dist-info/METADATA,sha256=t6r_2U8fyGCJ9M-_AJwOWUwXk1yXm-ZehyXyV_222LM,3913
-sinabs-3.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sinabs-3.1.0.dist-info/pbr.json,sha256=x8yH_fuaMueIOtdAfzIwTGqy2Zk3gtedaesdhHIselc,47
-sinabs-3.1.0.dist-info/top_level.txt,sha256=QOXGzf0ZeDjRnJ9OgAjkk6h5jrh66cwrwvtPJTyfDk8,7
-sinabs-3.1.0.dist-info/RECORD,,
+sinabs-3.1.1.dist-info/licenses/AUTHORS,sha256=rSozYAiy4aFshSqHdw7bV-CyqtIyWhcPo26R5JR6-MY,1847
+sinabs-3.1.1.dist-info/licenses/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+sinabs-3.1.1.dist-info/METADATA,sha256=N9GAqWDhMyzvI8UFqXW9IS1q5aKktZhm6rJleHlxDvg,3910
+sinabs-3.1.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sinabs-3.1.1.dist-info/pbr.json,sha256=Pzslm554HrYvSZWJBrXHy8I-yJ0JXOBa7BO2xjmMwAI,47
+sinabs-3.1.1.dist-info/top_level.txt,sha256=QOXGzf0ZeDjRnJ9OgAjkk6h5jrh66cwrwvtPJTyfDk8,7
+sinabs-3.1.1.dist-info/RECORD,,

sinabs-3.1.1.dist-info/pbr.json

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

sinabs-3.1.0.dist-info/pbr.json

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