1 from nmutil
.singlepipe
import ControlBase
2 from nmutil
.pipemodbase
import PipeModBaseChain
3 from soc
.fu
.alu
.input_stage
import ALUInputStage
4 from soc
.fu
.alu
.main_stage
import ALUMainStage
5 from soc
.fu
.alu
.output_stage
import ALUOutputStage
7 class ALUStages(PipeModBaseChain
):
9 inp
= ALUInputStage(self
.pspec
)
10 main
= ALUMainStage(self
.pspec
)
14 class ALUStageEnd(PipeModBaseChain
):
16 out
= ALUOutputStage(self
.pspec
)
20 class ALUBasePipe(ControlBase
):
21 def __init__(self
, pspec
):
22 ControlBase
.__init
__(self
)
24 self
.pipe1
= ALUStages(pspec
)
25 self
.pipe2
= ALUStageEnd(pspec
)
26 self
._eqs
= self
.connect([self
.pipe1
, self
.pipe2
])
28 def elaborate(self
, platform
):
29 m
= ControlBase
.elaborate(self
, platform
)
30 m
.submodules
.pipe1
= self
.pipe1
31 m
.submodules
.pipe2
= self
.pipe2