Skip to content

Commit 87928a4

Browse files
committed
--amend
1 parent 8fdd559 commit 87928a4

File tree

1 file changed

+0
-7
lines changed

1 file changed

+0
-7
lines changed

src/liger_kernel/ops/backends/_ascend/ops/multi_token_attention.py

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,3 @@
1-
"""
2-
Fused causal masking + softmax/sparsemax Triton kernels for NPU.
3-
4-
This implementation fuses causal masking with softmax and sparsemax forward and backward
5-
operations in single kernels to reduce memory traffic and improve performance.
6-
"""
7-
81
import torch
92
import torch.nn.functional as F
103
import triton

0 commit comments

Comments
 (0)