onnxruntime: limit nvcc threads

2024-04-14 19:39:56 +02:00
parent 852af4a5dc
commit db0727e8c0
1 changed files with 1 additions and 0 deletions
@@ -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 {