Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
36 changes: 5 additions & 31 deletions onedal/utils/_third_party.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,44 +28,18 @@

from daal4py.sklearn._utils import _package_check_version


@functools.lru_cache(maxsize=256, typed=False)
def is_dpctl_available(version=None):
"""Check availability of DPCtl package.

Parameters
----------
version : str or None, default=None
Minimum supported dpctl version if installed.
Secondary version check skipped if set to None.

Returns
-------
dpctl_available : bool
Flag describing import success.
"""
try:
import dpctl

dpctl_available = True
except ImportError:
dpctl_available = False
if dpctl_available and version is not None:
dpctl_available = _package_check_version(version, dpctl.__version__)
return dpctl_available


# Note: The dpctl package provides SYCL infrastructure (e.g. SyclQueue)
# which is loaded as normal as it is preferred over included backend
# replacements in the core onedal python module.
dpctl_available = is_dpctl_available()

if dpctl_available:
try:
from dpctl import SyclQueue
else:

dpctl_available = True
except ImportError:
from onedal import _default_backend as backend

SyclQueue = backend.SyclQueue
dpctl_available = False


def lazy_import(*module_names: str) -> Callable:
Expand Down
Loading