如何在Chisel3中指定平方两个无符号数的差值？

Question

Here is one way do it that seems ugly.

class DiffSquared extends Module {
  val inputWidth = 8
  val width = 16
  val io = IO(new Bundle {
    val X = Input(UInt(inputWidth.W))
    val M = Input(UInt(inputWidth.W))
    val Out = Output(UInt(width.W))
  })
  val x = Wire(UInt((inputWidth+1).W))
  val m = Wire(UInt((inputWidth+1).W))
  x := io.X
  m := io.M
  val diff = Wire(SInt((inputWidth+1).W))
  diff := x.asSInt - m.asSInt
  io.Out := (diff*diff).asUInt
}

What is a better way to zero-extend the UInt to a 9 bit SInt, do the difference, square it, and represent the result as a 16 bit UInt?

Answer 1

Since no one bit, here is a slight improvement. Should we add a zero extension method to UInt to make this better? Is there already one there?

class DiffSquared extends Module {
  val inputWidth = 8
  val width = 16
  val io = IO( new Bundle{
    val X = Input(UInt(inputWidth.W))
    val M = Input(UInt(inputWidth.W))
    val Out = Output(UInt(width.W))
  })
  def zX(w:UInt) = Wire(UInt((w.getWidth+1).W),init=w).asSInt
  val diff = Wire(init=zX(io.X)-zX(io.M))
  io.Out := (diff*diff).asUInt
}

Answer 2

There's a useful function here: zext that zero extends a UInt to an SInt of the UInt's width + 1. Thus you could write the code as:

class DiffSquared extends Module {
  val inputWidth = 8
  val width = 16
  val io = IO(new Bundle {
    val X = Input(UInt(inputWidth.W))
    val M = Input(UInt(inputWidth.W))
    val Out = Output(UInt(width.W))
  })
  val diff = io.X.zext() - io.M.zext()
  io.Out := (diff*diff).asUInt
}