Merge pull request #167741 from SomeoneSerge/fix-xgboost-cuda-11_4
xgboost: fix cmake for cuda>=11.4
commit7208e65e2600915579c37a74b6579dbc6cf038f8
authorSamuel Ainsworth <skainsworth@gmail.com>
dateFri 8 Apr 2022, 13:23 (5 years ago)
parent3cb2a2d325adc9
pkgs/development/libraries/xgboost/default.nix+1
@@ -33,6 +33,7 @@ stdenv.mkDerivation rec {
3333
3434 cmakeFlags = lib.optionals doCheck [ "-DGOOGLE_TEST=ON" ]
3535 ++ lib.optionals cudaSupport [ "-DUSE_CUDA=ON" "-DCUDA_HOST_COMPILER=${cudatoolkit.cc}/bin/cc" ]
36+ ++ lib.optionals (cudaSupport && lib.versionAtLeast cudatoolkit.version "11.4.0") [ "-DBUILD_WITH_CUDA_CUB=ON" ]
3637 ++ lib.optionals ncclSupport [ "-DUSE_NCCL=ON" ];
3738
3839 inherit doCheck;