Skip to content

Commit 2952983

Browse files
Fix copy-paste-error reported by Coverity
And save duplicate device<->host transfers
1 parent 71b1640 commit 2952983

File tree

4 files changed

+81
-40
lines changed

4 files changed

+81
-40
lines changed

dpctl/tests/elementwise/test_greater.py

Lines changed: 21 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -68,34 +68,40 @@ def test_greater_complex_matrix(op_dtype):
6868
sz = 127
6969
ar1_np_real = np.random.randint(0, 10, sz)
7070
ar1_np_imag = np.random.randint(0, 10, sz)
71-
ar1 = dpt.asarray(ar1_np_real + 1j * ar1_np_imag, dtype=op_dtype)
71+
ar1_np = ar1_np_real + 1j * ar1_np_imag
72+
ar1 = dpt.asarray(ar1_np, dtype=op_dtype)
7273

7374
ar2_np_real = np.random.randint(0, 10, sz)
7475
ar2_np_imag = np.random.randint(0, 10, sz)
75-
ar2 = dpt.asarray(ar2_np_real + 1j * ar2_np_imag, dtype=op_dtype)
76+
ar2_np = ar2_np_real + 1j * ar2_np_imag
77+
ar2 = dpt.asarray(ar2_np, dtype=op_dtype)
7678

7779
r = dpt.greater(ar1, ar2)
78-
expected = np.greater(dpt.asnumpy(ar1), dpt.asnumpy(ar2))
80+
expected = np.greater(ar1_np, ar2_np)
7981
assert _compare_dtypes(r.dtype, expected.dtype, sycl_queue=q)
8082
assert r.shape == expected.shape
8183
assert (dpt.asnumpy(r) == expected).all()
8284

8385
r1 = dpt.greater(ar1[::-2], ar2[::2])
84-
expected1 = np.greater(dpt.asnumpy(ar1[::-2]), dpt.asnumpy(ar2[::2]))
86+
expected1 = np.greater(ar1_np[::-2], ar2_np[::2])
8587
assert _compare_dtypes(r.dtype, expected1.dtype, sycl_queue=q)
8688
assert r1.shape == expected1.shape
8789
assert (dpt.asnumpy(r1) == expected1).all()
8890

8991
ar3 = dpt.asarray([1.0 + 9j, 2.0 + 0j, 2.0 + 1j, 2.0 + 2j], dtype=op_dtype)
9092
ar4 = dpt.asarray([2.0 + 0j, dpt.nan, dpt.inf, -dpt.inf], dtype=op_dtype)
93+
94+
ar3_np = dpt.asnumpy(ar3)
95+
ar4_np = dpt.asnumpy(ar4)
96+
9197
r2 = dpt.greater(ar3, ar4)
9298
with np.errstate(invalid="ignore"):
93-
expected2 = np.greater(dpt.asnumpy(ar3), dpt.asnumpy(ar4))
99+
expected2 = np.greater(ar3_np, ar4_np)
94100
assert (dpt.asnumpy(r2) == expected2).all()
95101

96102
r3 = dpt.greater(ar4, ar4)
97103
with np.errstate(invalid="ignore"):
98-
expected3 = np.greater(dpt.asnumpy(ar4), dpt.asnumpy(ar4))
104+
expected3 = np.greater(ar4_np, ar4_np)
99105
assert (dpt.asnumpy(r3) == expected3).all()
100106

101107

@@ -105,23 +111,28 @@ def test_greater_complex_float():
105111
ar1 = dpt.asarray([1.0 + 9j, 2.0 + 0j, 2.0 + 1j, 2.0 + 2j], dtype="c8")
106112
ar2 = dpt.full((4,), 2, dtype="f4")
107113

114+
ar1_np = dpt.asnumpy(ar1)
115+
ar2_np = dpt.asnumpy(ar2)
116+
108117
r = dpt.greater(ar1, ar2)
109-
expected = np.greater(dpt.asnumpy(ar1), dpt.asnumpy(ar2))
118+
expected = np.greater(ar1_np, ar2_np)
110119
assert (dpt.asnumpy(r) == expected).all()
111120

