File tree Expand file tree Collapse file tree 1 file changed +3
-1
lines changed
pytensor/link/numba/dispatch Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Original file line number Diff line number Diff line change 9
9
10
10
from pytensor .link .numba .dispatch import basic as numba_basic
11
11
from pytensor .link .numba .dispatch .basic import numba_funcify
12
- from pytensor .tensor .slinalg import Cholesky , BlockDiagonal , SolveTriangular
12
+ from pytensor .tensor .slinalg import BlockDiagonal , Cholesky , SolveTriangular
13
13
14
14
15
15
_PTR = ctypes .POINTER
@@ -299,6 +299,7 @@ def solve_triangular(a, b):
299
299
300
300
return solve_triangular
301
301
302
+
302
303
def _cholesky (a , lower = False , overwrite_a = False , check_finite = True ):
303
304
return linalg .cholesky (
304
305
a , lower = lower , overwrite_a = overwrite_a , check_finite = check_finite
@@ -357,6 +358,7 @@ def nb_cholesky(a):
357
358
358
359
return nb_cholesky
359
360
361
+
360
362
@numba_funcify .register (BlockDiagonal )
361
363
def numba_funcify_BlockDiagonal (op , node , ** kwargs ):
362
364
dtype = node .outputs [0 ].dtype
You can’t perform that action at this time.
0 commit comments