mirror of
https://github.com/ilyakooo0/nixpkgs.git
synced 2024-09-11 07:15:54 +03:00
Merge pull request #272084 from ConnorBaker/fix/ctranslate2-optional-cudnn-dep
ctranslate2: enable cuDNN only if it is available
This commit is contained in:
commit
2c7f3c0fb7
@ -60,7 +60,7 @@ stdenv.mkDerivation rec {
|
||||
cudaPackages.cuda_cudart
|
||||
cudaPackages.libcublas
|
||||
cudaPackages.libcurand
|
||||
] ++ lib.optionals withCuDNN [
|
||||
] ++ lib.optionals (withCUDA && withCuDNN) [
|
||||
cudaPackages.cudnn
|
||||
] ++ lib.optionals withOneDNN [
|
||||
oneDNN
|
||||
@ -87,5 +87,8 @@ stdenv.mkDerivation rec {
|
||||
changelog = "https://github.com/OpenNMT/CTranslate2/blob/${src.rev}/CHANGELOG.md";
|
||||
license = licenses.mit;
|
||||
maintainers = with maintainers; [ hexa misuzu ];
|
||||
broken =
|
||||
(lib.versionOlder cudaPackages.cudaVersion "11.4")
|
||||
|| !(withCuDNN -> withCUDA);
|
||||
};
|
||||
}
|
||||
|
@ -20935,7 +20935,8 @@ with pkgs;
|
||||
ctranslate2 = callPackage ../development/libraries/ctranslate2 rec {
|
||||
stdenv = if withCUDA then gcc11Stdenv else pkgs.stdenv;
|
||||
withCUDA = pkgs.config.cudaSupport;
|
||||
withCuDNN = pkgs.config.cudaSupport;
|
||||
withCuDNN = withCUDA && (cudaPackages ? cudnn);
|
||||
cudaPackages = pkgs.cudaPackages;
|
||||
};
|
||||
|
||||
ubus = callPackage ../development/libraries/ubus { };
|
||||
|
Loading…
Reference in New Issue
Block a user