112121
r1 = dpt.greater(ar2, ar1)
113-
expected1 = np.greater(dpt.asnumpy(ar2), dpt.asnumpy(ar1))
122+
expected1 = np.greater(ar2_np, ar1_np)
114123
assert (dpt.asnumpy(r1) == expected1).all()
115124
with np.errstate(invalid="ignore"):
116125
for tp in [dpt.nan, dpt.inf, -dpt.inf]:
117126

118127
ar3 = dpt.full((4,), tp)
128+
ar3_np = dpt.asnumpy(ar3)
129+
119130
r2 = dpt.greater(ar1, ar3)
120-
expected2 = np.greater(dpt.asnumpy(ar1), dpt.asnumpy(ar3))
131+
expected2 = np.greater(ar1_np, ar3_np)
121132
assert (dpt.asnumpy(r2) == expected2).all()
122133

123134
r3 = dpt.greater(ar3, ar1)
124-
expected3 = np.greater(dpt.asnumpy(ar3), dpt.asnumpy(ar1))
135+
expected3 = np.greater(ar3_np, ar1_np)
125136
assert (dpt.asnumpy(r3) == expected3).all()
126137

127138

dpctl/tests/elementwise/test_greater_equal.py

Lines changed: 19 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -68,34 +68,39 @@ def test_greater_equal_complex_matrix(op_dtype):
6868
sz = 127
6969
ar1_np_real = np.random.randint(0, 10, sz)
7070
ar1_np_imag = np.random.randint(0, 10, sz)
71-
ar1 = dpt.asarray(ar1_np_real + 1j * ar1_np_imag, dtype=op_dtype)
71+
ar1_np = ar1_np_real + 1j * ar1_np_imag
72+
ar1 = dpt.asarray(ar1_np, dtype=op_dtype)
7273

7374
ar2_np_real = np.random.randint(0, 10, sz)
7475
ar2_np_imag = np.random.randint(0, 10, sz)
75-
ar2 = dpt.asarray(ar2_np_real + 1j * ar2_np_imag, dtype=op_dtype)
76+
ar2_np = ar2_np_real + 1j * ar2_np_imag
77+
ar2 = dpt.asarray(ar2_np, dtype=op_dtype)
7678

7779
r = dpt.greater_equal(ar1, ar2)
78-
expected = np.greater_equal(dpt.asnumpy(ar1), dpt.asnumpy(ar2))
80+
expected = np.greater_equal(ar1_np, ar2_np)
7981
assert _compare_dtypes(r.dtype, expected.dtype, sycl_queue=q)
8082
assert r.shape == expected.shape
8183
assert (dpt.asnumpy(r) == expected).all()
8284

8385
r1 = dpt.greater_equal(ar1[::-2], ar2[::2])
84-
expected1 = np.greater_equal(dpt.asnumpy(ar1[::-2]), dpt.asnumpy(ar2[::2]))
86+
expected1 = np.greater_equal(ar1_np[::-2], ar2_np[::2])
8587
assert _compare_dtypes(r.dtype, expected1.dtype, sycl_queue=q)
8688
assert r1.shape == expected1.shape
8789
assert (dpt.asnumpy(r1) == expected1).all()
8890

8991
ar3 = dpt.asarray([1.0 + 9j, 2.0 + 0j, 2.0 + 1j, 2.0 + 2j], dtype=op_dtype)
9092
ar4 = dpt.asarray([2.0 + 0j, dpt.nan, dpt.inf, -dpt.inf], dtype=op_dtype)
93+
94+
ar3_np = dpt.asnumpy(ar3)
95+
ar4_np = dpt.asnumpy(ar4)
9196
r2 = dpt.greater_equal(ar3, ar4)
9297
with np.errstate(invalid="ignore"):
93-
expected2 = np.greater_equal(dpt.asnumpy(ar3), dpt.asnumpy(ar4))
98+
expected2 = np.greater_equal(ar3_np, ar4_np)
9499
assert (dpt.asnumpy(r2) == expected2).all()
95100

