Unverified Commit 73efae15 authored by Pol Dellaiera's avatar Pol Dellaiera Committed by GitHub
Browse files

python3Packages.onnxruntime: buildInputs dep on onnxruntime (#382903)

parents eb974fbf eb400924

pkgs/development/python-modules/onnxruntime/default.nix

+7 −0
Original line number Diff line number Diff line
@@ -50,6 +50,13 @@ buildPythonPackage { 
      oneDNN

      re2

      onnxruntime.protobuf



      # https://github.com/NixOS/nixpkgs/pull/357656 patches the onnx lib to ${pkgs.onnxruntime}/lib

      # but these files are copied into this package too. If the origional non-python onnxruntime

      # package is GC-ed, cuda support in this python package will break.

      # Two options, rebuild onnxruntime twice with the different paths hard-coded, or just hold a runtime

      # dependency between the two. Option 2, because onnxruntime takes forever to build with cuda support.

      onnxruntime

    ]

    ++ lib.optionals onnxruntime.passthru.cudaSupport (

      with onnxruntime.passthru.cudaPackages;