onnxruntime: limit nvcc threads

2024-04-06 19:25:35 +02:00 · 2024-04-06 19:25:35 +02:00 · db0727e8c0
parent 852af4a5dc
commit db0727e8c0
1 changed files with 1 additions and 0 deletions
--- a/pkgs/development/libraries/onnxruntime/default.nix
+++ b/pkgs/development/libraries/onnxruntime/default.nix
@ -191,6 +191,7 @@ effectiveStdenv.mkDerivation rec {
    (lib.cmakeFeature "FETCHCONTENT_SOURCE_DIR_CUTLASS" "${cutlass}")
    (lib.cmakeFeature "onnxruntime_CUDNN_HOME" "${cudaPackages.cudnn}")
    (lib.cmakeFeature "CMAKE_CUDA_ARCHITECTURES" cudaArchitecturesString)
+    (lib.cmakeFeature "onnxruntime_NVCC_THREADS" "1")
  ];

  env = lib.optionalAttrs effectiveStdenv.cc.isClang {