From 1e8c591e58363907a3003f9d80d749a81e3448b8 Mon Sep 17 00:00:00 2001 From: "Liu, Kaixuan" Date: Tue, 2 Dec 2025 17:03:21 +0800 Subject: [PATCH] add dnnl dll path for windows Signed-off-by: Liu, Kaixuan --- src/kernels/__init__.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) 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 (