Commit 68906573 authored by Connor Baker's avatar Connor Baker
Browse files

cudaPackages.nccl-tests: add passthru.updateScript

parent a06f6c29

pkgs/development/libraries/science/math/nccl/tests.nix

+7 −1
Original line number Diff line number Diff line
@@ -6,6 +6,7 @@ 
, mpi

, stdenv

, which

, gitUpdater

}:



cudaPackages.backendStdenv.mkDerivation (finalAttrs: {
@@ -46,12 +47,17 @@ cudaPackages.backendStdenv.mkDerivation (finalAttrs: { 
    cp -r build/* $out/bin/

  '';



  passthru.updateScript = gitUpdater {

    inherit (finalAttrs) pname version;

    rev-prefix = "v";

  };



  meta = with lib; {

    description = "Tests to check both the performance and the correctness of NVIDIA NCCL operations";

    homepage = "https://github.com/NVIDIA/nccl-tests";

    platforms = platforms.linux;

    license = licenses.bsd3;

    broken = !config.cudaSupport || (mpiSupport && mpi == null);

    maintainers = with maintainers; [ jmillerpdt ];

    maintainers = with maintainers; [ jmillerpdt ] ++ teams.cuda.members;

  };

})