We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent b68b973 commit cdb78f6Copy full SHA for cdb78f6
pytensor/link/numba/dispatch/basic.py
@@ -37,7 +37,7 @@
37
from pytensor.tensor.blas import BatchedDot
38
from pytensor.tensor.math import Dot
39
from pytensor.tensor.shape import Reshape, Shape, Shape_i, SpecifyShape
40
-from pytensor.tensor.slinalg import Solve
+from pytensor.tensor.slinalg import Cholesky, Solve
41
from pytensor.tensor.subtensor import (
42
AdvancedIncSubtensor,
43
AdvancedIncSubtensor1,
@@ -808,6 +808,7 @@ def softplus(x):
808
809
return softplus
810
811
+
812
@numba_funcify.register(Cholesky)
813
def numba_funcify_Cholesky(op, node, **kwargs):
814
lower = op.lower
0 commit comments