ignore for tc

fattorib · Nov 11, 2024 · b32cdd2 · b32cdd2
1 parent a555269
commit b32cdd2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,