xgboost: fix cmake for cuda>=11.4

2022-04-07 20:19:51 +03:00 · 2022-04-07 20:19:51 +03:00 · 325adc9280
commit 325adc9280
parent a7bbda4271
1 changed files with 1 additions and 0 deletions
--- a/pkgs/development/libraries/xgboost/default.nix
+++ b/pkgs/development/libraries/xgboost/default.nix
@ -33,6 +33,7 @@ stdenv.mkDerivation rec {

  cmakeFlags = lib.optionals doCheck [ "-DGOOGLE_TEST=ON" ]
    ++ lib.optionals cudaSupport [ "-DUSE_CUDA=ON" "-DCUDA_HOST_COMPILER=${cudatoolkit.cc}/bin/cc" ]
+    ++ lib.optionals (cudaSupport && lib.versionAtLeast cudatoolkit.version "11.4.0") [ "-DBUILD_WITH_CUDA_CUB=ON" ]
    ++ lib.optionals ncclSupport [ "-DUSE_NCCL=ON" ];

  inherit doCheck;