tensor: Doc comments

hw/rtl/fpu/VX_tensor_dpu.sv

@@ -97,8 +97,8 @@ module VX_tensor_dpu #(
     );
 
     // Split A_tile and C_tile by rows (0-1, 2-3) and parallelize in two
-    // threadgroups; B_tile is shared across the two threadgroups. See Figure
-    // 13 in paper
+    // threadgroup DPUs; B_tile is shared across the two threadgroups. See
+    // Figure 13 in paper
     VX_tensor_threadgroup #(
     ) threadgroup_0 (
         .clk   (clk),
@@ -196,7 +196,8 @@ module VX_tensor_threadgroup #(
     // + C_frag.  substep_in and substep_out keeps track of which cycle they're at
     // & when they have to pop from input queue and push to result queue.
     // Note that substep is different from the "step" defined in the HMMA
-    // instruction set; it is a purely microarchitectural construct.
+    // instruction set; it is similar in meaning to the substeps in
+    // VX_tensor_octet.
     //
     // substep_in == 0: FEDP uses first half from operand buffer
     // substep_in == 1: FEDP uses last half and pops from operand buffer
@@ -270,12 +271,21 @@ module VX_tensor_threadgroup #(
     for (genvar i = 0; i < 4; ++i) begin
         // at substep == 0, the 0th and 2nd columns of D begins compute;
         // at substep == 1, the 1st and 3rd columns of D begins compute.
-        // there are two row elements for each column, rounding out
-        // 4 elements being computed by 4 FEDPs at every cycle
+        // There are two row elements for each column, rounding out to
+        // 4 elements computed by 4 FEDPs at every cycle
         // (see Figure 10(b)).
 
-        // d_row: 0, 0, 1, 1
-        // d_col: 0, 2, 0, 2
+        //     i    : 0, 1, 2, 3
+        // d_row    : 0, 0, 1, 1
+        // d_col    : 0, 2, 0, 2
+        // d_col_sel: 1, 3, 1, 3
+        //
+        // substep 0:
+        // [ 0 x 2 x ]
+        // [ 1 x 3 x ]
+        // substep 1:
+        // [ x 0 x 2 ]
+        // [ x 1 x 3 ]
         localparam int d_row = i / 2;
         localparam int d_col = (i % 2) * 2;
         wire [31:0] d_col_sel = (substep_in == 1'b0) ? d_col : (d_col + 1);