Add helper test function to obtain device ID

Author: antonwolfy

dpnp/tests/conftest.py

@@ -41,6 +41,8 @@
 
 import dpnp
 
+from .helper import get_dev_id
+
 skip_mark = pytest.mark.skip(reason="Skipping test.")
 
 
@@ -129,7 +131,6 @@ def pytest_collection_modifyitems(config, items):
     test_exclude_file_cuda = os.path.join(test_path, "skipped_tests_cuda.tbl")
 
     dev = dpctl.select_default_device()
-    dev_id = dpctl.utils.intel_device_info(dev).get("device_id", 0)
     is_cpu = dev.is_cpu
     is_gpu = dev.is_gpu
     support_fp64 = dev.has_aspect_fp64
@@ -139,7 +140,7 @@ def pytest_collection_modifyitems(config, items):
     print(
         f"DPNP Test scope includes all integer dtypes: {bool(dtype_config.all_int_types)}"
     )
-    print(f"DPNP current device ID: {dev_id:#x}")
+    print(f"DPNP current device ID: {get_dev_id(dev):#x}")
     print(f"DPNP current device is CPU: {is_cpu}")
     print(f"DPNP current device is GPU: {is_gpu}")
     print(f"DPNP current device supports fp64: {support_fp64}")

dpnp/tests/helper.py

@@ -26,14 +26,8 @@ def _assert_shape(a, b):
         assert a.shape == (), f"{a.shape} != ()"
 
 
-def _get_dev_id(device=None):
-    dev = dpctl.select_default_device() if device is None else device
-    dev_info = dpctl.utils.intel_device_info(dev)
-    return dev_info.get("device_id", 0)
-
-
 def _get_dev_mask(device=None):
-    return _get_dev_id(device) & 0xFF00
+    return get_dev_id(device) & 0xFF00
 
 
 def assert_dtype_allclose(
@@ -309,6 +303,16 @@ def get_complex_dtypes(device=None):
     return dtypes
 
 
+def get_dev_id(device=None):
+    """
+    Obtain Intel Device ID for a device (the default device if not provided).
+    """
+
+    dev = dpctl.select_default_device() if device is None else device
+    dev_info = dpctl.utils.intel_device_info(dev)
+    return dev_info.get("device_id", 0)
+
+
 def get_float_dtypes(no_float16=True, device=None):
     """
     Build a list of floating types supported by DPNP based on device capabilities.