96101
r3 = dpt.greater_equal(ar4, ar4)
97102
with np.errstate(invalid="ignore"):
98-
expected3 = np.greater_equal(dpt.asnumpy(ar4), dpt.asnumpy(ar4))
103+
expected3 = np.greater_equal(ar4_np, ar4_np)
99104
assert (dpt.asnumpy(r3) == expected3).all()
100105

101106

@@ -105,23 +110,27 @@ def test_greater_equal_complex_float():
105110
ar1 = dpt.asarray([1.0 + 9j, 2.0 + 0j, 2.0 + 1j, 2.0 + 2j], dtype="c8")
106111
ar2 = dpt.full((4,), 2, dtype="f4")
107112

113+
ar1_np = dpt.asnumpy(ar1)
114+
ar2_np = dpt.asnumpy(ar2)
115+
108116
r = dpt.greater_equal(ar1, ar2)
109-
expected = np.greater_equal(dpt.asnumpy(ar1), dpt.asnumpy(ar2))
117+
expected = np.greater_equal(ar1_np, ar2_np)
110118
assert (dpt.asnumpy(r) == expected).all()
111119

112120
r1 = dpt.greater_equal(ar2, ar1)
113-
expected1 = np.greater_equal(dpt.asnumpy(ar2), dpt.asnumpy(ar1))
121+
expected1 = np.greater_equal(ar2_np, ar1_np)
114122
assert (dpt.asnumpy(r1) == expected1).all()
115123
with np.errstate(invalid="ignore"):
116124
for tp in [dpt.nan, dpt.inf, -dpt.inf]:
117125

118126
ar3 = dpt.full((4,), tp)
127+
ar3_np = dpt.asnumpy(ar3)
119128
r2 = dpt.greater_equal(ar1, ar3)
120-
expected2 = np.greater_equal(dpt.asnumpy(ar1), dpt.asnumpy(ar3))
129+
expected2 = np.greater_equal(ar1_np, ar3_np)
121130
assert (dpt.asnumpy(r2) == expected2).all()
122131

123132
r3 = dpt.greater_equal(ar3, ar1)
124-
expected3 = np.greater_equal(dpt.asnumpy(ar3), dpt.asnumpy(ar1))
133+
expected3 = np.greater_equal(ar3_np, ar1_np)
125134
assert (dpt.asnumpy(r3) == expected3).all()
126135

127136

dpctl/tests/elementwise/test_less.py

Lines changed: 21 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -68,34 +68,40 @@ def test_less_complex_matrix(op_dtype):
6868
sz = 127
6969
ar1_np_real = np.random.randint(0, 10, sz)
7070
ar1_np_imag = np.random.randint(0, 10, sz)
71-
ar1 = dpt.asarray(ar1_np_real + 1j * ar1_np_imag, dtype=op_dtype)
71+
ar1_np = ar1_np_real + 1j * ar1_np_imag
72+
ar1 = dpt.asarray(ar1_np, dtype=op_dtype)
7273

7374
ar2_np_real = np.random.randint(0, 10, sz)
7475
ar2_np_imag = np.random.randint(0, 10, sz)
75-
ar2 = dpt.asarray(ar2_np_real + 1j * ar2_np_imag, dtype=op_dtype)
76+
ar2_np = ar2_np_real + 1j * ar2_np_imag
77+
ar2 = dpt.asarray(ar2_np, dtype=op_dtype)
7678

7779
r = dpt.less(ar1, ar2)
78-
expected = np.less(dpt.asnumpy(ar1), dpt.asnumpy(ar2))
80+
expected = np.less(ar1_np, ar2_np)
7981
assert _compare_dtypes(r.dtype, expected.dtype, sycl_queue=q)
8082
assert r.shape == expected.shape
8183
assert (dpt.asnumpy(r) == expected).all()
8284

8385
r1 = dpt.less(ar1[::-2], ar2[::2])
84-
expected1 = np.less(dpt.asnumpy(ar1[::-2]), dpt.asnumpy(ar2[::2]))
86+
expected1 = np.less(ar1_np[::-2], ar2_np[::2])
8587
assert _compare_dtypes(r.dtype, expected1.dtype, sycl_queue=q)
8688
assert r1.shape == expected1.shape
8789
assert (dpt.asnumpy(r1) == expected1).all()
8890

