From c916a9d059b3aa772f06c2622854da093db8a13b Mon Sep 17 00:00:00 2001 From: Luke Kenneth Casson Leighton Date: Sun, 20 Feb 2022 22:20:21 +0000 Subject: [PATCH] nope, it's perfectly fine Revert "weird exception, oe not found in the shiftrot input record" This reverts commit 264cc7fd7d7547e1e19424b8f8fd0fbfea29cec5. --- src/soc/fu/shift_rot/pipeline.py | 16 +++++++++++----- 1 file changed, 11 insertions(+), 5 deletions(-) diff --git a/src/soc/fu/shift_rot/pipeline.py b/src/soc/fu/shift_rot/pipeline.py index 80e46038..67dc034c 100644 --- a/src/soc/fu/shift_rot/pipeline.py +++ b/src/soc/fu/shift_rot/pipeline.py @@ -4,11 +4,15 @@ from soc.fu.shift_rot.input_stage import ShiftRotInputStage from soc.fu.shift_rot.main_stage import ShiftRotMainStage from soc.fu.shift_rot.output_stage import ShiftRotOutputStage -class ShiftRotStages(PipeModBaseChain): +class ShiftRotStart(PipeModBaseChain): def get_chain(self): inp = ShiftRotInputStage(self.pspec) + return [inp] + +class ShiftRotStage(PipeModBaseChain): + def get_chain(self): main = ShiftRotMainStage(self.pspec) - return [inp, main] + return [main] class ShiftRotStageEnd(PipeModBaseChain): @@ -21,13 +25,15 @@ class ShiftRotBasePipe(ControlBase): def __init__(self, pspec): ControlBase.__init__(self) self.pspec = pspec - self.pipe1 = ShiftRotStages(pspec) - self.pipe2 = ShiftRotStageEnd(pspec) - self._eqs = self.connect([self.pipe1, self.pipe2]) + self.pipe1 = ShiftRotStart(pspec) + self.pipe2 = ShiftRotStage(pspec) + self.pipe3 = ShiftRotStageEnd(pspec) + self._eqs = self.connect([self.pipe1, self.pipe2, self.pipe3]) def elaborate(self, platform): m = ControlBase.elaborate(self, platform) m.submodules.pipe1 = self.pipe1 m.submodules.pipe2 = self.pipe2 + m.submodules.pipe3 = self.pipe3 m.d.comb += self._eqs return m -- 2.30.2