Use generator to write configuration to the switch

Xilinx · Dec 3, 2021 · 1d6dad1 · 1d6dad1
1 parent d4f26fa
commit 1d6dad1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pynq_composable/switch.py b/pynq_composable/switch.py
@@ -123,6 +123,6 @@ def _populateRouting(self):
         interfaces are held in reset.
         """
 
-        for i in range(len(self._pi)):
-            self.write(self._pi_offset + 4 * i, int(self._pi[i]))
+        for idx, offset in _mux_mi_gen(self.max_slots):
+            self.write(offset, int(self._pi[idx]))
         self.write(self._control_reg, self._reg_update)