8991
ar3 = dpt.asarray([1.0 + 9j, 2.0 + 0j, 2.0 + 1j, 2.0 + 2j], dtype=op_dtype)
9092
ar4 = dpt.asarray([2.0 + 0j, dpt.nan, dpt.inf, -dpt.inf], dtype=op_dtype)
93+
94+
ar3_np = dpt.asnumpy(ar3)
95+
ar4_np = dpt.asnumpy(ar4)
96+
9197
r2 = dpt.less(ar3, ar4)
9298
with np.errstate(invalid="ignore"):
93-
expected2 = np.less(dpt.asnumpy(ar3), dpt.asnumpy(ar4))
99+
expected2 = np.less(ar3_np, ar4_np)
94100
assert (dpt.asnumpy(r2) == expected2).all()
95101

96102
r3 = dpt.less(ar4, ar4)
97103
with np.errstate(invalid="ignore"):
98-
expected3 = np.less(dpt.asnumpy(ar4), dpt.asnumpy(ar4))
104+
expected3 = np.less(ar4_np, ar4_np)
99105
assert (dpt.asnumpy(r3) == expected3).all()
100106

101107

@@ -105,23 +111,28 @@ def test_less_complex_float():
105111
ar1 = dpt.asarray([1.0 + 9j, 2.0 + 0j, 2.0 + 1j, 2.0 + 2j], dtype="c8")
106112
ar2 = dpt.full((4,), 2, dtype="f4")
107113

114+
ar1_np = dpt.asnumpy(ar1)
115+
ar2_np = dpt.asnumpy(ar2)
116+
108117
r = dpt.less(ar1, ar2)
109-
expected = np.less(dpt.asnumpy(ar1), dpt.asnumpy(ar2))
118+
expected = np.less(ar1_np, ar2_np)
110119
assert (dpt.asnumpy(r) == expected).all()
111120

112121
r1 = dpt.less(ar2, ar1)
113-
expected1 = np.less(dpt.asnumpy(ar2), dpt.asnumpy(ar1))
122+
expected1 = np.less(ar2_np, ar1_np)
114123
assert (dpt.asnumpy(r1) == expected1).all()
115124
with np.errstate(invalid="ignore"):
116125
for tp in [dpt.nan, dpt.inf, -dpt.inf]:
117126

118127
ar3 = dpt.full((4,), tp)
128+
ar3_np = dpt.asnumpy(ar3)
129+
119130
r2 = dpt.less(ar1, ar3)
120-
expected2 = np.less(dpt.asnumpy(ar1), dpt.asnumpy(ar3))
131+
expected2 = np.less(ar1_np, ar3_np)
121132
assert (dpt.asnumpy(r2) == expected2).all()
122133

123134
r3 = dpt.less(ar3, ar1)
124-
expected3 = np.less(dpt.asnumpy(ar3), dpt.asnumpy(ar1))
135+
expected3 = np.less(ar3_np, ar1_np)
125136
assert (dpt.asnumpy(r3) == expected3).all()
126137

127138

dpctl/tests/elementwise/test_less_equal.py

Lines changed: 20 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -68,34 +68,40 @@ def test_less_equal_complex_matrix(op_dtype):
6868
sz = 127
6969
ar1_np_real = np.random.randint(0, 10, sz)
7070
ar1_np_imag = np.random.randint(0, 10, sz)
71-
ar1 = dpt.asarray(ar1_np_real + 1j * ar1_np_imag, dtype=op_dtype)
71+
ar1_np = ar1_np_real + 1j * ar1_np_imag
72+
ar1 = dpt.asarray(ar1_np, dtype=op_dtype)
7273

7374
ar2_np_real = np.random.randint(0, 10, sz)
7475
ar2_np_imag = np.random.randint(0, 10, sz)
75-
ar2 = dpt.asarray(ar2_np_real + 1j * ar2_np_imag, dtype=op_dtype)
76+
ar2_np = ar2_np_real + 1j * ar2_np_imag
77+
ar2 = dpt.asarray(ar2_np, dtype=op_dtype)
7678

