fix bug

AlexMontgomerie · May 11, 2024 · a274ef1 · a274ef1
1 parent b7c101c
commit a274ef1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/fpgaconvnet/models/layers/ConvolutionPointwiseSparseLayer.py b/fpgaconvnet/models/layers/ConvolutionPointwiseSparseLayer.py
@@ -216,5 +216,7 @@ def resource(self):
         rsc = super().resource()
         # when sparsity occurs, the crossbar in sparse_vector_dot already acts as a squeeze
         squeeze_rsc = self.modules['squeeze'].rsc()
-        rsc = { rsc_type: rsc[rsc_type] - squeeze_rsc[rsc_type] for rsc_type in ["LUT", "FF", "DSP", "BRAM"] }
+        for rsc_type in squeeze_rsc.keys():
+            if rsc_type in rsc.keys():
+                rsc[rsc_type] -= squeeze_rsc[rsc_type]
         return rsc
diff --git a/fpgaconvnet/models/layers/ConvolutionSparseLayer.py b/fpgaconvnet/models/layers/ConvolutionSparseLayer.py
@@ -233,6 +233,8 @@ def resource(self):
         rsc = super().resource()
         # when sparsity occurs, the crossbar in sparse_vector_dot already acts as a squeeze
         squeeze_rsc = self.modules['squeeze'].rsc()
-        rsc = { rsc_type: rsc[rsc_type] - squeeze_rsc[rsc_type] for rsc_type in ["LUT", "FF", "DSP", "BRAM"] }
+        for rsc_type in squeeze_rsc.keys():
+            if rsc_type in rsc.keys():
+                rsc[rsc_type] -= squeeze_rsc[rsc_type]
         return rsc