|
21 | 21 |
|
22 | 22 | import dpctl
|
23 | 23 | import dpctl.utils
|
| 24 | +from dpctl.enum_types import backend_type |
24 | 25 |
|
25 | 26 |
|
26 | 27 | def test_get_execution_queue_input_validation():
|
@@ -122,3 +123,38 @@ def test_onetrace_enabled():
|
122 | 123 | with dpctl.utils.onetrace_enabled():
|
123 | 124 | assert os.getenv(v_name, None) == "1"
|
124 | 125 | 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