1 from nmutil
.singlepipe
import ControlBase
2 from nmutil
.pipemodbase
import PipeModBaseChain
3 from soc
.fu
.branch
.main_stage
import BranchMainStage
5 class BranchStages(PipeModBaseChain
):
7 main
= BranchMainStage(self
.pspec
)
11 class BranchBasePipe(ControlBase
):
12 def __init__(self
, pspec
):
13 ControlBase
.__init
__(self
)
15 self
.pipe1
= BranchStages(pspec
)
16 self
._eqs
= self
.connect([self
.pipe1
])
18 def elaborate(self
, platform
):
19 m
= ControlBase
.elaborate(self
, platform
)
20 m
.submodules
.pipe
= self
.pipe1