add 2 more lut writes

Author: richardyrh

src/main/scala/gemmini/Controller.scala

@@ -216,10 +216,9 @@ class GemminiModule[T <: Data: Arithmetic, U <: Data, V <: Data]
     mx_io.scale_mem <> spad.module.io.scale_mem.get
 
     mx_io.requant_out <> mx_requantizer.get.io.requant_data_out
-    // TODO TODO TODO TODO TODO TODO lut1 and lut2
-    mx_requantizer.get.io.lut_write <> mx_io.lut0
-    mx_io.lut1.ready := false.B
-    mx_io.lut2.ready := false.B
+    mx_requantizer.get.io.lut0_write <> mx_io.lut0
+    mx_requantizer.get.io.lut1_write <> mx_io.lut1
+    mx_requantizer.get.io.lut2_write <> mx_io.lut2
 
     Seq(mx_io.requant_in_gpu, mx_io.requant_out, mx_io.lut0, mx_io.lut1, mx_io.lut2).foreach(dontTouch(_))
     //Seq( mx_io.requant_out).foreach(dontTouch(_))

src/main/scala/gemmini/MxRequantizer.scala

@@ -57,7 +57,9 @@ class MxRequantizerIO(
   val requant_data_in = Flipped(Decoupled(new RequantizerInBundle(inputnumLanes, inputdataWidth)))
   val scaleMem_write = Decoupled(new ScalingFactorWriteReq(scaleMem_addr_width, scaleMem_data_width)) 
   val requant_data_out = Decoupled(new RequantizerOutBundle(outputnumLanes))
-  val lut_write = Flipped(Decoupled(new QuantLutWriteBundle(lutConfig)))
+  val lut0_write = Flipped(Decoupled(new QuantLutWriteBundle(lutConfig)))
+  val lut1_write = Flipped(Decoupled(new QuantLutWriteBundle(lutConfig)))
+  val lut2_write = Flipped(Decoupled(new QuantLutWriteBundle(lutConfig)))
   val spad_projected_data = Vec(sp_banks, new ScratchpadReadIO(sp_bank_entries, sp_width_projected))
   val spad_deprojected_data = Vec(sp_banks, Flipped(new ScratchpadReadIO(sp_bank_entries, sp_width)))
   val fp8_mode = Input(Bool())  // true for 64-lane mode, false for 16-lane mode
@@ -271,7 +273,9 @@ class MxRequantizer[T <: Data: Arithmetic](
   //   quantLut.io.lut_write.ready := false.B
   // }
 
-  quantLut.io.lut_write <> io.lut_write
+  quantLut.io.lut_write <> io.lut0_write
+  io.lut1_write.ready := false.B
+  io.lut2_write.ready := false.B
 
   when(quantize_valid && (total_bits_per_element === 6.U)) {
     quantLut.io.quant_fp6.valid := true.B