Skip to content

Commit 004614e

Browse files
Merge pull request #1537 from IntelPython/fix-coverity-issues-in-tests
2 parents f9d3501 + bb81be5 commit 004614e

8 files changed

+156
-79
lines changed

dpctl/tests/elementwise/test_greater.py

+21-10
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

+19-10
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

+21-10
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

+20-10
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

dpctl/tests/elementwise/test_logical_and.py

+20-10
Original file line numberDiff line numberDiff line change
@@ -75,20 +75,22 @@ def test_logical_and_complex_matrix(op_dtype):
7575
sz = 127
7676
ar1_np_real = np.random.randint(0, 2, sz)
7777
ar1_np_imag = np.random.randint(0, 2, sz)
78-
ar1 = dpt.asarray(ar1_np_real + 1j * ar1_np_imag, dtype=op_dtype)
78+
ar1_np = ar1_np_real + 1j * ar1_np_imag
79+
ar1 = dpt.asarray(ar1_np, dtype=op_dtype)
7980

8081
ar2_np_real = np.random.randint(0, 2, sz)
8182
ar2_np_imag = np.random.randint(0, 2, sz)
82-
ar2 = dpt.asarray(ar2_np_real + 1j * ar2_np_imag, dtype=op_dtype)
83+
ar2_np = ar2_np_real + 1j * ar2_np_imag
84+
ar2 = dpt.asarray(ar2_np, dtype=op_dtype)
8385

8486
r = dpt.logical_and(ar1, ar2)
85-
expected = np.logical_and(dpt.asnumpy(ar1), dpt.asnumpy(ar2))
87+
expected = np.logical_and(ar1_np, ar2_np)
8688
assert _compare_dtypes(r.dtype, expected.dtype, sycl_queue=q)
8789
assert r.shape == expected.shape
8890
assert (dpt.asnumpy(r) == expected).all()
8991

9092
r1 = dpt.logical_and(ar1[::-2], ar2[::2])
91-
expected1 = np.logical_and(dpt.asnumpy(ar1[::-2]), dpt.asnumpy(ar2[::2]))
93+
expected1 = np.logical_and(ar1_np[::-2], ar2_np[::2])
9294
assert _compare_dtypes(r.dtype, expected1.dtype, sycl_queue=q)
9395
assert r1.shape == expected1.shape
9496
assert (dpt.asnumpy(r1) == expected1).all()
@@ -106,14 +108,18 @@ def test_logical_and_complex_matrix(op_dtype):
106108
dtype=op_dtype,
107109
)
108110
ar4 = dpt.full(ar3.shape, fill_value=1.0 + 2j, dtype=op_dtype)
111+
112+
ar3_np = dpt.asnumpy(ar3)
113+
ar4_np = dpt.asnumpy(ar4)
114+
109115
r2 = dpt.logical_and(ar3, ar4)
110116
with np.errstate(invalid="ignore"):
111-
expected2 = np.logical_and(dpt.asnumpy(ar3), dpt.asnumpy(ar4))
117+
expected2 = np.logical_and(ar3_np, ar4_np)
112118
assert (dpt.asnumpy(r2) == expected2).all()
113119

114120
r3 = dpt.logical_and(ar4, ar4)
115121
with np.errstate(invalid="ignore"):
116-
expected3 = np.logical_and(dpt.asnumpy(ar4), dpt.asnumpy(ar4))
122+
expected3 = np.logical_and(ar4_np, ar4_np)
117123
assert (dpt.asnumpy(r3) == expected3).all()
118124

119125

@@ -123,12 +129,15 @@ def test_logical_and_complex_float():
123129
ar1 = dpt.asarray([1j, 1.0 + 9j, 2.0 + 0j, 2.0 + 1j], dtype="c8")
124130
ar2 = dpt.full(ar1.shape, 2, dtype="f4")
125131

132+
ar1_np = dpt.asnumpy(ar1)
133+
ar2_np = dpt.asnumpy(ar2)
134+
126135
r = dpt.logical_and(ar1, ar2)
127-
expected = np.logical_and(dpt.asnumpy(ar1), dpt.asnumpy(ar2))
136+
expected = np.logical_and(ar1_np, ar2_np)
128137
assert (dpt.asnumpy(r) == expected).all()
129138

130139
r1 = dpt.logical_and(ar2, ar1)
131-
expected1 = np.logical_and(dpt.asnumpy(ar2), dpt.asnumpy(ar1))
140+
expected1 = np.logical_and(ar2_np, ar1_np)
132141
assert (dpt.asnumpy(r1) == expected1).all()
133142
with np.errstate(invalid="ignore"):
134143
for tp in [
@@ -140,12 +149,13 @@ def test_logical_and_complex_float():
140149
-dpt.inf * 1j,
141150
]:
142151
ar3 = dpt.full(ar1.shape, tp)
152+
ar3_np = dpt.asnumpy(ar3)
143153
r2 = dpt.logical_and(ar1, ar3)
144-
expected2 = np.logical_and(dpt.asnumpy(ar1), dpt.asnumpy(ar3))
154+
expected2 = np.logical_and(ar1_np, ar3_np)
145155
assert (dpt.asnumpy(r2) == expected2).all()
146156

147157
r3 = dpt.logical_and(ar3, ar1)
148-
expected3 = np.logical_and(dpt.asnumpy(ar3), dpt.asnumpy(ar1))
158+
expected3 = np.logical_and(ar3_np, ar1_np)
149159
assert (dpt.asnumpy(r3) == expected3).all()
150160

151161

0 commit comments

Comments
 (0)