Loading pkgs/development/libraries/science/math/nccl/tests.nix +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" ]; Loading Loading
pkgs/development/libraries/science/math/nccl/tests.nix +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" ]; Loading