Add required packages to frameworks

Author: ZzEeKkAa

dpbench/config/reader.py

@@ -13,6 +13,8 @@
 
 import tomli
 
+from dpbench.infrastructure.frameworks.fabric import get_framework_class
+
 from .benchmark import Benchmark, BenchmarkImplementation, Presets
 from .config import Config
 from .framework import Framework
@@ -117,7 +119,7 @@ def read_configs(  # noqa: C901: TODO: move modules into config
     for framework in config.frameworks:
         config.implementations += framework.postfixes
 
-    if implementations is None:
+    if implementations is None or len(implementations) == 0:
         implementations = {impl.postfix for impl in config.implementations}
 
     if load_implementations:
@@ -228,6 +230,15 @@ def read_frameworks(
         if len(framework.postfixes) == 0:
             continue
 
+        cls = get_framework_class(framework)
+        unavailable_pkgs = cls.get_missing_required_packages()
+        if len(unavailable_pkgs) > 0:
+            logging.warning(
+                f"Framework {framework.simple_name} unavailable "
+                + f"due to missing packages {unavailable_pkgs}"
+            )
+            continue
+
         config.frameworks.append(framework)
 
 

dpbench/infrastructure/frameworks/cupy_framework.py

@@ -19,6 +19,10 @@ def __init__(self, fname: str = None, config: cfg.Framework = None):
 
         super().__init__(fname, config)
 
+    @staticmethod
+    def required_packages() -> list[str]:
+        return ["cupy"]
+
     def copy_to_func(self) -> Callable:
         """Returns the copy-method that should be used
         for copying the benchmark arguments."""

dpbench/infrastructure/frameworks/dpnp_framework.py

@@ -35,6 +35,10 @@ def __init__(self, fname: str = None, config: cfg.Framework = None):
             )
             raise sdce
 
+    @staticmethod
+    def required_packages() -> list[str]:
+        return ["dpnp"]
+
     def device_filter_string(self) -> str:
         """Returns the sycl device's filter string if the framework has an
         associated sycl device."""

dpbench/infrastructure/frameworks/fabric.py

@@ -39,7 +39,7 @@ def build_framework_map() -> dict[str, Framework]:
     return result
 
 
-def build_framework(framework_config: cfg.Framework) -> Framework:
+def get_framework_class(framework_config: cfg.Framework) -> Framework:
     available_classes = [
         Framework,
         DpcppFramework,
@@ -61,4 +61,9 @@ def build_framework(framework_config: cfg.Framework) -> Framework:
         )
         constructor = Framework
 
+    return constructor
+
+
+def build_framework(framework_config: cfg.Framework) -> Framework:
+    constructor = get_framework_class(framework_config)
     return constructor(config=framework_config)

dpbench/infrastructure/frameworks/framework.py

@@ -5,6 +5,7 @@
 # SPDX-License-Identifier: BSD-3-Clause
 
 import logging
+from importlib.util import find_spec
 from typing import Any, Callable, Dict, final
 
 import pkg_resources
@@ -45,6 +46,20 @@ def __init__(
 
         self.device_info = cpuinfo.get_cpu_info().get("brand_raw")
 
+    @staticmethod
+    def required_packages() -> list[str]:
+        return []
+
+    @classmethod
+    def get_missing_required_packages(cls) -> None:
+        unavailable_packages = []
+        for pkg in cls.required_packages():
+            spec = find_spec(pkg)
+            if spec is None:
+                unavailable_packages.append(pkg)
+
+        return unavailable_packages
+
     def device_filter_string(self) -> str:
         """Returns the sycl device's filter string if the framework has an
         associated sycl device."""

dpbench/infrastructure/frameworks/numba_cuda_framework.py

@@ -19,6 +19,10 @@ def __init__(self, fname: str = None, config: cfg.Framework = None):
 
         super().__init__(fname, config)
 
+    @staticmethod
+    def required_packages() -> list[str]:
+        return ["cupy"]
+
     def copy_to_func(self) -> Callable:
         """Returns the copy-method that should be used
         for copying the benchmark arguments."""

dpbench/infrastructure/frameworks/numba_dpex_framework.py

@@ -33,6 +33,10 @@ def __init__(self, fname: str = None, config: cfg.Framework = None):
             )
             raise sdce
 
+    @staticmethod
+    def required_packages() -> list[str]:
+        return ["numba_dpex"]
+
     def device_filter_string(self) -> str:
         """Returns the sycl device's filter string if the framework has an
         associated sycl device."""

dpbench/infrastructure/frameworks/numba_framework.py

@@ -27,3 +27,7 @@ def __init__(self, fname: str = None, config: cfg.Framework = None):
         """
 
         super().__init__(fname, config)
+
+    @staticmethod
+    def required_packages() -> list[str]:
+        return ["numba"]

dpbench/infrastructure/frameworks/numba_mlir_framework.py

@@ -28,6 +28,10 @@ def __init__(self, fname: str = None, config: cfg.Framework = None):
 
             self.device_info = dpctl.SyclDevice(self.sycl_device).name
 
+    @staticmethod
+    def required_packages() -> list[str]:
+        return ["numba_mlir"]
+
     def copy_to_func(self) -> Callable:
         """Returns the copy-method that should be used
         for copying the benchmark arguments to device."""