ucx: fix with cudaSupport (#460294)
This commit is contained in:
@@ -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 \{} \;
|
||||
|
||||
Reference in New Issue
Block a user