src/soc/fu/common_output_stage.py

   1 # This stage is intended to handle the gating of carry out,
   2 # and updating the condition register
   3 from nmigen import (Module, Signal, Cat, Repl)
   4 from nmutil.pipemodbase import PipeModBase
   5 from ieee754.part.partsig import PartitionedSignal
   6 from soc.decoder.power_enums import InternalOp
   7
   8
   9 class CommonOutputStage(PipeModBase):
  10     def __init__(self, pspec):
  11         super().__init__(pspec, "output")
  12
  13     def elaborate(self, platform):
  14         m = Module()
  15         comb = m.d.comb
  16         op = self.i.ctx.op
  17
  18         # op requests inversion of the output
  19         o = Signal.like(self.i.o)
  20         with m.If(op.invert_out):
  21             comb += o.eq(~self.i.o)
  22         with m.Else():
  23             comb += o.eq(self.i.o)
  24
  25         # target register if 32-bit is only the 32 LSBs
  26         target = Signal(64, reset_less=True)
  27         with m.If(op.is_32bit):
  28             comb += target.eq(o[:32])
  29         with m.Else():
  30             comb += target.eq(o)
  31
  32         # Handle carry_out
  33         comb += self.o.xer_ca.data.eq(self.i.xer_ca.data)
  34         comb += self.o.xer_ca.ok.eq(op.output_carry)
  35
  36         # create condition register cr0 and sticky-overflow
  37         is_zero = Signal(reset_less=True)
  38         is_positive = Signal(reset_less=True)
  39         is_negative = Signal(reset_less=True)
  40         msb_test = Signal(reset_less=True) # set equal to MSB, invert if OP=CMP
  41         is_cmp = Signal(reset_less=True)   # true if OP=CMP
  42         self.so = Signal(1, reset_less=True)
  43         cr0 = Signal(4, reset_less=True)
  44
  45         # TODO: if o[63] is XORed with "operand == OP_CMP"
  46         # that can be used as a test
  47         # see https://bugs.libre-soc.org/show_bug.cgi?id=305#c60
  48
  49         comb += is_cmp.eq(op.insn_type == InternalOp.OP_CMP)
  50         comb += msb_test.eq(target[-1] ^ is_cmp)
  51         comb += is_zero.eq(target == 0)
  52         comb += is_positive.eq(~is_zero & ~msb_test)
  53         comb += is_negative.eq(~is_zero & msb_test)
  54
  55         with m.If(op.insn_type != InternalOp.OP_CMPEQB):
  56             comb += cr0.eq(Cat(self.so, is_zero, is_positive, is_negative))
  57         with m.Else():
  58             comb += cr0.eq(self.i.cr0)
  59
  60         # copy out [inverted] cr0, output, and context out
  61         comb += self.o.o.eq(o)
  62         comb += self.o.cr0.data.eq(cr0)
  63         comb += self.o.cr0.ok.eq(op.rc.rc & op.rc.rc_ok) # CR0 to be set
  64         comb += self.o.ctx.eq(self.i.ctx)
  65
  66         return m