From 569963789990a7294a72b8983d61674e081c1cc7 Mon Sep 17 00:00:00 2001
From: Liangfu Chen <liangfu.chen@icloud.com>
Date: Tue, 14 Jan 2020 10:39:56 +0800
Subject: [PATCH] [VTA] Fix an issue in updating uop_idx in the TensorGemm
 module (#4694)

---
 vta/hardware/chisel/src/main/scala/core/TensorGemm.scala | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/vta/hardware/chisel/src/main/scala/core/TensorGemm.scala b/vta/hardware/chisel/src/main/scala/core/TensorGemm.scala
index 7328c426978f..f982176c1e81 100644
--- a/vta/hardware/chisel/src/main/scala/core/TensorGemm.scala
+++ b/vta/hardware/chisel/src/main/scala/core/TensorGemm.scala
@@ -280,7 +280,7 @@ class TensorGemm(debug: Boolean = false)(implicit p: Parameters)
       (state === sExe &&
         uop_idx === uop_end - 1.U)) {
     uop_idx := dec.uop_begin
-  }.elsewhen(state === sExe) {
+  }.elsewhen(state === sExe && dec.uop_begin =/= uop_end) {
     uop_idx := uop_idx + 1.U
   }