Open
Description
Description
There still seems to be an issue with the gradient of gammainc in numba mode:
import pytensor
import pytensor.tensor as pt
x = pt.vector("x")
y = pt.math.gammainc(x, x)
dx = pytensor.grad(y.sum(), x)
func = pytensor.function([x], dx, mode="NUMBA")
func([1.])
This raises warnings about object mode for gammainc_grad_a
and gammainc_grad_b
. Those look like ScalarLoop Ops, and I thought we added those recently? Maybe something in the specific ops is raising an error?
original report in nutpie:
pymc-devs/nutpie#163 (comment)
Metadata
Metadata
Assignees
Labels
No labels