Skip to content

Commit fd7b3f9

Browse files
Add tests for new function in dpctl.utils
Adds a test for _is_gen9 helper utility useful for skipping tests known to fail on Gen9. Adds a test for intel_device_info function. Test that descriptor names do not have typos.
1 parent a80b015 commit fd7b3f9

File tree

1 file changed

+36
-0
lines changed

1 file changed

+36
-0
lines changed

dpctl/tests/test_utils.py

Lines changed: 36 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121

2222
import dpctl
2323
import dpctl.utils
24+
from dpctl.enum_types import backend_type
2425

2526

2627
def test_get_execution_queue_input_validation():
@@ -122,3 +123,38 @@ def test_onetrace_enabled():
122123
with dpctl.utils.onetrace_enabled():
123124
assert os.getenv(v_name, None) == "1"
124125
assert os.getenv(v_name, None) == v_v
126+
127+
128+
def test__is_gen9():
129+
try:
130+
d = dpctl.select_default_device()
131+
except dpctl.SyclDeviceCreationError:
132+
pytest.skip("Default device could not be created")
133+
u = dpctl.utils._is_gen9(d)
134+
assert isinstance(u, bool)
135+
136+
137+
def test_intel_device_info():
138+
try:
139+
d = dpctl.select_default_device()
140+
except dpctl.SyclDeviceCreationError:
141+
pytest.skip("Default device could not be created")
142+
descr = dpctl.utils.intel_device_info(d)
143+
assert isinstance(descr, dict)
144+
assert ("device_id" in descr) or (
145+
not d.has_aspect_cpu and not d.backend == backend_type.level_zero
146+
)
147+
allowed_names = [
148+
"device_id",
149+
"gpu_slices",
150+
"gpu_eu_count",
151+
"gpu_eu_simd_width",
152+
"gpu_hw_threads_per_eu",
153+
"gpu_subslices_per_slice",
154+
"gpu_eu_count_per_subslice",
155+
"max_mem_bandwidth",
156+
]
157+
for descriptor_name in descr.keys():
158+
test = descriptor_name in allowed_names
159+
err_msg = f"Key '{descriptor_name}' is not recognized"
160+
assert test, err_msg

0 commit comments

Comments
 (0)