Unverified Commit f9ffac5a authored by Pavol Rusnak's avatar Pavol Rusnak Committed by GitHub
Browse files

onnxruntime: add autoAddCudaCompatRunpath on Jetson (#357738)

parents 218681f7 ad5e8b76

pkgs/development/libraries/onnxruntime/default.nix

+4 −2
Original line number Diff line number Diff line
@@ -95,8 +95,8 @@ let 
    hash = "sha256-YJdZ0cMtUncH5Z6TtAWBH0xtAIu2UcbjnVcCM4tfg20=";

  };



  isCudaJetson = cudaSupport && cudaPackages.cudaFlags.isJetsonBuild;

in



effectiveStdenv.mkDerivation rec {

  pname = "onnxruntime";

  inherit src version;
@@ -128,6 +128,8 @@ effectiveStdenv.mkDerivation rec { 
  ]) ++ lib.optionals cudaSupport [

    cudaPackages.cuda_nvcc

    cudaPackages.cudnn-frontend

  ] ++ lib.optionals isCudaJetson [

    cudaPackages.autoAddCudaCompatRunpath

  ];



  buildInputs = [