There was an error while loading. Please reload this page.
1 parent 18dcd86 commit 49d7a52Copy full SHA for 49d7a52
1 file changed
src/flag_gems/runtime/backend/_kunlunxin/ops/digamma.py
@@ -20,8 +20,9 @@ def _digamma(x):
20
21
rinv = 1.0 / r
22
z = rinv * rinv
23
- poly = z * (1.0 / 12.0 + z * (-1.0 / 120.0 + z * (1.0 / 252.0
24
- + z * (-1.0 / 240.0))))
+ poly = z * (
+ 1.0 / 12.0 + z * (-1.0 / 120.0 + z * (1.0 / 252.0 + z * (-1.0 / 240.0)))
25
+ )
26
psi = tl.math.log(r) - 0.5 * rinv - poly - inv_sum
27
28
cot = tl.math.cos(PI * x) / tl.math.sin(PI * x)
0 commit comments