We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent a118066 commit 0f602cbCopy full SHA for 0f602cb
1 file changed
src/liger_kernel/ops/dyt.py
@@ -33,6 +33,9 @@ def _dyt_fwd_kernel(
33
BLOCK_SIZE: tl.constexpr,
34
):
35
"""
36
+ Reference:
37
+ https://arxiv.org/abs/2503.10622
38
+
39
Shapes:
40
- x: (BT, C)
41
- alpha: (1)
@@ -73,6 +76,9 @@ def _dyt_bwd_kernel(
73
76
74
77
75
78
79
80
81
82
83
84
0 commit comments