7779
r = dpt.less_equal(ar1, ar2)
78-
expected = np.less_equal(dpt.asnumpy(ar1), dpt.asnumpy(ar2))
80+
expected = np.less_equal(ar1_np, ar2_np)
7981
assert _compare_dtypes(r.dtype, expected.dtype, sycl_queue=q)
8082
assert r.shape == expected.shape
8183
assert (dpt.asnumpy(r) == expected).all()
8284

8385
r1 = dpt.less_equal(ar1[::-2], ar2[::2])
84-
expected1 = np.less_equal(dpt.asnumpy(ar1[::-2]), dpt.asnumpy(ar2[::2]))
86+
expected1 = np.less_equal(ar1_np[::-2], ar2_np[::2])
8587
assert _compare_dtypes(r.dtype, expected1.dtype, sycl_queue=q)
8688
assert r1.shape == expected1.shape
8789
assert (dpt.asnumpy(r1) == expected1).all()
8890

8991
ar3 = dpt.asarray([1.0 + 9j, 2.0 + 0j, 2.0 + 1j, 2.0 + 2j], dtype=op_dtype)
9092
ar4 = dpt.asarray([2.0 + 0j, dpt.nan, dpt.inf, -dpt.inf], dtype=op_dtype)
93+
94+
ar3_np = dpt.asnumpy(ar3)
95+
ar4_np = dpt.asnumpy(ar4)
96+
9197
r2 = dpt.less_equal(ar3, ar4)
9298
with np.errstate(invalid="ignore"):
93-
expected2 = np.less_equal(dpt.asnumpy(ar3), dpt.asnumpy(ar4))
99+
expected2 = np.less_equal(ar3_np, ar4_np)
94100
assert (dpt.asnumpy(r2) == expected2).all()
95101

96102
r3 = dpt.less_equal(ar4, ar4)
97103
with np.errstate(invalid="ignore"):
98-
expected3 = np.less_equal(dpt.asnumpy(ar4), dpt.asnumpy(ar4))
104+
expected3 = np.less_equal(ar4_np, ar4_np)
99105
assert (dpt.asnumpy(r3) == expected3).all()
100106

101107

@@ -105,23 +111,27 @@ def test_less_equal_complex_float():
105111
ar1 = dpt.asarray([1.0 + 9j, 2.0 + 0j, 2.0 + 1j, 2.0 + 2j], dtype="c8")
106112
ar2 = dpt.full((4,), 2, dtype="f4")
107113

114+
ar1_np = dpt.asnumpy(ar1)
115+
ar2_np = dpt.asnumpy(ar2)
116+
108117
r = dpt.less_equal(ar1, ar2)
109-
expected = np.less_equal(dpt.asnumpy(ar1), dpt.asnumpy(ar2))
118+
expected = np.less_equal(ar1_np, ar2_np)
110119
assert (dpt.asnumpy(r) == expected).all()
111120

112121
r1 = dpt.less_equal(ar2, ar1)
113-
expected1 = np.less_equal(dpt.asnumpy(ar2), dpt.asnumpy(ar1))
122+
expected1 = np.less_equal(ar2_np, ar1_np)
114123
assert (dpt.asnumpy(r1) == expected1).all()
115124
with np.errstate(invalid="ignore"):
116125
for tp in [dpt.nan, dpt.inf, -dpt.inf]:
117126

118127
ar3 = dpt.full((4,), tp)
128+
ar3_np = dpt.asnumpy(ar3)
119129
r2 = dpt.less_equal(ar1, ar3)
120-
expected2 = np.less_equal(dpt.asnumpy(ar1), dpt.asnumpy(ar3))
130+
expected2 = np.less_equal(ar1_np, ar3_np)
121131
assert (dpt.asnumpy(r2) == expected2).all()
122132

123133
r3 = dpt.less_equal(ar3, ar1)
124-
expected3 = np.less_equal(dpt.asnumpy(ar3), dpt.asnumpy(ar1))
134+
expected3 = np.less_equal(ar3_np, ar1_np)
125135
assert (dpt.asnumpy(r3) == expected3).all()
126136

127137

0 commit comments

Comments
 (0)