diff --git a/src/kernels/__init__.py b/src/kernels/__init__.py index 85057db..d4d7eb5 100644 --- a/src/kernels/__init__.py +++ b/src/kernels/__init__.py @@ -1,5 +1,21 @@ +import os +import platform import importlib.metadata + +if platform.system() == "Windows": + # Add Intel oneAPI directories for XPU support + _oneapi_paths = [ + r"C:\Program Files (x86)\Intel\oneAPI\dnnl\latest\bin", + ] + + for _path in _oneapi_paths: + if os.path.exists(_path): + try: + os.add_dll_directory(_path) + except Exception: + pass # Ignore if already added or permission issues + __version__ = importlib.metadata.version("kernels") from kernels.layer import (