From a175b672b6829cda06c66b3c028807424fe52c8e Mon Sep 17 00:00:00 2001 From: winston Date: Wed, 3 Jun 2026 19:24:36 +0200 Subject: [PATCH] python3Packages.onnxruntime: add openvino support --- pkgs/development/python-modules/onnxruntime/default.nix | 2 ++ pkgs/top-level/python-packages.nix | 1 + 2 files changed, 3 insertions(+) diff --git a/pkgs/development/python-modules/onnxruntime/default.nix b/pkgs/development/python-modules/onnxruntime/default.nix index 6d52ed8ca512..250bea6c2290 100644 --- a/pkgs/development/python-modules/onnxruntime/default.nix +++ b/pkgs/development/python-modules/onnxruntime/default.nix @@ -10,6 +10,7 @@ re2, # dependencies + openvino, coloredlogs, numpy, packaging, @@ -76,6 +77,7 @@ buildPythonPackage { ++ lib.optionals onnxruntime.passthru.ncclSupport [ nccl # libnccl.so.XX ] + ++ lib.optionals stdenv.hostPlatform.isLinux [ openvino ] ); dependencies = [ diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 7e4d095655c1..dd384f1b3452 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -11835,6 +11835,7 @@ self: super: with self; { python3Packages = self; pythonSupport = true; }; + inherit (pkgs) openvino; }; onnxruntime-tools = callPackage ../development/python-modules/onnxruntime-tools { };