From fdaa9493be2ac7dd885f7d92bf4855ba54b6bac1 Mon Sep 17 00:00:00 2001
From: Daniel Han-Chen <danielhanchen@gmail.com>
Date: Mon, 26 Feb 2024 03:09:37 +1100
Subject: [PATCH] Update cross_entropy_loss.py

---
 unsloth/kernels/cross_entropy_loss.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/unsloth/kernels/cross_entropy_loss.py b/unsloth/kernels/cross_entropy_loss.py
index a139fce49d..2010a4cbe4 100644
--- a/unsloth/kernels/cross_entropy_loss.py
+++ b/unsloth/kernels/cross_entropy_loss.py
@@ -217,7 +217,6 @@ class Fast_CrossEntropyLoss(torch.autograd.Function):
                 losses,
                 logsumexp,
                 labels,
-                n_rows,
                 VOCAB_SIZE = vocab_size,
                 BLOCK_SIZE = MAX_FUSED_SIZE,
                 num_warps  = 32,