From b32cdd249c80caf5794a2fc247a632eded9b93c9 Mon Sep 17 00:00:00 2001
From: Benjamin Fattori <ben.fattori@mail.utoronto.ca>
Date: Mon, 11 Nov 2024 17:42:00 -0500
Subject: [PATCH] ignore for tc

---
 mamba/fused_cross_entropy.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mamba/fused_cross_entropy.py b/mamba/fused_cross_entropy.py
index 00e9359..81d2ba3 100644
--- a/mamba/fused_cross_entropy.py
+++ b/mamba/fused_cross_entropy.py
@@ -64,7 +64,7 @@ def forward(ctx, weight: torch.Tensor, act: torch.Tensor, labels: torch.Tensor):
         return scale * out_loss
 
     @staticmethod
-    def backward(ctx, grad_output):
+    def backward(ctx, grad_output):  # type: ignore
 
         (
             weight_grad,