Commit 1eb39d0e authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files

onnxruntime: remove reference to nvcc in libonnxruntime_providers_cuda.so

parent 4a807ebf

pkgs/by-name/on/onnxruntime/package.nix

+9 −0
Original line number Diff line number Diff line
@@ -17,6 +17,7 @@ 
  perl,

  pkg-config,

  python3Packages,

  removeReferencesTo,

  re2,

  zlib,

  protobuf,
@@ -44,6 +45,7 @@ let 


  stdenv = throw "Use effectiveStdenv instead";

  effectiveStdenv = if cudaSupport then cudaPackages.backendStdenv else inputs.stdenv;

  inherit (cudaPackages) cuda_nvcc;



  cudaArchitecturesString = cudaPackages.flags.cmakeCudaArchitecturesString;
@@ -121,6 +123,7 @@ effectiveStdenv.mkDerivation rec { 
  ++ lib.optionals cudaSupport [

    cudaPackages.cuda_nvcc

    cudaPackages.cudnn-frontend

    removeReferencesTo

  ]

  ++ lib.optionals isCudaJetson [

    cudaPackages.autoAddCudaCompatRunpath
@@ -318,6 +321,12 @@ effectiveStdenv.mkDerivation rec { 
      ../include/onnxruntime/core/session/onnxruntime_*.h

  '';



  # See comments in `cudaPackages.nccl`

  postFixup = lib.optionalString cudaSupport ''

    remove-references-to -t "${lib.getBin cuda_nvcc}" ''${!outputLib}/lib/libonnxruntime_providers_cuda.so

  '';

  disallowedRequisites = [ (lib.getBin cuda_nvcc) ];



  passthru = {

    inherit cudaSupport cudaPackages ncclSupport; # for the python module

    inherit protobuf;