From 9c1d79725076a1be28b18f0d548edd487464c33d Mon Sep 17 00:00:00 2001
From: Hansung Kim <hansung_kim@berkeley.edu>
Date: Sun, 5 May 2024 18:36:15 -0700
Subject: [PATCH] tensor: add missing }

---
 hw/dpi/float_dpi.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/hw/dpi/float_dpi.cpp b/hw/dpi/float_dpi.cpp
index 76d6767b..29ca22df 100644
--- a/hw/dpi/float_dpi.cpp
+++ b/hw/dpi/float_dpi.cpp
@@ -345,7 +345,7 @@ void dpi_fmax(bool enable, int dst_fmt, int64_t a, int64_t b, int64_t* result, s
   }
 }
 
-// A is M * K, B is K * K * M, C is M * M, D is M * M
+// A is M * K, B is K * M, C is M * M, D is M * M
 #define M 4
 #define K 2
 
@@ -414,6 +414,7 @@ void dpi_hmma(bool enable, const svBitVecVal* A_tile, const svBitVecVal* B_tile,
   }
 
   write_float_array(D_tile, &c_D_tile[0][0], M, M);
+}
 
 // 1 copy per warp
 float A_tile_full[4][16][8];