Unverified Commit ad5e8b76 authored by Conroy Cheers's avatar Conroy Cheers Committed by Pavol Rusnak
Browse files

onnxruntime: add autoAddCudaCompatRunpath on Jetson

parent 46b88a98

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 = [