Skip to content

Commit 5775b11

Browse files
committed
Rename common.linalg to common._linalg
1 parent 2912c9e commit 5775b11

File tree

3 files changed

+34
-32
lines changed

3 files changed

+34
-32
lines changed

array_api_compat/cupy/linalg.py

Lines changed: 17 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -8,28 +8,29 @@
88
linalg_all = list(_n)
99
del _n
1010

11-
from ..common import linalg
11+
from ..common import _linalg
1212
from .._internal import get_xp
1313

1414
import cupy as cp
1515

16-
cross = get_xp(cp)(linalg.cross)
17-
diagonal = get_xp(cp)(linalg.diagonal)
18-
matmul = get_xp(cp)(linalg.matmul)
19-
cholesky = get_xp(cp)(linalg.cholesky)
20-
matrix_rank = get_xp(cp)(linalg.matrix_rank)
21-
pinv = get_xp(cp)(linalg.pinv)
22-
matrix_norm = get_xp(cp)(linalg.matrix_norm)
23-
matrix_transpose = get_xp(cp)(linalg.matrix_transpose)
24-
outer = get_xp(cp)(linalg.outer)
25-
svdvals = get_xp(cp)(linalg.svdvals)
26-
tensordot = get_xp(cp)(linalg.tensordot)
27-
trace = get_xp(cp)(linalg.trace)
28-
vecdot = get_xp(cp)(linalg.vecdot)
29-
vector_norm = get_xp(cp)(linalg.vector_norm)
16+
cross = get_xp(cp)(_linalg.cross)
17+
diagonal = get_xp(cp)(_linalg.diagonal)
18+
matmul = get_xp(cp)(_linalg.matmul)
19+
cholesky = get_xp(cp)(_linalg.cholesky)
20+
matrix_rank = get_xp(cp)(_linalg.matrix_rank)
21+
pinv = get_xp(cp)(_linalg.pinv)
22+
matrix_norm = get_xp(cp)(_linalg.matrix_norm)
23+
matrix_transpose = get_xp(cp)(_linalg.matrix_transpose)
24+
outer = get_xp(cp)(_linalg.outer)
25+
svdvals = get_xp(cp)(_linalg.svdvals)
26+
tensordot = get_xp(cp)(_linalg.tensordot)
27+
trace = get_xp(cp)(_linalg.trace)
28+
vecdot = get_xp(cp)(_linalg.vecdot)
29+
vector_norm = get_xp(cp)(_linalg.vector_norm)
3030

31-
__all__ = linalg_all + linalg.__all__
31+
__all__ = linalg_all + _linalg.__all__
3232

3333
del get_xp
3434
del cp
3535
del linalg_all
36+
del _linalg

array_api_compat/numpy/linalg.py

Lines changed: 17 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -1,28 +1,29 @@
11
from numpy.linalg import *
22
from numpy.linalg import __all__ as linalg_all
33

4-
from ..common import linalg
4+
from ..common import _linalg
55
from .._internal import get_xp
66

77
import numpy as np
88

9-
cross = get_xp(np)(linalg.cross)
10-
diagonal = get_xp(np)(linalg.diagonal)
11-
matmul = get_xp(np)(linalg.matmul)
12-
cholesky = get_xp(np)(linalg.cholesky)
13-
matrix_rank = get_xp(np)(linalg.matrix_rank)
14-
pinv = get_xp(np)(linalg.pinv)
15-
matrix_norm = get_xp(np)(linalg.matrix_norm)
16-
matrix_transpose = get_xp(np)(linalg.matrix_transpose)
17-
outer = get_xp(np)(linalg.outer)
18-
svdvals = get_xp(np)(linalg.svdvals)
19-
tensordot = get_xp(np)(linalg.tensordot)
20-
trace = get_xp(np)(linalg.trace)
21-
vecdot = get_xp(np)(linalg.vecdot)
22-
vector_norm = get_xp(np)(linalg.vector_norm)
9+
cross = get_xp(np)(_linalg.cross)
10+
diagonal = get_xp(np)(_linalg.diagonal)
11+
matmul = get_xp(np)(_linalg.matmul)
12+
cholesky = get_xp(np)(_linalg.cholesky)
13+
matrix_rank = get_xp(np)(_linalg.matrix_rank)
14+
pinv = get_xp(np)(_linalg.pinv)
15+
matrix_norm = get_xp(np)(_linalg.matrix_norm)
16+
matrix_transpose = get_xp(np)(_linalg.matrix_transpose)
17+
outer = get_xp(np)(_linalg.outer)
18+
svdvals = get_xp(np)(_linalg.svdvals)
19+
tensordot = get_xp(np)(_linalg.tensordot)
20+
trace = get_xp(np)(_linalg.trace)
21+
vecdot = get_xp(np)(_linalg.vecdot)
22+
vector_norm = get_xp(np)(_linalg.vector_norm)
2323

24-
__all__ = linalg_all + linalg.__all__
24+
__all__ = linalg_all + _linalg.__all__
2525

2626
del get_xp
2727
del np
2828
del linalg_all
29+
del _linalg

0 commit comments

Comments
 (0)