Fixed UIntToOH(x, 1) invocation with x.width == 0 (#778)

1 files changed, 2 insertions, 1 deletions

diff --git a/src/main/scala/chisel3/util/OneHot.scala b/src/main/scala/chisel3/util/OneHot.scala
index 8a0bb9fc..a6af0d99 100644
--- a/src/main/scala/chisel3/util/OneHot.scala
+++ b/src/main/scala/chisel3/util/OneHot.scala
@@ -45,9 +45,10 @@ object UIntToOH {
   def apply(in: UInt): UInt = 1.U << in
   def apply(in: UInt, width: Int): UInt = width match {
     case 0 => 0.U(0.W)
+    case 1 => 1.U(1.W)
     case _ =>
       val shiftAmountWidth = log2Ceil(width)
-      val shiftAmount = in.pad(shiftAmountWidth)((shiftAmountWidth - 1) max 0, 0)
+      val shiftAmount = in.pad(shiftAmountWidth)(shiftAmountWidth - 1, 0)
       (1.U << shiftAmount)(width - 1, 0)
   }
 }