From 31f0c9c08d5b4e076f88697a24c902000e5e6f22 Mon Sep 17 00:00:00 2001
From: Daniel Han-Chen <danielhanchen@gmail.com>
Date: Mon, 22 Jan 2024 22:31:38 +1100
Subject: [PATCH] Update utils.py

---
 unsloth/kernels/utils.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/unsloth/kernels/utils.py b/unsloth/kernels/utils.py
index 141814c332..5f5e7ac5e7 100644
--- a/unsloth/kernels/utils.py
+++ b/unsloth/kernels/utils.py
@@ -180,6 +180,7 @@ pass
 def fast_linear_forward(proj, X, temp_lora = None, out = None):
     W, W_quant, lora_A, lora_B, lora_S = get_lora_parameters(proj)
     out = fast_gemv(X, W, W_quant, out = out)
+    print(X.shape, W.shape, out.shape)
     if lora_A is not None:
         dtype = X.dtype
         temp_lora = torch.matmul(X, lora_A.to(dtype).t(), out = temp_lora)