Fix (gptq): typo in name variable (#691)

Xilinx · Jul 31, 2023 · 461a0e0 · 461a0e0
1 parent 3c5ca54
commit 461a0e0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/brevitas/graph/gptq.py b/src/brevitas/graph/gptq.py
@@ -373,7 +373,7 @@ def single_layer_update(self, percdamp=.01):
                     perm = permutation_list[group_index]
                     q = q_groups[group_index]  # [OC/groups]
                     w = weight[group_index, :, perm[i1:i2][i]].to(torch.float32)  # [OC/groups]
-                    d = h_inv[group_index, i, i]  # [1]
+                    d = h_inv_block[group_index, i, i]  # [1]
                     error = (w - q) / d  # [OC/groups]
                     error_block[group_index, :, i] = error
                     # We need to update the original weights