Skip to content

Commit 7c9ff55

Browse files
committed
--amend
1 parent 1a0ed36 commit 7c9ff55

1 file changed

Lines changed: 0 additions & 7 deletions

File tree

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)