Skip to content

Commit 0277672

Browse files
Update _testing.py in dpctl_ext/tensor
1 parent 8adaa68 commit 0277672

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

dpctl_ext/tensor/_testing.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -39,10 +39,10 @@
3939

4040

4141
def _allclose_complex_fp(z1, z2, atol, rtol, equal_nan):
42-
z1r = dpt.real(z1)
43-
z1i = dpt.imag(z1)
44-
z2r = dpt.real(z2)
45-
z2i = dpt.imag(z2)
42+
z1r = dpt_ext.real(z1)
43+
z1i = dpt_ext.imag(z1)
44+
z2r = dpt_ext.real(z2)
45+
z2i = dpt_ext.imag(z2)
4646
if equal_nan:
4747
check1 = dpt_ext.all(
4848
dpt_ext.isnan(z1r) == dpt_ext.isnan(z2r)

0 commit comments

Comments
 (0)