Skip to content

Commit 85d468c

Browse files
Migrated tests for sqrt into elementwise folder
1 parent dc386eb commit 85d468c

File tree

1 file changed

+1
-22
lines changed

1 file changed

+1
-22
lines changed

dpctl/tests/test_tensor_sqrt.py renamed to dpctl/tests/elementwise/test_sqrt.py

Lines changed: 1 addition & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -5,30 +5,9 @@
55
from numpy.testing import assert_equal
66

77
import dpctl.tensor as dpt
8-
import dpctl.tensor._type_utils as tu
98
from dpctl.tests.helper import get_queue_or_skip, skip_if_dtype_not_supported
109

11-
_all_dtypes = [
12-
"b1",
13-
"i1",
14-
"u1",
15-
"i2",
16-
"u2",
17-
"i4",
18-
"u4",
19-
"i8",
20-
"u8",
21-
"f2",
22-
"f4",
23-
"f8",
24-
"c8",
25-
"c16",
26-
]
27-
_usm_types = ["device", "shared", "host"]
28-
29-
30-
def _map_to_device_dtype(dt, dev):
31-
return tu._to_device_supported_dtype(dt, dev)
10+
from .utils import _all_dtypes, _map_to_device_dtype, _usm_types
3211

3312

3413
@pytest.mark.parametrize("dtype", _all_dtypes)

0 commit comments

Comments
 (0)