diff --git a/pkgs/development/libraries/rccl/default.nix b/pkgs/development/libraries/rccl/default.nix new file mode 100644 index 000000000000..440705b2c017 --- /dev/null +++ b/pkgs/development/libraries/rccl/default.nix @@ -0,0 +1,84 @@ +{ lib +, stdenv +, fetchFromGitHub +, cmake +, rocm-cmake +, rocm-runtime +, rocm-device-libs +, rocm-comgr +, rocm-smi +, hip +, gtest +, chrpath ? null +, buildTests ? false +}: + +assert buildTests -> chrpath != null; + +stdenv.mkDerivation rec { + pname = "rccl"; + rocmVersion = "5.3.1"; + version = "2.12.10-${rocmVersion}"; + + outputs = [ + "out" + ] ++ lib.optionals buildTests [ + "test" + ]; + + src = fetchFromGitHub { + owner = "ROCmSoftwarePlatform"; + repo = "rccl"; + rev = "rocm-${rocmVersion}"; + hash = "sha256-whRXGD8oINDYhFs8+hEWKWoGNqacGlyy7xi8peA8Qsk="; + }; + + nativeBuildInputs = [ + cmake + rocm-cmake + hip + ]; + + buildInputs = [ + rocm-runtime + rocm-device-libs + rocm-comgr + rocm-smi + gtest + ] ++ lib.optionals buildTests [ + chrpath + ]; + + cmakeFlags = [ + "-DCMAKE_C_COMPILER=hipcc" + "-DCMAKE_CXX_COMPILER=hipcc" + # Manually define CMAKE_INSTALL_ + # See: https://github.com/NixOS/nixpkgs/pull/197838 + "-DCMAKE_INSTALL_BINDIR=bin" + "-DCMAKE_INSTALL_LIBDIR=lib" + "-DCMAKE_INSTALL_INCLUDEDIR=include" + ] ++ lib.optionals buildTests [ + "-DBUILD_TESTS=ON" + ]; + + # Replace the manually set parallel jobs to NIX_BUILD_CORES + postPatch = '' + substituteInPlace CMakeLists.txt \ + --replace "8 P" "$NIX_BUILD_CORES P" \ + --replace "8)" "$NIX_BUILD_CORES)" + ''; + + postInstall = lib.optionalString buildTests '' + mkdir -p $test/bin + mv $out/bin/* $test/bin + rmdir $out/bin + ''; + + meta = with lib; { + description = "ROCm communication collectives library"; + homepage = "https://github.com/ROCmSoftwarePlatform/rccl"; + license = with licenses; [ bsd2 bsd3 ]; + maintainers = with maintainers; [ Madouura ]; + broken = rocmVersion != hip.version; + }; +} diff --git a/pkgs/top-level/all-packages.nix b/pkgs/top-level/all-packages.nix index 5e024086f90e..2a37c75ba7e6 100644 --- a/pkgs/top-level/all-packages.nix +++ b/pkgs/top-level/all-packages.nix @@ -14853,6 +14853,8 @@ with pkgs; hipsparse = callPackage ../development/libraries/hipsparse { }; + rccl = callPackage ../development/libraries/rccl { }; + rocm-cmake = callPackage ../development/tools/build-managers/rocm-cmake { }; rocm-comgr = callPackage ../development/libraries/rocm-comgr {