use output from align as input to add0

diff --git a/src/add/nmigen_add_experiment.py b/src/add/nmigen_add_experiment.py
index 0bc19a333d503b9cfe5019a5327f65f45cfcf816..1b9fca1f1765d88bd0b6afd79029b401607c1ad4 100644 (file)
--- a/src/add/nmigen_add_experiment.py
+++ b/src/add/nmigen_add_experiment.py
@@ -811,9 +811,9 @@ class FPADD:
         m.submodules.align = alm.mod
 
         add0 = self.add_state(FPAddStage0(self.width))
-        add0.set_inputs({"a": a, "b": b})
+        add0.set_inputs({"a": alm.out_a, "b": alm.out_b})
         add0.set_outputs({"z": z})
-        add0.mod.setup(m, a, b, z, add0.out_z, add0.out_tot)
+        add0.mod.setup(m, alm.out_a, alm.out_b, z, add0.out_z, add0.out_tot)
         m.submodules.add0 = add0.mod
 
         add1 = self.add_state(FPAddStage1(self.width))