ucc: drop the cudatoolkit runfile

This commit is contained in:
Someone Serge
2023-12-04 20:24:32 +00:00
parent 9cc210a783
commit 3e37f3c983
+11 -7
View File
@@ -1,7 +1,7 @@
{ stdenv, lib, fetchFromGitHub, libtool, automake, autoconf, ucx
, config
, enableCuda ? config.cudaSupport
, cudatoolkit
, cudaPackages
, enableAvx ? stdenv.hostPlatform.avxSupport
, enableSse41 ? stdenv.hostPlatform.sse4_1Support
, enableSse42 ? stdenv.hostPlatform.sse4_2Support
@@ -30,19 +30,23 @@ stdenv.mkDerivation rec {
done
'';
nativeBuildInputs = [ libtool automake autoconf ]
++ lib.optionals enableCuda [ cudaPackages.cuda_nvcc ];
buildInputs = [ ucx ]
++ lib.optionals enableCuda [
cudaPackages.cuda_cccl
cudaPackages.cuda_cudart
];
preConfigure = ''
./autogen.sh
'';
nativeBuildInputs = [ libtool automake autoconf ];
buildInputs = [ ucx ]
++ lib.optional enableCuda cudatoolkit;
configureFlags = [ ]
++ lib.optional enableSse41 "--with-sse41"
++ lib.optional enableSse42 "--with-sse42"
++ lib.optional enableAvx "--with-avx"
++ lib.optional enableCuda "--with-cuda=${cudatoolkit}";
++ lib.optional enableCuda "--with-cuda=${cudaPackages.cuda_cudart}";
postInstall = ''
find $out/lib/ -name "*.la" -exec rm -f \{} \;