Merge pull request #167741 from SomeoneSerge/fix-xgboost-cuda-11_4

xgboost: fix cmake for cuda>=11.4
2022-04-08 13:23:21 -07:00 · 2022-04-08 13:23:21 -07:00 · 7208e65e26
parent 3cb2a2d199 325adc9280
commit 7208e65e26
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;