add process function to modules, about to use StageChain

diff --git a/src/add/nmigen_add_experiment.py b/src/add/nmigen_add_experiment.py
index f340897fdcd9ed8de6c981db22446c01d942c6ff..662f1fb81f3a06ea126ed9fc261ab60706cd029b 100644 (file)
--- a/src/add/nmigen_add_experiment.py
+++ b/src/add/nmigen_add_experiment.py
@@ -1021,6 +1021,9 @@ class FPNorm1ModSingle:
         m.submodules.normalise_1 = self
         m.d.comb += self.i.eq(i)
 
+    def process(self, i):
+        return self.o
+
     def elaborate(self, platform):
         m = Module()
 
@@ -1306,6 +1309,9 @@ class FPRoundMod:
     def ospec(self):
         return FPRoundData(self.width, self.id_wid)
 
+    def process(self, i):
+        return self.out_z
+
     def setup(self, m, i):
         m.submodules.roundz = self
         m.d.comb += self.i.eq(i)
@@ -1360,6 +1366,9 @@ class FPCorrectionsMod:
     def ospec(self):
         return FPRoundData(self.width, self.id_wid)
 
+    def process(self, i):
+        return self.out_z
+
     def setup(self, m, i):
         """ links module to inputs and outputs
         """
@@ -1425,6 +1434,9 @@ class FPPackMod:
     def ospec(self):
         return FPPackData(self.width, self.id_wid)
 
+    def process(self, i):
+        return self.o
+
     def setup(self, m, in_z):
         """ links module to inputs and outputs
         """