Skip to content

Commit 0f602cb

Browse files
committed
Add reference
1 parent a118066 commit 0f602cb

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

src/liger_kernel/ops/dyt.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,9 @@ def _dyt_fwd_kernel(
3333
BLOCK_SIZE: tl.constexpr,
3434
):
3535
"""
36+
Reference:
37+
https://arxiv.org/abs/2503.10622
38+
3639
Shapes:
3740
- x: (BT, C)
3841
- alpha: (1)
@@ -73,6 +76,9 @@ def _dyt_bwd_kernel(
7376
BLOCK_SIZE: tl.constexpr,
7477
):
7578
"""
79+
Reference:
80+
https://arxiv.org/abs/2503.10622
81+
7682
Shapes:
7783
- x: (BT, C)
7884
- alpha: (1)

0 commit comments

Comments
 (0)