Add select_device_with_aspects func

Author: vlad-perevezentsev

dpctl/__init__.py

@@ -64,6 +64,7 @@
     set_global_queue,
 )
 
+from ._device_selection import select_device_with_aspects
 from ._version import get_versions
 from .enum_types import backend_type, device_type, event_status_type
 
@@ -80,6 +81,7 @@
     "select_default_device",
     "select_gpu_device",
     "select_host_device",
+    "select_device_with_aspects",
     "get_num_devices",
     "has_cpu_devices",
     "has_gpu_devices",

dpctl/_device_selection.py

@@ -0,0 +1,29 @@
+from . import SyclDevice, get_devices
+
+
+def select_device_with_aspects(aspect_list, deny_list=[]):
+    check_list = aspect_list + deny_list
+    for asp in check_list:
+        if type(asp) != str:
+            raise TypeError("The list objects must be of a string type")
+        if not hasattr(SyclDevice, "has_aspect_" + asp):
+            raise ValueError(f"The {asp} aspect is not supported in dpctl")
+    devs = get_devices()
+    max_score = 0
+    selected_dev = None
+
+    for dev in devs:
+        aspect_status = True
+        for asp in aspect_list:
+            has_aspect = "dev.has_aspect_" + asp
+            if not eval(has_aspect):
+                aspect_status = False
+        for deny in deny_list:
+            has_aspect = "dev.has_aspect_" + deny
+            if eval(has_aspect):
+                aspect_status = False
+        if aspect_status and dev.default_selector_score > max_score:
+            max_score = dev.default_selector_score
+            selected_dev = dev
+
+    return selected_dev

dpctl/tests/test_sycl_device.py

@@ -645,3 +645,53 @@ def test_hashing_of_device():
     """
     device_dict = {dpctl.SyclDevice(): "default_device"}
     assert device_dict
+
+
+list_of_valid_aspects = [
+    "cpu",
+    "gpu",
+    "accelerator",
+    "custom",
+    "fp16",
+    "fp64",
+    "image",
+    "online_compiler",
+    "online_linker",
+    "queue_profiling",
+    "usm_device_allocations",
+    "usm_host_allocations",
+    "usm_shared_allocations",
+    "usm_system_allocator",
+]
+
+list_of_invalid_aspects = [
+    "emulated",
+    "host_debuggable",
+    "atomic64",
+    "usm_atomic_host_allocations",
+    "usm_atomic_shared_allocations",
+]
+
+
+@pytest.fixture(params=list_of_valid_aspects)
+def valid_aspects(request):
+    return request.param
+
+
+@pytest.fixture(params=list_of_invalid_aspects)
+def invalid_aspects(request):
+    return request.param
+
+
+def test_valid_aspects(valid_aspects):
+    dpctl.select_device_with_aspects([valid_aspects])
+
+
+def test_invalid_aspects(invalid_aspects):
+    try:
+        dpctl.select_device_with_aspects([invalid_aspects])
+        raise AttributeError(
+            f"The {invalid_aspects} aspect is supported in dpctl"
+        )
+    except ValueError:
+        pytest.skip(f"The {invalid_aspects} aspect is not supported in dpctl")