Skip to content

Commit 7a12565

Browse files
Add tests for _get_default_device_type utility
1 parent 2be09e5 commit 7a12565

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

dpctl/tests/elementwise/test_type_utils.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -112,6 +112,17 @@ def _denier_fn(dt):
112112
)
113113

114114

115+
def test_type_utils_get_device_default_type():
116+
try:
117+
dev = dpctl.SyclDevice()
118+
except dpctl.SyclDeviceCreationError:
119+
pytest.skip("No SYCL devices available")
120+
for k in ["b", "i", "u", "f", "c"]:
121+
dt = tu._get_device_default_dtype(k, dev)
122+
assert isinstance(dt, dpt.dtype)
123+
assert dt.kind == k
124+
125+
115126
def test_type_utils_find_buf_dtype2():
116127
def _denier_fn(dt1, dt2):
117128
return False

0 commit comments

Comments
 (0)