chore: update cross_entropy.py (#84)

linkedin · Aug 25, 2024 · a19c667 · a19c667
1 parent 016257a
commit a19c667
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/liger_kernel/ops/cross_entropy.py b/src/liger_kernel/ops/cross_entropy.py
@@ -196,7 +196,7 @@ def forward(ctx, _input, target, ignore_index):
             ignore_index=ignore_index,
             BLOCK_SIZE=BLOCK_SIZE,
             # TODO: 32 seems to give the best performance
-            # Performance is quite sentitive to num_warps
+            # Performance is quite sensitive to num_warps
             num_warps=32,
         )