ucx: fix with cudaSupport (#460294)

This commit is contained in:
Connor Baker
2025-11-10 20:11:47 +00:00
committed by GitHub
+4 -3
View File
@@ -47,8 +47,8 @@ stdenv.mkDerivation (finalAttrs: {
src = fetchFromGitHub {
owner = "openucx";
repo = "ucx";
rev = "v${finalAttrs.version}";
sha256 = "sha256-n3xJmbvUXZzfhotOBJRyH2OEL4NFZIKyB808HwEQSYo=";
tag = "v${finalAttrs.version}";
hash = "sha256-n3xJmbvUXZzfhotOBJRyH2OEL4NFZIKyB808HwEQSYo=";
};
outputs = [
@@ -77,6 +77,7 @@ stdenv.mkDerivation (finalAttrs: {
]
++ lib.optionals enableCuda [
cudaPackages.cuda_cudart
cudaPackages.cuda_nvcc
cudaPackages.cuda_nvml_dev
]
@@ -102,7 +103,7 @@ stdenv.mkDerivation (finalAttrs: {
"--with-verbs=${lib.getDev rdma-core}"
]
++ lib.optionals enableCuda [ "--with-cuda=${cudaPackages.cuda_cudart}" ]
++ lib.optional enableRocm "--with-rocm=${rocm}";
++ lib.optionals enableRocm [ "--with-rocm=${rocm}" ];
postInstall = ''
find $out/lib/ -name "*.la" -exec rm -f \{} \;