Unverified Commit 0f978ad6 authored by Connor Baker's avatar Connor Baker Committed by GitHub
Browse files

ucx: fix with cudaSupport (#460294)

parents c0d0db90 3f5a60fb

pkgs/by-name/uc/ucx/package.nix

+4 −3
Original line number Diff line number Diff line
@@ -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 \{} \;