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 c5c68ef commit 02ca580Copy full SHA for 02ca580
tests/test_linalg.py
@@ -5,11 +5,11 @@
5
import numpy
6
7
8
-def vvsort(val, vec, size):
+def vvsort(val, vec, size, xp):
9
for i in range(size):
10
imax = i
11
for j in range(i + 1, size):
12
- if numpy.abs(val[imax]) < numpy.abs(val[j]):
+ if xp.abs(val[imax]) < xp.abs(val[j]):
13
imax = j
14
15
temp = val[i]
@@ -86,10 +86,10 @@ def test_eig_arange(type, size):
86
np_val, np_vec = numpy.linalg.eig(symm)
87
88
# DPNP sort val/vec by abs value
89
- vvsort(dpnp_val, dpnp_vec, size)
+ vvsort(dpnp_val, dpnp_vec, size, inp)
90
91
# NP sort val/vec by abs value
92
- vvsort(np_val, np_vec, size)
+ vvsort(np_val, np_vec, size, numpy)
93
94
# NP change sign of vectors
95
for i in range(np_vec.shape[1]):
0 commit comments