ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update cross_entropy_loss.py

Browse source
This commit is contained in:
Daniel Han 2024-11-03 21:27:37 -08:00
commit ca37a2e950
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
unsloth/kernels/cross_entropy_loss.py
View file

@ -32,7 +32,7 @@ from unsloth_zoo.loss_utils import (
@triton.jit
def _cross_entropy_forward(
logits_ptr ,
logits_row_stride : tl.constexpr(tl.int64),
logits_row_stride ,
loss_ptr ,
logsumexp_ptr ,
labels_ptr ,
@ -105,7 +105,7 @@ pass
@triton.jit
def _chunked_cross_entropy_forward(
logits_ptr ,
logits_row_stride : tl.constexpr(tl.int64),
logits_row_stride ,
loss_ptr ,
logsumexp_ptr ,
labels_ptr ,
@ -188,7 +188,7 @@ pass
@triton.jit
def _cross_entropy_backward(
logits_ptr ,
logits_row_stride : tl.constexpr(tl.int64),
logits_row_stride ,
dloss_ptr ,
dloss_row_stride ,
logsumexp_ptr ,