Check if the decouple pin is valid. RapidWright integration

Xilinx · Feb 1, 2022 · 5ea3059 · 5ea3059
1 parent a688f46
commit 5ea3059
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/pynq_composable/composable.py b/pynq_composable/composable.py
@@ -542,10 +542,11 @@ def loadIP(self, dfx_list: list) -> None:
                                                   bitname, pr))
 
         path = os.path.dirname(self._bitfile) + '/'
+        decoupler = self._dfx_control[self._dfx_dict[pr]['decouple']]
         for pr in bit_dict:
             if not bit_dict[pr]['loaded']:
-                if self._dfx_control:
-                    self._dfx_control[self._dfx_dict[pr]['decouple']].write(1)
+                if self._dfx_control and decoupler:
+                    decoupler.write(1)
                 for i in range(5):
                     try:
                         self._pr_download(pr, path + bit_dict[pr]['bitstream'])
@@ -557,8 +558,8 @@ def loadIP(self, dfx_list: list) -> None:
                                            "downloaded"
                                            .format(bit_dict[pr]['bitstream']))
 
-                if self._dfx_control:
-                    self._dfx_control[self._dfx_dict[pr]['decouple']].write(0)
+                if self._dfx_control and decoupler:
+                    decoupler.write(0)
 
     def remove(self, iplist: list = None) -> None:
         """Remove IP object from the current pipeline