diff options
author | Samuel Ainsworth <skainsworth@gmail.com> | 2022-04-08 13:23:21 -0700 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-04-08 13:23:21 -0700 |
commit | 7208e65e2600915579c37a74b6579dbc6cf038f8 (patch) | |
tree | 77338e360db251a70891c9bd1a411795329d2d1f | |
parent | 3cb2a2d199ac560272f6d2b7ea4cf56e8f1996b6 (diff) | |
parent | 325adc928057d9d73406e0014844c4c2ecbaa8c9 (diff) |
Merge pull request #167741 from SomeoneSerge/fix-xgboost-cuda-11_4
xgboost: fix cmake for cuda>=11.4
-rw-r--r-- | pkgs/development/libraries/xgboost/default.nix | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/pkgs/development/libraries/xgboost/default.nix b/pkgs/development/libraries/xgboost/default.nix index b9f35d407b39e..6186661a12c2d 100644 --- 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; |