|
|
|
@ -25,7 +25,13 @@ stdenv.mkDerivation rec { |
|
|
|
|
sha256 = "sha256-h7zcHCOxe1h7HRB6idtjf4HUBEoHC4V2pqbN9hpe00g="; |
|
|
|
|
}; |
|
|
|
|
|
|
|
|
|
nativeBuildInputs = [ cmake ] ++ lib.optional stdenv.isDarwin llvmPackages.openmp; |
|
|
|
|
nativeBuildInputs = [ |
|
|
|
|
cmake |
|
|
|
|
] ++ lib.optionals stdenv.isDarwin [ |
|
|
|
|
llvmPackages.openmp |
|
|
|
|
] ++ lib.optionals cudaSupport [ |
|
|
|
|
cudaPackages.autoAddOpenGLRunpathHook |
|
|
|
|
]; |
|
|
|
|
|
|
|
|
|
buildInputs = [ gtest ] ++ lib.optional cudaSupport cudaPackages.cudatoolkit |
|
|
|
|
++ lib.optional ncclSupport cudaPackages.nccl; |
|
|
|
|