Add m.d.sync return to get_a and get_b cases

diff --git a/src/add/fmul.py b/src/add/fmul.py
index a65700eb0da0388b647a69a1d401af9c729d7436..25944a83ef1ca136aa2868a609e847d8408171fe 100644 (file)
--- a/src/add/fmul.py
+++ b/src/add/fmul.py
@@ -3,6 +3,7 @@ from nmigen.cli import main, verilog
 
 from fpbase import FPNumIn, FPNumOut, FPOp, Overflow, FPBase, FPState
 from fpcommon.getop import FPGetOp
+from singlepipe import eq
 
 
 class FPMUL(FPBase):
@@ -15,6 +16,12 @@ class FPMUL(FPBase):
         self.in_b  = FPOp(width)
         self.out_z = FPOp(width)
 
+       self.states = []
+
+    def add_state(self, state):
+       self.states.append(state)
+       return state
+
     def get_fragment(self, platform=None):
         """ creates the HDL code-fragment for FPMUL
         """
@@ -34,19 +41,24 @@ class FPMUL(FPBase):
         m.submodules.b = b
         m.submodules.z = z
 
+        m.d.comb += a.v.eq(self.a.v)
+        m.d.comb += b.v.eq(self.b.v)
+
         with m.FSM() as fsm:
 
             # ******
             # gets operand a
 
             with m.State("get_a"):
-                self.get_op(m, self.in_a, a, "get_b")
+                res = self.get_op(m, self.in_a, a, "get_b")
+               m.d.sync += eq([a, self.in_a.ack], res)
 
             # ******
             # gets operand b
 
             with m.State("get_b"):
-                self.get_op(m, self.in_b, b, "special_cases")
+                res = self.get_op(m, self.in_b, b, "special_cases")
+               m.d.sync += eq([b, self.in_b.ack], res)
 
             # ******
             # special cases