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 dc386eb commit 85d468cCopy full SHA for 85d468c
dpctl/tests/test_tensor_sqrt.py renamed to dpctl/tests/elementwise/test_sqrt.py
@@ -5,30 +5,9 @@
5
from numpy.testing import assert_equal
6
7
import dpctl.tensor as dpt
8
-import dpctl.tensor._type_utils as tu
9
from dpctl.tests.helper import get_queue_or_skip, skip_if_dtype_not_supported
10
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)
+from .utils import _all_dtypes, _map_to_device_dtype, _usm_types
32
33
34
@pytest.mark.parametrize("dtype", _all_dtypes)
0 commit comments