hooray, p_o_ready works

[ieee754fpu.git] / src / add / singlepipe.py

diff --git a/src/add/singlepipe.py b/src/add/singlepipe.py
index e32a0c67a2665176bd92c4c95247621c5ed139c5..2cb7d1ab06bacdc11f95ac14848d0b2b70aed308 100644 (file)
--- a/src/add/singlepipe.py
+++ b/src/add/singlepipe.py
@@ -197,8 +197,12 @@ class PrevControl:
         vlen = len(self.i_valid)
         if vlen > 1: # multi-bit case: valid only when i_valid is all 1s
             all1s = Const(-1, (len(self.i_valid), False))
+            if self.stage_ctl:
+                return self.i_valid == all1s & self.s_o_ready
             return self.i_valid == all1s
         # single-bit i_valid case
+        if self.stage_ctl:
+            return self.i_valid & self.s_o_ready
         return self.i_valid