From 7cf69dd9f2cd848985a9715dd12ccfc05032249d Mon Sep 17 00:00:00 2001
From: Daniel Han-Chen <danielhanchen@gmail.com>
Date: Mon, 26 Feb 2024 16:19:39 +1100
Subject: [PATCH] Update cross_entropy_loss.py

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

diff --git a/unsloth/kernels/cross_entropy_loss.py b/unsloth/kernels/cross_entropy_loss.py
index 86e7bd168e..3775d78e5e 100644
--- a/unsloth/kernels/cross_entropy_loss.py
+++ b/unsloth/kernels/cross_entropy_loss.py
@@ -251,7 +251,7 @@ class Fast_CrossEntropyLoss(torch.autograd.Function):
             dlosses, dlosses.stride(0),
             logsumexp,
             labels,
-            vocab_size,
+            VOCAB_SIZE = vocab_size,
             BLOCK_SIZE = BLOCK_SIZE,
             num_warps  = 8,
         )