From 23f3153b14fd11a357935ce214fb78aa17c2b999 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <maltanar@gmail.com>
Date: Tue, 11 Aug 2020 16:40:02 +0200
Subject: [PATCH] [Driver] fix wait condition for zynq-iodma

---
 src/finn/transformation/fpgadataflow/templates.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/finn/transformation/fpgadataflow/templates.py b/src/finn/transformation/fpgadataflow/templates.py
index 036286046..eaeadc7b3 100644
--- a/src/finn/transformation/fpgadataflow/templates.py
+++ b/src/finn/transformation/fpgadataflow/templates.py
@@ -211,8 +211,10 @@ class FINNAccelDriver():
             self.idma.write(0x00, 1)
             self.odma.write(0x00, 1)
             # wait until output IODMA is finished
-            while self.odma.read(0x00) and 2 == 0:
-                pass
+            status = self.odma.read(0x00)
+            while status & 0x2 == 0:
+                status = self.odma.read(0x00)
+
         elif self.platform == "alveo":
             self.ibuf_packed_device.sync_to_device()
             self.idma.start(self.ibuf_packed_device, self.N)
-- 
GitLab