nordic-dev.net/nixpkgs
nordic-dev.net/nixpkgs
2
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2025-01-31 17:23:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

onnxruntime: add ncclSupport flag (#350393)

Browse Source
This commit is contained in:
Aleksana 2024-10-27 12:12:44 +08:00 committed by GitHub
commit 3eced873b3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
pkgs/development/libraries/onnxruntime/default.nix
View File

@ -23,6 +23,7 @@
, protobuf_21
, pythonSupport ? true
, cudaSupport ? config.cudaSupport
, ncclSupport ? config.cudaSupport
, cudaPackages ? {}
}@inputs:
@ -160,8 +161,9 @@ effectiveStdenv.mkDerivation rec {
libcufft # cufft.h
cudnn # cudnn.h
cuda_cudart
] ++ lib.optionals (cudaSupport && ncclSupport) (with cudaPackages; [
nccl
]);
]));
nativeCheckInputs = [
gtest
@ -198,7 +200,7 @@ effectiveStdenv.mkDerivation rec {
"-Donnxruntime_ENABLE_LTO=ON"
"-Donnxruntime_USE_FULL_PROTOBUF=OFF"
(lib.cmakeBool "onnxruntime_USE_CUDA" cudaSupport)
(lib.cmakeBool "onnxruntime_USE_NCCL" cudaSupport)
(lib.cmakeBool "onnxruntime_USE_NCCL" (cudaSupport && ncclSupport))
] ++ lib.optionals pythonSupport [
"-Donnxruntime_ENABLE_PYTHON=ON"
] ++ lib.optionals cudaSupport [