Commit 7e9527b3 authored by Connor Baker's avatar Connor Baker
Browse files

cudaPackages.nccl-tests: 2.13.6 -> 2.13.8

parent 68906573
Loading
Loading
Loading
Loading
+16 −10
Original line number Diff line number Diff line
{ config
, cudaPackages
, cuda_cccl
, cuda_cudart
, cuda_nvcc
, cudaVersion
, backendStdenv
, fetchFromGitHub
, lib
, mpiSupport ? false
, mpi
, stdenv
, nccl
, which
, gitUpdater
}:

cudaPackages.backendStdenv.mkDerivation (finalAttrs: {
backendStdenv.mkDerivation (finalAttrs: {

  pname = "nccl-tests";
  version = "2.13.6";
  version = "2.13.8";

  src = fetchFromGitHub {
    owner = "NVIDIA";
    repo = finalAttrs.pname;
    rev = "v${finalAttrs.version}";
    hash = "sha256-3gSBQ0g6mnQ/MFXGflE+BqqrIUoiBgp8+fWRQOvLVkw=";
    hash = "sha256-dxLoflsTHDBnZRTzoXdm30OyKpLlRa73b784YWALBHg=";
  };

  strictDeps = true;

  nativeBuildInputs = [
    cudaPackages.cuda_nvcc
    cuda_nvcc
    which
  ];

  buildInputs = [
    cudaPackages.cuda_cudart
    cudaPackages.nccl
    cuda_cudart
    nccl
  ] ++ lib.optionals (lib.versionAtLeast cudaVersion "12.0") [
    cuda_cccl.dev # <nv/target>
  ] ++ lib.optional mpiSupport mpi;

  makeFlags = [
    "CUDA_HOME=${cudaPackages.cuda_nvcc}"
    "NCCL_HOME=${cudaPackages.nccl}"
    "CUDA_HOME=${cuda_nvcc}"
    "NCCL_HOME=${nccl}"
  ] ++ lib.optionals mpiSupport [
    "MPI=1"
  ];