Did get_a state in nmigen

diff --git a/src/add/fmul.py b/src/add/fmul.py
index 7abf072a8680337173c9e81a6d1e3919a8985536..d8192c6bfb7877f87667bcf2556513ae68391933 100644 (file)
--- a/src/add/fmul.py
+++ b/src/add/fmul.py
@@ -30,8 +30,17 @@ class FPMUL(FPBase):
 
         with m.FSM() as fsm:
 
- """
-   always @(posedge clk)
+               with m.State("get_a"):
+                       m.next += "get_b"
+                       m.d.sync += s.in_a.ack.eq(1)
+                       with m.If(s.in_a.ack & in_a.stb):
+                               m.d.sync += [
+                               a.eq(in_a),
+                               s.in_a.ack(0)
+                       ]
+
+"""
+always @(posedge clk)
   begin
 
     case(state)