diff --git a/modules/sdnq/kernels/openvino_mm.py b/modules/sdnq/kernels/openvino_mm.py index 5190266cc..51b5f392a 100644 --- a/modules/sdnq/kernels/openvino_mm.py +++ b/modules/sdnq/kernels/openvino_mm.py @@ -6,7 +6,7 @@ from openvino.properties import hint as ov_hints core = ov.Core() -OV_DEVICE: str = os.environ.get("SDNQ_OPENVINO_DEVICE", "CPU") +OV_DEVICE: str = os.environ.get("SDNQ_OPENVINO_DEVICE", "HETERO:NPU,CPU" if "NPU" in core.get_available_devices() else "CPU") OV_COMPILED_CACHE: dict[tuple[str, tuple[int,int] | None, str, tuple[int,int] | None], tuple[ov.InferRequest, str]] = {} if OV_DEVICE == "NPU":